From 75dce909f018fed5310acea50afcfd37c21635c1 Mon Sep 17 00:00:00 2001
From: Smartsheet-JB-Brown <jb.brown@smartsheet.com>
Date: Mon, 14 Apr 2025 23:00:21 -0700
Subject: [PATCH] Add default source on first load

---
 .../PackageManagerViewStateManager.ts         | 20 ++++++++-
 .../PackageManagerViewStateManager.test.ts    | 42 ++++++++++++++++++-
 2 files changed, 60 insertions(+), 2 deletions(-)

diff --git a/webview-ui/src/components/package-manager/PackageManagerViewStateManager.ts b/webview-ui/src/components/package-manager/PackageManagerViewStateManager.ts
index e7b9f956f7..b4338f9d2b 100644
--- a/webview-ui/src/components/package-manager/PackageManagerViewStateManager.ts
+++ b/webview-ui/src/components/package-manager/PackageManagerViewStateManager.ts
@@ -53,7 +53,7 @@ export class PackageManagerViewStateManager {
 			isFetching: false,
 			activeTab: "browse",
 			refreshingUrls: [],
-			sources: [],
+			sources: [DEFAULT_PACKAGE_MANAGER_SOURCE],
 			filters: {
 				type: "",
 				search: "",
@@ -66,6 +66,14 @@ export class PackageManagerViewStateManager {
 		}
 	}
 
+	public initialize(): void {
+		// Send initial sources to extension
+		vscode.postMessage({
+			type: "packageManagerSources",
+			sources: [DEFAULT_PACKAGE_MANAGER_SOURCE],
+		} as WebviewMessage)
+	}
+
 	public onStateChange(handler: StateChangeHandler): () => void {
 		this.stateChangeHandlers.add(handler)
 		return () => this.stateChangeHandlers.delete(handler)
@@ -192,6 +200,16 @@ export class PackageManagerViewStateManager {
 			case "SET_ACTIVE_TAB": {
 				const { tab } = transition.payload as TransitionPayloads["SET_ACTIVE_TAB"]
 				this.state.activeTab = tab
+
+				// Add default source when switching to sources tab if no sources exist
+				if (tab === "sources" && this.state.sources.length === 0) {
+					this.state.sources = [DEFAULT_PACKAGE_MANAGER_SOURCE]
+					vscode.postMessage({
+						type: "packageManagerSources",
+						sources: [DEFAULT_PACKAGE_MANAGER_SOURCE],
+					} as WebviewMessage)
+				}
+
 				this.notifyStateChange()
 				if (tab === "browse") {
 					void this.transition({ type: "FETCH_ITEMS" })
diff --git a/webview-ui/src/components/package-manager/__tests__/PackageManagerViewStateManager.test.ts b/webview-ui/src/components/package-manager/__tests__/PackageManagerViewStateManager.test.ts
index 4c8aa86b29..6b70664bb3 100644
--- a/webview-ui/src/components/package-manager/__tests__/PackageManagerViewStateManager.test.ts
+++ b/webview-ui/src/components/package-manager/__tests__/PackageManagerViewStateManager.test.ts
@@ -5,6 +5,7 @@ import {
 	PackageManagerItem,
 	PackageManagerSource,
 } from "../../../../../src/services/package-manager/types"
+import { DEFAULT_PACKAGE_MANAGER_SOURCE } from "../../../../../src/services/package-manager/constants"
 
 const createTestItem = (overrides = {}): PackageManagerItem => ({
 	name: "test",
@@ -39,6 +40,7 @@ describe("PackageManagerViewStateManager", () => {
 		jest.clearAllMocks()
 		jest.useFakeTimers()
 		manager = new PackageManagerViewStateManager()
+		manager.initialize() // Send initial sources
 	})
 
 	afterEach(() => {
@@ -54,7 +56,7 @@ describe("PackageManagerViewStateManager", () => {
 				isFetching: false,
 				activeTab: "browse",
 				refreshingUrls: [],
-				sources: [],
+				sources: [DEFAULT_PACKAGE_MANAGER_SOURCE],
 				filters: {
 					type: "",
 					search: "",
@@ -66,10 +68,45 @@ describe("PackageManagerViewStateManager", () => {
 				},
 			})
 		})
+
+		it("should send initial sources when initialized", () => {
+			manager.initialize()
+			expect(vscode.postMessage).toHaveBeenCalledWith({
+				type: "packageManagerSources",
+				sources: [DEFAULT_PACKAGE_MANAGER_SOURCE],
+			})
+		})
+
+		it("should initialize with default source", () => {
+			const manager = new PackageManagerViewStateManager()
+
+			// Initial state should include default source
+			const state = manager.getState()
+			expect(state.sources).toEqual([
+				{
+					url: "https://github.com/RooVetGit/Roo-Code-Packages",
+					name: "Roo Code Package Manager Template",
+					enabled: true,
+				},
+			])
+
+			// Verify initial message was sent to update sources
+			expect(vscode.postMessage).toHaveBeenCalledWith({
+				type: "packageManagerSources",
+				sources: [
+					{
+						url: "https://github.com/RooVetGit/Roo-Code-Packages",
+						name: "Roo Code Package Manager Template",
+						enabled: true,
+					},
+				],
+			})
+		})
 	})
 
 	describe("Fetch Transitions", () => {
 		it("should handle FETCH_ITEMS transition", async () => {
+			jest.clearAllMocks() // Clear mock to ignore initialize() call
 			await manager.transition({ type: "FETCH_ITEMS" })
 
 			expect(vscode.postMessage).toHaveBeenCalledWith({
@@ -82,6 +119,7 @@ describe("PackageManagerViewStateManager", () => {
 		})
 
 		it("should not start a new fetch if one is in progress", async () => {
+			jest.clearAllMocks() // Clear mock to ignore initialize() call
 			// Start first fetch
 			await manager.transition({ type: "FETCH_ITEMS" })
 
@@ -491,6 +529,8 @@ describe("PackageManagerViewStateManager", () => {
 		})
 
 		it("should prevent concurrent fetches during timeout period", async () => {
+			jest.clearAllMocks() // Clear mock to ignore initialize() call
+
 			// Start first fetch
 			await manager.transition({ type: "FETCH_ITEMS" })