server.ts

import http from "node:http";
import { cwd as getPwd } from "node:process";
import path from "node:path";
import fs from "node:fs";

import chalk from "chalk";
import boxen from "boxen";
import clipboard from "clipboardy";
import urlencode from "urlencode";

import { ip } from "./ip.js";
import { port } from "./port.js";
import { sendDirectory } from "./directory.js";
import { sendFile } from "./file.js";
import { args } from "./args.js";

// https://stackoverflow.com/a/70106896
import manifest from "./package.json" assert { type: "json" };

let server;

function createServer() {
  // Base Path of Directory
  let basePath = path.join(getPwd(), args["dir"] as string);

  // Request Handler
  server = http.createServer((req, res) => {
    if (!req.url) {
      req.url = "/";
    }

    // Decode the encoded URL
    req.url = urlencode.decode(req.url);

    // Full path to requested file or directory
    const fullPath = path.join(basePath, ...req.url!.split("/"));

    if (!fs.existsSync(fullPath)) {
      return res.end("Invalid File Path");
    }

    const stats = fs.statSync(fullPath);

    if (stats.isDirectory()) {
      return sendDirectory(res, fullPath);
    } else if (stats.isFile()) {
      return sendFile(res, fullPath);
    } else {
      return res.end("Invalid Path");
    }
  });

  // Output when server is ready
  server.listen(port, ip, () => {
    const url = `http://${ip}:${port}`;

    clipboard.writeSync(url);

    console.log(
      boxen(
        chalk.yellow("Server started on: ") +
          chalk.bold.underline.green(url) +
          "\n" +
          "\n" +
          chalk.dim.red(`Exposed Directory: ${basePath}`) +
          "\n" +
          chalk.gray("URL copied to clipboard"),
        {
          title: `${chalk.bold.rgb(160, 32, 240)("Sirin")} ${chalk.blue(
            `v${manifest.version}`
          )}`,
          textAlignment: "center",
          padding: 1,
          margin: 1,
          borderColor: "green",
        }
      ),
      "\n",
      chalk.rgb(
        218,
        52,
        80
      )(`
By Saurav Pal,
Cloud & DevOps Moderator,
GDSC NIT Silchar 2022-23`)
    );
  });

  server.on("error", () => {
    console.error("Error");
  });

  // Graceful Exit
  process.on("SIGINT", serverClose(server));
}

function serverClose(server: http.Server) {
  return () => {
    console.log(chalk.red("Closing Server..."));
    server.close();
    console.log(chalk.green("Server Closed!"));
  };
}

export { createServer };