scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-09-07 08:26:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add default source on first load

Browse source
This commit is contained in:
Smartsheet-JB-Brown 2025-04-14 23:00:21 -07:00
parent ad91533205
commit 75dce909f0
2 changed files with 60 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
webview-ui/src/components/package-manager/PackageManagerViewStateManager.ts
View file

@ -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" })

42
webview-ui/src/components/package-manager/__tests__/PackageManagerViewStateManager.test.ts
View file

@ -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" })