feat(settings): add rules management UI (Zoo-Code-Org#657)

* feat(settings): add rules management UI

* fix: address rules settings review feedback

* test: increase rules settings coverage

* test: improve rules settings patch coverage

* fix: address rules settings review feedback

Author: ivanarifin

packages/types/src/index.ts

@@ -22,6 +22,7 @@ export * from "./provider-settings.js"
 export * from "./task.js"
 export * from "./todo.js"
 export * from "./skills.js"
+export * from "./rules.js"
 export * from "./marketplace.js"
 export * from "./telemetry.js"
 export * from "./terminal.js"

packages/types/src/rules.ts

@@ -0,0 +1,34 @@
+export type RuleScope = "global" | "project"
+
+export type RuleKind = "generic" | "mode"
+
+export interface RuleMetadata {
+	id: string
+	name: string
+	scope: RuleScope
+	kind: RuleKind
+	modeSlug?: string
+	modeName?: string
+	filePath: string
+	relativePath: string
+	directoryPath: string
+	description?: string
+	isSymlink?: boolean
+}
+
+export interface CreateRuleInput {
+	scope: RuleScope
+	kind: RuleKind
+	modeSlug?: string
+	fileName: string
+}
+
+export interface DeleteRuleInput {
+	id?: string
+	scope: RuleScope
+	kind: RuleKind
+	modeSlug?: string
+	relativePath: string
+}
+
+export type RuleLookupInput = DeleteRuleInput

packages/types/src/vscode-extension-host.ts

@@ -15,6 +15,7 @@ import type { McpServer } from "./mcp.js"
 import type { ModelInfo, ModelRecord, RouterModels } from "./model.js"
 import type { OpenAiCodexRateLimitInfo } from "./providers/openai-codex-rate-limits.js"
 import type { SkillMetadata } from "./skills.js"
+import type { RuleMetadata } from "./rules.js"
 import type { TelemetrySetting } from "./telemetry.js"
 import type { WorktreeIncludeStatus } from "./worktree.js"
 
@@ -99,6 +100,7 @@ export interface ExtensionMessage {
 		| "branchWorktreeIncludeResult"
 		| "folderSelected"
 		| "skills"
+		| "rules"
 		| "fileContent"
 		| "rooHistoryImportProgress"
 	text?: string
@@ -178,6 +180,7 @@ export interface ExtensionMessage {
 	list?: string[] // For dismissedUpsells
 	tools?: SerializedCustomToolDefinition[] // For customToolsResult
 	skills?: SkillMetadata[] // For skills response
+	rules?: RuleMetadata[] // For rules response
 	modes?: { slug: string; name: string }[] // For modes response
 	rooHistoryImportProgress?: {
 		status: "starting" | "copying" | "finished" | "failed"
@@ -613,6 +616,12 @@ export interface WebviewMessage {
 		| "moveSkill"
 		| "updateSkillModes"
 		| "openSkillFile"
+		// Rules messages
+		| "requestRules"
+		| "createRule"
+		| "deleteRule"
+		| "openRuleFile"
+		| "openRulesDirectory"
 	text?: string
 	taskId?: string
 	editedMessageContent?: string

src/core/webview/__tests__/rulesMessageHandler.spec.ts

@@ -0,0 +1,291 @@
+import type { RuleMetadata, WebviewMessage } from "@roo-code/types"
+import type { ClineProvider } from "../ClineProvider"
+
+vi.mock("vscode", () => ({
+	window: {
+		showErrorMessage: vi.fn(),
+		showWarningMessage: vi.fn(),
+	},
+}))
+
+vi.mock("../../../integrations/misc/open-file", () => ({
+	openFile: vi.fn(),
+}))
+
+vi.mock("../../../services/rules/rules", () => ({
+	getRules: vi.fn(),
+	createRule: vi.fn(),
+	deleteRule: vi.fn(),
+	resolveRuleFile: vi.fn(),
+	getRulesDirectoryPath: vi.fn(),
+}))
+
+import * as vscode from "vscode"
+import { openFile } from "../../../integrations/misc/open-file"
+import { createRule, deleteRule, getRules, getRulesDirectoryPath, resolveRuleFile } from "../../../services/rules/rules"
+import {
+	handleCreateRule,
+	handleDeleteRule,
+	handleOpenRuleFile,
+	handleOpenRulesDirectory,
+	handleRequestRules,
+} from "../rulesMessageHandler"
+
+const mockRules: RuleMetadata[] = [
+	{
+		id: "global:generic:generic:rule.md",
+		name: "rule.md",
+		scope: "global",
+		kind: "generic",
+		filePath: "/home/.roo/rules/rule.md",
+		relativePath: "rule.md",
+		directoryPath: "/home/.roo/rules",
+	},
+]
+
+describe("rulesMessageHandler", () => {
+	const mockLog = vi.fn()
+	const mockPostMessageToWebview = vi.fn()
+	const mockGetModes = vi.fn()
+
+	const createMockProvider = (): ClineProvider =>
+		({
+			log: mockLog,
+			postMessageToWebview: mockPostMessageToWebview,
+			getModes: mockGetModes,
+		}) as unknown as ClineProvider
+
+	beforeEach(() => {
+		vi.clearAllMocks()
+		mockGetModes.mockResolvedValue([{ slug: "code", name: "Code" }])
+		vi.mocked(getRules).mockResolvedValue(mockRules)
+	})
+
+	it("handleRequestRules posts rules", async () => {
+		const provider = createMockProvider()
+
+		const result = await handleRequestRules(provider, "/workspace")
+
+		expect(result).toEqual(mockRules)
+		expect(getRules).toHaveBeenCalledWith("/workspace", { modes: [{ slug: "code", name: "Code" }] })
+		expect(mockPostMessageToWebview).toHaveBeenCalledWith({ type: "rules", rules: mockRules })
+	})
+
+	it("handleRequestRules posts an empty list on failure", async () => {
+		const provider = createMockProvider()
+		vi.mocked(getRules).mockRejectedValue(new Error("scan failed"))
+
+		const result = await handleRequestRules(provider, "/workspace")
+
+		expect(result).toEqual([])
+		expect(mockLog).toHaveBeenCalled()
+		expect(mockPostMessageToWebview).toHaveBeenCalledWith({ type: "rules", rules: [] })
+	})
+
+	it("handleCreateRule creates, opens, and posts refreshed rules", async () => {
+		const provider = createMockProvider()
+		vi.mocked(createRule).mockResolvedValue("/workspace/.roo/rules/new.md")
+
+		const result = await handleCreateRule(provider, "/workspace", {
+			type: "createRule",
+			values: { scope: "project", kind: "generic", fileName: "new.md" },
+		} as WebviewMessage)
+
+		expect(result).toEqual(mockRules)
+		expect(createRule).toHaveBeenCalledWith("/workspace", { scope: "project", kind: "generic", fileName: "new.md" })
+		expect(openFile).toHaveBeenCalledWith("/workspace/.roo/rules/new.md")
+		expect(mockPostMessageToWebview).toHaveBeenCalledWith({ type: "rules", rules: mockRules })
+	})
+
+	it("handleDeleteRule deletes and posts refreshed rules", async () => {
+		const provider = createMockProvider()
+
+		const result = await handleDeleteRule(provider, "/workspace", {
+			type: "deleteRule",
+			values: { scope: "global", kind: "generic", relativePath: "rule.md" },
+		} as WebviewMessage)
+
+		expect(result).toEqual(mockRules)
+		expect(deleteRule).toHaveBeenCalledWith("/workspace", {
+			scope: "global",
+			kind: "generic",
+			relativePath: "rule.md",
+		})
+		expect(mockPostMessageToWebview).toHaveBeenCalledWith({ type: "rules", rules: mockRules })
+	})
+
+	it("handleOpenRuleFile safely resolves and opens the rule file", async () => {
+		const provider = createMockProvider()
+		vi.mocked(resolveRuleFile).mockResolvedValue("/workspace/.roo/rules/rule.md")
+
+		await handleOpenRuleFile(provider, "/workspace", {
+			type: "openRuleFile",
+			values: { scope: "project", kind: "generic", relativePath: "rule.md" },
+		} as WebviewMessage)
+
+		expect(resolveRuleFile).toHaveBeenCalledWith("/workspace", {
+			scope: "project",
+			kind: "generic",
+			relativePath: "rule.md",
+		})
+		expect(openFile).toHaveBeenCalledWith("/workspace/.roo/rules/rule.md")
+	})
+
+	it("handleDeleteRule shows an error when required delete values are missing", async () => {
+		const provider = createMockProvider()
+
+		const result = await handleDeleteRule(provider, "/workspace", {
+			type: "deleteRule",
+			values: { scope: "global", kind: "generic" },
+		} as WebviewMessage)
+
+		expect(result).toBeUndefined()
+		expect(deleteRule).not.toHaveBeenCalled()
+		expect(vscode.window.showErrorMessage).toHaveBeenCalledWith(
+			"Failed to delete rule: Missing required fields: scope, kind, or relativePath",
+		)
+	})
+
+	it("handleOpenRuleFile shows an error when the resolved rule file is missing", async () => {
+		const provider = createMockProvider()
+		vi.mocked(resolveRuleFile).mockResolvedValue(undefined)
+
+		await handleOpenRuleFile(provider, "/workspace", {
+			type: "openRuleFile",
+			values: { scope: "project", kind: "generic", relativePath: "missing.md" },
+		} as WebviewMessage)
+
+		expect(openFile).not.toHaveBeenCalled()
+		expect(vscode.window.showErrorMessage).toHaveBeenCalledWith("Failed to open rule file: Rule file not found")
+	})
+
+	it("handleOpenRulesDirectory shows an error when directory input is invalid", async () => {
+		const provider = createMockProvider()
+		vi.mocked(getRulesDirectoryPath).mockImplementation(() => {
+			throw new Error("Invalid rule scope")
+		})
+
+		await handleOpenRulesDirectory(provider, "/workspace", {
+			type: "openRulesDirectory",
+			values: { scope: "team", kind: "generic" },
+		} as WebviewMessage)
+
+		expect(openFile).not.toHaveBeenCalled()
+		expect(vscode.window.showErrorMessage).toHaveBeenCalledWith(
+			"Failed to open rules directory: Invalid rule scope",
+		)
+	})
+
+	it("handleCreateRule shows an error for missing required values before creating", async () => {
+		const provider = createMockProvider()
+
+		const result = await handleCreateRule(provider, "/workspace", {
+			type: "createRule",
+			values: { scope: "project", kind: "generic" },
+		} as WebviewMessage)
+
+		expect(result).toBeUndefined()
+		expect(createRule).not.toHaveBeenCalled()
+		expect(vscode.window.showErrorMessage).toHaveBeenCalledWith(
+			"Failed to create rule: Missing required fields: scope, kind, or fileName",
+		)
+	})
+
+	it("handleCreateRule validates scope and kind before creating", async () => {
+		const provider = createMockProvider()
+
+		const result = await handleCreateRule(provider, "/workspace", {
+			type: "createRule",
+			values: { scope: "team", kind: "workspace", fileName: "new.md" },
+		} as WebviewMessage)
+
+		expect(result).toBeUndefined()
+		expect(createRule).not.toHaveBeenCalled()
+		expect(vscode.window.showErrorMessage).toHaveBeenCalledWith(
+			"Failed to create rule: Missing required fields: scope, kind, or fileName",
+		)
+	})
+
+	it("handleCreateRule shows an error for missing workspace project rules and does not refresh", async () => {
+		const provider = createMockProvider()
+		vi.mocked(createRule).mockRejectedValue(new Error("Workspace rules require an open workspace"))
+
+		const result = await handleCreateRule(provider, "", {
+			type: "createRule",
+			values: { scope: "project", kind: "generic", fileName: "new.md" },
+		} as WebviewMessage)
+
+		expect(result).toBeUndefined()
+		expect(vscode.window.showErrorMessage).toHaveBeenCalledWith(
+			"Failed to create rule: Workspace rules require an open workspace",
+		)
+		expect(mockPostMessageToWebview).not.toHaveBeenCalled()
+	})
+	it("handleRequestRules safely handles nullish errors", async () => {
+		const provider = createMockProvider()
+		vi.mocked(getRules).mockRejectedValue(null)
+
+		const result = await handleRequestRules(provider, "/workspace")
+
+		expect(result).toEqual([])
+		expect(mockLog).toHaveBeenCalledWith("Error fetching rules: null")
+		expect(mockPostMessageToWebview).toHaveBeenCalledWith({ type: "rules", rules: [] })
+	})
+
+	it("handleCreateRule warns when refresh fails after creation succeeds", async () => {
+		const provider = createMockProvider()
+		vi.mocked(createRule).mockResolvedValue("/workspace/.roo/rules/new.md")
+		vi.mocked(getRules).mockRejectedValue(new Error("refresh failed"))
+
+		const result = await handleCreateRule(provider, "/workspace", {
+			type: "createRule",
+			values: { scope: "project", kind: "generic", fileName: "new.md" },
+		} as WebviewMessage)
+
+		expect(result).toBeUndefined()
+		expect(createRule).toHaveBeenCalledWith("/workspace", { scope: "project", kind: "generic", fileName: "new.md" })
+		expect(openFile).toHaveBeenCalledWith("/workspace/.roo/rules/new.md")
+		expect(vscode.window.showErrorMessage).not.toHaveBeenCalled()
+		expect(vscode.window.showWarningMessage).toHaveBeenCalledWith(
+			"Rule created, but refreshing the rules list failed.",
+		)
+	})
+
+	it("handleDeleteRule warns when refresh fails after deletion succeeds", async () => {
+		const provider = createMockProvider()
+		vi.mocked(getRules).mockRejectedValue(new Error("refresh failed"))
+
+		const result = await handleDeleteRule(provider, "/workspace", {
+			type: "deleteRule",
+			values: { scope: "global", kind: "generic", relativePath: "rule.md" },
+		} as WebviewMessage)
+
+		expect(result).toBeUndefined()
+		expect(deleteRule).toHaveBeenCalledWith("/workspace", {
+			scope: "global",
+			kind: "generic",
+			relativePath: "rule.md",
+		})
+		expect(vscode.window.showErrorMessage).not.toHaveBeenCalled()
+		expect(vscode.window.showWarningMessage).toHaveBeenCalledWith(
+			"Rule deleted, but refreshing the rules list failed.",
+		)
+	})
+
+	it("handleOpenRulesDirectory resolves and opens the requested rules directory", async () => {
+		const provider = createMockProvider()
+		vi.mocked(getRulesDirectoryPath).mockReturnValue("/workspace/.roo/rules-code")
+
+		await handleOpenRulesDirectory(provider, "/workspace", {
+			type: "openRulesDirectory",
+			values: { scope: "project", kind: "mode", modeSlug: "code" },
+		} as WebviewMessage)
+
+		expect(getRulesDirectoryPath).toHaveBeenCalledWith("/workspace", {
+			scope: "project",
+			kind: "mode",
+			modeSlug: "code",
+		})
+		expect(openFile).toHaveBeenCalledWith("/workspace/.roo/rules-code")
+	})
+})