apibara 2.0.0-beta.9 → 2.1.0-beta.3

package/src/create/utils.ts

@@ -0,0 +1,422 @@
+import fs from "node:fs";
+import path, { basename } from "node:path";
+import * as prettier from "prettier";
+import prompts from "prompts";
+import { blue, cyan, red, yellow } from "./colors";
+import { dnaUrls, networks } from "./constants";
+import type { Chain, Language, Network, PkgInfo } from "./types";
+
+export function isEmpty(path: string) {
+  const files = fs.readdirSync(path);
+  return files.length === 0 || (files.length === 1 && files[0] === ".git");
+}
+export function emptyDir(dir: string) {
+  if (!fs.existsSync(dir)) {
+    return;
+  }
+  for (const file of fs.readdirSync(dir)) {
+    if (file === ".git") {
+      continue;
+    }
+    fs.rmSync(path.resolve(dir, file), { recursive: true, force: true });
+  }
+}
+
+export function validateLanguage(language?: string, throwError = false) {
+  if (!language) {
+    return false;
+  }
+
+  if (
+    language === "typescript" ||
+    language === "ts" ||
+    language === "javascript" ||
+    language === "js"
+  ) {
+    return true;
+  }
+
+  if (throwError) {
+    throw new Error(
+      `Invalid language ${cyan("(--language | -l)")}: ${red(language)}. Options: ${blue("typescript, ts")} or ${yellow("javascript, js")} | default: ${cyan("typescript")}`,
+    );
+  }
+
+  return false;
+}
+
+export function getLanguageFromAlias(alias: string): Language {
+  if (alias === "ts" || alias === "typescript") {
+    return "typescript";
+  }
+  if (alias === "js" || alias === "javascript") {
+    return "javascript";
+  }
+
+  throw new Error(
+    `Invalid language ${cyan("(--language | -l)")}: ${red(alias)}. Options: ${blue("typescript, ts")} or ${yellow("javascript, js")}`,
+  );
+}
+
+export function validateIndexerId(indexerId?: string, throwError = false) {
+  if (!indexerId) {
+    return false;
+  }
+  if (!/^[a-z0-9-]+$/.test(indexerId)) {
+    if (throwError) {
+      throw new Error(
+        `Invalid indexer ID ${cyan("(--indexer-id)")}: ${red(indexerId)}. Indexer ID must contain only lowercase letters, numbers, and hyphens.`,
+      );
+    }
+    return false;
+  }
+  return true;
+}
+
+export function validateChain(chain?: string, throwError = false) {
+  if (!chain) {
+    return false;
+  }
+  if (chain) {
+    if (chain === "starknet" || chain === "ethereum" || chain === "beaconchain")
+      return true;
+    if (throwError) {
+      throw new Error(
+        `Invalid chain ${cyan("(--chain)")}: ${red(chain)}. Chain must be one of ${blue("starknet, ethereum, beaconchain")}.`,
+      );
+    }
+    return false;
+  }
+  return true;
+}
+
+export function validateNetwork(
+  chain?: string,
+  network?: string,
+  throwError = false,
+) {
+  if (!network) {
+    return false;
+  }
+
+  if (network === "other") {
+    return true;
+  }
+
+  if (chain) {
+    if (chain === "starknet") {
+      if (network === "mainnet" || network === "sepolia") {
+        return true;
+      }
+      if (throwError) {
+        throw new Error(
+          `Invalid network ${cyan("(--network)")}: ${red(network)}. For chain ${blue("starknet")}, network must be one of ${blue("mainnet, sepolia, other")}.`,
+        );
+      }
+      return false;
+    }
+    if (chain === "ethereum") {
+      if (network === "mainnet" || network === "goerli") {
+        return true;
+      }
+      if (throwError) {
+        throw new Error(
+          `Invalid network ${cyan("(--network)")}: ${red(network)}. For chain ${blue("ethereum")}, network must be one of ${blue("mainnet, goerli, other")}.`,
+        );
+      }
+      return false;
+    }
+    if (chain === "beaconchain") {
+      if (network === "mainnet") {
+        return true;
+      }
+      if (throwError) {
+        throw new Error(
+          `Invalid network ${cyan("(--network)")}: ${red(network)}. For chain ${blue("beaconchain")}, network must be ${blue("mainnet, other")}.`,
+        );
+      }
+      return false;
+    }
+  }
+
+  if (networks.find((n) => n.name === network)) {
+    return true;
+  }
+
+  if (throwError) {
+    throw new Error(
+      `Invalid network ${cyan("(--network)")}: ${red(network)}. Network must be one of ${blue("mainnet, sepolia, goerli, other")}.`,
+    );
+  }
+  return false;
+}
+
+export function validateStorage(storage?: string, throwError = false) {
+  if (!storage) {
+    return false;
+  }
+  if (storage === "postgres" || storage === "none") {
+    return true;
+  }
+  if (throwError) {
+    throw new Error(
+      `Invalid storage ${cyan("(--storage)")}: ${red(storage)}. Storage must be one of ${blue("postgres, none")}.`,
+    );
+  }
+  return false;
+}
+
+export function validateDnaUrl(dnaUrl?: string, throwError = false) {
+  if (!dnaUrl) {
+    return false;
+  }
+  if (!dnaUrl.startsWith("https://") && !dnaUrl.startsWith("http://")) {
+    if (throwError) {
+      throw new Error(
+        `Invalid DNA URL ${cyan("(--dna-url)")}: ${red(dnaUrl)}. DNA URL must start with ${blue("https:// or http://")}.`,
+      );
+    }
+    return false;
+  }
+  return true;
+}
+
+export function hasApibaraConfig(cwd: string): boolean {
+  const configPathJS = path.join(cwd, "apibara.config.js");
+  const configPathTS = path.join(cwd, "apibara.config.ts");
+
+  return fs.existsSync(configPathJS) || fs.existsSync(configPathTS);
+}
+
+export function getApibaraConfigLanguage(cwd: string): Language {
+  const configPathJS = path.join(cwd, "apibara.config.js");
+  const configPathTS = path.join(cwd, "apibara.config.ts");
+
+  if (fs.existsSync(configPathJS)) {
+    return "javascript";
+  }
+  if (fs.existsSync(configPathTS)) {
+    return "typescript";
+  }
+
+  throw new Error(red("✖") + " No apibara.config found");
+}
+
+export function getDnaUrl(chain: Chain, network: Network) {
+  if (chain === "ethereum") {
+    if (network === "mainnet") {
+      return dnaUrls.ethereum;
+    }
+    if (network === "sepolia") {
+      return dnaUrls.ethereumSepolia;
+    }
+  }
+
+  if (chain === "beaconchain") {
+    if (network === "mainnet") {
+      return dnaUrls.beaconchain;
+    }
+  }
+
+  if (chain === "starknet") {
+    if (network === "mainnet") {
+      return dnaUrls.starknet;
+    }
+    if (network === "sepolia") {
+      return dnaUrls.starknetSepolia;
+    }
+  }
+
+  throw new Error(red("✖") + " Invalid chain or network");
+}
+
+/**
+ * Converts a kebab-case string to camelCase.
+ *
+ * Examples:
+ * - "hello-world" → "helloWorld"
+ * - "my-long-variable-name" → "myLongVariableName"
+ * - "MY-CAPS" → "myCaps"
+ * - "-leading-dash" → "leadingDash"
+ * - "trailing-dash-" → "trailingDash"
+ * - "double--dash" → "doubleDash"
+ * - "hello---world" → "helloWorld"
+ * - "mixed_dash-and_underscore" → "mixedDashAndUnderscore"
+ *
+ * @param str The kebab-case string to convert
+ * @returns The camelCase version of the string
+ */
+export function convertKebabToCamelCase(_str: string): string {
+  let str = _str;
+
+  // Handle empty or invalid input
+  if (!str || typeof str !== "string") {
+    return "";
+  }
+
+  // Check if already camelCase
+  if (/^[a-z][a-zA-Z0-9]*$/.test(str)) {
+    return str;
+  }
+
+  // Trim leading/trailing dashes and spaces
+  str = str.trim().replace(/^-+|-+$/g, "");
+
+  // Handle empty string after trim
+  if (!str) {
+    return "";
+  }
+
+  return (
+    str
+      // Replace multiple consecutive dashes/underscores with a single dash
+      .replace(/[-_]+/g, "-")
+      // Split on dash
+      .split("-")
+      // Filter out empty strings (from consecutive dashes)
+      .filter(Boolean)
+      // Convert each word
+      .map((word, index) => {
+        // Convert word to lowercase
+        const _word = word.toLowerCase();
+
+        // Capitalize first letter if not the first word
+        if (index > 0) {
+          return _word.charAt(0).toUpperCase() + _word.slice(1);
+        }
+
+        return _word;
+      })
+      .join("")
+  );
+}
+
+export async function checkFileExists(
+  path: string,
+  options?: {
+    askPrompt?: boolean;
+    fileName?: string;
+    allowIgnore?: boolean;
+  },
+): Promise<{
+  exists: boolean;
+  overwrite: boolean;
+}> {
+  const { askPrompt = false, fileName, allowIgnore = false } = options ?? {};
+
+  if (!fs.existsSync(path)) {
+    return {
+      exists: false,
+      overwrite: false,
+    };
+  }
+
+  if (askPrompt) {
+    const { overwrite } = await prompts({
+      type: "select",
+      name: "overwrite",
+      message: `${fileName ?? basename(path)} already exists. Please choose how to proceed:`,
+      initial: 0,
+      choices: [
+        ...(allowIgnore
+          ? [
+              {
+                title: "Keep original file",
+                value: "ignore",
+              },
+            ]
+          : []),
+        {
+          title: "Cancel operation",
+          value: "no",
+        },
+        {
+          title: "Overwrite file",
+          value: "yes",
+        },
+      ],
+    });
+
+    if (overwrite === "no") {
+      cancelOperation();
+    }
+
+    if (overwrite === "ignore") {
+      return {
+        exists: true,
+        overwrite: false,
+      };
+    }
+
+    return {
+      exists: true,
+      overwrite: true,
+    };
+  }
+
+  return {
+    exists: true,
+    overwrite: false,
+  };
+}
+
+export function cancelOperation(message?: string) {
+  throw new Error(red("✖") + (message ?? " Operation cancelled"));
+}
+
+export function getPackageManager(): PkgInfo {
+  const userAgent = process.env.npm_config_user_agent;
+  const pkgInfo = pkgFromUserAgent(userAgent);
+  if (pkgInfo) {
+    return pkgInfo;
+  }
+  return {
+    name: "npm",
+  };
+}
+
+/**
+ 
+https://github.com/vitejs/vite/blob/07091a1e804e5934208ef0b6324a04317dd0d815/packages/create-vite/src/index.ts#L585
+
+MIT License
+
+Copyright (c) 2019-present, VoidZero Inc. and Vite contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+
+ */
+function pkgFromUserAgent(userAgent: string | undefined): PkgInfo | undefined {
+  if (!userAgent) return undefined;
+  const pkgSpec = userAgent.split(" ")[0];
+  const pkgSpecArr = pkgSpec.split("/");
+  return {
+    name: pkgSpecArr[0],
+    version: pkgSpecArr[1],
+  };
+}
+
+export async function formatFile(path: string) {
+  const file = fs.readFileSync(path, "utf8");
+  const formatted = await prettier.format(file, {
+    filepath: path,
+    tabWidth: 2,
+  });
+  fs.writeFileSync(path, formatted);
+}

