stagent 0.9.6 → 0.11.0

package/src/app/api/workflows/[id]/resume/route.ts

@@ -0,0 +1,59 @@
+import { NextRequest, NextResponse } from "next/server";
+import { db } from "@/lib/db";
+import { workflows } from "@/lib/db/schema";
+import { eq } from "drizzle-orm";
+import { resumeWorkflow } from "@/lib/workflows/engine";
+
+/**
+ * POST /api/workflows/[id]/resume
+ *
+ * Manually resume a workflow that was paused at a delay step. Called by the
+ * "Resume Now" button in WorkflowStatusView to skip the remaining delay. The
+ * scheduler tick also calls resumeWorkflow() directly (not through this route)
+ * when workflows.resume_at is reached.
+ *
+ * Response codes:
+ *   202 Accepted   — resume dispatched (fire-and-forget)
+ *   404 Not Found  — workflow does not exist
+ *   409 Conflict   — workflow is not in paused state (already resumed, racing
+ *                    scheduler, or was never paused). resumeWorkflow handles
+ *                    this internally with its atomic status transition, so the
+ *                    conflict is reported here for correct UX feedback.
+ */
+export async function POST(
+  _req: NextRequest,
+  { params }: { params: Promise<{ id: string }> },
+) {
+  const { id } = await params;
+
+  const [workflow] = await db
+    .select()
+    .from(workflows)
+    .where(eq(workflows.id, id));
+
+  if (!workflow) {
+    return NextResponse.json({ error: "Workflow not found" }, { status: 404 });
+  }
+
+  if (workflow.status !== "paused") {
+    return NextResponse.json(
+      {
+        error: `Workflow is not paused (current status: ${workflow.status})`,
+        status: workflow.status,
+      },
+      { status: 409 },
+    );
+  }
+
+  // Fire-and-forget. resumeWorkflow performs its own atomic status check, so
+  // if the scheduler tick beats this request by microseconds, the DB UPDATE
+  // will match zero rows and resumeWorkflow returns silently without harm.
+  resumeWorkflow(id).catch((error) => {
+    console.error(`Workflow ${id} resume failed:`, error);
+  });
+
+  return NextResponse.json(
+    { status: "resuming", workflowId: id },
+    { status: 202 },
+  );
+}

package/src/app/api/workflows/[id]/status/route.ts

@@ -3,6 +3,11 @@ import { db } from "@/lib/db";
 import { workflows, tasks, documents } from "@/lib/db/schema";
 import { eq, and, inArray, count, desc, sql as drizzleSql } from "drizzle-orm";
 import { parseWorkflowState } from "@/lib/workflows/engine";
