From 16ee7133addf3bc428c73a2bed4db40baf8117ec Mon Sep 17 00:00:00 2001
From: suguanYang <wangbinqi77@gmail.com>
Date: Fri, 21 Aug 2026 01:04:35 +0800
Subject: [PATCH] fix: keep Official Library demo sources selected on citation
 click

Tesla and SpaceX seed citations point at catalog rows tagged as Official
Library. Resolving those ids back to null left the parsed panel on the
guest empty login state instead of the cited chunk.

Co-authored-by: Cursor <cursoragent@cursor.com>
---
 src/components/workspace-shell.test.ts        | 91 +++++++++++++++++++
 src/components/workspace-source-state.test.ts |  4 +-
 src/components/workspace-source-state.ts      |  2 +-
 3 files changed, 94 insertions(+), 3 deletions(-)

diff --git a/src/components/workspace-shell.test.ts b/src/components/workspace-shell.test.ts
index e8a4fcb..660471f 100644
--- a/src/components/workspace-shell.test.ts
+++ b/src/components/workspace-shell.test.ts
@@ -449,6 +449,97 @@ describe("WorkspaceShell", () => {
     ).toBe(false);
   });
 
+  it("focuses guest Official Library demo citations instead of the empty login state", async () => {
+    const fetch = vi.fn<typeof globalThis.fetch>(async (input) => {
+      const url = getRequestURL(input);
+
+      if (url.pathname === "/api/sources/demo-tsla-q4-2025/chunks") {
+        return Response.json({
+          chunks: [
+            {
+              chunkId: "demo-tsla-q4-2025:chunk_1",
+              documentId: "demo-doc-tsla-q4-2025",
+              sectionPath: "TSLA-Q4-2025-Update.pdf/OTHER UPDATES",
+              type: "text",
+              content: "Tesla entered into an agreement to invest approximately",
+              sourceTitle: "TSLA-Q4-2025-Update.pdf",
+            },
+          ],
+          pagination: {
+            page: Number(url.searchParams.get("page") ?? "1"),
+            pageSize: 100,
+            total: 1,
+            totalPages: 1,
+          },
+        });
+      }
+
+      return Response.json({ message: "Unexpected request" }, { status: 404 });
+    });
+    vi.stubGlobal("fetch", fetch);
+
+    render(
+      React.createElement(C, {
+        isGuest: true,
+        sources: [
+          {
+            id: "demo-tsla-q4-2025",
+            kind: "demo",
+            demoSourceId: "demo-tsla-q4-2025",
+            title: "TSLA-Q4-2025-Update.pdf",
+            status: "ready",
+            mimeType: "application/pdf",
+            documentId: "demo-doc-tsla-q4-2025",
+            officialLibrary: {
+              librarySourceId: "financial-tsla-q4-2025",
+              categoryId: "financial-reports",
+              sourceUrl: "https://example.com/tsla-q4-2025.pdf",
+            },
+          },
+        ],
+        chatMessages: [
+          {
+            id: "assistant_1",
+            role: "assistant",
+            content: "Tesla invested in xAI. [[cite:1]]",
+            citations: [
+              {
+                content: "Tesla entered into an agreement to invest approximately",
+                description: "xAI investment",
+                chunkType: "text",
+                score: 0.95,
+                pageCitationPageNumber: 12,
+                source: {
+                  documentId: "demo-doc-tsla-q4-2025",
+                  sourceFileName: "TSLA-Q4-2025-Update.pdf",
+                  sectionPath: "TSLA-Q4-2025-Update.pdf/OTHER UPDATES",
+                },
+              },
+            ],
+          },
+        ],
+      }),
+    );
+
+    const citationButton = await findStableConnectedElement(() => {
+      const desktopChatPanel = within(screen.getByTestId("desktop-chat-panel"));
+      return desktopChatPanel.getByRole("button", {
+        name: "Open source TSLA-Q4-2025-Update.pdf/p12",
+      });
+    });
+    fireEvent.click(citationButton);
+
+    await waitFor(() => {
+      const chunksPanel = screen.getByTestId("desktop-chunks-panel");
+      expect(within(chunksPanel).queryByText("Log in to add documents")).toBeNull();
+      const topRow = chunksPanel.querySelector<HTMLElement>('[data-index="0"]');
+      expect(topRow?.getAttribute("data-chunk-id")).toBe(
+        "demo-tsla-q4-2025:chunk_1",
+      );
+      expect(topRow?.getAttribute("data-focused-chunk")).toBe("true");
+    });
+  });
+
   it("focuses guest citations from the mobile chat panel", async () => {
     const fetch = vi.fn<typeof globalThis.fetch>(async (input) => {
       const url = getRequestURL(input);
diff --git a/src/components/workspace-source-state.test.ts b/src/components/workspace-source-state.test.ts
index 5345646..03d9319 100644
--- a/src/components/workspace-source-state.test.ts
+++ b/src/components/workspace-source-state.test.ts
@@ -58,7 +58,7 @@ describe("workspaceSourceState", () => {
     );
   });
 
-  it("does not resolve an unmaterialized Official Library row as the selected Source", () => {
+  it("keeps an explicitly selected Official Library demo source for citation jumps", () => {
     const sources: readonly SourceView[] = [
       {
         id: "demo-spacex-s1",
@@ -88,7 +88,7 @@ describe("workspaceSourceState", () => {
         sources,
         "demo-spacex-s1",
       ),
-    ).toBeNull();
+    ).toBe("demo-spacex-s1");
   });
 
   it("selects a preferred document source when opening a chunk-tree link", () => {
diff --git a/src/components/workspace-source-state.ts b/src/components/workspace-source-state.ts
index 5d33744..41607f3 100644
--- a/src/components/workspace-source-state.ts
+++ b/src/components/workspace-source-state.ts
@@ -74,7 +74,7 @@ function getResolvedSelectedSourceId(
 
   const selectedSource = sources.find((source) => source.id === selectedSourceId)
   if (selectedSource) {
-    return isVisibleSource(selectedSource) ? selectedSource.id : null
+    return selectedSource.id
   }
 
   const selectedDocumentId = getRemoteSourceDocumentId(selectedSourceId)