Add http2 on the same port with websocket

tshemsedinov · tshemsedinov · commit 4248678a8d7d · 2023-06-01T23:21:12.000+03:00
diff --git a/http2-ws/README.md b/http2-ws/README.md
@@ -0,0 +1,22 @@
+# Start HTTP server
+
+## With self-signed certificate (for testing)
+
+- Generate certificates, run: `./cert/generate.sh`
+- Start server: `node server.js`
+- Open in browser: `https://127.0.0.1:8000/`
+
+## With certbot (for production)
+
+- Let's Encrypt is a free certificate authority: https://letsencrypt.org/
+- Use Certbot (free tool for automatically generatinging Let’s Encrypt
+  certificates to enable HTTPS): https://certbot.eff.org/
+  - Install: `dnf -y install certbot`
+- Generate certificate:
+  `certbot certonly --standalone -d www.domain.com -d domain.com -m your.name@domain.com --agree-tos --no-eff-email`
+- Copy certificate:
+  `cp /etc/letsencrypt/live/domain.com/fullchain.pem ./cert/cert.pem`
+- Copy private key:
+  `cp /etc/letsencrypt/live/domain.com/privkey.pem ./cert/key.pem`
+- Or use your web server for challenge exchange:
+  `certbot certonly --webroot -w ./ -d domain.com -m your.name@domain.com --agree-tos --no-eff-email`
diff --git a/http2-ws/cert/generate.ext b/http2-ws/cert/generate.ext
@@ -0,0 +1,9 @@
+authorityKeyIdentifier=keyid,issuer
+basicConstraints=CA:FALSE
+keyUsage = digitalSignature, nonRepudiation, keyEncipherment, dataEncipherment
+subjectAltName = @alt_names
+
+[alt_names]
+DNS.1 = example.com
+IP.1 = 127.0.0.1
+IP.2 = ::1
diff --git a/http2-ws/cert/generate.sh b/http2-ws/cert/generate.sh
@@ -0,0 +1,5 @@
+cd "$(dirname "$0")"
+openssl genrsa -out key.pem 3072
+openssl req -new -out self.pem -key key.pem -subj '/CN=localhost'
+openssl req -text -noout -in self.pem
+openssl x509 -req -days 1024 -in self.pem -signkey key.pem -out cert.pem -extfile generate.ext
diff --git a/http2-ws/index.html b/http2-ws/index.html
@@ -0,0 +1,34 @@
+<!DOCTYPE html>
+<html>
+<body>
+<h1>Page example</h1>
+<div id="log"></div>
+<style>
+input { border: 1px solid green; }
+</style>
+<script>
+  const log = document.getElementById('log');
+
+  const writeLine = (text) => {
+    const line = document.createElement('div');
+    line.innerHTML = `<p>${text}</p>`;
+    log.appendChild(line);
+  };
+
+  const socket = new WebSocket('wss://127.0.0.1:8000');
+
+  socket.addEventListener('open', () => {
+    socket.send('Hello!');
+    writeLine('connected');
+  });
+
+  socket.addEventListener('close', () => {
+    writeLine('closed');
+  });
+
+  socket.addEventListener('message', ({ data }) => {
+    writeLine(data);
+  });
+</script>
+</body>
+</html>
diff --git a/http2-ws/package-lock.json b/http2-ws/package-lock.json
diff --git a/http2-ws/package.json b/http2-ws/package.json
@@ -0,0 +1,10 @@
+{
+  "name": "http2WebsocketExample",
+  "version": "1.0.0",
+  "author": "Timur Shemsedinov <timur.shemsedinov@gmail.com>",
+  "description": "HTTP2 and Websocket Example",
+  "license": "MIT",
+  "dependencies": {
+    "ws": "^8.13.0"
+  }
+}
diff --git a/http2-ws/server.js b/http2-ws/server.js
@@ -0,0 +1,62 @@
+'use strict';
+
+const fs = require('node:fs');
+const http2 = require('node:http2');
+const WebSocket = require('ws');
+
+const index = fs.readFileSync('./index.html', 'utf8');
+const user = { name: 'jura', age: 22 };
+
+const routing = {
+  '/': index,
+  '/user': user,
+  '/user/name': () => user.name.toUpperCase(),
+  '/user/age': () => user.age,
+  '/hello': { hello: 'world', andArray: [1, 2, 3, 4, 5, 6, 7] },
+  '/api/method1': (req, res) => {
+    console.log(req.url + ' ' + res.statusCode);
+    return { status: res.statusCode };
+  },
+  '/api/method2': (req) => ({
+    user,
+    url: req.url,
+    cookie: req.headers.cookie
+  }),
+};
+
+const types = {
+  object: JSON.stringify,
+  string: (s) => s,
+  undefined: () => 'not found',
+  function: (fn, req, res) => JSON.stringify(fn(req, res)),
+};
+
+const key = fs.readFileSync('./cert/key.pem');
+const cert = fs.readFileSync('./cert/cert.pem');
+const options = { key, cert };
+
+const server = http2.createSecureServer(options, (req, res) => {
+  const data = routing[req.url];
+  const type = typeof data;
+  const serializer = types[type];
+  const result = serializer(data, req, res);
+  res.end(result);
+});
+
+server.listen(8000);
+console.log('Open: https://127.0.0.1:8000');
+
+const ws = new WebSocket.Server({ server });
+
+ws.on('connection', (connection, req) => {
+  const ip = req.socket.remoteAddress;
+  console.log(`Connected ${ip}`);
+  connection.on('message', (message) => {
+    console.log('Received: ' + message);
+  });
+  connection.on('close', () => {
+    console.log(`Disconnected ${ip}`);
+  });
+});
+
+setInterval(() => user.age++, 2000);
