Move goose2 (#8516)

Signed-off-by: Jack Amadeo <jackamadeo@squareup.com> Co-authored-by: block-open-source[bot] <201011344+block-open-source[bot]@users.noreply.github.com> Co-authored-by: block-open-source[bot] <1159699+block-open-source[bot]@users.noreply.github.com> Co-authored-by: Wes <wesbillman@users.noreply.github.com> Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com> Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com> Co-authored-by: Matt Toohey <contact@matttoohey.com> Co-authored-by: tulsi <tulsi@block.xyz> Co-authored-by: morgmart <98432065+morgmart@users.noreply.github.com> Co-authored-by: Bradley Axen <baxen@squareup.com> Co-authored-by: Alex Hancock <alexhancock@block.xyz> Co-authored-by: Taylor Ho <taylorkmho@gmail.com> Co-authored-by: Nahiyan Khan <nahiyan.khan@gmail.com> Co-authored-by: Lifei Zhou <lifei@squareup.com>
2026-04-14 10:39:30 -04:00
parent 17a404d4f9
commit 482f1962c1
503 changed files with 89093 additions and 105 deletions
@@ -0,0 +1,194 @@
+import { describe, expect, it } from "vitest";
+import {
+  buildProjectSystemPrompt,
+  composeSystemPrompt,
+  defaultArtifactsDir,
+  getProjectArtifactRoots,
+  getProjectFolderName,
+  getProjectFolderOption,
+  resolveEffectiveWorkingDir,
+  resolveProjectWorkingDir,
+} from "./chatProjectContext";
+
+describe("chatProjectContext", () => {
+  it("builds project instructions from stored project settings", () => {
+    const systemPrompt = buildProjectSystemPrompt({
+      id: "project-1",
+      name: "Goose2",
+      description: "Desktop app",
+      prompt: "Always read AGENTS.md before editing.",
+      icon: "folder",
+      color: "#000000",
+      preferredProvider: "goose",
+      preferredModel: "claude-sonnet-4",
+      workingDirs: ["/Users/wesb/dev/goose2"],
+      artifactsDir: "/Users/wesb/.goose/projects/goose2/artifacts",
+      useWorktrees: true,
+      order: 0,
+      archivedAt: null,
+      createdAt: "now",
+      updatedAt: "now",
+    });
+
+    expect(systemPrompt).toContain("<project-settings>");
+    expect(systemPrompt).toContain("Project name: Goose2");
+    expect(systemPrompt).toContain(
+      "Working directories: /Users/wesb/dev/goose2",
+    );
+    expect(systemPrompt).toContain(
+      "Artifact directory: /Users/wesb/dev/goose2/artifacts",
+    );
+    expect(systemPrompt).toContain("Preferred provider: goose");
+    expect(systemPrompt).toContain(
+      "Use git worktrees for branch isolation: yes",
+    );
+    expect(systemPrompt).toContain("<project-file-policy>");
+    expect(systemPrompt).toContain(
+      "Write newly generated files to /Users/wesb/dev/goose2/artifacts by default.",
+    );
+    expect(systemPrompt).toContain("<project-instructions>");
+    expect(systemPrompt).toContain("Always read AGENTS.md before editing.");
+  });
+
+  it("combines persona and project prompts without empty sections", () => {
+    expect(
+      composeSystemPrompt("Persona prompt", undefined, "Project prompt"),
+    ).toBe("Persona prompt\n\nProject prompt");
+  });
+
+  it("extracts the folder name from a path", () => {
+    expect(getProjectFolderName("/Users/wesb/dev/goose2")).toBe("goose2");
+    expect(getProjectFolderName("C:\\Users\\wesb\\goose2\\")).toBe("goose2");
+  });
+
+  it("creates folder options from the project's working directories", () => {
+    expect(
+      getProjectFolderOption({
+        workingDirs: ["/Users/wesb/dev/goose2", "/Users/wesb/dev/other"],
+        artifactsDir: "/Users/wesb/.goose/projects/goose2/artifacts",
+      }),
+    ).toEqual([
+      {
+        id: "/Users/wesb/dev/goose2/artifacts",
+        name: "artifacts",
+        path: "/Users/wesb/dev/goose2/artifacts",
+      },
+      {
+        id: "/Users/wesb/dev/other/artifacts",
+        name: "artifacts",
+        path: "/Users/wesb/dev/other/artifacts",
+      },
+    ]);
+  });
+
+  it("returns an empty array when workingDirs is empty", () => {
+    expect(
+      getProjectFolderOption({
+        workingDirs: [],
+        artifactsDir: "/Users/wesb/.goose/projects/sample-project/artifacts",
+      }),
+    ).toEqual([
+      {
+        id: "/Users/wesb/.goose/projects/sample-project/artifacts",
+        name: "artifacts",
+        path: "/Users/wesb/.goose/projects/sample-project/artifacts",
+      },
+    ]);
+  });
+
+  it("returns an empty array when project is null", () => {
+    expect(getProjectFolderOption(null)).toEqual([]);
+  });
+
+  it("returns only artifact subdirectories for working dirs", () => {
+    expect(
+      getProjectArtifactRoots({
+        workingDirs: ["/Users/wesb/dev/goose2", "/Users/wesb/dev/other"],
+        artifactsDir: "/Users/wesb/.goose/projects/goose2/artifacts",
+      }),
+    ).toEqual([
+      "/Users/wesb/dev/goose2/artifacts",
+      "/Users/wesb/dev/other/artifacts",
+    ]);
+  });
+
+  it("resolves the first working directory to an artifacts subfolder", () => {
+    expect(
+      resolveProjectWorkingDir({
+        workingDirs: ["/Users/wesb/dev/goose2", "/Users/wesb/dev/other"],
+        artifactsDir: "/Users/wesb/.goose/projects/goose2/artifacts",
+      }),
+    ).toBe("/Users/wesb/dev/goose2/artifacts");
+  });
+
+  it("falls back to the project artifacts dir when no working dirs exist", () => {
+    expect(
+      resolveProjectWorkingDir({
+        workingDirs: [],
+        artifactsDir: "/Users/wesb/.goose/projects/sample-project/artifacts",
+      }),
+    ).toBe("/Users/wesb/.goose/projects/sample-project/artifacts");
+  });
+
+  describe("defaultArtifactsDir", () => {
+    it("normalises path separators and appends .goose/artifacts", () => {
+      expect(defaultArtifactsDir("/Users/wesb")).toBe(
+        "/Users/wesb/.goose/artifacts",
+      );
+    });
+
+    it("normalises backslashes on Windows-style paths", () => {
+      expect(defaultArtifactsDir("C:\\Users\\wesb\\")).toBe(
+        "C:/Users/wesb/.goose/artifacts",
+      );
+    });
+
+    it("strips trailing slashes", () => {
+      expect(defaultArtifactsDir("/Users/wesb/")).toBe(
+        "/Users/wesb/.goose/artifacts",
+      );
+    });
+  });
+
+  describe("resolveEffectiveWorkingDir", () => {
+    it("returns the project working dir without requiring homeDir", () => {
+      expect(
+        resolveEffectiveWorkingDir({
+          workingDirs: ["/Users/wesb/dev/goose2"],
+          artifactsDir: "/Users/wesb/.goose/projects/goose2/artifacts",
+        }),
+      ).toBe("/Users/wesb/dev/goose2/artifacts");
+    });
+
+    it("returns the project working dir when available", () => {
+      expect(
+        resolveEffectiveWorkingDir(
+          {
+            workingDirs: ["/Users/wesb/dev/goose2"],
+            artifactsDir: "/Users/wesb/.goose/projects/goose2/artifacts",
+          },
+          "/Users/wesb",
+        ),
+      ).toBe("/Users/wesb/dev/goose2/artifacts");
+    });
+
+    it("returns undefined when a project exists but has no working dirs", () => {
+      expect(
+        resolveEffectiveWorkingDir(
+          { workingDirs: [], artifactsDir: "" },
+          "/Users/wesb",
+        ),
+      ).toBeUndefined();
+    });
+
+    it("falls back to home artifacts dir when no project", () => {
+      expect(resolveEffectiveWorkingDir(null, "/Users/wesb")).toBe(
+        "/Users/wesb/.goose/artifacts",
+      );
+    });
+
+    it("does not resolve a non-project fallback without homeDir", () => {
+      expect(resolveEffectiveWorkingDir(null)).toBeUndefined();
+    });
+  });
+});