apibara 2.0.0-beta.4 → 2.0.0-beta.40

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,187 +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 from "consola";
-import { defineCommand, runMain } from "citty";
-import config from './${configPath}';
-
-${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",
-      "@apibara/cli",
-    ],
-  };
-};
+  );
+
+  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/plugins/config.ts

@@ -0,0 +1,12 @@
+import virtual from "@rollup/plugin-virtual";
+import type { Apibara } from "apibara/types";
+
+export function appConfig(apibara: Apibara) {
+  return virtual({
+    "#apibara-internal-virtual/config": `
+    import * as projectConfig from '${apibara.options._c12.configFile}';
+
+    export const config = projectConfig.default;
+    `,
+  });
+}

package/src/rollup/plugins/esm-shim.ts

@@ -0,0 +1,69 @@
+import MagicString from "magic-string";
+import type { Plugin } from "rollup";
+
+/**
+ * An alternative to @rollup/plugin-esm-shim
+ */
+export function esmShim(): Plugin {
+  const ESMShim = `
+// -- Shims --
+import cjsUrl from 'node:url';
+import cjsPath from 'node:path';
+const __filename = cjsUrl.fileURLToPath(import.meta.url);
+const __dirname = cjsPath.dirname(__filename);
+// -- End Shims --
+`;
+
+  const CJSyntaxRegex = /__filename|__dirname/;
+
+  return {
+    name: "esm-shim",
+
+    renderChunk(code, _chunk, opts) {
+      if (opts.format === "es") {
+        if (code.includes(ESMShim) || !CJSyntaxRegex.test(code)) {
+          return null;
+        }
+
+        let endIndexOfLastImport = -1;
+
+        // Find the last import statement and its ending index
+        for (const match of code.matchAll(/^import\s.*';$/gm)) {
+          if (match.length === 0 || typeof match.index !== "number") {
+            continue;
+          }
+
+          endIndexOfLastImport = match.index + match[0].length;
+        }
+
+        const s = new MagicString(code);
+        s.appendRight(endIndexOfLastImport, ESMShim);
+
+        const sourceMap = s.generateMap({
+          includeContent: true,
+        });
+
+        let sourcesContent: string[] | undefined;
+        if (Array.isArray(sourceMap.sourcesContent)) {
+          sourcesContent = [];
+          for (let i = 0; i < sourceMap.sourcesContent.length; i++) {
+            const content = sourceMap.sourcesContent[i];
+            if (typeof content === "string") {
+              sourcesContent.push(content);
+            }
+          }
+        }
+
+        return {
+          code: s.toString(),
+          map: {
+            ...sourceMap,
+            sourcesContent,
+          },
+        };
+      }
+
+      return null;
+    },
+  };
+}

package/src/rollup/plugins/indexers.ts

@@ -0,0 +1,17 @@
+import virtual from "@rollup/plugin-virtual";
+import type { Apibara } from "apibara/types";
+import { hash } from "ohash";
+
+export function indexers(apibara: Apibara) {
+  const indexers = [...new Set(apibara.indexers)];
+
+  return virtual({
+    "#apibara-internal-virtual/indexers": `
+    ${indexers.map((i) => `import _${hash(i)} from '${i.indexer}';`).join("\n")}
+
+    export const indexers = [
+      ${indexers.map((i) => `{ name: "${i.name}", indexer: _${hash(i)} }`).join(",\n")}
+    ];
+    `,
+  });
+}

package/src/runtime/dev.ts

@@ -0,0 +1,64 @@
+import { runWithReconnect } from "@apibara/indexer";
+import { createClient } from "@apibara/protocol";
+import { defineCommand, runMain } from "citty";
+import { availableIndexers, createIndexer } from "./internal/app";
+
+const startCommand = defineCommand({
+  meta: {
+    name: "start",
+    description: "Start the indexer",
+  },
+  args: {
+    indexers: {
+      type: "string",
+      description: "Which indexers to run",
+    },
+    preset: {
+      type: "string",
+      description: "Preset to use",
+    },
+  },
+  async run({ args }) {
+    const { indexers: indexersArgs, preset } = args;
+
+    let selectedIndexers = availableIndexers;
+    if (indexersArgs) {
+      selectedIndexers = indexersArgs.split(",");
+    }
+
+    for (const indexer of selectedIndexers) {
+      if (!availableIndexers.includes(indexer)) {
+        throw new Error(
+          `Specified indexer "${indexer}" but it was not defined`,
+        );
+      }
+    }
+
+    await Promise.all(
+      selectedIndexers.map(async (indexer) => {
+        const indexerInstance = createIndexer(indexer, preset);
+
+        const client = createClient(
+          indexerInstance.streamConfig,
+          indexerInstance.options.streamUrl,
+        );
+
+        await runWithReconnect(client, indexerInstance);
+      }),
+    );
+  },
+});
+
+export const mainCli = defineCommand({
+  meta: {
+    name: "indexer-dev-runner",
+    description: "Run indexer in dev mode",
+  },
+  subCommands: {
+    start: () => startCommand,
+  },
+});
+
+runMain(mainCli);
+
+export default {};

package/src/runtime/index.ts

@@ -0,0 +1,2 @@
+export { createIndexer } from "./internal/app";
+export { createLogger } from "./internal/logger";

package/src/runtime/internal/app.ts

@@ -0,0 +1,78 @@
+import { createIndexer as _createIndexer } from "@apibara/indexer";
+import {
+  type InternalContext,
+  internalContext,
+} from "@apibara/indexer/internal/plugins";
+import {
+  type ConsolaReporter,
+  inMemoryPersistence,
+  logger,
+} from "@apibara/indexer/plugins";
+import { config } from "#apibara-internal-virtual/config";
+import { indexers } from "#apibara-internal-virtual/indexers";
+import { createLogger } from "./logger";
+
+export const availableIndexers = indexers.map((i) => i.name);
+
+export function createIndexer(indexerName: string, preset?: string) {
+  let runtimeConfig: Record<string, unknown> = { ...config.runtimeConfig };
+
+  if (preset) {
+    if (config.presets === undefined) {
+      throw new Error(
+        `Specified preset "${preset}" but no presets were defined`,
+      );
+    }
+
+    if (config.presets[preset] === undefined) {
+      throw new Error(`Specified preset "${preset}" but it was not defined`);
+    }
+
+    const presetValue = config.presets[preset] as {
+      runtimeConfig: Record<string, unknown>;
+    };
+    runtimeConfig = { ...runtimeConfig, ...presetValue.runtimeConfig };
+  }
+
+  const indexerDefinition = indexers.find((i) => i.name === indexerName);
+
+  if (indexerDefinition === undefined) {
+    throw new Error(
+      `Specified indexer "${indexerName}" but it was not defined`,
+    );
+  }
+
+  const definition =
+    typeof indexerDefinition.indexer === "function"
+      ? indexerDefinition.indexer(runtimeConfig)
+      : indexerDefinition.indexer;
+
+  let reporter: ConsolaReporter = createLogger({
+    indexer: indexerName,
+    preset,
+    indexers: availableIndexers,
+  });
+
+  if (config.logger) {
+    reporter = config.logger({
+      indexer: indexerName,
+      preset,
+      indexers: availableIndexers,
+    });
+  }
+
+  // Put the in-memory persistence plugin first so that it can be overridden by any user-defined
+  // persistence plugin.
+  // Put the logger last since we want to override any user-defined logger.
+  definition.plugins = [
+    internalContext({
+      indexerName,
+      availableIndexers,
+    } as InternalContext),
+    inMemoryPersistence(),
+    ...(definition.plugins ?? []),
+    logger({ logger: reporter }),
+  ];
+
+  return _createIndexer(definition);
+}

package/src/runtime/internal/logger.ts

@@ -0,0 +1,70 @@
+import type {
+  ConsolaOptions,
+  ConsolaReporter,
+  LogLevel,
+  LogObject,
+  LogType,
+} from "consola";
+import { type ColorName, colors, getColor } from "consola/utils";
+import { murmurHash } from "ohash";
+
+const INDEXER_COLOR_MAP = [
+  colors.red,
+  colors.green,
+  colors.yellow,
+  colors.blue,
+  colors.magenta,
+  colors.cyan,
+];
+
+const TYPE_COLOR_MAP: { [k in LogType]?: string } = {
+  info: "cyan",
+  fail: "red",
+  success: "green",
+  ready: "green",
+  start: "magenta",
+};
+
+const LEVEL_COLOR_MAP: { [k in LogLevel]?: string } = {
+  0: "red",
+  1: "yellow",
+};
+
+const MAX_INDEXER_NAME_LENGTH = 20;
+
+class DefaultReporter implements ConsolaReporter {
+  private tag: string;
+
+  constructor(indexer: string, indexers: string[], preset?: string) {
+    const color =
+      INDEXER_COLOR_MAP[murmurHash(indexer) % INDEXER_COLOR_MAP.length];
+
+    const presetLength = preset ? preset.length : 0;
+
+    const longestIndexerName =
+      Math.max(...indexers.map((i) => i.length), indexer.length) + presetLength;
+
+    const paddedIndexer = `${indexer}${preset ? `:${preset} ` : ""}`
+      .padEnd(longestIndexerName, " ")
+      .slice(0, Math.min(longestIndexerName, MAX_INDEXER_NAME_LENGTH));
+
+    this.tag = color(`${paddedIndexer} |`);
+  }
+
+  log(logObj: LogObject, ctx: { options: ConsolaOptions }) {
+    const { args } = logObj;
+    const typeColor =
+      TYPE_COLOR_MAP[logObj.type] || LEVEL_COLOR_MAP[logObj.level] || "gray";
+
+    const type = getColor(typeColor as ColorName, "white")(logObj.type);
+    console.log(`${this.tag} ${type}`, ...args);
+  }
+}
+
+export function createLogger({
+  indexer,
+  indexers,
+  preset,
+}: { indexer: string; indexers: string[]; preset?: string }) {
+  return new DefaultReporter(indexer, indexers, preset);
+}

package/src/runtime/start.ts

@@ -0,0 +1,48 @@
+import { runWithReconnect } from "@apibara/indexer";
+import { createClient } from "@apibara/protocol";
+import { defineCommand, runMain } from "citty";
+import { createIndexer } from "./internal/app";
+
+const startCommand = defineCommand({
+  meta: {
+    name: "start",
+    description: "Start the indexer",
+  },
+  args: {
+    indexer: {
+      type: "string",
+      description: "Indexer name",
+      required: true,
+    },
+    preset: {
+      type: "string",
+      description: "Preset to use",
+    },
+  },
+  async run({ args }) {
+    const { indexer, preset } = args;
+
+    const indexerInstance = createIndexer(indexer, preset);
+
+    const client = createClient(
+      indexerInstance.streamConfig,
+      indexerInstance.options.streamUrl,
+    );
+
+    await runWithReconnect(client, indexerInstance);
+  },
+});
+
+export const mainCli = defineCommand({
+  meta: {
+    name: "indexer-runner",
+    description: "Run an indexer",
+  },
+  subCommands: {
+    start: () => startCommand,
+  },
+});
+
+runMain(mainCli);
+
+export default {};

package/src/types/apibara.ts

@@ -3,9 +3,17 @@ import type { Hookable } from "hookable";
 import type { ApibaraDynamicConfig, ApibaraOptions } from "./config";
 import type { ApibaraHooks } from "./hooks";
 
+export type IndexerDefinition = {
+  // Name of the indexer.
+  name: string;
+  // Path to the indexer file.
+  indexer: string;
+};
+
 export interface Apibara {
   options: ApibaraOptions;
   hooks: Hookable<ApibaraHooks>;
+  indexers: IndexerDefinition[];
   logger: ConsolaInstance;
   close: () => Promise<void>;
   updateConfig: (config: ApibaraDynamicConfig) => void | Promise<void>;