chore: Adds lom with basic commands parsing.

lom/src/index.ts

@@ -1 +1,146 @@
-console.log("Hello, world!");
+import { readFile, writeFile, mkdir } from "node:fs/promises";
+
+import parseArgsStringToArgv from "string-argv";
+import baseAxios from "axios";
+import ssh2 from "ssh2";
+
+import { commands } from "./commands.js";
+
+let keyFile: Buffer | string | undefined;
+
+const serverBaseURL: string =
+  process.env.SERVER_BASE_URL ?? "http://127.0.0.1:3000/";
+
+const axios = baseAxios.create({
+  baseURL: serverBaseURL,
+  validateStatus(status) {
+    return true;
+  },
+});
+
+try {
+  keyFile = await readFile("../keys/host.key");
+} catch (e) {
+  console.log("Error reading host key file! Creating new keypair...");
+  await mkdir("../keys").catch(() => null);
+
+  const keyPair: { private: string; public: string } = await new Promise(
+    resolve =>
+      ssh2.utils.generateKeyPair("ed25519", (err, keyPair) => resolve(keyPair)),
+  );
+
+  await writeFile("../keys/host.key", keyPair.private);
+  await writeFile("../keys/host.pub", keyPair.public);
+
+  keyFile = keyPair.private;
+}
+
+if (!keyFile) throw new Error("Somehow failed to fetch the key file!");
+
+const server: ssh2.Server = new ssh2.Server({
+  hostKeys: [keyFile],
+
+  banner: "NextNet-LOM (c) NextNet project et al.",
+  greeting: "NextNet LOM (beta)",
+});
+
+server.on("connection", client => {
+  let token: string = "";
+
+  client.on("authentication", async auth => {
+    if (auth.method != "password") return auth.reject(["password"]); // We need to know the password to auth with the API
+
+    const response = await axios.post("/api/v1/users/login", {
+      username: auth.username,
+      password: auth.password,
+    });
+
+    if (response.status == 403) {
+      return auth.reject(["password"]);
+    }
+
+    token = response.data.token;
+    auth.accept();
+  });
+
+  client.on("ready", () => {
+    client.on("session", (accept, reject) => {
+      const conn = accept();
+
+      // We're dumb. We don't really care.
+      conn.on("pty", accept => accept());
+      conn.on("window-change", accept => accept());
+
+      conn.on("shell", async accept => {
+        const stream = accept();
+        stream.write(
+          "Welcome to NextNet LOM. Run 'help' to see commands.\r\n\r\n~$ ",
+        );
+
+        let line = "";
+
+        function println(...str: string[]) {
+          stream.write(str.join(" ").replace("\n", "\r\n"));
+        }
+
+        async function eventLoop(): Promise<any> {
+          const readStreamData = stream.read();
+          if (readStreamData == null) return setTimeout(eventLoop, 5);
+
+          if (readStreamData.includes("\r") || readStreamData.includes("\n")) {
+            line = line.replace("\r", "");
+
+            if (line == "") {
+              stream.write(`\r\n~$ `);
+              return setTimeout(eventLoop, 5);
+            }
+
+            const argv = parseArgsStringToArgv(line);
+            line = "";
+
+            if (argv[0] == "exit") {
+              stream.close();
+            } else {
+              stream.write("\r\n");
+              const command = commands.find(i => i.name == argv[0]);
+
+              if (!command) {
+                stream.write(
+                  `Unknown command ${argv[0]}. Run 'help' to see commands.\r\n~$ `,
+                );
+                return setTimeout(eventLoop, 5);
+              }
+
+              await command.run(argv, println, axios, token);
+              stream.write(`~$ `);
+            }
+          } else if (readStreamData.includes("\x7F")) {
+            // \x7F = Ascii escape code for backspace (client side)
+            if (line == "" || line == "\r") return setTimeout(eventLoop, 5);
+
+            line = line.substring(0, line.length - 1);
+
+            // Ascii excape code for backspace (server side)
+            stream.write("\u0008 \u0008");
+          } else if (!readStreamData.includes("\x1B")) {
+            // (hacky) include all input but client sided ascii movement
+            // TODO: implement
+            line += readStreamData;
+            stream.write(readStreamData);
+          }
+
+          setTimeout(eventLoop, 5);
+        }
+
+        eventLoop();
+      });
+    });
+  });
+});
+
+server.listen(
+  2222,
+  process.env.NODE_ENV == "production" ? "0.0.0.0" : "127.0.0.1",
+);
+
+console.log("Started server at ::2222");