fix(migrations): Preserve host state adapter identity

Pass the real core state adapter into TypeScript migrations so Redis-native index and lock detection continues to work.

Add multiline import guard coverage and remove the stale generated temp-path comment from the history migration.

Co-Authored-By: GPT-5.6 Codex <noreply@openai.com>

Author: dcramer

packages/junior-migrations/src/journal.ts

@@ -86,7 +86,7 @@ function validateTypeScriptSource(tag: string, source: string): void {
     }
   };
   const imports = source.matchAll(
-    /^\s*import\s+([^;]+?)\s+from\s+["']([^"']+)["'];?/gm,
+    /^\s*import\s+([\s\S]*?)\s+from\s+["']([^"']+)["'];?/gm,
   );
   for (const match of imports) {
     const clause = match[1]?.trim();
@@ -97,7 +97,7 @@ function validateTypeScriptSource(tag: string, source: string): void {
     validateRuntimeSpecifier(specifier);
   }
   const exports = source.matchAll(
-    /^\s*export\s+([^;]+?)\s+from\s+["']([^"']+)["'];?/gm,
+    /^\s*export\s+([\s\S]*?)\s+from\s+["']([^"']+)["'];?/gm,
   );
   for (const match of exports) {
     const clause = match[1]?.trim();

packages/junior-migrations/tests/journal.test.ts

@@ -62,6 +62,18 @@ describe("resolveMigrations", () => {
     );
   });
 
+  it("rejects multiline imports of Junior runtime modules", async () => {
+    const folder = await migrationFolder(["0000_multiline"]);
+    await writeFile(
+      join(folder, "0000_multiline.ts"),
+      'import {\n  getChatConfig,\n  getDb,\n} from "@sentry/junior/internal";\nexport default { apiVersion: 1, async up() {} };\n',
+    );
+
+    await expect(resolveMigrations(folder)).rejects.toThrow(
+      "cannot import application runtime code",
+    );
+  });
+
   it("allows versioned Junior migration helpers", async () => {
     const folder = await migrationFolder(["0000_helpers"]);
     await writeFile(

packages/junior/migrations/0008_conversation_history_to_sql.ts

@@ -1034,7 +1034,6 @@ async function importConversationFromLegacy(conversationId, deps) {
   return { imported };
 }
 
-// ../../../../../../private/tmp/0008-entry.ts
 import {
   createMigrationSqlConversationMessageStore,
   createMigrationStateConversationStore,

packages/junior/src/chat/conversations/sql/migrations.ts

@@ -10,7 +10,6 @@ import {
 } from "@sentry/junior-migrations";
 import type { StateAdapter } from "chat";
 import type { RedisStateAdapter } from "@chat-adapter/state-redis";
-import { createMigrationStateV1 } from "@/chat/migrations/state-v1";
 import type { JuniorSqlMigrationExecutor } from "@/db/db";
 import { juniorSqlSchema as schema } from "@/db/schema";
 
@@ -155,7 +154,7 @@ export async function migrateSchema(
             },
           }
         : {}),
-      state: createMigrationStateV1(options.stateAdapter),
+      state: options.stateAdapter,
     }),
     loadTypeScript: options.loadTypeScript,
     mode: "all",

packages/junior/src/chat/migrations/state-v1.ts

@@ -2,6 +2,7 @@ import { mkdtempSync, rmSync, writeFileSync } from "node:fs";
 import { tmpdir } from "node:os";
 import path from "node:path";
 import type { RedisStateAdapter } from "@chat-adapter/state-redis";
+import type { MigrationContextV1 } from "@sentry/junior-migrations";
 import { createJiti } from "jiti";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 import { getChatConfig } from "@/chat/config";
@@ -379,6 +380,35 @@ export const plugins = {
     }
   }, 15_000);
 
+  it("passes the host state adapter through without wrapping its identity", async () => {
+    const stateAdapter = getStateAdapter();
+    await stateAdapter.connect();
+    const fixture = await createLocalJuniorSqlFixture();
+    const observedStates: unknown[] = [];
+
+    try {
+      await migrateSchema(fixture.sql, {
+        loadTypeScript: async () => ({
+          default: {
+            apiVersion: 1,
+            async up(context: MigrationContextV1) {
+              observedStates.push(context.state);
+            },
+          },
+        }),
+        mode: "all",
+        stateAdapter,
+      });
+
+      expect(observedStates).toHaveLength(5);
+      expect(observedStates.every((state) => state === stateAdapter)).toBe(
+        true,
+      );
+    } finally {
+      await fixture.close();
+    }
+  });
+
   it("migrates legacy conversation work before SQL conversation backfill", async () => {
     const stateAdapter = getStateAdapter();
     await stateAdapter.connect();