chore(references): migrate ai-chat to onBoot

Author: ericallam

references/ai-chat/src/trigger/chat.ts

@@ -267,9 +267,13 @@ export const aiChat = chat
 
     // --- Lifecycle hooks ---
 
-    // #region onPreload — eagerly initialize before the user's first message
-    onPreload: async ({ chatId, runId, chatAccessToken, clientData }) => {
-      if (!clientData) return;
+    // #region onBoot — per-process setup that runs on EVERY fresh worker
+    //
+    // Fires for the initial run, preloaded runs, AND reactive continuation
+    // runs (post-cancel/crash/endRun/upgrade). The single place to initialize
+    // `chat.local` and per-process resources so they're ready in `run()`
+    // regardless of how the run was triggered.
+    onBoot: async ({ clientData }) => {
       const user = await prisma.user.upsert({
         where: { id: clientData.userId },
         create: { id: clientData.userId, name: "User" },
@@ -289,13 +293,18 @@ export const aiChat = chat
       });
       chat.prompt.set(resolved);
       chat.skills.set([await timeUtilsSkill.local()]);
+    },
+    // #endregion
 
+    // #region onPreload — eagerly create chat/session DB rows before the first message
+    onPreload: async ({ chatId, chatAccessToken, clientData }) => {
+      if (!clientData) return;
       await prisma.chat.upsert({
         where: { id: chatId },
         create: {
           id: chatId,
           title: "New chat",
-          userId: user.id,
+          userId: clientData.userId,
           model: clientData?.model ?? DEFAULT_MODEL,
         },
         update: {},
@@ -308,50 +317,24 @@ export const aiChat = chat
     },
     // #endregion
 
-    // #region onChatStart — fallback init when not preloaded
-    onChatStart: async ({
-      chatId,
-      runId,
-      chatAccessToken,
-      clientData,
-      continuation,
-      preloaded,
-    }) => {
+    // #region onChatStart — first-message chat/session DB rows when not preloaded
+    //
+    // Fires once per chat (on the very first message of the chat's lifetime).
+    // Per-process state initialization lives in `onBoot`; this hook is only
+    // for chat-scoped DB work that's a no-op on continuation runs.
+    onChatStart: async ({ chatId, chatAccessToken, clientData, preloaded }) => {
       if (preloaded) return;
 
-      const user = await prisma.user.upsert({
-        where: { id: clientData.userId },
-        create: { id: clientData.userId, name: "User" },
+      await prisma.chat.upsert({
+        where: { id: chatId },
+        create: {
+          id: chatId,
+          title: "New chat",
+          userId: clientData.userId,
+          model: clientData.model ?? DEFAULT_MODEL,
+        },
         update: {},
       });
-      userContext.init({
-        userId: user.id,
-        name: user.name,
-        plan: user.plan as "free" | "pro",
-        preferredModel: user.preferredModel,
-        messageCount: user.messageCount,
-      });
-
-      const resolved = await systemPrompt.resolve({
-        name: user.name,
-        plan: user.plan as string,
-      });
-      chat.prompt.set(resolved);
-      chat.skills.set([await timeUtilsSkill.local()]);
-
-      if (!continuation) {
-        await prisma.chat.upsert({
-          where: { id: chatId },
-          create: {
-            id: chatId,
-            title: "New chat",
-            userId: user.id,
-            model: clientData.model ?? DEFAULT_MODEL,
-          },
-          update: {},
-        });
-      }
-
       await prisma.chatSession.upsert({
         where: { id: chatId },
         create: { id: chatId, publicAccessToken: chatAccessToken },