prasi-api/pkgs/api/_deploy.ts

import { $ } from "bun";
import * as fs from "fs";
import { dirAsync, readAsync, removeAsync, writeAsync } from "fs-jetpack";
import { apiContext } from "service-srv";
import { deploy } from "utils/deploy";
import { dir } from "utils/dir";
import { g } from "utils/global";
import { genEnv, parseEnv } from "utils/parse-env";
import { restartServer } from "utils/restart";

export const _ = {
  url: "/_deploy",
  async api(
    action: (
      | { type: "check" }
      | { type: "db-update"; url: string }
      | { type: "db-pull" }
      | { type: "db-gen" }
      | { type: "db-ver" }
      | { type: "db-sync"; url: string }
      | { type: "restart" }
      | { type: "domain-add"; domain: string }
      | { type: "domain-del"; domain: string }
      | { type: "deploy-del"; ts: string }
      | { type: "deploy"; dlurl: string }
      | { type: "deploy-status" }
      | { type: "redeploy"; ts: string }
    ) & {
      id_site: string;
    },
  ) {
    const { res } = apiContext(this);

    const path = dir(`app/web/`);
    await dirAsync(path);

    switch (action.type) {
      case "check":
        const deploys = fs.readdirSync(dir(`/app/web/deploy`));

        return {
          now: Date.now(),
          current: parseInt(g.deploy.config.deploy.ts),
          deploys: deploys
            .filter((e) => e.endsWith(".gz"))
            .map((e) => parseInt(e.replace(".gz", ""))),
          db: {
            url: g.dburl || "-",
          },
        };
      case "db-ver":
        {
          let file = Bun.file(dir(`${g.datadir}/db-ver`));
          if (!(await file.exists())) {
            await Bun.write(dir(`${g.datadir}/db-ver`), Date.now().toString());
            file = Bun.file(dir(`${g.datadir}/db-ver`));
          }
          const res = await file.text();
          return res;
        }
        break;
      case "db-sync": {
        const res = await fetch(action.url);
        const text = await res.text();
        if (text) {
          await Bun.write(dir("app/db/prisma/schema.prisma"), text);
          await Bun.write(dir(`${g.datadir}/db-ver`), Date.now().toString());
        }
        return "ok";
      }
      case "db-update":
        if (action.url) {
          g.dburl = action.url;
          const env = genEnv({
            ...parseEnv(await Bun.file(dir(".env")).text()),
            DATABASE_URL: action.url,
          });
          await Bun.write(dir(".env"), env);
          await Bun.write(dir("app/db/.env"), env);
        }
        return "ok";
      case "db-gen":
        {
          await $`bun prisma generate`.cwd(dir("app/db"));

          res.send("ok");
          setTimeout(() => {
            restartServer();
          }, 300);
        }
        break;
      case "db-pull":
        {
          let env = await readAsync(dir("app/db/.env"));
          if (env) {
            const ENV = parseEnv(env);
            if (typeof ENV.DATABASE_URL === "string") {
              const type = ENV.DATABASE_URL.split("://").shift();
              if (type) {
                await writeAsync(
                  dir("app/db/prisma/schema.prisma"),
                  `\
    generator client {
      provider = "prisma-client-js"
    }

    datasource db {
      provider = "${type}"
      url      = env("DATABASE_URL")
    }`,
                );

                try {
                  await Bun.write(
                    dir("app/db/.env"),
                    `DATABASE_URL=${ENV.DATABASE_URL}`,
                  );
                  await $`bun install`.cwd(dir("app/db"));
                  await $`bun prisma db pull --force`.cwd(dir("app/db"));
                  await $`bun prisma generate`.cwd(dir("app/db"));
                  await Bun.write(
                    dir(`${g.datadir}/db-ver`),
                    Date.now().toString(),
                  );
                } catch (e) {
                  console.error(e);
                }
                res.send("ok");
                setTimeout(() => {
                  restartServer();
                }, 300);
              }
            }
          }
        }
        break;
      case "restart":
        {
          res.send("ok");
          setTimeout(() => {
            restartServer();
          }, 300);
        }
        break;
      case "deploy-del":
        {
          await removeAsync(dir(`/app/web/deploy/${action.ts}.gz`));
          const deploys = fs.readdirSync(dir(`/app/web/deploy`));

          return {
            now: Date.now(),
            current: parseInt(deploy.config.deploy.ts),
            deploys: deploys
              .filter((e) => e.endsWith(".gz"))
              .map((e) => parseInt(e.replace(".gz", ""))),
          };
        }
        break;
      case "deploy-status":
        break;
      case "deploy":
        {
          deploy.config.site_id = action.id_site;

          await deploy.saveConfig();
          deploy.config.deploy.ts = Date.now() + "";
          await deploy.init();
          const deploys = fs.readdirSync(dir(`/app/web/deploy`));

          if (g.mode === "prod") {
            setTimeout(() => {
              process.exit();
            }, 500);
          }

          return {
            now: Date.now(),
            current: parseInt(deploy.config.deploy.ts),
            deploys: deploys
              .filter((e) => e.endsWith(".gz"))
              .map((e) => parseInt(e.replace(".gz", ""))),
          };
        }
        break;
      case "redeploy":
        {
          deploy.config.deploy.ts = action.ts;
          await deploy.saveConfig();
          await deploy.load(action.ts);
          const deploys = fs.readdirSync(dir(`/app/web/deploy`));

          return {
            now: Date.now(),
            current: parseInt(deploy.config.deploy.ts),
            deploys: deploys
              .filter((e) => e.endsWith(".gz"))
              .map((e) => parseInt(e.replace(".gz", ""))),
          };
        }
        break;
    }
  },
};

export const downloadFile = async (
  url: string,
  filePath: string,
  progress?: (rec: number, total: number) => void,
) => {
  try {
    const _url = new URL(url);
    if (_url.hostname === "localhost") {
      _url.hostname = "127.0.0.1";
    }
    g.log.info(`Downloading ${url} to ${filePath}`);
    const res = await fetch(_url as any);
    if (res.body) {
      const file = Bun.file(filePath);

      const writer = file.writer();
      const reader = res.body.getReader();

      // Step 3: read the data
      let receivedLength = 0; // received that many bytes at the moment
      let chunks = []; // array of received binary chunks (comprises the body)
      while (true) {
        const { done, value } = await reader.read();

        if (done) {
          writer.end();
          break;
        }

        chunks.push(value);
        writer.write(value);
        receivedLength += value.length;

        if (progress) {
          progress(
            receivedLength,
            parseInt(res.headers.get("content-length") || "0"),
          );
        }
      }
    }
    return true;
  } catch (e) {
    console.log(e);
    return false;
  }
};