arri 0.43.2 → 0.45.0

package/dist/shared/arri.547ce62e.cjs

@@ -0,0 +1,436 @@
+'use strict';
+
+const fs = require('node:fs');
+const fs$1 = require('node:fs/promises');
+const codegenUtils = require('@arrirpc/codegen-utils');
+const listhen = require('@joshmossas/listhen');
+const c12 = require('c12');
+const citty = require('citty');
+const esbuild = require('esbuild');
+const ofetch = require('ofetch');
+const path = require('pathe');
+const consola = require('consola');
+const prettier = require('prettier');
+const scule = require('scule');
+
+function _interopDefaultCompat (e) { return e && typeof e === 'object' && 'default' in e ? e.default : e; }
+
+function _interopNamespaceCompat(e) {
+    if (e && typeof e === 'object' && 'default' in e) return e;
+    const n = Object.create(null);
+    if (e) {
+        for (const k in e) {
+            n[k] = e[k];
+        }
+    }
+    n.default = e;
+    return n;
+}
+
+const fs__default = /*#__PURE__*/_interopDefaultCompat(fs$1);
+const esbuild__namespace = /*#__PURE__*/_interopNamespaceCompat(esbuild);
+const path__default = /*#__PURE__*/_interopDefaultCompat(path);
+const prettier__default = /*#__PURE__*/_interopDefaultCompat(prettier);
+
+function isArriConfig(input) {
+  if (typeof input !== "object" || input === null) {
+    return false;
+  }
+  if ("generators" in input) {
+    if (!Array.isArray(input.generators)) {
+      return false;
+    }
+  }
+  if ("port" in input) {
+    if (typeof input.port !== "number") {
+      return false;
+    }
+    if (Number.isNaN(input.port)) {
+      return false;
+    }
+  }
+  if ("entry" in input && typeof input.entry !== "string") {
+    return false;
+  }
+  return true;
+}
+function isResolvedArriConfig(input) {
+  if (!isArriConfig(input)) {
+    return false;
+  }
+  return typeof input.rootDir === "string" && typeof input.srcDir === "string" && (typeof input.procedureDir === "string" || typeof input.procedureDir === "boolean") && Array.isArray(input.procedureGlobPatterns) && Array.isArray(input.generators);
+}
+const defaultConfig = {
+  port: 3e3,
+  rootDir: ".",
+  srcDir: "src",
+  entry: "app.ts",
+  serverEntry: "",
+  procedureDir: "procedures",
+  procedureGlobPatterns: ["**/*.rpc.ts"],
+  generators: [],
+  buildDir: ".arri",
+  esbuild: {},
+  https: false,
+  http2: false,
+  devServer: {}
+};
+function defineConfig(config) {
+  return {
+    port: config.port ?? defaultConfig.port,
+    rootDir: config.rootDir ?? defaultConfig.rootDir,
+    srcDir: config.srcDir ?? defaultConfig.srcDir,
+    entry: config.entry ?? defaultConfig.entry,
+    procedureDir: config.procedureDir ?? defaultConfig.procedureDir,
+    procedureGlobPatterns: config.procedureGlobPatterns ?? defaultConfig.procedureGlobPatterns,
+    generators: config.generators ?? defaultConfig.generators,
+    buildDir: config.buildDir ?? defaultConfig.buildDir,
+    esbuild: config.esbuild ?? defaultConfig.esbuild,
+    serverEntry: config.serverEntry ?? "",
+    https: config.https ?? false,
+    http2: config.http2 ?? false,
+    devServer: config.devServer ?? {}
+  };
+}
+
+const logger = consola.createConsola().withTag("arri");
+const GEN_APP_FILE = "__arri_app.ts";
+const GEN_SERVER_ENTRY_FILE = "__arri_server.ts";
+const OUT_APP_FILE = "app.mjs";
+const OUT_SERVER_ENTRY = "server.mjs";
+const OUT_CODEGEN = "codegen.mjs";
+async function setupWorkingDir(config) {
+  const arriDir = path__default.resolve(config.rootDir, config.buildDir);
+  const outDir = path__default.resolve(config.rootDir, ".output");
+  if (fs.existsSync(arriDir)) {
+    await fs__default.rm(arriDir, { recursive: true, force: true });
+  }
+  if (fs.existsSync(outDir)) {
+    await fs__default.rm(outDir, { recursive: true, force: true });
+  }
+  await fs__default.mkdir(arriDir);
+  await fs__default.mkdir(outDir);
+}
+async function createAppWithRoutesModule(config) {
+  const glob = await import('globby');
+  const appModule = path__default.resolve(config.rootDir, config.srcDir, config.entry);
+  const appImportParts = path__default.relative(path__default.resolve(config.rootDir, config.buildDir), appModule).split(".");
+  appImportParts.pop();
+  const routes = [];
+  const existingRoutes = [];
+  await Promise.all(
+    config.procedureGlobPatterns.map(async (pattern) => {
+      const results = await getFsRouteBatch(glob.globby, pattern, config);
+      for (const result of results) {
+        if (!existingRoutes.includes(result.name)) {
+          routes.push(result);
+          existingRoutes.push(result.name);
+        }
+      }
+    })
+  );
+  routes.sort((a, b) => a.name < b.name ? -1 : 1);
+  const module = await prettier__default.format(
+    `import sourceMapSupport from 'source-map-support';
+        sourceMapSupport.install();
+        import app from '${appImportParts.join(".")}';
+        ${routes.map(
+      (route) => `import ${route.importName} from '${route.importPath}';`
+    ).join("\n")}
+        const routes: {id: string; route: any}[] = [${routes.map(
+      (route) => `{ id: '${route.name}', route: ${route.importName} }`
+    ).join(",\n")}
+        ];
+        for(const route of routes) {
+            if(route.route.transport === 'http') {
+                app.rpc({
+                    name: route.id,
+                    ...route.route,
+                });
+                continue;
+            }
+            if (route.route.transport === 'ws') {
+                app.wsRpc({
+                    name: route.id,
+                    ...route.route,
+                });
+                continue;
+            }
+        }
+        export default app`,
+    { parser: "typescript", tabWidth: 4 }
+  );
+  await fs__default.writeFile(
+    path__default.resolve(config.rootDir, config.buildDir, GEN_APP_FILE),
+    module
+  );
+}
+async function getFsRouteBatch(glob, globPattern, config) {
+  if (!config.procedureDir) {
+    return [];
+  }
+  const target = `${config.rootDir}/${config.srcDir}/${config.procedureDir}/${globPattern}`.split("//").join("/").split("//").join("/");
+  const files = await glob(target, {
+    onlyFiles: true
+  });
+  const routes = [];
+  for (const file of files) {
+    const meta = getRpcMetaFromPath(config, file);
+    if (meta) {
+      const importParts = path__default.relative(path__default.resolve(config.rootDir, config.buildDir), file).split(".");
+      importParts.pop();
+      importParts.push("js");
+      routes.push({
+        name: meta.id,
+        importName: scule.camelCase(meta.id.split(".").join("_")),
+        importPath: `./${importParts.join(".")}`
+      });
+    }
+  }
+  return routes;
+}
+const disallowedNameChars = "~`!@#$%^&*()-+=[]{}\\|:;\"'<>,./";
+const getRpcMetaFromPath = (config, filePath) => {
+  if (!config.procedureDir) {
+    return void 0;
+  }
+  const resolvedFilePath = filePath.split("./").join("/");
+  const parts = resolvedFilePath.split("/");
+  const reversedParts = [...parts].reverse();
+  const nameParts = [];
+  for (const part of reversedParts) {
+    if (part === config.procedureDir) {
+      break;
+    }
+    nameParts.push(part);
+  }
+  const fileName = nameParts.reverse().pop() ?? "";
+  const serviceParts = nameParts.filter((part) => part.trim().length > 0).map((part) => codegenUtils.removeDisallowedChars(part, disallowedNameChars));
+  const fileNameParts = fileName.split(".");
+  if (!fileNameParts.length) {
+    return void 0;
+  }
+  const rpcName = codegenUtils.removeDisallowedChars(
+    fileNameParts[0],
+    disallowedNameChars
+  );
+  const httpParts = [
+    ...serviceParts.map((part) => scule.kebabCase(part)),
+    scule.kebabCase(rpcName)
+  ];
+  return {
+    id: [...serviceParts, rpcName].join("."),
+    httpPath: `/${httpParts.join("/")}`
+  };
+};
+function isInsideDir(dir, parentDir) {
+  if (path__default.resolve(dir).startsWith(path__default.resolve(parentDir))) {
+    return true;
+  }
+  return false;
+}
+
+const DEV_ENDPOINT_ROOT = `/__arri_dev__`;
+const DEV_DEFINITION_ENDPOINT = `${DEV_ENDPOINT_ROOT}/__definition`;
+const dev = citty.defineCommand({
+  meta: {
+    name: "dev",
+    description: "Start the arri dev server"
+  },
+  args: {
+    config: {
+      type: "string",
+      description: "Path to the arri config file",
+      alias: "c",
+      default: "./arri.config.ts"
+    }
+  },
+  async run({ args }) {
+    process.env.ARRI_DEV_MODE = "true";
+    const { config } = await c12.loadConfig({
+      configFile: path__default.resolve(args.config)
+    });
+    if (!isResolvedArriConfig(config)) {
+      throw new Error("Unable to find config");
+    }
+    const appEntry = path__default.resolve(
+      config.rootDir,
+      config.srcDir,
+      config.entry
+    );
+    if (!fs.existsSync(appEntry)) {
+      throw new Error(`Unable to find entry at ${appEntry}`);
+    }
+    await startDevServer(config);
+  }
+});
+const startListener = (config, showQr = false) => listhen.listenAndWatch(path__default.resolve(config.rootDir, ".output", OUT_SERVER_ENTRY), {
+  public: true,
+  port: config.port,
+  logger,
+  qr: showQr,
+  https: config.https,
+  http2: config.http2,
+  ws: true
+});
+async function bundleFilesContext(config) {
+  const serverContent = `import app from './${OUT_APP_FILE}';
+
+export default app.h3App;`;
+  await fs__default.writeFile(
+    path__default.resolve(config.rootDir, ".output", OUT_SERVER_ENTRY),
+    serverContent
+  );
+  return await esbuild__namespace.context({
+    ...config.esbuild,
+    entryPoints: [
+      path__default.resolve(config.rootDir, config.buildDir, GEN_APP_FILE)
+    ],
+    outfile: path__default.resolve(config.rootDir, ".output", OUT_APP_FILE),
+    format: "esm",
+    bundle: true,
+    packages: "external",
+    sourcemap: config.esbuild.sourcemap ?? true,
+    target: config.esbuild.target ?? "node20",
+    platform: config.esbuild.platform ?? "node"
+  });
+}
+async function startDevServer(config) {
+  await setupWorkingDir(config);
+  const watcher = await import('chokidar');
+  let fileWatcher;
+  await Promise.all([
+    createEntryModule(config),
+    createAppWithRoutesModule(config)
+  ]);
+  const context = await bundleFilesContext(config);
+  await context.rebuild();
+  const listener = await startListener(config, true);
+  setTimeout(async () => {
+    await generateClients(config);
+  }, 1e3);
+  const cleanExit = async () => {
+    process.exit();
+  };
+  process.on("exit", async () => {
+    await Promise.allSettled([
+      listener.close(),
+      fileWatcher?.close(),
+      context.dispose()
+    ]);
+  });
+  process.on("SIGINT", cleanExit);
+  process.on("SIGTERM", cleanExit);
+  async function load(isRestart, reason) {
+    if (fileWatcher) {
+      await fileWatcher.close();
+    }
+    const srcDir = path__default.resolve(config.rootDir ?? "", config.srcDir);
+    const dirsToWatch = [srcDir];
+    if (config.esbuild.alias) {
+      for (const key of Object.keys(config.esbuild.alias)) {
+        const alias = config.esbuild.alias[key];
+        if (alias) {
+          if (!isInsideDir(alias, srcDir)) {
+            dirsToWatch.push(alias);
+          }
+        }
+      }
+    }
+    if (config.devServer.additionalWatchDirs?.length) {
+      for (const dir of config.devServer.additionalWatchDirs) {
+        dirsToWatch.push(dir);
+      }
+    }
+    const buildDir = path__default.resolve(config.rootDir, config.buildDir);
+    const outDir = path__default.resolve(config.rootDir, ".output");
+    fileWatcher = watcher.watch(dirsToWatch, {
+      ignoreInitial: true,
+      ignored: [
+        buildDir,
+        outDir,
+        "**/.output",
+        "**/dist/**",
+        "**/node_modules/**",
+        "**/.git/**"
+      ]
+    });
+    fileWatcher.on("all", async (eventName, path2) => {
+      if (eventName === "addDir" || eventName === "add") {
+        return;
+      }
+      await createAppWithRoutesModule(config);
+      let bundleCreated = false;
+      try {
+        logger.log(
+          "Change detected. Bundling files and restarting server...."
+        );
+        await context.rebuild();
+        bundleCreated = true;
+      } catch (err) {
+        logger.error("ERROR", err);
+        bundleCreated = false;
+      }
+      if (bundleCreated && config.generators.length) {
+        setTimeout(async () => {
+          await generateClients(config);
+          bundleCreated = false;
+        }, 1e3);
+      }
+    });
+  }
+  await load();
+}
+async function generateClients(config) {
+  try {
+    const protocol = config.https === true || typeof config.https === "object" ? "https" : "http";
+    if (protocol === "https") {
+      process.env.NODE_TLS_REJECT_UNAUTHORIZED = "0";
+    }
+    const result = await ofetch.ofetch(
+      `${protocol}://127.0.0.1:${config.port}${DEV_DEFINITION_ENDPOINT}`
+    );
+    if (!codegenUtils.isAppDefinition(result)) {
+      return;
+    }
+    logger.log(`Generating client code...`);
+    const clientCount = config.generators.length;
+    await Promise.allSettled(
+      config.generators.map(
+        (generator) => generator.generator(result, true)
+      )
+    );
+    logger.success(
+      `Generated ${clientCount} client${clientCount === 1 ? "" : "s"}`
+    );
+  } catch (err) {
+    logger.error(err);
+  }
+}
+async function createEntryModule(config) {
+  const appModule = path__default.resolve(config.rootDir, config.srcDir, config.entry);
+  const appImportParts = path__default.relative(path__default.resolve(config.rootDir, config.srcDir), appModule).split(".");
+  appImportParts.pop();
+  const virtualEntry = `import { toNodeListener } from 'h3';
+import app from './${GEN_APP_FILE}';
+
+export default app.h3App;`;
+  await fs__default.writeFile(
+    path__default.resolve(config.rootDir, config.buildDir, GEN_SERVER_ENTRY_FILE),
+    virtualEntry
+  );
+}
+
+exports.DEV_DEFINITION_ENDPOINT = DEV_DEFINITION_ENDPOINT;
+exports.GEN_APP_FILE = GEN_APP_FILE;
+exports.OUT_APP_FILE = OUT_APP_FILE;
+exports.OUT_CODEGEN = OUT_CODEGEN;
+exports.OUT_SERVER_ENTRY = OUT_SERVER_ENTRY;
+exports.createAppWithRoutesModule = createAppWithRoutesModule;
+exports.defaultConfig = defaultConfig;
+exports.defineConfig = defineConfig;
+exports.dev = dev;
+exports.isArriConfig = isArriConfig;
+exports.isResolvedArriConfig = isResolvedArriConfig;
+exports.logger = logger;
+exports.setupWorkingDir = setupWorkingDir;