+import type {
+  WorkflowStatusResponse,
+  NonLoopPattern,
+  StepWithState,
+} from "@/lib/workflows/types";
 
 /** Collect output documents for workflow step tasks + input documents from parent task */
 async function getWorkflowDocuments(
@@ -97,15 +102,18 @@ export async function GET(
   const sourceTaskId: string | undefined = definition.sourceTaskId;
   const { stepDocuments, parentDocuments } = await getWorkflowDocuments(state, sourceTaskId);
 
-  // Loop pattern returns loop-specific data instead of step states
+  // Loop pattern returns loop-specific data instead of step states.
+  // The `satisfies` annotation enforces the TDR-031 contract: the loop arm
+  // of WorkflowStatusResponse cannot emit `workflowState` or `resumeAt`, and
+  // its `steps` field is raw WorkflowStep[] (not StepWithState[]).
   if (definition.pattern === "loop") {
-    return NextResponse.json({
+    const loopBody = {
       id: workflow.id,
       name: workflow.name,
       status: workflow.status,
       projectId: workflow.projectId,
       definition: workflow.definition,
-      pattern: definition.pattern,
+      pattern: "loop" as const,
       loopConfig: definition.loopConfig,
       swarmConfig: definition.swarmConfig,
       loopState,
@@ -114,18 +122,23 @@ export async function GET(
       parentDocuments,
       runNumber: workflow.runNumber,
       runHistory,
-    });
+    } satisfies WorkflowStatusResponse;
+    return NextResponse.json(loopBody);
   }
 
-  return NextResponse.json({
+  // Non-loop arm: sequence, parallel, swarm, planner-executor, checkpoint all
+  // share the step-state rendering path. `satisfies` enforces that this branch
+  // cannot accidentally emit `loopState`, and that every step has `.state`.
+  const nonLoopBody = {
     id: workflow.id,
     name: workflow.name,
     status: workflow.status,
+    resumeAt: workflow.resumeAt ?? null,
     projectId: workflow.projectId,
     definition: workflow.definition,
-    pattern: definition.pattern,
+    pattern: definition.pattern as NonLoopPattern,
     swarmConfig: definition.swarmConfig,
-    steps: definition.steps.map((step, i) => ({
+    steps: definition.steps.map((step, i): StepWithState => ({
       ...step,
       state: state?.stepStates[i] ?? { stepId: step.id, status: "pending" },
     })),
@@ -134,5 +147,6 @@ export async function GET(
     parentDocuments,
     runNumber: workflow.runNumber,
     runHistory,
-  });
+  } satisfies WorkflowStatusResponse;
+  return NextResponse.json(nonLoopBody);
 }

package/src/app/api/workspace/context/route.ts

@@ -1,6 +1,8 @@
 import { NextResponse } from "next/server";
 import { getWorkspaceContext } from "@/lib/environment/workspace-context";
 
+export const dynamic = "force-dynamic";
+
 export function GET() {
   const context = getWorkspaceContext();
   return NextResponse.json(context, {

package/src/app/api/workspace/fix-data-dir/route.ts

@@ -0,0 +1,81 @@
+import { NextResponse } from "next/server";
+import { basename, join } from "path";
+import { homedir } from "os";
+import { existsSync, readFileSync, writeFileSync, mkdirSync } from "fs";
+import Database from "better-sqlite3";
+import { getLaunchCwd } from "@/lib/environment/workspace-context";
+import { isDevMode, isPrivateInstance } from "@/lib/instance/detect";
+import { bootstrapStagentDatabase } from "@/lib/db/bootstrap";
+
+export const dynamic = "force-dynamic";
+
+/**
+ * POST /api/workspace/fix-data-dir
+ *
+ * Fixes a data-dir mismatch for domain clones by:
+ * 1. Deriving the correct STAGENT_DATA_DIR from the folder name
+ * 2. Writing it to .env.local (alongside STAGENT_CLOUD_DISABLED=true)
+ * 3. Creating the data dir + bootstrapping an empty database there
+ *
+ * Requires a dev server restart to take effect.
+ */
+export async function POST() {
+  const cwd = getLaunchCwd();
+
+  // Guard: main repo doesn't need fixing
+  if (isDevMode(cwd)) {
+    return NextResponse.json(
+      { error: "Main dev repo does not need a data-dir fix" },
+      { status: 400 }
+    );
+  }
+
+  // Guard: already isolated
+  if (isPrivateInstance()) {
+    return NextResponse.json(
+      { error: "STAGENT_DATA_DIR is already set to a non-default path" },
+      { status: 400 }
+    );
+  }
+
+  const folderName = basename(cwd);
+  const home = homedir();
+  // stagent-wealth → ~/.stagent-wealth, stagent-growth → ~/.stagent-growth
+  const dataDir = join(home, `.${folderName}`);
+  const displayDataDir = `~/.${folderName}`;
+
+  // --- 1. Update .env.local ---
+  const envLocalPath = join(cwd, ".env.local");
+  let envContent = "";
+  if (existsSync(envLocalPath)) {
+    envContent = readFileSync(envLocalPath, "utf-8");
+  }
+
+  // Replace or append STAGENT_DATA_DIR
+  if (/^STAGENT_DATA_DIR=.*/m.test(envContent)) {
+    envContent = envContent.replace(
+      /^STAGENT_DATA_DIR=.*/m,
+      `STAGENT_DATA_DIR=${dataDir}`
+    );
+  } else {
+    envContent = envContent.trimEnd() + `\nSTAGENT_DATA_DIR=${dataDir}\n`;
+  }
+
+  writeFileSync(envLocalPath, envContent, "utf-8");
+
+  // --- 2. Create data dir + bootstrap DB ---
+  mkdirSync(dataDir, { recursive: true });
+  const dbPath = join(dataDir, "stagent.db");
+  const sqlite = new Database(dbPath);
+  sqlite.pragma("journal_mode = WAL");
+  sqlite.pragma("foreign_keys = ON");
+  bootstrapStagentDatabase(sqlite);
+  sqlite.close();
+
+  return NextResponse.json({
+    success: true,
+    dataDir: displayDataDir,
+    envLocalPath,
+    needsRestart: true,
+  });
+}

package/src/app/chat/page.tsx

@@ -1,5 +1,6 @@
 import { listConversations } from "@/lib/data/chat";
 import { getPromptCategories } from "@/lib/chat/suggested-prompts";
+import { reconcileStreamingMessages } from "@/lib/chat/reconcile";
 import { ChatShell } from "@/components/chat/chat-shell";
 
 export const dynamic = "force-dynamic";
@@ -10,6 +11,16 @@ export default async function ChatPage({
   searchParams: Promise<{ c?: string }>;
 }) {
   const params = await searchParams;
+
+  // Safety-net sweep: finalize any orphaned streaming rows from crashes,
+  // disconnects, or prior releases. Fire-and-forget-ish — we await so the
+  // UI never renders stale `streaming` placeholders, but errors are swallowed.
+  try {
+    await reconcileStreamingMessages();
+  } catch (err) {
+    console.error("[chat] reconcileStreamingMessages failed:", err);
+  }
+
   const [conversations, promptCategories] = await Promise.all([
     listConversations({ status: "active" }),
     getPromptCategories(),

package/src/app/documents/page.tsx

@@ -1,3 +1,4 @@
+import { Suspense } from "react";
 import { db } from "@/lib/db";
 import { documents, tasks, projects, workflows } from "@/lib/db/schema";
 import { desc, eq } from "drizzle-orm";
@@ -47,7 +48,9 @@ export default async function DocumentsPage() {
 
   return (
     <PageShell title="Documents">
-      <DocumentBrowser initialDocuments={docs} projects={projectList} />
+      <Suspense fallback={null}>
+        <DocumentBrowser initialDocuments={docs} projects={projectList} />
+      </Suspense>
     </PageShell>
   );
 }

package/src/app/inbox/page.tsx

@@ -4,6 +4,10 @@ import { desc, eq, and, gte } from "drizzle-orm";
 import { InboxList } from "@/components/notifications/inbox-list";
 import { GovernanceStats } from "@/components/notifications/governance-stats";
 import { PageShell } from "@/components/shared/page-shell";
+import {
+  buildDefaultNotificationVisibilityCondition,
+  filterDefaultVisibleNotifications,
+} from "@/lib/notifications/visibility";
 
 export const dynamic = "force-dynamic";
 
@@ -16,6 +20,7 @@ export default async function InboxPage() {
       db
         .select()
         .from(notifications)
+        .where(buildDefaultNotificationVisibilityCondition())
         .orderBy(desc(notifications.createdAt))
         .limit(100),
       db
@@ -53,11 +58,13 @@ export default async function InboxPage() {
     ]);
 
   // Serialize Date objects for client component consumption
-  const initialNotifications = rows.map((n) => ({
-    ...n,
-    createdAt: n.createdAt.toISOString(),
-    respondedAt: n.respondedAt?.toISOString() ?? null,
-  }));
+  const initialNotifications = filterDefaultVisibleNotifications(
+    rows.map((n) => ({
+      ...n,
+      createdAt: n.createdAt.toISOString(),
+      respondedAt: n.respondedAt?.toISOString() ?? null,
+    }))
+  );
 
   return (
     <PageShell

package/src/app/layout.tsx

@@ -1,5 +1,5 @@
 import type { Metadata } from "next";
-import Script from "next/script";
+import { cookies } from "next/headers";
 import { Inter, JetBrains_Mono } from "next/font/google";
 import { SidebarProvider, SidebarInset } from "@/components/ui/sidebar";
 import { TooltipProvider } from "@/components/ui/tooltip";
@@ -8,6 +8,13 @@ import { CommandPalette } from "@/components/shared/command-palette";
 import { PendingApprovalHost } from "@/components/notifications/pending-approval-host";
 import { GlobalShortcuts } from "@/components/shared/global-shortcuts";
 import { Toaster } from "@/components/ui/sonner";
+import { ChatSessionProvider } from "@/components/chat/chat-session-provider";
+import {
+  DEFAULT_THEME,
+  THEME_COOKIE,
+  isResolvedTheme,
+  type ResolvedTheme,
+} from "@/lib/theme";
 import "./globals.css";
 
 const inter = Inter({
@@ -55,22 +62,34 @@ const CRITICAL_THEME_CSS = `
   html { background: var(--background); font-size: 14px; }
 `.replace(/\s+/g, " ").trim();
 
-// Light-first: defaults to 'light' when no localStorage preference exists.
-// Static theme initialization script — no user input, safe from XSS.
-const THEME_INIT_SCRIPT = `(function(){try{var d=document.documentElement;var s=localStorage.getItem('stagent-theme');var t=s==='dark'||s==='light'?s:'light';d.classList.toggle('dark',t==='dark');d.dataset.theme=t;d.style.colorScheme=t;d.style.backgroundColor=t==='dark'?'oklch(0.14 0.02 250)':'oklch(0.985 0.004 250)';document.cookie='stagent-theme='+t+';path=/;max-age=31536000;SameSite=Lax';}catch(e){}})()`;
-
-export default function RootLayout({
+export default async function RootLayout({
   children,
 }: {
   children: React.ReactNode;
 }) {
+  // Resolve theme server-side from the stagent-theme cookie. Every client-side
+  // theme toggle writes this cookie (see src/lib/theme.ts), so SSR stays in
+  // sync with the user's preference and there is no FOUC — and no pre-hydration
+  // <script> tag, which is what React 19 warns about.
+  const cookieStore = await cookies();
+  const cookieValue = cookieStore.get(THEME_COOKIE)?.value;
+  const theme: ResolvedTheme = isResolvedTheme(cookieValue)
+    ? cookieValue
+    : DEFAULT_THEME;
+  const htmlClass = theme === "dark" ? "dark" : undefined;
+  const htmlBackground =
+    theme === "dark" ? "oklch(0.14 0.02 250)" : "oklch(0.985 0.004 250)";
   return (
-    <html lang="en" suppressHydrationWarning>
+    <html
+      lang="en"
+      className={htmlClass}
+      data-theme={theme}
+      style={{ colorScheme: theme, backgroundColor: htmlBackground }}
+      suppressHydrationWarning
+    >
       <head>
         {/* Static CSS — no user input, safe from XSS */}
         <style dangerouslySetInnerHTML={{ __html: CRITICAL_THEME_CSS }} />
-        {/* Theme bootstrap — runs before paint to prevent FOUC */}
-        <Script id="theme-init" strategy="beforeInteractive">{THEME_INIT_SCRIPT}</Script>
       </head>
       <body
         className={`${inter.variable} ${jetbrainsMono.variable} font-sans antialiased bg-background text-foreground`}
@@ -79,18 +98,20 @@ export default function RootLayout({
           Skip to main content
         </a>
         <TooltipProvider>
-          <SidebarProvider>
-            <AppSidebar />
-            <SidebarInset className="min-w-0">
-              <main id="main-content">
-                {children}
-              </main>
-            </SidebarInset>
-          </SidebarProvider>
-          <PendingApprovalHost />
-          <CommandPalette />
-          <GlobalShortcuts />
-          <Toaster />
+          <ChatSessionProvider>
+            <SidebarProvider>
+              <AppSidebar />
+              <SidebarInset className="min-w-0">
+                <main id="main-content">
+                  {children}
+                </main>
+              </SidebarInset>
+            </SidebarProvider>
+            <PendingApprovalHost />
+            <CommandPalette />
+            <GlobalShortcuts />
+            <Toaster />
+          </ChatSessionProvider>
         </TooltipProvider>
       </body>
     </html>

package/src/app/page.tsx

@@ -12,7 +12,6 @@ import { QuickActions } from "@/components/dashboard/quick-actions";
 import { RecentProjects } from "@/components/dashboard/recent-projects";
 import type { RecentProject } from "@/components/dashboard/recent-projects";
 import { WelcomeLanding } from "@/components/dashboard/welcome-landing";
-import { EmailCaptureCard } from "@/components/onboarding/email-capture-card";
 import { ActivationChecklist } from "@/components/onboarding/activation-checklist";
 import {
   getCompletionsByDay,
@@ -107,7 +106,6 @@ export default async function HomePage() {
     return (
       <div className="bg-background min-h-screen p-4 sm:p-6">
         <div className="surface-page-shell min-h-[calc(100dvh-2rem)] rounded-xl p-5 sm:p-6 lg:p-7 space-y-6">
-          <EmailCaptureCard />
           <WelcomeLanding />
         </div>
       </div>

package/src/app/settings/page.tsx

@@ -1,7 +1,3 @@
-import { SubscriptionSection } from "@/components/settings/subscription-section";
-import { CloudAccountSection } from "@/components/settings/cloud-account-section";
-import { CloudSyncSection } from "@/components/settings/cloud-sync-section";
-import { TelemetrySection } from "@/components/settings/telemetry-section";
 import { ProvidersAndRuntimesSection } from "@/components/settings/providers-runtimes-section";
 import { PermissionsSections } from "@/components/settings/permissions-sections";
 import { DataManagementSection } from "@/components/settings/data-management-section";
@@ -11,32 +7,35 @@ import { ChatSettingsSection } from "@/components/settings/chat-settings-section
 import { RuntimeTimeoutSection } from "@/components/settings/runtime-timeout-section";
 import { BrowserToolsSection } from "@/components/settings/browser-tools-section";
 import { WebSearchSection } from "@/components/settings/web-search-section";
+import { EnvironmentSection } from "@/components/settings/environment-section";
 import { LearningContextSection } from "@/components/settings/learning-context-section";
 import { OllamaSection } from "@/components/settings/ollama-section";
 import { ChannelsSection } from "@/components/settings/channels-section";
+import { InstanceSection } from "@/components/instance/instance-section";
 import { PageShell } from "@/components/shared/page-shell";
 
 export const dynamic = "force-dynamic";
 
 export default function SettingsPage() {
   return (
-    <PageShell title="Settings" description="Manage your Stagent configuration">
+    <PageShell
+      title="Settings"
+      description="Manage your Stagent configuration"
+    >
       <div className="space-y-6">
-        <SubscriptionSection />
-        <CloudAccountSection />
-        <CloudSyncSection />
+        <InstanceSection />
         <ProvidersAndRuntimesSection />
         <OllamaSection />
         <ChatSettingsSection />
         <RuntimeTimeoutSection />
         <LearningContextSection />
         <WebSearchSection />
+        <EnvironmentSection />
         <BrowserToolsSection />
         <ChannelsSection />
         <BudgetGuardrailsSection />
         <PermissionsSections />
         <DatabaseSnapshotsSection />
-        <TelemetrySection />
         <DataManagementSection />
       </div>
     </PageShell>

package/src/components/chat/__tests__/capability-banner.test.tsx

@@ -0,0 +1,38 @@
+import { describe, it, expect, beforeEach } from "vitest";
+import { render, screen, fireEvent } from "@testing-library/react";
+import { CapabilityBanner } from "../capability-banner";
+
+describe("CapabilityBanner", () => {
+  beforeEach(() => {
+    sessionStorage.clear();
+  });
+
+  it("is hidden on claude-code runtime", () => {
+    render(<CapabilityBanner runtimeId="claude-code" />);
+    expect(screen.queryByRole("status")).toBeNull();
+  });
+
+  it("is hidden on openai-codex-app-server runtime", () => {
+    render(<CapabilityBanner runtimeId="openai-codex-app-server" />);
+    expect(screen.queryByRole("status")).toBeNull();
+  });
+
+  it("is visible on ollama runtime with capability message", () => {
+    render(<CapabilityBanner runtimeId="ollama" />);
+    const status = screen.getByRole("status");
+    expect(status.textContent).toContain("file read/write");
+  });
+
+  it("hides on dismiss and persists to sessionStorage", () => {
+    render(<CapabilityBanner runtimeId="ollama" />);
+    fireEvent.click(screen.getByRole("button", { name: /dismiss/i }));
+    expect(screen.queryByRole("status")).toBeNull();
+    expect(sessionStorage.getItem("stagent.capability-banner.dismissed.ollama")).toBe("1");
+  });
+
+  it("stays dismissed on remount if sessionStorage flag set", () => {
+    sessionStorage.setItem("stagent.capability-banner.dismissed.ollama", "1");
+    render(<CapabilityBanner runtimeId="ollama" />);
+    expect(screen.queryByRole("status")).toBeNull();
+  });
+});