Updated 7 files and added 4 files (automated)

1 files changed, 75 insertions, 0 deletions

diff --git a/server/platforms.js b/server/platforms.js
new file mode 100644
index 0000000..01773ed
--- /dev/null
+++ b/server/platforms.js
@@ -0,0 +1,75 @@
+const secrets = require('./secrets.json');
+const WebSocketServer = require('ws').WebSocketServer;
+const wss = new WebSocketServer({ port: 19219 });
+
+async function processRequest(ws, data) {
+    if (data.method === "wolfram") {
+        let query = "https://api.wolframalpha.com/v1/simple?appid=" + secrets.wolfram + "&i=" + encodeURIComponent(data.query ?? "");
+
+        ws.send({
+            state: "REQUEST_RESPONSE",
+            data: {
+                _id: data._id ?? null,
+                data: Buffer.from(await (await fetch(query)).arrayBuffer()).toString("base64")
+            }
+        });
+
+        return;
+    }
+
+    ws.send({
+        state: "REQUEST_RESPONSE",
+        data: null
+    });
+}
+
+wss.on('connection', function connection(ws) {
+    ws.authenticated = false;
+    ws.token = null;
+    ws.auth = null;
+
+    ws._send = ws.send;
+    ws.send = (d) => {
+        return ws._send(JSON.stringify(d));
+    }
+
+    ws.on('error', console.error);
+
+    ws.on('message', async (_data) => {
+        try {
+            let data = JSON.parse(_data);
+
+            if (data.state) {
+                switch (data.state) {
+                    case "AUTHENTICATION_RESPONSE":
+                        if (data.token) {
+                            let userInfo = (await (await fetch("https://school.equestria.dev/_matrix/client/v3/account/whoami", { headers: { 'Authorization': 'Bearer ' + data.token } })).json());
+                            console.log(userInfo);
+
+                            if (userInfo['user_id']) {
+                                ws.auth = userInfo;
+                                ws.token = data.token;
+                                ws.authenticated = true;
+
+                                ws.send({
+                                    state: "AUTHENTICATION_OK"
+                                });
+                            } else {
+                                ws.close();
+                            }
+                        }
+                        break;
+                }
+            }
+
+            if (!ws.authenticated) ws.close();
+            if (!data.state || data.state !== "AUTHENTICATION_RESPONSE") processRequest(ws, data);
+        } catch (e) {
+            console.error(e);
+        }
+    });
+
+    ws.send({
+        state: "NEEDS_AUTHENTICATION"
+    });
+});