create-interview-cockpit 0.5.0 → 0.6.0

package/template/client/src/store.ts

@@ -1,8 +1,36 @@
 import { create } from "zustand";
-import type { Topic, Question, CodeSnippet, WorkspaceMeta } from "./types";
+import type {
+  Topic,
+  Question,
+  CodeSnippet,
+  WorkspaceMeta,
+  InfraLabWorkspace,
+  FrontendLabWorkspace,
+} from "./types";
 import type { AiSettings } from "./api";
 import * as api from "./api";
 
+// Default Express server used when opening a React/Next.js sandbox with no prior server code
+const DEFAULT_SERVER_CODE = `import express from 'express';
+const app = express();
+app.use(express.json());
+// Allow requests from the React preview iframe
+app.use((_req, res, next) => {
+  res.setHeader('Access-Control-Allow-Origin', '*');
+  res.setHeader('Access-Control-Allow-Methods', 'GET,POST,PUT,DELETE,OPTIONS');
+  res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
+  if (_req.method === 'OPTIONS') { res.sendStatus(204); return; }
+  next();
+});
+
+app.get('/api/hello', (_req, res) => {
+  res.json({ message: 'Hello from Express!', time: Date.now() });
+});
+
+const port = Number(process.env.PORT);
+app.listen(port, () => console.log('Server on :' + port));
+`;
+
 const DEFAULT_AI_SETTINGS: AiSettings = {
   systemPrompt: "",
   responseProfiles: {
@@ -11,6 +39,7 @@ const DEFAULT_AI_SETTINGS: AiSettings = {
     normal: { maxOutputTokens: 3000, maxSteps: 5 },
   },
   vizGuide: "",
+  plotGuide: "",
   alwaysSendPrefsDefault: false,
   thinkingBudget: 0,
   promptGroups: {
@@ -52,6 +81,19 @@ const DEFAULT_AI_SETTINGS: AiSettings = {
           "When using technical terms or abbreviations, immediately expand their meaning in square brackets right after the term — e.g. 'TCP [Transmission Control Protocol — a connection-oriented transport protocol]' or 'idempotent [an operation that produces the same result no matter how many times it is applied]'. Do this throughout your response so I never need to look anything up.",
       },
     },
+    "diagram-use": {
+      label: "Diagram Usage",
+      description: "Which visual formats to use and how often.",
+      default: "none",
+      options: {
+        none: "",
+        vizcraft:
+          "Prioritize using vizcraft (viz) diagrams as much as possible. *NB:* Character limits do not apply to these diagrams",
+        mermaid:
+          "Prioritize using mermaid diagrams as much as possible. *NB:* Character limits do not apply to these diagrams",
+        plot: "Prioritize using plot blocks for graphs, curves, distributions, and data charts when they would make the explanation clearer. *NB:* Character limits do not apply to these plots",
+      },
+    },
   },
 };
 
@@ -108,19 +150,30 @@ interface Store {
   createDriveSubfolder: (
     id: string,
     name: string,
-  ) => Promise<import("./api").DriveFolder>;
+  ) => Promise<
+    import("./api").DriveFolder | { needsAuth: true; authUrl: string }
+  >;
 
   fetchTopics: () => Promise<void>;
   fetchQuestions: (topicId: string) => Promise<void>;
   addTopic: (name: string) => Promise<void>;
   removeTopic: (id: string) => Promise<void>;
   renameTopic: (id: string, name: string) => Promise<void>;
+  updateTopicSystemContext: (
+    id: string,
+    systemContext: string,
+  ) => Promise<void>;
   addQuestion: (topicId: string, title: string) => Promise<void>;
   addChildQuestion: (
     topicId: string,
     parentQuestionId: string,
     title: string,
   ) => Promise<void>;
