Add a global New workspace picker (#1324)

* Add global New workspace project picker * Clean up New workspace picker state
2026-06-04 15:16:41 +08:00
parent 0b62e3bd7f
commit 38f8179bc1
28 changed files with 1392 additions and 424 deletions
@@ -0,0 +1,205 @@
+import { describe, expect, it } from "vitest";
+import type { WorkspaceDescriptor } from "@/stores/session-store";
+import type { WorkspaceStructureProject } from "@/projects/workspace-structure";
+import {
+  buildHostProjectList,
+  canCreateWorktreeForProjectKind,
+  hostProjectFromRoute,
+  hostProjectFromWorkspace,
+  resolveInitialWorktreeProject,
+  resolveSelectedHostProject,
+  type HostProjectListItem,
+} from "./host-project-model";
+
+function structureProject(input: Partial<WorkspaceStructureProject>): WorkspaceStructureProject {
+  return {
+    projectKey: input.projectKey ?? "project-a",
+    projectName: input.projectName ?? "Project A",
+    projectKind: input.projectKind ?? "git",
+    iconWorkingDir: input.iconWorkingDir ?? "/repo/a",
+    workspaceKeys: input.workspaceKeys ?? ["workspace-a"],
+  };
+}
+
+function hostProject(input: Partial<HostProjectListItem>): HostProjectListItem {
+  return {
+    serverId: input.serverId ?? "host-a",
+    projectKey: input.projectKey ?? "project-a",
+    projectName: input.projectName ?? "Project A",
+    projectKind: input.projectKind ?? "git",
+    iconWorkingDir: input.iconWorkingDir ?? "/repo/a",
+    workspaceKeys: input.workspaceKeys ?? ["workspace-a"],
+    canCreateWorktree: input.canCreateWorktree ?? true,
+  };
+}
+
+function workspace(input: Partial<WorkspaceDescriptor>): WorkspaceDescriptor {
+  return {
+    id: input.id ?? "workspace-a",
+    projectId: input.projectId ?? "project-a",
+    projectDisplayName: input.projectDisplayName ?? "Project A",
+    projectRootPath: input.projectRootPath ?? "/repo/a",
+    workspaceDirectory: input.workspaceDirectory ?? "/repo/a",
+    projectKind: input.projectKind ?? "git",
+    workspaceKind: input.workspaceKind ?? "local_checkout",
+    name: input.name ?? "main",
+    status: input.status ?? "done",
+    statusEnteredAt: input.statusEnteredAt ?? null,
+    archivingAt: input.archivingAt ?? null,
+    diffStat: input.diffStat ?? null,
+    scripts: input.scripts ?? [],
+  };
+}
+
+const routeProject = hostProject({
+  projectKey: "route-project",
+  projectName: "Route Project",
+  iconWorkingDir: "/repo/route",
+});
+const lastActiveProject = hostProject({
+  projectKey: "last-project",
+  projectName: "Last Project",
+  iconWorkingDir: "/repo/last",
+});
+const firstProject = hostProject({
+  projectKey: "first-project",
+  projectName: "First Project",
+  iconWorkingDir: "/repo/first",
+});
+
+describe("host project list", () => {
+  it("preserves workspace-structure order and project metadata", () => {
+    expect(
+      buildHostProjectList({
+        serverId: "host-a",
+        projects: [
+          structureProject({
+            projectKey: "project-b",
+            projectName: "Project B",
+            projectKind: "directory",
+            iconWorkingDir: "/repo/b",
+            workspaceKeys: ["workspace-b"],
+          }),
+          structureProject({
+            projectKey: "project-a",
+            projectName: "Project A",
+            projectKind: "git",
+            iconWorkingDir: "/repo/a",
+            workspaceKeys: ["workspace-a"],
+          }),
+        ],
+      }),
+    ).toEqual([
+      {
+        serverId: "host-a",
+        projectKey: "project-b",
+        projectName: "Project B",
+        projectKind: "directory",
+        iconWorkingDir: "/repo/b",
+        workspaceKeys: ["workspace-b"],
+        canCreateWorktree: false,
+      },
+      {
+        serverId: "host-a",
+        projectKey: "project-a",
+        projectName: "Project A",
+        projectKind: "git",
+        iconWorkingDir: "/repo/a",
+        workspaceKeys: ["workspace-a"],
+        canCreateWorktree: true,
+      },
+    ]);
+  });
+
+  it("keeps worktree capability separate from project listability", () => {
+    expect(canCreateWorktreeForProjectKind("git")).toBe(true);
+    expect(canCreateWorktreeForProjectKind("directory")).toBe(false);
+  });
+
+  it("uses route project before last active project when it can create worktrees", () => {
+    expect(
+      resolveInitialWorktreeProject({
+        routeProject,
+        lastActiveProject,
+        projects: [firstProject],
+      }),
+    ).toEqual(routeProject);
+  });
+
+  it("skips non-worktree route and last-active projects", () => {
+    expect(
+      resolveInitialWorktreeProject({
+        routeProject: { ...routeProject, projectKind: "directory", canCreateWorktree: false },
+        lastActiveProject: {
+          ...lastActiveProject,
+          projectKind: "directory",
+          canCreateWorktree: false,
+        },
+        projects: [
+          { ...firstProject, projectKind: "directory", canCreateWorktree: false },
+          hostProject({ projectKey: "git-project", projectName: "Git Project" }),
+        ],
+      }),
+    ).toMatchObject({ projectKey: "git-project" });
+  });
+
+  it("leaves the project empty when no worktree-capable project is available", () => {
+    expect(
+      resolveInitialWorktreeProject({
+        routeProject: null,
+        lastActiveProject: null,
+        projects: [{ ...firstProject, projectKind: "directory", canCreateWorktree: false }],
+      }),
+    ).toBeNull();
+  });
+
+  it("keeps a selected route project available before project hydration", () => {
+    expect(
+      resolveSelectedHostProject({
+        selectedProjectKey: routeProject.projectKey,
+        projects: [],
+        routeProject,
+        lastActiveProject: null,
+      }),
+    ).toEqual(routeProject);
+  });
+
+  it("converts route project only when it has a key and source directory", () => {
+    expect(
+      hostProjectFromRoute({
+        serverId: "host-a",
+        projectId: "project-a",
+        displayName: "Project A",
+        sourceDirectory: "/repo/a",
+      }),
+    ).toEqual({
+      serverId: "host-a",
+      projectKey: "project-a",
+      projectName: "Project A",
+      projectKind: "git",
+      iconWorkingDir: "/repo/a",
+      workspaceKeys: [],
+      canCreateWorktree: true,
+    });
+    expect(hostProjectFromRoute({ serverId: "host-a", projectId: "project-a" })).toBeNull();
+  });
+
+  it("converts last active workspaces with matching worktree capability", () => {
+    expect(hostProjectFromWorkspace({ serverId: "host-a", workspace: workspace({}) })).toEqual({
+      serverId: "host-a",
+      projectKey: "project-a",
+      projectName: "Project A",
+      projectKind: "git",
+      iconWorkingDir: "/repo/a",
+      workspaceKeys: ["workspace-a"],
+      canCreateWorktree: true,
+    });
+
+    expect(
+      hostProjectFromWorkspace({
+        serverId: "host-a",
+        workspace: workspace({ projectKind: "directory" }),
+      }),
+    ).toMatchObject({ projectKind: "directory", canCreateWorktree: false });
+  });
+});