index.ts

import { execSync } from "node:child_process";
import fs from "node:fs";
import { createServer } from "node:http";
import type { Socket } from "node:net";
import path from "node:path";
import fastifyMiddie from "@fastify/middie";
import fastifyStatic from "@fastify/static";
// @ts-expect-error
import { server as wisp } from "@mercuryworkshop/wisp-js/server";
import { build } from "astro";
import Fastify from "fastify";
import INConfig from "./config";
import { Main, Revert } from "./randomize";

async function Start() {
  const FirstRun = process.env.FIRST === "true";

  if (!fs.existsSync("dist")) {
    Main();
    console.log("Interstellar's not built yet! Building now...");

    await build({}).catch((err) => {
      console.error(err);
      process.exit(1);
    });

    await Revert();

    if (FirstRun) {
      console.log("Restarting Server...");
      execSync("pnpm disable && pnpm start", { stdio: "inherit" });
      process.exit(0);
    }
  }

  const port = INConfig.server?.port || 8080;

  const app = Fastify({
    serverFactory: (handler) =>
      createServer(handler).on("upgrade", (req, socket: Socket, head) =>
        req.url?.startsWith("/f")
          ? wisp.routeRequest(req, socket, head)
          : socket.destroy(),
      ),
  });

  await app.register(import("@fastify/compress"), {
    encodings: ["br", "gzip", "deflate"],
  });

  if (INConfig.auth?.challenge) {
    await app.register(import("@fastify/basic-auth"), {
      authenticate: true,
      validate(username, password, _req, _reply, done) {
        for (const [user, pass] of Object.entries(INConfig.auth?.users || {})) {
          if (username === user && password === pass) {
            return done();
          }
        }
        return done(new Error("Invalid credentials"));
      },
    });
    await app.after();
    app.addHook("onRequest", app.basicAuth);
  }

  // @ts-ignore
  const { handler } = await import("./dist/server/entry.mjs");
  await app
    .register(fastifyStatic, {
      root: path.join(import.meta.dirname, "dist", "client"),
    })
    .register(fastifyMiddie);
  app.use(handler);
  app.listen({ port }, (err, addr) => {
    if (err) {
      console.error(err);
      process.exit(1);
    }
    console.log("Listening on %s", addr);
  });
}

process.env.FIRST = process.env.FIRST || "true";
await Start();