+  moveQuestion: (
+    questionId: string,
+    topicId: string,
+    parentQuestionId: string | null,
+  ) => Promise<void>;
   removeQuestion: (questionId: string, topicId: string) => Promise<void>;
   renameQuestion: (
     questionId: string,
@@ -159,10 +212,14 @@ interface Store {
     code: string,
     language: string,
     label: string,
-    origin: "user" | "ai" | "sandbox",
+    origin: "user" | "ai" | "sandbox" | "infra" | "react" | "nextjs",
   ) => Promise<import("./types").ContextFile>;
   clearMessages: (questionId: string) => Promise<void>;
 
+  // ── Linked Conversations ─────────────────────────────────────
+  linkConversation: (questionId: string, linkedId: string) => Promise<void>;
+  unlinkConversation: (questionId: string, linkedId: string) => Promise<void>;
+
   // ── Workspace Context Files ──────────────────────────────────
   workspaceFiles: import("./types").ContextFile[];
   fetchWorkspaceFiles: () => Promise<void>;
@@ -215,14 +272,25 @@ interface Store {
     clientCode: string;
     clientLang: string;
     fileId?: string;
+    /** If set, the client panel opens in React or Next.js preview mode instead of script mode */
+    clientType?: "script" | "react" | "nextjs";
+    reactFiles?: Record<string, string> | null;
+    reactActiveFile?: string | null;
   } | null;
-  openCodeRunner: (code?: string, language?: string) => void;
+  /** When set, the script runner opens with this file pre-loaded — Save will overwrite it */
+  runnerInitialFileId: string | null;
+  openCodeRunner: (code?: string, language?: string, fileId?: string) => void;
   openSandbox: (
     serverCode: string,
     serverLang: string,
     clientCode: string,
     clientLang: string,
     fileId?: string,
+    opts?: {
+      clientType?: "script" | "react" | "nextjs";
+      reactFiles?: Record<string, string>;
+      reactActiveFile?: string;
+    },
   ) => void;
   overwriteContextFileContent: (
     questionId: string,
@@ -235,6 +303,27 @@ interface Store {
     label: string,
   ) => Promise<void>;
   closeCodeRunner: () => void;
+
+  // ── Infra Lab ────────────────────────────────────────────────
+  showInfraLab: boolean;
+  runnerInitialInfra: InfraLabWorkspace | null;
+  runnerInitialInfraFileId: string | null;
+  openInfraLab: (workspace?: InfraLabWorkspace, fileId?: string) => void;
+  closeInfraLab: () => void;
+
+  // ── Frontend Labs (React / Next.js) — open inside the sandbox ──
+  openReactLab: (
+    workspace?: FrontendLabWorkspace,
+    fileId?: string,
+    serverCode?: string,
+    serverLang?: string,
+  ) => void;
+  openNextLab: (
+    workspace?: FrontendLabWorkspace,
+    fileId?: string,
+    serverCode?: string,
+    serverLang?: string,
+  ) => void;
 }
 
 export const useStore = create<Store>((set, get) => ({
@@ -259,6 +348,10 @@ export const useStore = create<Store>((set, get) => ({
   runnerInitialCode: "",
   runnerInitialLanguage: "typescript",
   runnerInitialSandbox: null,
+  runnerInitialFileId: null,
+  showInfraLab: false,
+  runnerInitialInfra: null,
+  runnerInitialInfraFileId: null,
 
   // ── Workspaces ───────────────────────────────────────────────
   workspaces: [],
@@ -448,6 +541,13 @@ export const useStore = create<Store>((set, get) => ({
     }));
   },
 
+  updateTopicSystemContext: async (id, systemContext) => {
+    await api.updateTopic(id, { systemContext });
+    set((s) => ({
+      topics: s.topics.map((t) => (t.id === id ? { ...t, systemContext } : t)),
+    }));
+  },
+
   addQuestion: async (topicId, title) => {
     const question = await api.createQuestion(topicId, title);
     set((s) => ({
@@ -468,6 +568,29 @@ export const useStore = create<Store>((set, get) => ({
     }));
   },
 
+  moveQuestion: async (questionId, topicId, parentQuestionId) => {
+    await api.updateQuestion(questionId, {
+      parentQuestionId,
+    });
+    set((s) => ({
+      questionsByTopic: {
+        ...s.questionsByTopic,
+        [topicId]: (s.questionsByTopic[topicId] || []).map((q) =>
+          q.id === questionId
+            ? { ...q, parentQuestionId: parentQuestionId ?? undefined }
+            : q,
+        ),
+      },
+      currentQuestion:
+        s.currentQuestion?.id === questionId
+          ? {
+              ...s.currentQuestion,
+              parentQuestionId: parentQuestionId ?? undefined,
+            }
+          : s.currentQuestion,
+    }));
+  },
+
   removeQuestion: async (questionId, topicId) => {
     await api.deleteQuestion(questionId);
     set((s) => ({
@@ -682,6 +805,38 @@ export const useStore = create<Store>((set, get) => ({
     }));
   },
 
+  linkConversation: async (questionId, linkedId) => {
+    const q = get().currentQuestion;
+    const current =
+      q?.id === questionId ? q : await api.fetchQuestion(questionId);
+    const existing = current.linkedConversationIds ?? [];
+    if (existing.includes(linkedId)) return;
+    const updated = [...existing, linkedId];
+    await api.updateQuestion(questionId, { linkedConversationIds: updated });
+    set((s) => ({
+      currentQuestion:
+        s.currentQuestion?.id === questionId
+          ? { ...s.currentQuestion, linkedConversationIds: updated }
+          : s.currentQuestion,
+    }));
+  },
+
+  unlinkConversation: async (questionId, linkedId) => {
+    const q = get().currentQuestion;
+    const current =
+      q?.id === questionId ? q : await api.fetchQuestion(questionId);
+    const updated = (current.linkedConversationIds ?? []).filter(
+      (id) => id !== linkedId,
+    );
+    await api.updateQuestion(questionId, { linkedConversationIds: updated });
+    set((s) => ({
+      currentQuestion:
+        s.currentQuestion?.id === questionId
+          ? { ...s.currentQuestion, linkedConversationIds: updated }
+          : s.currentQuestion,
+    }));
+  },
+
   fetchWorkspaceFiles: async () => {
     const files = await api.fetchWorkspaceFiles();
     set({ workspaceFiles: files });
@@ -727,16 +882,26 @@ export const useStore = create<Store>((set, get) => ({
   openDocViewer: (fileId, quote, fileName) =>
     set({ viewingDoc: { fileId, quote, fileName } }),
   closeDocViewer: () => set({ viewingDoc: null }),
-  openCodeRunner: (code = "", language = "typescript") =>
+  openCodeRunner: (code = "", language = "typescript", fileId?) =>
     set({
       showCodeRunner: true,
+      showInfraLab: false,
       runnerInitialCode: code,
       runnerInitialLanguage: language,
       runnerInitialSandbox: null,
+      runnerInitialFileId: fileId ?? null,
     }),
-  openSandbox: (serverCode, serverLang, clientCode, clientLang, fileId?) =>
+  openSandbox: (
+    serverCode,
+    serverLang,
+    clientCode,
+    clientLang,
+    fileId?,
+    opts?,
+  ) =>
     set({
       showCodeRunner: true,
+      showInfraLab: false,
       runnerInitialCode: "",
       runnerInitialLanguage: "typescript",
       runnerInitialSandbox: {
@@ -745,7 +910,54 @@ export const useStore = create<Store>((set, get) => ({
         clientCode,
         clientLang,
         fileId,
+        clientType: opts?.clientType,
+        reactFiles: opts?.reactFiles,
+        reactActiveFile: opts?.reactActiveFile,
+      },
+      runnerInitialFileId: null,
+    }),
+  openInfraLab: (workspace, fileId?) =>
+    set({
+      showInfraLab: true,
+      showCodeRunner: false,
+      runnerInitialInfra: workspace ?? null,
+      runnerInitialInfraFileId: fileId ?? null,
+    }),
+  openReactLab: (workspace?, fileId?, serverCode?, serverLang?) =>
+    set({
+      showCodeRunner: true,
+      showInfraLab: false,
+      runnerInitialCode: "",
+      runnerInitialLanguage: "typescript",
+      runnerInitialSandbox: {
+        serverCode: serverCode ?? DEFAULT_SERVER_CODE,
+        serverLang: serverLang ?? "typescript",
+        clientCode: "",
+        clientLang: "javascript",
+        fileId,
+        clientType: "react",
+        reactFiles: workspace?.files ?? null,
+        reactActiveFile: workspace?.activeFile ?? null,
+      },
+      runnerInitialFileId: null,
+    }),
+  openNextLab: (workspace?, fileId?, serverCode?, serverLang?) =>
+    set({
+      showCodeRunner: true,
+      showInfraLab: false,
+      runnerInitialCode: "",
+      runnerInitialLanguage: "typescript",
+      runnerInitialSandbox: {
+        serverCode: serverCode ?? DEFAULT_SERVER_CODE,
+        serverLang: serverLang ?? "typescript",
+        clientCode: "",
+        clientLang: "javascript",
+        fileId,
+        clientType: "nextjs",
+        reactFiles: workspace?.files ?? null,
+        reactActiveFile: workspace?.activeFile ?? null,
       },
+      runnerInitialFileId: null,
     }),
 
   overwriteContextFileContent: async (questionId, fileId, content) => {
@@ -767,6 +979,7 @@ export const useStore = create<Store>((set, get) => ({
     }));
   },
   closeCodeRunner: () => set({ showCodeRunner: false }),
+  closeInfraLab: () => set({ showInfraLab: false }),
 
   fetchAiSettings: async () => {
     const settings = await api.fetchAiSettings();

package/template/client/src/types.ts

@@ -1,3 +1,12 @@
+export type ContextFileOrigin =
+  | "user"
+  | "ai"
+  | "upload"
+  | "sandbox"
+  | "infra"
+  | "react"
+  | "nextjs";
+
 export interface ContextFile {
   id: string;
   name: string;
@@ -6,14 +15,33 @@ export interface ContextFile {
   createdAt: string;
   /** Distinguishes how this file was added. 'upload' = user-uploaded doc,
    *  'user' = code saved from Code Runner, 'ai' = AI-generated code block,
-   *  'sandbox' = paired server+client sandbox saved as JSON. */
-  origin?: "user" | "ai" | "upload" | "sandbox";
+   *  'sandbox' = paired server+client sandbox saved as JSON,
+   *  'infra' = Terraform-style infra lab workspace saved as JSON. */
+  origin?: ContextFileOrigin;
   /** Language hint for code snippets (e.g. 'typescript', 'javascript'). */
   language?: string;
   /** Short display label for code snippets. */
   label?: string;
 }
 
+export interface FrontendLabWorkspace {
+  version: 1;
+  label: string;
+  /** Determines defaults, entry file, and file-tree conventions. */
+  type: "react" | "nextjs";
+  activeFile: string;
+  files: Record<string, string>;
+}
+
+export interface InfraLabWorkspace {
+  version: 1;
+  label: string;
+  provider: "aws";
+  executionMode: "plan-only" | "localstack";
+  activeFile: string;
+  files: Record<string, string>;
+}
+
 export interface WorkspaceMeta {
   id: string;
   name: string;
@@ -37,6 +65,8 @@ export interface Topic {
   id: string;
   name: string;
   contextFiles: ContextFile[];
+  /** Topic-wide system prompt prepended to every question's context in this topic. */
+  systemContext?: string;
   createdAt: string;
 }
 
@@ -82,7 +112,7 @@ export interface ReadingBookmark {
 export interface Question {
   id: string;
   topicId: string;
-  parentQuestionId?: string;
+  parentQuestionId?: string | null;
   title: string;
   systemContext: string;
   codeContextFiles: string[];
@@ -90,5 +120,7 @@ export interface Question {
   messages: Message[];
   annotations?: Annotation[];
   readingBookmark?: ReadingBookmark;
+  /** IDs of other questions in the same topic whose conversation history is injected as context. */
+  linkedConversationIds?: string[];
   createdAt: string;
 }

package/template/client/tsconfig.tsbuildinfo

@@ -1 +1 @@
-{"root":["./src/app.tsx","./src/api.ts","./src/main.tsx","./src/store.ts","./src/types.ts","./src/vite-env.d.ts","./src/components/chatmessage.tsx","./src/components/chatview.tsx","./src/components/codecontextpanel.tsx","./src/components/fileattachments.tsx","./src/components/fileviewermodal.tsx","./src/components/markdownrenderer.tsx","./src/components/mermaiddiagram.tsx","./src/components/sidebar.tsx","./src/components/textannotator.tsx"],"version":"5.9.3"}
+{"root":["./src/app.tsx","./src/api.ts","./src/main.tsx","./src/store.ts","./src/types.ts","./src/vite-env.d.ts","./src/components/aisettingsmodal.tsx","./src/components/annotationdialog.tsx","./src/components/chatmessage.tsx","./src/components/chatview.tsx","./src/components/codecontextpanel.tsx","./src/components/codelineannotationpopup.tsx","./src/components/coderunnermodal.tsx","./src/components/docrefmodal.tsx","./src/components/fileattachments.tsx","./src/components/filepickermodal.tsx","./src/components/fileviewermodal.tsx","./src/components/linkedconvospicker.tsx","./src/components/markdownrenderer.tsx","./src/components/mermaiddiagram.tsx","./src/components/plotembed.tsx","./src/components/sidebar.tsx","./src/components/textannotator.tsx","./src/components/vizcraftembed.tsx","./src/components/workspaceswitcher.tsx"],"version":"5.9.3"}

package/template/server/src/google-drive.ts

@@ -357,7 +357,10 @@ export async function syncWorkspace(
                       cs.content &&
                       (cs.origin === "user" ||
                         cs.origin === "ai" ||
-                        cs.origin === "sandbox")
+                        cs.origin === "sandbox" ||
+                        cs.origin === "react" ||
+                        cs.origin === "nextjs" ||
+                        cs.origin === "infra")
                     ) {
                       try {
                         await fs.mkdir(ctxDir, { recursive: true });
@@ -562,7 +565,35 @@ async function getExportDriveClient(): Promise<drive_v3.Drive> {
       /* ok */
     }
   });
-  return google.drive({ version: "v3", auth: client });
+
+  // Wrap the drive client so any invalid_grant errors clear the stored token
+  // and surface a friendlier error that the caller can detect.
+  const drive = google.drive({ version: "v3", auth: client });
+  const originalFiles = drive.files as any;
+  const patchMethod = (obj: any, method: string) => {
+    const orig = obj[method].bind(obj);
+    obj[method] = async (...args: any[]) => {
+      try {
+        return await orig(...args);
+      } catch (err: any) {
+        if (err?.message?.includes("invalid_grant") || err?.code === 400) {
+          try {
+            await fs.unlink(EXPORT_TOKENS_FILE);
+          } catch {
+            /* already gone */
+          }
+          const e = new Error("NEEDS_REAUTH");
+          (e as any).needsReauth = true;
+          throw e;
+        }
+        throw err;
+      }
+    };
+  };
+  ["create", "update", "delete", "list", "get"].forEach((m) =>
+    patchMethod(originalFiles, m),
+  );
+  return drive;
 }
 
 export interface ExportResult {
@@ -736,7 +767,10 @@ export async function exportWorkspace(
                 if (
                   cf.origin === "user" ||
                   cf.origin === "ai" ||
-                  cf.origin === "sandbox"
+                  cf.origin === "sandbox" ||
+                  cf.origin === "react" ||
+                  cf.origin === "nextjs" ||
+                  cf.origin === "infra"
                 ) {
                   try {
                     const content = await fs.readFile(