package/src/rollup/config.ts

@@ -1,17 +1,19 @@
-import { existsSync } from "node:fs";
 import { builtinModules } from "node:module";
 import commonjs from "@rollup/plugin-commonjs";
 import json from "@rollup/plugin-json";
 import { nodeResolve } from "@rollup/plugin-node-resolve";
-import typescript from "@rollup/plugin-typescript";
 import type { Apibara, RollupConfig } from "apibara/types";
-import fsExtra from "fs-extra";
-import { basename, join } from "pathe";
+import { join } from "pathe";
+import type { OutputPluginOption } from "rollup";
 
-export const getRollupConfig = (
-  apibara: Apibara,
-  dev = false,
-): RollupConfig => {
+import defu from "defu";
+import { appConfig } from "./plugins/config";
+import { esmShim } from "./plugins/esm-shim";
+import { indexers } from "./plugins/indexers";
+
+const runtimeDependencies = ["better-sqlite3", "@electric-sql/pglite"];
+
+export function getRollupConfig(apibara: Apibara): RollupConfig {
   const extensions: string[] = [
     ".ts",
     ".mjs",
@@ -22,188 +24,64 @@ export const getRollupConfig = (
     ".jsx",
   ];
 
-  const indexerDir = join(apibara.options.rootDir, "indexers");
-  const configPath = join(apibara.options.rootDir, "./apibara.config.ts");
-
-  // Check if the indexers directory and config file exist
-  if (!existsSync(indexerDir)) {
-    throw new Error(`Indexers directory not found: ${indexerDir}`);
-  }
-  if (!existsSync(configPath)) {
-    throw new Error(`Config file not found: ${configPath}`);
-  }
-
-  // Check if the indexers directory exists and is not empty
-  let indexerFiles: string[] = [];
-  try {
-    indexerFiles = fsExtra
-      .readdirSync(indexerDir)
-      .filter((file) => file.endsWith(".indexer.ts"));
-    if (indexerFiles.length === 0) {
-      console.warn(`No indexer files found in ${indexerDir}`);
-    }
-  } catch (error) {
-    console.error(`Error reading indexers directory: ${error}`);
-  }
-
-  const indexerImports = indexerFiles
-    .map(
-      (file, index) =>
-        `import indexer${index} from '${join(indexerDir, file)}';`,
-    )
-    .join("\n");
-
-  // Generate main.ts content
-  const mainContent = `
-import { createClient } from "@apibara/protocol";
-import { createIndexer, run } from "@apibara/indexer";
-import { consola as _consola } from "apibara/internal/consola";
-import { defineCommand, runMain } from "apibara/internal/citty";
-import config from './${configPath}';
-
-const consola = _consola.withTag("Apibara | ");
-
-${indexerImports}
-
-const indexers = {
-  ${indexerFiles
-    .map(
-      (file, index) => `'${basename(file, ".indexer.ts")}': indexer${index},`,
-    )
-    .join("\n  ")}
-};
-
-const command = defineCommand({
-  meta: {
-    name: "run-indexers",
-    description: "Run Apibara indexers",
-  },
-  args: {
-    indexers: {
-      type: "string",
-      description: "Comma-separated list of indexers to run",
-    },
-    preset: {
-      type: "string",
-      description: "Preset to use",
-    },
-    sink: {
-      type: "string",
-      description: "Sink to use",
-    },
-  },
-  async run({ args }) {
-    const selectedIndexers = args.indexers ? args.indexers.split(',') : Object.keys(indexers);
-    const preset = args.preset || config.preset || 'default';
-    const sinkName = args.sink || 'default';
-
-    // Apply preset
-    let runtimeConfig = { ...config.runtimeConfig };
-    if (preset && config.presets && config.presets[preset]) {
-      runtimeConfig = { ...runtimeConfig, ...config.presets[preset].runtimeConfig };
-    }
-
-    // Get sink function
-    const sinkFunction = config.sink?.[sinkName];
-    if (!sinkFunction) {
-      throw new Error(\`Sink \${sinkName} not found\`);
-    }
-
-    await Promise.all(selectedIndexers.map(async (name) => {
-      if (!indexers[name]) {
-        console.error(\`Indexer \${name} not found\`);
-        return;
-      }
-
-      const indexerConfig = indexers[name]
-      const indexer = typeof indexerConfig === 'function'
-        ? await createIndexer(indexerConfig(runtimeConfig))
-        : createIndexer(indexerConfig);
-
-      const client = createClient(indexer.streamConfig, indexer.options.streamUrl);
-      const sink = sinkFunction();
-
-      try {
-        consola.log("Running Indexer: ", name);
-        await run(client, indexer, sink);
-      } catch (error) {
-        consola.error(\`Error in indexer \${name}:\`, error);
-      }
-    }));
-  },
-});
-
-runMain(command);
-`;
-
-  return {
-    input: {
-      main: "virtual:main.ts",
-    },
-    output: {
-      dir: join(apibara.options.outputDir || "./.apibara/build"),
-      format: "esm",
-      exports: "auto",
-      entryFileNames: "[name].mjs",
-      chunkFileNames: "chunks/[name]-[hash].mjs",
-      generatedCode: {
-        constBindings: true,
-      },
-      sourcemap: true,
-      sourcemapExcludeSources: true,
-      sourcemapIgnoreList(relativePath, sourcemapPath) {
-        return relativePath.includes("node_modules");
-      },
-    },
-    plugins: [
-      {
-        name: "virtual",
-        resolveId(id) {
-          if (id === "virtual:main.ts") {
-            return id;
-          }
-          return null;
+  const rollupConfig: RollupConfig & { plugins: OutputPluginOption[] } = defu(
+    // biome-ignore lint/suspicious/noExplicitAny: apibara.options.rollupConfig is typed
+    apibara.options.rollupConfig as any,
+    <RollupConfig>{
+      input: apibara.options.entry,
+      output: {
+        dir: join(apibara.options.outputDir || "./.apibara/build"),
+        format: "esm",
+        exports: "auto",
+        entryFileNames: "[name].mjs",
+        chunkFileNames: "chunks/[name]-[hash].mjs",
+        generatedCode: {
+          constBindings: true,
         },
-        load(id) {
-          if (id === "virtual:main.ts") {
-            return mainContent;
-          }
-          return null;
+        sourcemap: true,
+        sourcemapExcludeSources: true,
+        sourcemapIgnoreList(relativePath, sourcemapPath) {
+          return relativePath.includes("node_modules");
         },
       },
-      nodeResolve({
-        extensions,
-        preferBuiltins: true,
-      }),
-      commonjs(),
-      json(),
-      typescript({
-        tsconfig: join(apibara.options.rootDir, "./tsconfig.json"),
-        compilerOptions: {
-          outDir: join(apibara.options.outputDir || "./.apibara/build"),
-          declarationDir: join(apibara.options.outputDir || "./.apibara/build"),
-          noEmit: false,
-          types: ["node"],
-        },
-      }),
-    ],
-    onwarn(warning, rollupWarn) {
-      if (
-        !["CIRCULAR_DEPENDENCY", "EVAL"].includes(warning.code || "") &&
-        !warning.message.includes("Unsupported source map comment") &&
-        !warning.message.includes("@__PURE__")
-      ) {
-        rollupWarn(warning);
-      }
+      plugins: [],
+      onwarn(warning, rollupWarn) {
+        if (
+          !["CIRCULAR_DEPENDENCY", "EVAL", "THIS_IS_UNDEFINED"].includes(
+            warning.code || "",
+          ) &&
+          !warning.message.includes("Unsupported source map comment") &&
+          !warning.message.includes("@__PURE__") &&
+          !warning.message.includes("/*#__PURE__*/")
+        ) {
+          rollupWarn(warning);
+        }
+      },
+      treeshake: true,
+      external: [...builtinModules, ...runtimeDependencies],
     },
-    treeshake: true,
-    external: [
-      ...builtinModules,
-      "@apibara/indexer",
-      "@apibara/protocol",
-      "@apibara/evm",
-      "@apibara/starknet",
-      "@apibara/beaconchain",
-    ],
-  };
-};
+  );
+
+  rollupConfig.plugins.push(esmShim());
+  rollupConfig.plugins.push(json());
+
+  rollupConfig.plugins.push(
+    commonjs({
+      strictRequires: true,
+      requireReturnsDefault: "auto",
+      ...apibara.options.commonJS,
+    }),
+  );
+
+  rollupConfig.plugins.push(
+    nodeResolve({
+      extensions,
+      preferBuiltins: true,
+      mainFields: ["main"],
+    }),
+  );
+  rollupConfig.plugins.push(indexers(apibara));
+  rollupConfig.plugins.push(appConfig(apibara));
+
+  return rollupConfig;
+}

package/src/rollup/index.ts

@@ -1 +1,2 @@
 export * from "./config";
+export type { Plugin } from "rollup";