pengyejiatu/IC-Coder-Plugin
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix:修复启用/禁用个人规则失效的bug

Browse Source
This commit is contained in:
Roe-xin
2026-03-20 15:19:24 +08:00
parent 1207d2b91a
commit 69f86dbc0d
3 changed files with 116 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
src/panels/helpers/messageRouter.ts
View File

@ -27,11 +27,15 @@ import {
savePersonalRule,
updatePersonalRule,
deletePersonalRule,
updatePersonalRulesEnabled,
} from "../../utils/personalRulesManager";
import { compactDialog } from "../../services/apiClient";
import { ChatHistoryManager } from "../../utils/chatHistoryManager";
import { getCachedUserInfo } from "../../services/userService";
import { loadConversationHistory, selectConversation } from "./conversationHelper";
import {
loadConversationHistory,
selectConversation,
} from "./conversationHelper";
import { getVCDFileInfo } from "./vcdHelper";
import {
handleAddContextFile,
@ -138,16 +142,16 @@ export async function handleWebviewMessage(
break;
case "loadConversationHistory":
loadConversationHistory(
panel,
message.offset || 0,
message.limit || 10,
);
loadConversationHistory(panel, message.offset || 0, message.limit || 10);
break;
case "selectConversation":
if (message.conversationId) {
selectConversation(panel, message.conversationId, context.extensionPath);
selectConversation(
panel,
message.conversationId,
context.extensionPath,
);
}
break;
@ -261,7 +265,9 @@ export async function handleWebviewMessage(
verified: true,
});
} else {
const { InvitationService } = require("../../services/invitationService");
const {
InvitationService,
} = require("../../services/invitationService");
const isVerified = await InvitationService.isVerified(context);
panel.webview.postMessage({
command: "invitationCodeStatus",
@ -292,7 +298,9 @@ export async function handleWebviewMessage(
case "checkTrialExpiration":
{
const { TrialExpirationService } = require("../../services/trialExpirationService");
const {
TrialExpirationService,
} = require("../../services/trialExpirationService");
const trialService = new TrialExpirationService(context, panel);
await trialService.checkExpiration();
}
@ -300,7 +308,9 @@ export async function handleWebviewMessage(
case "verifyInvitationCode":
{
const { InvitationService } = require("../../services/invitationService");
const {
InvitationService,
} = require("../../services/invitationService");
const result = await InvitationService.verifyCode(message.code);
if (result.success) {
@ -390,7 +400,7 @@ export async function handleWebviewMessage(
case "openContextSettings":
panel.webview.postMessage({
command: "openSettingsTab",
tab: "context"
tab: "context",
});
break;
@ -401,7 +411,7 @@ export async function handleWebviewMessage(
canSelectFolders: false,
openLabel: "选择文件",
filters: {
"支持的文件": ["md", "txt", "v", "sv", "pdf"],
支持的文件: ["md", "txt", "v", "sv", "pdf"],
},
});
@ -557,6 +567,19 @@ export async function handleWebviewMessage(
}
break;
case "updatePersonalRulesEnabled":
{
const success = await updatePersonalRulesEnabled(message.enabled);
if (success) {
const data = loadPersonalRules();
panel.webview.postMessage({
command: "personalRulesLoaded",
data: data,
});
}
}
break;
case "loadDocumentSets":
const { getDocumentSets } = await import("./contextHelper");
panel.webview.postMessage({

103
src/utils/personalRulesManager.ts
View File

@ -5,16 +5,16 @@
* 使用场景：保存和加载用户的个人规则
*/
import * as vscode from 'vscode';
import * as fs from 'fs';
import * as path from 'path';
import * as os from 'os';
import * as vscode from "vscode";
import * as fs from "fs";
import * as path from "path";
import * as os from "os";
/**
* 获取规则目录路径
*/
function getRulesDir(): string {
return path.join(os.homedir(), '.iccoder', 'rules');
return path.join(os.homedir(), ".iccoder", "rules");
}
/**
@ -31,18 +31,22 @@ function ensureRulesDir(): void {
* 从文件内容中提取规则名称
*/
function extractRuleName(content: string): string {
const lines = content.split('\n');
const lines = content.split("\n");
const firstLine = lines[0]?.trim();
if (firstLine && firstLine.startsWith('# ')) {
if (firstLine && firstLine.startsWith("# ")) {
return firstLine.substring(2).trim();
}
return content.substring(0, 30) + (content.length > 30 ? '...' : '');
return content.substring(0, 30) + (content.length > 30 ? "..." : "");
}
/**
* 保存新规则
*/
export async function savePersonalRule(name: string, content: string, enabled: boolean): Promise<boolean> {
export async function savePersonalRule(
name: string,
content: string,
enabled: boolean,
): Promise<boolean> {
try {
ensureRulesDir();
@ -51,11 +55,17 @@ export async function savePersonalRule(name: string, content: string, enabled: b
const filePath = path.join(getRulesDir(), filename);
const fileContent = `# ${name}\n\n${content}`;
fs.writeFileSync(filePath, fileContent, 'utf-8');
fs.writeFileSync(filePath, fileContent, "utf-8");
await vscode.workspace.getConfiguration('ic-coder').update('personalRulesEnabled', enabled, vscode.ConfigurationTarget.Global);
await vscode.workspace
.getConfiguration("ic-coder")
.update(
"personalRulesEnabled",
enabled,
vscode.ConfigurationTarget.Global,
);
vscode.window.showInformationMessage('规则已保存');
vscode.window.showInformationMessage("规则已保存");
return true;
} catch (error) {
vscode.window.showErrorMessage(`保存规则失败: ${error}`);
@ -66,15 +76,26 @@ export async function savePersonalRule(name: string, content: string, enabled: b
/**
* 更新规则
*/
export async function updatePersonalRule(filename: string, name: string, content: string, enabled: boolean): Promise<boolean> {
export async function updatePersonalRule(
filename: string,
name: string,
content: string,
enabled: boolean,
): Promise<boolean> {
try {
const filePath = path.join(getRulesDir(), filename);
const fileContent = `# ${name}\n\n${content}`;
fs.writeFileSync(filePath, fileContent, 'utf-8');
fs.writeFileSync(filePath, fileContent, "utf-8");
await vscode.workspace.getConfiguration('ic-coder').update('personalRulesEnabled', enabled, vscode.ConfigurationTarget.Global);
await vscode.workspace
.getConfiguration("ic-coder")
.update(
"personalRulesEnabled",
enabled,
vscode.ConfigurationTarget.Global,
);
vscode.window.showInformationMessage('规则已更新');
vscode.window.showInformationMessage("规则已更新");
return true;
} catch (error) {
vscode.window.showErrorMessage(`更新规则失败: ${error}`);
@ -90,7 +111,7 @@ export async function deletePersonalRule(filename: string): Promise<boolean> {
const filePath = path.join(getRulesDir(), filename);
if (fs.existsSync(filePath)) {
fs.unlinkSync(filePath);
vscode.window.showInformationMessage('规则已删除');
vscode.window.showInformationMessage("规则已删除");
return true;
}
return false;
@ -103,8 +124,13 @@ export async function deletePersonalRule(filename: string): Promise<boolean> {
/**
* 加载所有规则
*/
export function loadPersonalRules(): { rules: Array<{ filename: string; name: string; content: string }>; enabled: boolean } {
const enabled = vscode.workspace.getConfiguration('ic-coder').get<boolean>('personalRulesEnabled', true);
export function loadPersonalRules(): {
rules: Array<{ filename: string; name: string; content: string }>;
enabled: boolean;
} {
const enabled = vscode.workspace
.getConfiguration("ic-coder")
.get<boolean>("personalRulesEnabled", true);
const dir = getRulesDir();
if (!fs.existsSync(dir)) {
@ -112,16 +138,16 @@ export function loadPersonalRules(): { rules: Array<{ filename: string; name: st
}
try {
const files = fs.readdirSync(dir).filter(f => f.endsWith('.md'));
const rules = files.map(filename => {
const content = fs.readFileSync(path.join(dir, filename), 'utf-8');
const lines = content.split('\n');
let name = '';
const files = fs.readdirSync(dir).filter((f) => f.endsWith(".md"));
const rules = files.map((filename) => {
const content = fs.readFileSync(path.join(dir, filename), "utf-8");
const lines = content.split("\n");
let name = "";
let actualContent = content;
if (lines[0]?.trim().startsWith('# ')) {
if (lines[0]?.trim().startsWith("# ")) {
name = lines[0].substring(2).trim();
actualContent = lines.slice(2).join('\n').trim();
actualContent = lines.slice(2).join("\n").trim();
} else {
name = extractRuleName(content);
}
@ -130,11 +156,32 @@ export function loadPersonalRules(): { rules: Array<{ filename: string; name: st
});
return { rules, enabled };
} catch (error) {
console.error('读取规则失败:', error);
console.error("读取规则失败:", error);
return { rules: [], enabled };
}
}
/**
* 更新个人规则启用状态
*/
export async function updatePersonalRulesEnabled(
enabled: boolean,
): Promise<boolean> {
try {
await vscode.workspace
.getConfiguration("ic-coder")
.update(
"personalRulesEnabled",
enabled,
vscode.ConfigurationTarget.Global,
);
return true;
} catch (error) {
console.error("更新规则启用状态失败:", error);
return false;
}
}
/**
* 获取当前生效的所有规则内容
*/
@ -143,5 +190,5 @@ export function getActiveRules(): string | null {
if (!enabled || rules.length === 0) {
return null;
}
return rules.map(r => r.content).join('\n\n');
return rules.map((r) => r.content).join("\n\n");
}

6
src/views/rulesSettingsComponent.ts
View File

@ -400,6 +400,12 @@ export function getRulesSettingsComponentScript(): string {
}
}
//监听启用个人规则开关变化
document.getElementById('enablePersonalRulesCheckbox').addEventListener('change', function() {
const enabled = this.checked;
vscode.postMessage({ command: 'updatePersonalRulesEnabled', enabled: enabled });
});
// 页面加载时请求规则数据
vscode.postMessage({ command: 'loadPersonalRules' });
`;