stagent 0.1.7 → 0.1.9

package/dist/cli.js

@@ -276,33 +276,25 @@ function bootstrapStagentDatabase(sqlite2) {
     CREATE INDEX IF NOT EXISTS idx_learned_context_profile_version ON learned_context(profile_id, version);
     CREATE INDEX IF NOT EXISTS idx_learned_context_change_type ON learned_context(change_type);
   `);
-  try {
-    sqlite2.exec(`ALTER TABLE tasks ADD COLUMN agent_profile TEXT;`);
-  } catch {
-  }
+  const addColumnIfMissing = (ddl) => {
+    try {
+      sqlite2.exec(ddl);
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      if (!msg.includes("duplicate column")) {
+        console.error("[bootstrap] ALTER TABLE failed:", msg);
+      }
+    }
+  };
+  addColumnIfMissing(`ALTER TABLE tasks ADD COLUMN agent_profile TEXT;`);
   sqlite2.exec(`CREATE INDEX IF NOT EXISTS idx_tasks_agent_profile ON tasks(agent_profile);`);
-  try {
-    sqlite2.exec(`ALTER TABLE tasks ADD COLUMN workflow_id TEXT REFERENCES workflows(id);`);
-  } catch {
-  }
+  addColumnIfMissing(`ALTER TABLE tasks ADD COLUMN workflow_id TEXT REFERENCES workflows(id);`);
   sqlite2.exec(`CREATE INDEX IF NOT EXISTS idx_tasks_workflow_id ON tasks(workflow_id);`);
-  try {
-    sqlite2.exec(`ALTER TABLE tasks ADD COLUMN schedule_id TEXT REFERENCES schedules(id);`);
-  } catch {
-  }
+  addColumnIfMissing(`ALTER TABLE tasks ADD COLUMN schedule_id TEXT REFERENCES schedules(id);`);
   sqlite2.exec(`CREATE INDEX IF NOT EXISTS idx_tasks_schedule_id ON tasks(schedule_id);`);
-  try {
-    sqlite2.exec(`ALTER TABLE projects ADD COLUMN working_directory TEXT;`);
-  } catch {
-  }
-  try {
-    sqlite2.exec(`ALTER TABLE schedules ADD COLUMN assigned_agent TEXT;`);
-  } catch {
-  }
-  try {
-    sqlite2.exec(`ALTER TABLE documents ADD COLUMN version INTEGER NOT NULL DEFAULT 1;`);
-  } catch {
-  }
+  addColumnIfMissing(`ALTER TABLE projects ADD COLUMN working_directory TEXT;`);
+  addColumnIfMissing(`ALTER TABLE schedules ADD COLUMN assigned_agent TEXT;`);
+  addColumnIfMissing(`ALTER TABLE documents ADD COLUMN version INTEGER NOT NULL DEFAULT 1;`);
 }
 function hasLegacyStagentTables(sqlite2) {
   const placeholders = STAGENT_TABLES.map(() => "?").join(", ");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "stagent",
-  "version": "0.1.7",
+  "version": "0.1.9",
   "description": "Governed AI agent workspace for supervised local execution, workflows, documents, and provider runtimes.",
   "keywords": [
     "ai",

package/src/app/api/documents/route.ts

@@ -3,6 +3,11 @@ import { db } from "@/lib/db";
 import { documents, tasks, projects } from "@/lib/db/schema";
 import { eq, and, like, or, desc, sql } from "drizzle-orm";
 
+const VALID_DOC_STATUSES = ["uploaded", "processing", "ready", "error"] as const;
+const VALID_DOC_DIRECTIONS = ["input", "output"] as const;
+type DocStatus = typeof VALID_DOC_STATUSES[number];
+type DocDirection = typeof VALID_DOC_DIRECTIONS[number];
+
 export async function GET(req: NextRequest) {
   const url = new URL(req.url);
   const taskId = url.searchParams.get("taskId");
@@ -11,12 +16,26 @@ export async function GET(req: NextRequest) {
   const direction = url.searchParams.get("direction");
   const search = url.searchParams.get("search");
 
+  if (status && !VALID_DOC_STATUSES.includes(status as DocStatus)) {
+    return NextResponse.json(
+      { error: `Invalid status. Must be one of: ${VALID_DOC_STATUSES.join(", ")}` },
+      { status: 400 }
+    );
+  }
+
+  if (direction && !VALID_DOC_DIRECTIONS.includes(direction as DocDirection)) {
+    return NextResponse.json(
+      { error: `Invalid direction. Must be one of: ${VALID_DOC_DIRECTIONS.join(", ")}` },
+      { status: 400 }
+    );
+  }
+
   const conditions = [];
 
   if (taskId) conditions.push(eq(documents.taskId, taskId));
   if (projectId) conditions.push(eq(documents.projectId, projectId));
-  if (status) conditions.push(eq(documents.status, status as "uploaded" | "processing" | "ready" | "error"));
-  if (direction) conditions.push(eq(documents.direction, direction as "input" | "output"));
+  if (status) conditions.push(eq(documents.status, status as DocStatus));
+  if (direction) conditions.push(eq(documents.direction, direction as DocDirection));
 
   if (search) {
     conditions.push(

package/src/app/api/tasks/route.ts

@@ -6,14 +6,24 @@ import { createTaskSchema } from "@/lib/validators/task";
 import { processDocument } from "@/lib/documents/processor";
 import { validateRuntimeProfileAssignment } from "@/lib/agents/profiles/assignment-validation";
 
+const VALID_TASK_STATUSES = ["planned", "queued", "running", "completed", "failed", "cancelled"] as const;
+type TaskStatus = typeof VALID_TASK_STATUSES[number];
+
 export async function GET(req: NextRequest) {
   const url = new URL(req.url);
   const projectId = url.searchParams.get("projectId");
   const status = url.searchParams.get("status");
 
+  if (status && !VALID_TASK_STATUSES.includes(status as TaskStatus)) {
+    return NextResponse.json(
+      { error: `Invalid status. Must be one of: ${VALID_TASK_STATUSES.join(", ")}` },
+      { status: 400 }
+    );
+  }
+
   const conditions = [];
   if (projectId) conditions.push(eq(tasks.projectId, projectId));
-  if (status) conditions.push(eq(tasks.status, status as typeof tasks.status.enumValues[number]));
+  if (status) conditions.push(eq(tasks.status, status as TaskStatus));
 
   const result = await db
     .select()
@@ -70,10 +80,13 @@ export async function POST(req: NextRequest) {
           .where(eq(documents.id, fileId));
 
         // Trigger processing if not already done (fire-and-forget)
-        processDocument(fileId).catch(() => {});
+        processDocument(fileId).catch((err) => {
+          console.error(`[tasks] processDocument failed for ${fileId}:`, err);
+        });
       }
-    } catch {
+    } catch (err) {
       // File association is best-effort — don't fail task creation
+      console.error("[tasks] File association failed:", err);
     }
   }
 

package/src/app/api/uploads/route.ts

@@ -3,6 +3,7 @@ import { writeFile, mkdir } from "fs/promises";
 import { join } from "path";
 import { db } from "@/lib/db";
 import { documents } from "@/lib/db/schema";
+import { eq } from "drizzle-orm";
 import { processDocument } from "@/lib/documents/processor";
 import { getStagentUploadsDir } from "@/lib/utils/stagent-paths";
 
@@ -48,9 +49,22 @@ export async function POST(req: NextRequest) {
   });
 
   // Fire-and-forget: trigger async document processing
-  processDocument(id).catch((err) =>
-    console.error(`[upload] Processing failed for ${id}:`, err)
-  );
+  processDocument(id).catch(async (err) => {
+    console.error(`[upload] Processing failed for ${id}:`, err);
+    // Ensure document doesn't stay stuck in "processing" state
+    try {
+      await db
+        .update(documents)
+        .set({
+          status: "error",
+          processingError: err instanceof Error ? err.message : String(err),
+          updatedAt: new Date(),
+        })
+        .where(eq(documents.id, id));
+    } catch (dbErr) {
+      console.error(`[upload] Failed to update error status for ${id}:`, dbErr);
+    }
+  });
 
   return NextResponse.json(
     {

package/src/app/globals.css

@@ -659,6 +659,40 @@
   border: 1px solid color-mix(in oklab, var(--border) 75%, transparent);
 }
 
+/* --- Progress slide animation (AI Assist) --- */
+@keyframes progress-slide {
+  0% { transform: translateX(-100%); }
+  50% { transform: translateX(0%); }
+  100% { transform: translateX(100%); }
+}
+
+/* --- Card exit animation (ghost card deletion) --- */
+@keyframes card-exit {
+  0% {
+    opacity: 1;
+    transform: scale(1);
+    max-height: 300px;
+  }
+  40% {
+    opacity: 0;
+    transform: scale(0.97);
+    max-height: 300px;
+  }
+  100% {
+    opacity: 0;
+    transform: scale(0.95);
+    max-height: 0;
+    margin-top: 0;
+    margin-bottom: 0;
+    overflow: hidden;
+  }
+}
+
+.animate-card-exit {
+  animation: card-exit 500ms ease-out forwards;
+  overflow: hidden;
+}
+
 /* --- Glass shimmer animation --- */
 @keyframes glass-shimmer {
   0% { background-position: -200% 0; }

package/src/app/tasks/new/page.tsx

@@ -7,7 +7,12 @@ import { TaskCreatePanel } from "@/components/tasks/task-create-panel";
 
 export const dynamic = "force-dynamic";
 
-export default async function NewTaskPage() {
+export default async function NewTaskPage({
+  searchParams,
+}: {
+  searchParams: Promise<{ project?: string }>;
+}) {
+  const params = await searchParams;
   const allProjects = await db
     .select({ id: projects.id, name: projects.name })
     .from(projects)
@@ -21,7 +26,10 @@ export default async function NewTaskPage() {
           Back to Dashboard
         </Button>
       </Link>
-      <TaskCreatePanel projects={allProjects} />
+      <TaskCreatePanel
+        projects={allProjects}
+        defaultProjectId={params.project}
+      />
     </div>
   );
 }

package/src/components/tasks/__tests__/kanban-board-persistence.test.tsx

@@ -0,0 +1,124 @@
+import { render, screen } from "@testing-library/react";
+import { KanbanBoard, compareTasks } from "@/components/tasks/kanban-board";
+import type { TaskItem } from "@/components/tasks/task-card";
+
+vi.mock("next/navigation", () => ({
+  useRouter: () => ({
+    push: vi.fn(),
+    refresh: vi.fn(),
+  }),
+}));
+
+vi.mock("sonner", () => ({
+  toast: { success: vi.fn(), error: vi.fn(), info: vi.fn() },
+}));
+
+class ResizeObserverMock {
+  observe() {}
+  disconnect() {}
+  unobserve() {}
+}
+
+const makeTask = (overrides: Partial<TaskItem> = {}): TaskItem => ({
+  id: "task-1",
+  title: "Test task",
+  description: "A test task",
+  status: "planned",
+  priority: 2,
+  assignedAgent: null,
+  agentProfile: null,
+  projectId: "project-1",
+  projectName: "Project A",
+  result: null,
+  sessionId: null,
+  resumeCount: 0,
+  createdAt: new Date("2026-03-12T09:00:00Z").toISOString(),
+  updatedAt: new Date("2026-03-12T09:00:00Z").toISOString(),
+  ...overrides,
+});
+
+const defaultProjects = [
+  { id: "project-1", name: "Project A" },
+  { id: "project-2", name: "Project B" },
+];
+
+describe("kanban board persistence", () => {
+  beforeAll(() => {
+    vi.stubGlobal("ResizeObserver", ResizeObserverMock);
+  });
+
+  afterAll(() => {
+    vi.unstubAllGlobals();
+  });
+
+  beforeEach(() => {
+    localStorage.clear();
+    sessionStorage.clear();
+  });
+
+  it("resets stale project filter to 'all'", () => {
+    // Pre-set a project ID that doesn't exist in the projects list
+    localStorage.setItem("stagent-project-filter", "deleted-project-999");
+
+    render(
+      <KanbanBoard
+        initialTasks={[makeTask()]}
+        projects={defaultProjects}
+      />
+    );
+
+    // The task should be visible (not filtered out by a stale project ID)
+    expect(screen.getByText("Test task")).toBeInTheDocument();
+  });
+
+  it("includes ?project= in New Task link when project is filtered", () => {
+    localStorage.setItem("stagent-project-filter", "project-1");
+
+    render(
+      <KanbanBoard
+        initialTasks={[makeTask()]}
+        projects={defaultProjects}
+      />
+    );
+
+    const newTaskLink = screen.getByRole("link", { name: /new task/i });
+    expect(newTaskLink).toHaveAttribute("href", "/tasks/new?project=project-1");
+  });
+
+  it("has plain /tasks/new link when filter is 'all'", () => {
+    render(
+      <KanbanBoard
+        initialTasks={[makeTask()]}
+        projects={defaultProjects}
+      />
+    );
+
+    const newTaskLink = screen.getByRole("link", { name: /new task/i });
+    expect(newTaskLink).toHaveAttribute("href", "/tasks/new");
+  });
+});
+
+describe("compareTasks", () => {
+  const taskA = makeTask({ id: "a", title: "Alpha", priority: 0, createdAt: new Date("2026-03-10").toISOString() });
+  const taskB = makeTask({ id: "b", title: "Bravo", priority: 2, createdAt: new Date("2026-03-12").toISOString() });
+
+  it("sorts by priority ascending", () => {
+    expect(compareTasks(taskA, taskB, "priority")).toBeLessThan(0);
+    expect(compareTasks(taskB, taskA, "priority")).toBeGreaterThan(0);
+  });
+
+  it("sorts by created-desc (newer first)", () => {
+    expect(compareTasks(taskB, taskA, "created-desc")).toBeLessThan(0);
+    expect(compareTasks(taskA, taskB, "created-desc")).toBeGreaterThan(0);
+  });
+
+  it("sorts by created-asc (older first)", () => {
+    expect(compareTasks(taskA, taskB, "created-asc")).toBeLessThan(0);
+    expect(compareTasks(taskB, taskA, "created-asc")).toBeGreaterThan(0);
+  });
+
+  it("sorts by title-asc (alphabetical)", () => {
+    expect(compareTasks(taskA, taskB, "title-asc")).toBeLessThan(0);
+    expect(compareTasks(taskB, taskA, "title-asc")).toBeGreaterThan(0);
+  });
+});

package/src/components/tasks/__tests__/task-create-panel.test.tsx

@@ -0,0 +1,58 @@
+import { render, screen } from "@testing-library/react";
+import { TaskCreatePanel } from "@/components/tasks/task-create-panel";
+
+vi.mock("next/navigation", () => ({
+  useRouter: () => ({
+    push: vi.fn(),
+    refresh: vi.fn(),
+  }),
+}));
+
+vi.mock("sonner", () => ({
+  toast: { success: vi.fn(), error: vi.fn(), info: vi.fn() },
+}));
+
+const projects = [
+  { id: "proj-1", name: "Alpha Project" },
+  { id: "proj-2", name: "Beta Project" },
+];
+
+describe("TaskCreatePanel defaultProjectId", () => {
+  beforeEach(() => {
+    // Mock the /api/profiles fetch
+    global.fetch = vi.fn().mockResolvedValue({
+      ok: true,
+      json: () => Promise.resolve([]),
+    });
+  });
+
+  afterEach(() => {
+    vi.restoreAllMocks();
+  });
+
+  it("pre-selects project when defaultProjectId is provided", () => {
+    render(
+      <TaskCreatePanel projects={projects} defaultProjectId="proj-1" />
+    );
+
+    // The select-value span should show the project name
+    const selectValues = screen.getAllByText("Alpha Project");
+    const triggerValue = selectValues.find(
+      (el) => el.getAttribute("data-slot") === "select-value"
+    );
+    expect(triggerValue).toBeDefined();
+  });
+
+  it("shows 'None' when no defaultProjectId is provided", () => {
+    render(
+      <TaskCreatePanel projects={projects} />
+    );
+
+    // The select-value span should show "None"
+    const noneElements = screen.getAllByText("None");
+    const triggerValue = noneElements.find(
+      (el) => el.getAttribute("data-slot") === "select-value"
+    );
+    expect(triggerValue).toBeDefined();
+  });
+});

package/src/components/tasks/ai-assist-panel.tsx

@@ -1,10 +1,10 @@
 "use client";
 
-import { useState } from "react";
+import { useState, useEffect } from "react";
 import { Button } from "@/components/ui/button";
 import { Card, CardContent } from "@/components/ui/card";
 import { Badge } from "@/components/ui/badge";
-import { Sparkles, Loader2, Check, X } from "lucide-react";
+import { Sparkles, Check, X } from "lucide-react";
 
 interface TaskSuggestion {
   title: string;
@@ -42,6 +42,45 @@ const complexityColors: Record<string, string> = {
   complex: "text-complexity-complex",
 };
 
+const ACTIVITY_MESSAGES = [
+  "Connecting to AI...",
+  "Analyzing task complexity...",
+  "Generating suggestions...",
+  "Finalizing...",
+];
+
+function ProgressBar({ loading }: { loading: boolean }) {
+  const [messageIndex, setMessageIndex] = useState(0);
+
+  useEffect(() => {
+    if (!loading) {
+      setMessageIndex(0);
+      return;
+    }
+
+    const interval = setInterval(() => {
+      setMessageIndex((prev) =>
+        prev < ACTIVITY_MESSAGES.length - 1 ? prev + 1 : prev
+      );
+    }, 3000);
+
+    return () => clearInterval(interval);
+  }, [loading]);
+
+  if (!loading) return null;
+
+  return (
+    <div className="space-y-1.5">
+      <div className="h-1.5 w-full rounded-full bg-muted overflow-hidden">
+        <div className="h-full w-full rounded-full bg-primary animate-[progress-slide_1.5s_ease-in-out_infinite]" />
+      </div>
+      <p className="text-xs text-muted-foreground text-center">
+        {ACTIVITY_MESSAGES[messageIndex]}
+      </p>
+    </div>
+  );
+}
+
 export function AIAssistPanel({
   title,
   description,
@@ -87,7 +126,7 @@ export function AIAssistPanel({
 
   if (!result) {
     return (
-      <div className="pt-2">
+      <div className="pt-2 space-y-2">
         <Button
           type="button"
           variant="outline"
@@ -96,17 +135,16 @@ export function AIAssistPanel({
           disabled={loading || (!title.trim() && !description.trim())}
           className="w-full"
         >
-          {loading ? (
-            <Loader2 className="h-3 w-3 mr-1 animate-spin" />
-          ) : (
-            <Sparkles className="h-3 w-3 mr-1" />
-          )}
-          {loading ? "Analyzing..." : "AI Assist"}
+          <Sparkles className="h-3 w-3 mr-1" />
+          AI Assist
         </Button>
+        <ProgressBar loading={loading} />
         {error && <p className="text-xs text-destructive mt-1">{error}</p>}
-        <p className="text-xs text-muted-foreground text-center mt-1.5">
-          Suggests improved descriptions, sub-task breakdowns, and workflow patterns
-        </p>
+        {!loading && (
+          <p className="text-xs text-muted-foreground text-center mt-1.5">
+            Suggests improved descriptions, sub-task breakdowns, and workflow patterns
+          </p>
+        )}
       </div>
     );
   }