vibes

Author: ericallam

packages/cli-v3/src/build/bundle.ts

@@ -47,6 +47,7 @@ export type BundleResult = {
   runControllerEntryPoint: string | undefined;
   indexWorkerEntryPoint: string | undefined;
   indexControllerEntryPoint: string | undefined;
+  initEntryPoint: string | undefined;
   stop: (() => Promise<void>) | undefined;
 };
 
@@ -229,6 +230,7 @@ export async function getBundleResultFromBuild(
   let runControllerEntryPoint: string | undefined;
   let indexWorkerEntryPoint: string | undefined;
   let indexControllerEntryPoint: string | undefined;
+  let initEntryPoint: string | undefined;
 
   const configEntryPoint = resolvedConfig.configFile
     ? relative(resolvedConfig.workingDir, resolvedConfig.configFile)
@@ -254,6 +256,8 @@ export async function getBundleResultFromBuild(
         indexControllerEntryPoint = $outputPath;
       } else if (isIndexWorkerForTarget(outputMeta.entryPoint, target)) {
         indexWorkerEntryPoint = $outputPath;
+      } else if (outputMeta.entryPoint.endsWith("init.ts")) {
+        initEntryPoint = $outputPath;
       } else {
         if (
           !outputMeta.entryPoint.startsWith("..") &&
@@ -280,6 +284,7 @@ export async function getBundleResultFromBuild(
     runControllerEntryPoint,
     indexWorkerEntryPoint,
     indexControllerEntryPoint,
+    initEntryPoint,
     contentHash: hasher.digest("hex"),
   };
 }

packages/cli-v3/src/entryPoints/dev-run-worker.ts

@@ -270,6 +270,18 @@ const zodIpc = new ZodIpcConnection({
             async () => {
               const beforeImport = performance.now();
               resourceCatalog.setCurrentFileContext(taskManifest.entryPoint, taskManifest.filePath);
+
+              // Load init file if it exists
+              if (workerManifest.initEntryPoint) {
+                try {
+                  await import(normalizeImportPath(workerManifest.initEntryPoint));
+                  log(`Loaded init file from ${workerManifest.initEntryPoint}`);
+                } catch (err) {
+                  logError(`Failed to load init file`, err);
+                  throw err;
+                }
+              }
+
               await import(normalizeImportPath(taskManifest.entryPoint));
               resourceCatalog.clearCurrentFileContext();
               const durationMs = performance.now() - beforeImport;

packages/core/src/v3/schemas/build.ts

@@ -38,6 +38,7 @@ export const BuildManifest = z.object({
   indexWorkerEntryPoint: z.string(), // Dev & Deploy has a indexWorkerEntryPoint
   indexControllerEntryPoint: z.string().optional(), // Only deploy has a indexControllerEntryPoint
   loaderEntryPoint: z.string().optional(),
+  initEntryPoint: z.string().optional(), // Optional init.ts entry point
   configPath: z.string(),
   externals: BuildExternal.array().optional(),
   build: z.object({
@@ -85,6 +86,7 @@ export const WorkerManifest = z.object({
   workerEntryPoint: z.string(),
   controllerEntryPoint: z.string().optional(),
   loaderEntryPoint: z.string().optional(),
+  initEntryPoint: z.string().optional(), // Optional init.ts entry point
   runtime: BuildRuntime,
   customConditions: z.array(z.string()).optional(),
   otelImportHook: z