+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| File | ++ | Statements | ++ | Branches | ++ | Functions | ++ | Lines | ++ |
|---|---|---|---|---|---|---|---|---|---|
| main/data/repositories | +
+
+ |
+ 100% | +20/20 | +100% | +8/8 | +100% | +12/12 | +100% | +20/20 | +
| main/ipc | +
+
+ |
+ 100% | +11/11 | +75% | +3/4 | +100% | +4/4 | +100% | +11/11 | +
| main/services | +
+
+ |
+ 40.75% | +97/238 | +21.27% | +20/94 | +55.26% | +21/38 | +41.31% | +88/213 | +
| main/utils | +
+
+ |
+ 80.64% | +25/31 | +100% | +8/8 | +66.66% | +6/9 | +78.57% | +22/28 | +
| shared | +
+
+ |
+ 100% | +23/23 | +100% | +0/0 | +100% | +4/4 | +100% | +23/23 | +
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 +25 +26 +27 +28 +29 +30 +31 +32 +33 +34 +35 +36 +37 +38 +39 +40 +41 +42 +43 +44 +45 +46 +47 +48 +49 +50 +51 +52 +53 +54 +55 +56 +57 +58 +59 +60 +61 +62 | + + + + + + + + + + + + + +6x + + +1x + + + + +1x + + + + + + +1x + + + +2x + + +2x + + + +2x + + +2x + + + +1x + + + +3x + + + + + + + + + + | import type Database from 'better-sqlite3';
+import { BackupSnapshot } from '../../../shared/types';
+import { BackupType } from '../../../shared/enums';
+
+interface DbRow {
+ id: number;
+ name: string;
+ creation_time: string;
+ type: string;
+ menu_items_json: string;
+ sha256_checksum: string;
+}
+
+export class BackupSnapshotRepo {
+ constructor(private readonly db: Database.Database) {}
+
+ insert(snapshot: Omit<BackupSnapshot, 'id'>): BackupSnapshot {
+ const stmt = this.db.prepare(`
+ INSERT INTO backup_snapshots
+ (name, creation_time, type, menu_items_json, sha256_checksum)
+ VALUES (?, ?, ?, ?, ?)
+ `);
+ const result = stmt.run(
+ snapshot.name,
+ snapshot.creationTime,
+ snapshot.type,
+ snapshot.menuItemsJson,
+ snapshot.sha256Checksum
+ );
+ return { ...snapshot, id: result.lastInsertRowid as number };
+ }
+
+ findAll(): BackupSnapshot[] {
+ const rows = this.db
+ .prepare('SELECT * FROM backup_snapshots ORDER BY creation_time DESC')
+ .all() as DbRow[];
+ return rows.map(this.toModel);
+ }
+
+ findById(id: number): BackupSnapshot | null {
+ const row = this.db
+ .prepare('SELECT * FROM backup_snapshots WHERE id = ?')
+ .get(id) as DbRow | undefined;
+ return row ? this.toModel(row) : null;
+ }
+
+ delete(id: number): void {
+ this.db.prepare('DELETE FROM backup_snapshots WHERE id = ?').run(id);
+ }
+
+ private toModel(row: DbRow): BackupSnapshot {
+ return {
+ id: row.id,
+ name: row.name,
+ creationTime: row.creation_time,
+ type: row.type as BackupType,
+ menuItemsJson: row.menu_items_json,
+ sha256Checksum: row.sha256_checksum,
+ };
+ }
+}
+ |
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 +25 +26 +27 +28 +29 +30 +31 +32 +33 +34 +35 +36 +37 +38 +39 +40 +41 +42 +43 +44 +45 +46 +47 +48 +49 +50 +51 +52 +53 +54 +55 +56 +57 +58 +59 +60 +61 +62 +63 +64 +65 | + + + + + + + + + + + + + + +6x + + +2x + + + + +2x + + + + + + + +2x + + + +1x + + +1x + + + +2x + + +2x + + + +1x + + + +3x + + + + + + + + + + + | import type Database from 'better-sqlite3';
+import { OperationRecord } from '../../../shared/types';
+import { OperationType } from '../../../shared/enums';
+
+interface DbRow {
+ id: number;
+ timestamp: string;
+ operation_type: string;
+ target_name: string;
+ registry_path: string;
+ old_value: string | null;
+ new_value: string | null;
+}
+
+export class OperationRecordRepo {
+ constructor(private readonly db: Database.Database) {}
+
+ insert(record: Omit<OperationRecord, 'id'>): OperationRecord {
+ const stmt = this.db.prepare(`
+ INSERT INTO operation_records
+ (timestamp, operation_type, target_name, registry_path, old_value, new_value)
+ VALUES (?, ?, ?, ?, ?, ?)
+ `);
+ const result = stmt.run(
+ record.timestamp,
+ record.operationType,
+ record.targetEntryName,
+ record.registryPath,
+ record.oldValue ?? null,
+ record.newValue ?? null
+ );
+ return { ...record, id: result.lastInsertRowid as number };
+ }
+
+ findAll(): OperationRecord[] {
+ const rows = this.db
+ .prepare('SELECT * FROM operation_records ORDER BY timestamp DESC')
+ .all() as DbRow[];
+ return rows.map(this.toModel);
+ }
+
+ findById(id: number): OperationRecord | null {
+ const row = this.db
+ .prepare('SELECT * FROM operation_records WHERE id = ?')
+ .get(id) as DbRow | undefined;
+ return row ? this.toModel(row) : null;
+ }
+
+ deleteAll(): void {
+ this.db.prepare('DELETE FROM operation_records').run();
+ }
+
+ private toModel(row: DbRow): OperationRecord {
+ return {
+ id: row.id,
+ timestamp: row.timestamp,
+ operationType: row.operation_type as OperationType,
+ targetEntryName: row.target_name,
+ registryPath: row.registry_path,
+ oldValue: row.old_value,
+ newValue: row.new_value,
+ };
+ }
+}
+ |
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| File | ++ | Statements | ++ | Branches | ++ | Functions | ++ | Lines | ++ |
|---|---|---|---|---|---|---|---|---|---|
| BackupSnapshotRepo.ts | +
+
+ |
+ 100% | +10/10 | +100% | +2/2 | +100% | +6/6 | +100% | +10/10 | +
| OperationRecordRepo.ts | +
+
+ |
+ 100% | +10/10 | +100% | +6/6 | +100% | +6/6 | +100% | +10/10 | +
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| File | ++ | Statements | ++ | Branches | ++ | Functions | ++ | Lines | ++ |
|---|---|---|---|---|---|---|---|---|---|
| registry.ts | +
+
+ |
+ 100% | +11/11 | +75% | +3/4 | +100% | +4/4 | +100% | +11/11 | +
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 +25 +26 +27 +28 +29 +30 +31 +32 +33 +34 +35 +36 +37 +38 +39 +40 +41 +42 +43 +44 +45 +46 +47 | + + + + + + + +7x + + +1x + + + +7x + + +1x + + + + + + + + + + +1x +1x + + + +7x + + +2x +1x + +1x + +2x + + + + | import { ipcMain } from 'electron';
+import { IPC } from '../../shared/ipc-channels';
+import { MenuScene, MenuItemType } from '../../shared/enums';
+import { ToggleItemParams, BatchToggleParams } from '../../shared/types';
+import { MenuManagerService } from '../services/MenuManagerService';
+import { wrapHandler } from '../utils/ipcWrapper';
+
+export function registerRegistryHandlers(menuManager: MenuManagerService): void {
+ ipcMain.handle(
+ IPC.REGISTRY_GET_ITEMS,
+ wrapHandler((_event: unknown, scene: MenuScene) =>
+ menuManager.getMenuItems(scene)
+ )
+ );
+
+ ipcMain.handle(
+ IPC.REGISTRY_TOGGLE,
+ wrapHandler(async (_event: unknown, params: ToggleItemParams) => {
+ const item = {
+ id: -1,
+ name: params.name,
+ command: '',
+ iconPath: null,
+ isEnabled: params.isEnabled,
+ source: '',
+ menuScene: params.menuScene,
+ registryKey: params.registryKey,
+ type: params.type ?? MenuItemType.System,
+ };
+ const result = await menuManager.toggleItem(item);
+ return { newState: !params.isEnabled, newRegistryKey: result.newRegistryKey };
+ })
+ );
+
+ ipcMain.handle(
+ IPC.REGISTRY_BATCH,
+ wrapHandler(async (_event: unknown, params: BatchToggleParams) => {
+ if (params.enable) {
+ await menuManager.batchEnable(params.items);
+ } else {
+ await menuManager.batchDisable(params.items);
+ }
+ return true;
+ })
+ );
+}
+ |
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 +25 +26 +27 +28 +29 +30 +31 +32 +33 +34 +35 +36 +37 +38 +39 +40 +41 +42 +43 +44 +45 +46 +47 +48 +49 +50 +51 +52 +53 +54 +55 +56 +57 +58 +59 +60 +61 +62 +63 +64 +65 +66 +67 +68 +69 +70 +71 +72 +73 +74 +75 +76 +77 +78 +79 +80 +81 +82 +83 +84 +85 +86 +87 +88 +89 +90 +91 +92 +93 +94 +95 +96 +97 +98 +99 +100 +101 +102 +103 +104 +105 +106 +107 +108 +109 +110 +111 +112 +113 +114 +115 +116 +117 +118 +119 +120 +121 +122 +123 +124 +125 +126 +127 +128 +129 +130 +131 +132 +133 +134 +135 +136 +137 +138 +139 +140 +141 +142 +143 +144 +145 +146 +147 +148 +149 +150 +151 +152 +153 +154 +155 +156 +157 +158 +159 +160 +161 +162 +163 | + + + + + + + + + + + + + + + + + + +4x +4x +4x + + + +2x +2x +12x +12x + + +2x +2x + +2x + + + + + + + +2x +2x +2x + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +1x + + + +1x + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + | import { createHash } from 'crypto';
+import { promises as fs } from 'fs';
+import { dialog, BrowserWindow } from 'electron';
+import { BackupType, MenuScene, OperationType } from '../../shared/enums';
+import { BackupSnapshot, MenuItemEntry, RestoreDiffItem } from '../../shared/types';
+import { BackupSnapshotRepo } from '../data/repositories/BackupSnapshotRepo';
+import { MenuManagerService } from './MenuManagerService';
+import { OperationHistoryService } from './OperationHistoryService';
+import log from '../utils/logger';
+
+/** 去掉 ShellExt registryKey 末段的 '-' 前缀，用于跨状态匹配 */
+function normalizeKey(key: string): string {
+ const sep = key.lastIndexOf('\\');
+ if (sep === -1) return key.replace(/^-+/, '');
+ return key.substring(0, sep + 1) + key.substring(sep + 1).replace(/^-+/, '');
+}
+
+export class BackupService {
+ constructor(
+ private readonly repo: BackupSnapshotRepo,
+ private readonly menuManager: MenuManagerService,
+ private readonly history: OperationHistoryService
+ ) {}
+
+ async createBackup(name: string, type = BackupType.Manual): Promise<BackupSnapshot> {
+ const allItems: MenuItemEntry[] = [];
+ for (const scene of Object.values(MenuScene)) {
+ const items = await this.menuManager.getMenuItems(scene);
+ allItems.push(...items);
+ }
+
+ const jsonData = JSON.stringify(allItems);
+ const checksum = createHash('sha256').update(jsonData).digest('hex');
+
+ const snapshot = this.repo.insert({
+ name,
+ creationTime: new Date().toISOString(),
+ type,
+ menuItemsJson: jsonData,
+ sha256Checksum: checksum,
+ });
+
+ this.history.recordOperation(OperationType.Backup, name, '', '', checksum);
+ log.info(`Backup created: ${name} (${allItems.length} items)`);
+ return snapshot;
+ }
+
+ async restoreBackup(snapshotId: number): Promise<void> {
+ const snapshot = this.repo.findById(snapshotId);
+ if (!snapshot) throw new Error('找不到备份快照');
+
+ const expectedChecksum = createHash('sha256')
+ .update(snapshot.menuItemsJson)
+ .digest('hex');
+ if (expectedChecksum !== snapshot.sha256Checksum) {
+ throw new Error('备份校验失败，文件可能已被篡改');
+ }
+
+ // 还原前先自动创建备份
+ await this.createBackup(
+ `AutoBackup_BeforeRestore_${new Date().toISOString().replace(/[:.]/g, '-')}`,
+ BackupType.Auto
+ );
+
+ const itemsToRestore: MenuItemEntry[] = JSON.parse(snapshot.menuItemsJson);
+ if (!itemsToRestore.length) throw new Error('备份文件中没有有效的菜单项数据');
+
+ const allCurrentItems: MenuItemEntry[] = [];
+ for (const scene of Object.values(MenuScene)) {
+ allCurrentItems.push(...(await this.menuManager.getMenuItems(scene)));
+ }
+
+ const toEnable: MenuItemEntry[] = [];
+ const toDisable: MenuItemEntry[] = [];
+
+ for (const backupItem of itemsToRestore) {
+ const current = allCurrentItems.find((i) => normalizeKey(i.registryKey) === normalizeKey(backupItem.registryKey));
+ if (current && current.isEnabled !== backupItem.isEnabled) {
+ current.isEnabled = backupItem.isEnabled;
+ if (backupItem.isEnabled) toEnable.push(current);
+ else toDisable.push(current);
+ }
+ }
+
+ if (toEnable.length) await this.menuManager.batchEnable(toEnable);
+ if (toDisable.length) await this.menuManager.batchDisable(toDisable);
+
+ this.history.recordOperation(OperationType.Restore, snapshot.name, '', '', snapshotId.toString());
+ log.info(`Restore completed from backup: ${snapshot.name}`);
+ }
+
+ async deleteBackup(id: number): Promise<void> {
+ this.repo.delete(id);
+ }
+
+ getAllBackups(): BackupSnapshot[] {
+ return this.repo.findAll();
+ }
+
+ async previewRestoreDiff(snapshotId: number): Promise<RestoreDiffItem[]> {
+ const snapshot = this.repo.findById(snapshotId);
+ if (!snapshot) throw new Error('找不到备份快照');
+
+ const backupItems: MenuItemEntry[] = JSON.parse(snapshot.menuItemsJson);
+ const currentItems: MenuItemEntry[] = [];
+ for (const scene of Object.values(MenuScene)) {
+ currentItems.push(...(await this.menuManager.getMenuItems(scene)));
+ }
+
+ const diff: RestoreDiffItem[] = [];
+ for (const backupItem of backupItems) {
+ const current = currentItems.find((i) => normalizeKey(i.registryKey) === normalizeKey(backupItem.registryKey));
+ if (current && current.isEnabled !== backupItem.isEnabled) {
+ diff.push({ current, backup: backupItem });
+ }
+ }
+ return diff;
+ }
+
+ async exportBackup(snapshotId: number, win: BrowserWindow): Promise<void> {
+ const snapshot = this.repo.findById(snapshotId);
+ if (!snapshot) throw new Error('找不到备份快照');
+
+ const { filePath, canceled } = await dialog.showSaveDialog(win, {
+ title: '导出备份文件',
+ defaultPath: `${snapshot.name.replace(/[\\/:*?"<>|]/g, '_')}.cmbackup`,
+ filters: [{ name: 'ContextMaster Backup', extensions: ['cmbackup'] }],
+ });
+
+ if (canceled || !filePath) return;
+ await fs.writeFile(filePath, snapshot.menuItemsJson, 'utf-8');
+ log.info(`Backup exported to: ${filePath}`);
+ }
+
+ async importBackup(win: BrowserWindow): Promise<BackupSnapshot> {
+ const { filePaths, canceled } = await dialog.showOpenDialog(win, {
+ title: '导入备份文件',
+ filters: [
+ { name: 'ContextMaster Backup', extensions: ['cmbackup'] },
+ { name: 'JSON Files', extensions: ['json'] },
+ ],
+ properties: ['openFile'],
+ });
+
+ if (canceled || !filePaths.length) throw new Error('未选择文件');
+
+ const filePath = filePaths[0];
+ const jsonData = await fs.readFile(filePath, 'utf-8');
+ const checksum = createHash('sha256').update(jsonData).digest('hex');
+
+ const snapshot = this.repo.insert({
+ name: `导入 · ${require('path').basename(filePath, '.cmbackup')}`,
+ creationTime: new Date().toISOString(),
+ type: BackupType.Manual,
+ menuItemsJson: jsonData,
+ sha256Checksum: checksum,
+ });
+
+ log.info(`Backup imported from: ${filePath}`);
+ return snapshot;
+ }
+}
+ |
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 +25 +26 +27 +28 +29 +30 +31 +32 +33 +34 +35 +36 +37 +38 +39 +40 +41 +42 +43 +44 +45 +46 +47 +48 +49 +50 +51 +52 +53 +54 +55 +56 +57 +58 +59 +60 +61 +62 +63 +64 +65 +66 +67 +68 +69 +70 +71 +72 +73 +74 +75 +76 +77 +78 +79 +80 +81 +82 +83 +84 +85 +86 +87 +88 +89 | + + + + + + + +13x +13x + + + +1x + + + +6x +5x +4x +4x +4x + + + + + + +4x +4x + + + +6x +5x +4x +4x +4x + + + + + + +4x +4x + + + +2x +1x + +1x + + + + +4x +3x + +2x +2x +2x +3x + +1x + +1x +1x + + + + +4x +3x + +2x +2x +2x +3x + +1x + +1x +1x + + + + | import { MenuScene, OperationType } from '../../shared/enums';
+import { MenuItemEntry } from '../../shared/types';
+import { RegistryService } from './RegistryService';
+import { OperationHistoryService } from './OperationHistoryService';
+import log from '../utils/logger';
+
+export class MenuManagerService {
+ constructor(
+ private readonly registry: RegistryService,
+ private readonly history: OperationHistoryService
+ ) {}
+
+ async getMenuItems(scene: MenuScene): Promise<MenuItemEntry[]> {
+ return this.registry.getMenuItems(scene);
+ }
+
+ async enableItem(item: MenuItemEntry): Promise<{ newRegistryKey?: string }> {
+ if (item.isEnabled) return {};
+ const result = await this.registry.setItemEnabled(item.registryKey, true);
+ Iif (result.newRegistryKey) item.registryKey = result.newRegistryKey;
+ item.isEnabled = true;
+ this.history.recordOperation(
+ OperationType.Enable,
+ item.name,
+ item.registryKey,
+ 'false',
+ 'true'
+ );
+ log.info(`Enabled: ${item.name}`);
+ return result;
+ }
+
+ async disableItem(item: MenuItemEntry): Promise<{ newRegistryKey?: string }> {
+ if (!item.isEnabled) return {};
+ const result = await this.registry.setItemEnabled(item.registryKey, false);
+ Iif (result.newRegistryKey) item.registryKey = result.newRegistryKey;
+ item.isEnabled = false;
+ this.history.recordOperation(
+ OperationType.Disable,
+ item.name,
+ item.registryKey,
+ 'true',
+ 'false'
+ );
+ log.info(`Disabled: ${item.name}`);
+ return result;
+ }
+
+ async toggleItem(item: MenuItemEntry): Promise<{ newRegistryKey?: string }> {
+ if (item.isEnabled) {
+ return this.disableItem(item);
+ } else {
+ return this.enableItem(item);
+ }
+ }
+
+ async batchEnable(items: MenuItemEntry[]): Promise<void> {
+ const targets = items.filter((i) => !i.isEnabled);
+ if (!targets.length) return;
+
+ this.registry.createRollbackPoint(targets);
+ try {
+ for (const item of targets) {
+ await this.enableItem(item);
+ }
+ this.registry.commitTransaction();
+ } catch (e) {
+ await this.registry.rollback();
+ throw new Error(`批量启用失败，已回滚: ${(e as Error).message}`);
+ }
+ }
+
+ async batchDisable(items: MenuItemEntry[]): Promise<void> {
+ const targets = items.filter((i) => i.isEnabled);
+ if (!targets.length) return;
+
+ this.registry.createRollbackPoint(targets);
+ try {
+ for (const item of targets) {
+ await this.disableItem(item);
+ }
+ this.registry.commitTransaction();
+ } catch (e) {
+ await this.registry.rollback();
+ throw new Error(`批量禁用失败，已回滚: ${(e as Error).message}`);
+ }
+ }
+}
+ |
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 +25 +26 +27 +28 +29 +30 +31 +32 +33 +34 +35 +36 +37 +38 +39 +40 +41 +42 +43 +44 +45 +46 +47 +48 +49 +50 +51 +52 +53 +54 +55 +56 +57 +58 +59 +60 +61 +62 +63 +64 +65 +66 +67 +68 +69 +70 +71 +72 +73 +74 +75 +76 +77 +78 +79 +80 +81 +82 +83 +84 +85 +86 +87 +88 +89 +90 +91 +92 +93 +94 +95 +96 +97 +98 +99 +100 +101 +102 +103 +104 +105 +106 +107 +108 +109 +110 +111 +112 +113 +114 +115 +116 +117 +118 +119 +120 +121 +122 +123 +124 +125 +126 +127 +128 +129 +130 +131 +132 +133 +134 +135 +136 +137 +138 +139 +140 +141 +142 +143 +144 +145 +146 +147 +148 +149 +150 +151 +152 +153 +154 +155 +156 +157 +158 +159 +160 +161 +162 +163 +164 +165 +166 +167 +168 +169 +170 +171 +172 +173 +174 +175 +176 +177 +178 +179 +180 +181 +182 +183 +184 +185 +186 +187 +188 +189 +190 +191 +192 +193 +194 +195 +196 +197 +198 +199 +200 +201 +202 +203 +204 +205 +206 +207 +208 +209 +210 +211 +212 +213 +214 +215 +216 +217 +218 +219 +220 +221 +222 +223 +224 +225 +226 +227 +228 +229 +230 +231 +232 +233 +234 +235 +236 +237 +238 +239 +240 +241 +242 +243 +244 +245 +246 +247 +248 +249 +250 +251 +252 +253 +254 +255 +256 +257 +258 +259 +260 +261 +262 +263 +264 +265 +266 +267 +268 +269 +270 +271 +272 +273 +274 +275 +276 +277 +278 +279 +280 +281 +282 +283 +284 +285 +286 +287 +288 +289 +290 +291 +292 +293 +294 +295 +296 +297 +298 +299 +300 +301 +302 +303 +304 +305 +306 +307 +308 +309 +310 +311 +312 +313 +314 +315 +316 +317 +318 +319 +320 +321 +322 +323 +324 +325 +326 +327 +328 +329 +330 +331 +332 +333 +334 +335 +336 +337 +338 +339 +340 +341 +342 +343 +344 +345 +346 +347 +348 +349 +350 +351 +352 +353 +354 +355 +356 +357 +358 +359 +360 +361 +362 +363 +364 +365 +366 +367 +368 +369 +370 +371 +372 +373 +374 | + + + + + + + + +1x + +1x +1x + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +1x + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +2x +2x +1x + + + + + + + + + + + + +1x + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + | import { execFile } from 'child_process';
+import { promisify } from 'util';
+import * as os from 'os';
+import * as fs from 'fs';
+import * as path from 'path';
+import * as crypto from 'crypto';
+import { isAdmin } from '../utils/AdminHelper';
+import log from '../utils/logger';
+
+const execFileAsync = promisify(execFile);
+
+const PWSH7_PATH = 'C:\\Program Files\\PowerShell\\7\\pwsh.exe';
+const PS_EXE = fs.existsSync(PWSH7_PATH) ? PWSH7_PATH : 'powershell.exe';
+
+export class PowerShellBridge {
+ /**
+ * 执行 PowerShell 脚本并将 stdout 解析为 JSON
+ */
+ async execute<T>(script: string): Promise<T> {
+ log.debug('[PS] execute:', script.substring(0, 200));
+ const { stdout, stderr } = await execFileAsync(
+ PS_EXE,
+ ['-NonInteractive', '-NoProfile', '-OutputFormat', 'Text', '-Command', script],
+ { maxBuffer: 10 * 1024 * 1024, timeout: 30000 }
+ );
+
+ if (stderr) {
+ log.warn('[PS] stderr:', stderr);
+ }
+
+ const trimmed = stdout.trim();
+ if (!trimmed) return [] as unknown as T;
+
+ try {
+ return JSON.parse(trimmed) as T;
+ } catch (e) {
+ log.error('[PS] JSON parse error. stdout:', trimmed.substring(0, 500));
+ throw new Error(`PowerShell 输出 JSON 解析失败: ${String(e)}`);
+ }
+ }
+
+ /**
+ * 以提权方式执行脚本（非管理员时弹出 UAC 对话框）
+ * 管理员身份下直接 fallback 到 execute()
+ */
+ async executeElevated<T>(script: string): Promise<T> {
+ if (isAdmin()) {
+ return this.execute<T>(script);
+ }
+
+ const uid = crypto.randomUUID();
+ const opScript = path.join(os.tmpdir(), `cm_op_${uid}.ps1`);
+ const resultFile = path.join(os.tmpdir(), `cm_res_${uid}.json`);
+
+ // 包装原始脚本：捕获原始 JSON 输出（脚本本身已输出 JSON），写入 resultFile
+ const resultFilePs = resultFile.replace(/'/g, "''");
+ const wrappedScript2 = `
+$ErrorActionPreference = 'Stop'
+try {
+ $__out = & {
+${script}
+ } | Out-String
+ $__out = $__out.Trim()
+ if (-not $__out) { $__out = 'null' }
+ Set-Content -LiteralPath '${resultFilePs}' -Value $__out -Encoding UTF8
+} catch {
+ $__err = (@{ __error = $_.Exception.Message } | ConvertTo-Json -Compress)
+ Set-Content -LiteralPath '${resultFilePs}' -Value $__err -Encoding UTF8
+ exit 1
+}
+`.trim();
+
+ fs.writeFileSync(opScript, wrappedScript2, 'utf8');
+
+ // 通过非提权 PS 启动提权子进程（弹 UAC）
+ const launchScript = `Start-Process '${PS_EXE.replace(/'/g, "''")}' -Verb RunAs -Wait -WindowStyle Hidden -ArgumentList @('-NonInteractive','-NoProfile','-ExecutionPolicy','Bypass','-File','${opScript.replace(/'/g, "''")}')`;
+
+ log.debug('[PS] executeElevated: launching UAC process');
+ try {
+ await execFileAsync(
+ PS_EXE,
+ ['-NonInteractive', '-NoProfile', '-ExecutionPolicy', 'Bypass', '-Command', launchScript],
+ { maxBuffer: 1 * 1024 * 1024, timeout: 120000 }
+ );
+ } finally {
+ try { fs.unlinkSync(opScript); } catch { /* ignore */ }
+ }
+
+ if (!fs.existsSync(resultFile)) {
+ throw new Error('操作已取消（UAC 提权被拒绝）');
+ }
+
+ let resultJson: string;
+ try {
+ resultJson = fs.readFileSync(resultFile, 'utf8').trim();
+ fs.unlinkSync(resultFile);
+ } catch {
+ throw new Error('读取操作结果失败');
+ }
+
+ let parsed: unknown;
+ try {
+ parsed = JSON.parse(resultJson);
+ } catch {
+ throw new Error(`结果 JSON 解析失败: ${resultJson.substring(0, 200)}`);
+ }
+
+ if (parsed && typeof parsed === 'object' && '__error' in parsed) {
+ throw new Error(String((parsed as Record<string, unknown>).__error));
+ }
+
+ return parsed as T;
+ }
+
+ /**
+ * 构建扫描指定注册表路径下所有子键的脚本
+ * 返回 JSON 数组，每项含菜单条目信息
+ */
+ buildGetItemsScript(hkcrSubPath: string): string {
+ // PS 单对象会返回哈希表而非数组，用 @(...) 强制数组
+ return `
+$ErrorActionPreference = 'SilentlyContinue'
+New-PSDrive -Name HKCR -PSProvider Registry -Root HKEY_CLASSES_ROOT -ErrorAction SilentlyContinue | Out-Null
+$basePath = 'HKCR:\\${hkcrSubPath}'
+if (-not (Test-Path -LiteralPath $basePath)) { Write-Output '[]'; exit }
+$subKeys = Get-ChildItem -LiteralPath $basePath | Where-Object { $_.PSIsContainer }
+$result = @($subKeys | ForEach-Object {
+ $key = $_
+ $keyName = $key.PSChildName
+ $name = $key.GetValue('')
+ if (-not $name) { $name = $keyName }
+ $iconPath = $key.GetValue('Icon')
+ $isEnabled = ($key.GetValue('LegacyDisable') -eq $null)
+ $commandSubKey = Join-Path $key.PSPath 'command'
+ $command = ''
+ if (Test-Path -LiteralPath $commandSubKey) {
+ $command = (Get-Item -LiteralPath $commandSubKey).GetValue('')
+ if (-not $command) { $command = '' }
+ }
+ $regKey = '${hkcrSubPath}\\' + $keyName
+ [PSCustomObject]@{
+ name = [string]$name
+ command = [string]$command
+ iconPath = if ($iconPath) { [string]$iconPath } else { $null }
+ isEnabled = [bool]$isEnabled
+ source = ''
+ registryKey = [string]$regKey
+ subKeyName = [string]$keyName
+ }
+})
+$result | ConvertTo-Json -Compress -Depth 3
+`.trim();
+ }
+
+ /**
+ * 构建启用/禁用单个菜单项的脚本
+ * enable=true → Remove-ItemProperty LegacyDisable
+ * enable=false → Set-ItemProperty LegacyDisable -Value ''
+ */
+ buildSetEnabledScript(hkcrRelativeKey: string, enable: boolean): string {
+ const psPath = `HKCR:\\${hkcrRelativeKey}`;
+ if (enable) {
+ return `
+$ErrorActionPreference = 'Stop'
+New-PSDrive -Name HKCR -PSProvider Registry -Root HKEY_CLASSES_ROOT -ErrorAction SilentlyContinue | Out-Null
+$keyPath = '${psPath}'
+if (Test-Path -LiteralPath $keyPath) {
+ $prop = Get-ItemProperty -LiteralPath $keyPath -Name 'LegacyDisable' -ErrorAction SilentlyContinue
+ if ($prop -ne $null) {
+ Remove-ItemProperty -LiteralPath $keyPath -Name 'LegacyDisable' -Force
+ }
+}
+Write-Output '{"ok":true}'
+`.trim();
+ } else {
+ return `
+$ErrorActionPreference = 'Stop'
+New-PSDrive -Name HKCR -PSProvider Registry -Root HKEY_CLASSES_ROOT -ErrorAction SilentlyContinue | Out-Null
+$keyPath = '${psPath}'
+if (-not (Test-Path -LiteralPath $keyPath)) {
+ throw "注册表项不存在: ${hkcrRelativeKey}"
+}
+Set-ItemProperty -LiteralPath $keyPath -Name 'LegacyDisable' -Value '' -Type String -Force
+Write-Output '{"ok":true}'
+`.trim();
+ }
+ }
+
+ /**
+ * 构建枚举 shellex\ContextMenuHandlers 下所有 Shell 扩展的脚本
+ * 使用四级级联策略解析本地化名称：
+ * 1. LocalizedString/FriendlyTypeName → SHLoadIndirectString（解析 @DLL,-ID 格式）
+ * 2. InprocServer32 DLL 字符串表 → 通用字符串质量筛选（LoadLibraryEx + LoadString）
+ * 3. CLSID 默认值
+ * 4. 处理程序键名（最终兜底）
+ * CmHelper.dll 编译后缓存至 %LOCALAPPDATA%\ContextMaster\，避免重复编译开销
+ */
+ buildGetShellExtItemsScript(shellexSubPath: string): string {
+ return `
+$ErrorActionPreference = 'SilentlyContinue'
+New-PSDrive -Name HKCR -PSProvider Registry -Root HKEY_CLASSES_ROOT -ErrorAction SilentlyContinue | Out-Null
+$cmDir = Join-Path $env:LOCALAPPDATA 'ContextMaster'
+$cmDll = Join-Path $cmDir 'CmHelper.dll'
+$helperLoaded = $false
+if (Test-Path $cmDll) {
+ try { Add-Type -Path $cmDll -ErrorAction Stop; $helperLoaded = $true } catch {}
+}
+if (-not $helperLoaded) {
+ $src = @'
+using System;
+using System.Runtime.InteropServices;
+using System.Text;
+using System.Collections.Generic;
+public class CmHelper {
+ const uint LOAD_AS_DATA = 2u;
+ [DllImport("shlwapi.dll", CharSet = CharSet.Unicode)]
+ static extern int SHLoadIndirectString(string s, StringBuilder buf, int cap, IntPtr r);
+ [DllImport("kernel32.dll", CharSet = CharSet.Unicode)]
+ static extern IntPtr LoadLibraryEx(string p, IntPtr h, uint f);
+ [DllImport("kernel32.dll")]
+ static extern bool FreeLibrary(IntPtr h);
+ [DllImport("user32.dll", CharSet = CharSet.Unicode)]
+ static extern int LoadString(IntPtr h, uint id, StringBuilder buf, int cap);
+ public static string ResolveIndirect(string s) {
+ if (string.IsNullOrEmpty(s) || !s.StartsWith("@")) return null;
+ var sb = new StringBuilder(512);
+ return SHLoadIndirectString(s, sb, 512, IntPtr.Zero) == 0 ? sb.ToString() : null;
+ }
+ public static string[] ReadDllStrings(string dll, uint from, uint to) {
+ var list = new List<string>();
+ var hMod = LoadLibraryEx(dll, IntPtr.Zero, LOAD_AS_DATA);
+ if (hMod == IntPtr.Zero) return list.ToArray();
+ try {
+ for (uint i = from; i <= to; i++) {
+ var sb = new StringBuilder(512);
+ if (LoadString(hMod, i, sb, 512) > 0) list.Add(sb.ToString());
+ }
+ } finally { FreeLibrary(hMod); }
+ return list.ToArray();
+ }
+}
+'@
+ if (-not (Test-Path $cmDir)) { New-Item -Path $cmDir -ItemType Directory -Force | Out-Null }
+ if (Test-Path $cmDll) { Remove-Item -Path $cmDll -Force -ErrorAction SilentlyContinue }
+ try {
+ Add-Type -TypeDefinition $src -OutputAssembly $cmDll -ErrorAction Stop
+ $helperLoaded = $true
+ } catch {
+ try { Add-Type -TypeDefinition $src -ErrorAction Stop; $helperLoaded = $true } catch {}
+ }
+}
+function Resolve-ExtName($clsid, $fallback) {
+ if ($clsid -match '^\\{[0-9A-Fa-f-]+\\}$') {
+ $clsidPath = 'HKCR:\\CLSID\\' + $clsid
+ if (Test-Path -LiteralPath $clsidPath) {
+ $clsidKey = Get-Item -LiteralPath $clsidPath
+ foreach ($valName in @('LocalizedString', 'FriendlyTypeName')) {
+ $raw = $clsidKey.GetValue($valName)
+ if ($raw) {
+ if ($raw.StartsWith('@')) {
+ try {
+ $resolved = [CmHelper]::ResolveIndirect($raw)
+ if ($resolved -and $resolved.Length -ge 2) { return $resolved }
+ } catch {}
+ } elseif ($raw.Length -ge 2) {
+ return $raw
+ }
+ }
+ }
+ $inprocPath = Join-Path $clsidPath 'InprocServer32'
+ if (Test-Path -LiteralPath $inprocPath) {
+ $dllPath = (Get-Item -LiteralPath $inprocPath).GetValue('')
+ # Fix 1: 展开 %SystemRoot% 等环境变量，否则 Test-Path 永远返回 $false
+ if ($dllPath) {
+ $dllPath = [System.Environment]::ExpandEnvironmentVariables($dllPath)
+ }
+ if ($dllPath -and (Test-Path -LiteralPath $dllPath)) {
+ # Level 2: 通用字符串质量筛选（过滤后取第一条，无硬编码词表，无长度限制）
+ try {
+ $candidates = [CmHelper]::ReadDllStrings($dllPath, 1, 1000) |
+ Where-Object {
+ $_.Length -ge 2 -and
+ $_ -notmatch '[\\\\/:*?<>|]' -and
+ $_ -notmatch '^\\{' -and
+ $_ -notmatch '^https?://' -and
+ $_ -notmatch '%[0-9A-Za-z]' -and
+ $_ -notmatch '\\{[0-9]+\\}' -and
+ $_ -notmatch '[\\r\\n\\t]' -and
+ $_ -notmatch '^[0-9]' -and
+ $_ -notmatch '[\\u3002\\uff01\\uff1f]' -and
+ $_ -notmatch '[.!?]$'
+ }
+ $pool = $candidates | Where-Object { $_ -match '[^\\x00-\\x7F]' }
+ if (-not $pool) { $pool = $candidates }
+ $best = $pool | Select-Object -First 1
+ if ($best) { return $best }
+ } catch {}
+ # Level 2.5: DLL VersionInfo（适用于英文/日文等非中文软件）
+ try {
+ $ver = [System.Diagnostics.FileVersionInfo]::GetVersionInfo($dllPath)
+ $desc = $null
+ if ($ver.FileDescription -and $ver.FileDescription.Length -ge 2) {
+ $desc = $ver.FileDescription
+ } elseif ($ver.ProductName -and $ver.ProductName.Length -ge 2) {
+ $desc = $ver.ProductName
+ }
+ if ($desc -and $desc.Length -le 80 -and $desc -notmatch '^\\{' -and $desc -notmatch '[\\\\/:*?<>|]') {
+ return $desc
+ }
+ } catch {}
+ }
+ }
+ $def = $clsidKey.GetValue('')
+ if ($def) { return [string]$def }
+ }
+ }
+ return $fallback
+}
+$shellexPath = 'HKCR:\\${shellexSubPath}'
+if (-not (Test-Path -LiteralPath $shellexPath)) { Write-Output '[]'; exit }
+$handlers = Get-ChildItem -LiteralPath $shellexPath | Where-Object { $_.PSIsContainer }
+$result = @($handlers | ForEach-Object {
+ $handlerKeyName = $_.PSChildName
+ $clsid = $_.GetValue('')
+ if (-not $clsid) { $clsid = $handlerKeyName }
+ $cleanName = $handlerKeyName -replace '^-+', ''
+ $displayName = Resolve-ExtName $clsid $cleanName
+ $isEnabled = -not $handlerKeyName.StartsWith('-')
+ $regKey = '${shellexSubPath}\\' + $cleanName
+ [PSCustomObject]@{
+ name = [string]$displayName
+ command = [string]$clsid
+ iconPath = $null
+ isEnabled = [bool]$isEnabled
+ source = [string]$handlerKeyName
+ registryKey = [string]$regKey
+ subKeyName = [string]$handlerKeyName
+ itemType = 'ShellExt'
+ }
+})
+$result | ConvertTo-Json -Compress -Depth 3
+`.trim();
+ }
+
+ /**
+ * 构建启用/禁用 Shell 扩展的脚本（通过重命名键名添加/去除 '-' 前缀）
+ * enable=true → 将 '-Name' 重命名为 'Name'
+ * enable=false → 将 'Name' 重命名为 '-Name'
+ */
+ buildShellExtToggleScript(hkcrRelativeKey: string, enable: boolean): string {
+ const lastSlash = hkcrRelativeKey.lastIndexOf('\\');
+ const parentRelPath = hkcrRelativeKey.substring(0, lastSlash);
+ const keyName = hkcrRelativeKey.substring(lastSlash + 1);
+ const cleanName = keyName.replace(/^-+/, '');
+ const psParentPath = `HKCR:\\${parentRelPath}`;
+ // enable: 找 -cleanName 改为 cleanName；disable: 找 cleanName 改为 -cleanName
+ const psCurrentKeyName = enable ? `-${cleanName}` : cleanName;
+ const psNewKeyName = enable ? cleanName : `-${cleanName}`;
+ return `
+$ErrorActionPreference = 'Stop'
+New-PSDrive -Name HKCR -PSProvider Registry -Root HKEY_CLASSES_ROOT -ErrorAction SilentlyContinue | Out-Null
+$parentPath = '${psParentPath}'
+$currentKey = '${psCurrentKeyName}'
+$newKey = '${psNewKeyName}'
+$fullPath = Join-Path $parentPath $currentKey
+if (-not (Test-Path -LiteralPath $fullPath)) {
+ throw "ShellExt key not found: $fullPath"
+}
+Rename-Item -LiteralPath $fullPath -NewName $newKey -Force
+Write-Output '{"ok":true}'
+`.trim();
+ }
+}
+ |
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 +25 +26 +27 +28 +29 +30 +31 +32 +33 +34 +35 +36 +37 +38 +39 +40 +41 +42 +43 +44 +45 +46 +47 +48 +49 +50 +51 +52 +53 +54 +55 +56 +57 +58 +59 +60 +61 +62 +63 +64 +65 +66 +67 +68 +69 +70 +71 +72 +73 +74 +75 +76 +77 +78 +79 +80 +81 +82 +83 +84 +85 +86 +87 +88 +89 +90 +91 +92 +93 +94 +95 +96 +97 +98 +99 +100 +101 +102 +103 +104 +105 +106 +107 +108 +109 +110 +111 +112 +113 +114 +115 +116 +117 +118 +119 +120 +121 +122 +123 +124 +125 +126 +127 +128 +129 +130 +131 +132 +133 +134 +135 +136 +137 +138 +139 +140 +141 +142 +143 +144 +145 +146 +147 +148 +149 +150 +151 +152 +153 +154 +155 +156 +157 +158 +159 +160 +161 +162 +163 +164 +165 +166 +167 +168 +169 +170 +171 +172 +173 +174 +175 +176 +177 +178 +179 +180 +181 +182 | + + + + + +1x + + + + + + + + + +1x + + + + + + + + + +1x + + + + + + + + + + + + + + + +4x +4x +4x + + +4x + + + + + + +2x +2x +2x + +2x +2x +2x + + +2x +2x +2x +2x +2x + + + + +2x + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +2x +2x +3x + +2x + + + + + + + + + + + + + + + + + + + + + + +1x +1x + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +1x +1x + + + | import { MenuScene, MenuItemType } from '../../shared/enums';
+import { MenuItemEntry } from '../../shared/types';
+import { PowerShellBridge } from './PowerShellBridge';
+import log from '../utils/logger';
+
+// 与 C# RegistryService._sceneRegistryPaths 完全一致
+const SCENE_REGISTRY_PATHS: Record<MenuScene, string> = {
+ [MenuScene.Desktop]: 'DesktopBackground\\Shell',
+ [MenuScene.File]: '*\\shell',
+ [MenuScene.Folder]: 'Directory\\shell',
+ [MenuScene.Drive]: 'Drive\\shell',
+ [MenuScene.DirectoryBackground]:'Directory\\Background\\shell',
+ [MenuScene.RecycleBin]: 'CLSID\\{645FF040-5081-101B-9F08-00AA002F954E}\\shell',
+};
+
+// Shell 扩展（COM）注册路径：shellex\ContextMenuHandlers
+const SCENE_SHELLEX_PATHS: Record<MenuScene, string> = {
+ [MenuScene.Desktop]: 'DesktopBackground\\shellex\\ContextMenuHandlers',
+ [MenuScene.File]: '*\\shellex\\ContextMenuHandlers',
+ [MenuScene.Folder]: 'Directory\\shellex\\ContextMenuHandlers',
+ [MenuScene.Drive]: 'Drive\\shellex\\ContextMenuHandlers',
+ [MenuScene.DirectoryBackground]:'Directory\\Background\\shellex\\ContextMenuHandlers',
+ [MenuScene.RecycleBin]: 'CLSID\\{645FF040-5081-101B-9F08-00AA002F954E}\\shellex\\ContextMenuHandlers',
+};
+
+// 完整 HKCR 前缀（用于显示）
+const HKCR_PREFIX = 'HKEY_CLASSES_ROOT';
+
+interface PsMenuItemRaw {
+ name: string;
+ command: string;
+ iconPath: string | null;
+ isEnabled: boolean;
+ source: string;
+ registryKey: string;
+ subKeyName: string;
+ itemType?: string; // 'ShellExt' for shell extensions
+}
+
+export class RegistryService {
+ private readonly ps: PowerShellBridge;
+ /** 事务回滚数据：registryKey → 原始 isEnabled */
+ private rollbackData = new Map<string, boolean>();
+ private inTransaction = false;
+ private nextId = 1;
+
+ constructor(ps: PowerShellBridge) {
+ this.ps = ps;
+ }
+
+ /**
+ * 获取指定场景下的所有菜单条目（Classic Shell + Shell 扩展）
+ */
+ async getMenuItems(scene: MenuScene): Promise<MenuItemEntry[]> {
+ const basePath = SCENE_REGISTRY_PATHS[scene];
+ const shellexPath = SCENE_SHELLEX_PATHS[scene];
+ try {
+ // 读取 Classic Shell 命令
+ const script = this.ps.buildGetItemsScript(basePath);
+ const raw = await this.ps.execute<PsMenuItemRaw[]>(script);
+ const items = Array.isArray(raw) ? raw : (raw ? [raw] : []);
+
+ // 读取 Shell 扩展（COM ContextMenuHandlers），失败不阻断主流程
+ let shellexItems: PsMenuItemRaw[] = [];
+ try {
+ const shellexScript = this.ps.buildGetShellExtItemsScript(shellexPath);
+ const shellexRaw = await this.ps.execute<PsMenuItemRaw[]>(shellexScript);
+ shellexItems = Array.isArray(shellexRaw) ? shellexRaw : (shellexRaw ? [shellexRaw] : []);
+ } catch (e) {
+ log.warn(`getMenuItems shellex(${scene}) failed (non-fatal):`, e);
+ }
+
+ return [...items, ...shellexItems].map((r) => ({
+ id: this.nextId++,
+ name: r.name,
+ command: r.command,
+ iconPath: r.iconPath,
+ isEnabled: r.isEnabled,
+ source: r.source || this.inferSource(r.subKeyName),
+ menuScene: scene,
+ registryKey: r.registryKey,
+ type: this.determineType(r.itemType),
+ }));
+ } catch (e) {
+ log.error(`getMenuItems(${scene}) failed:`, e);
+ throw new Error(`读取注册表场景 ${scene} 失败: ${(e as Error).message}`);
+ }
+ }
+
+ /**
+ * 启用或禁用单个菜单条目
+ * ShellExt 通过重命名键（±前缀）实现；Classic Shell 通过 LegacyDisable 值实现
+ * ShellExt 通过重命名键（±前缀）实现，registryKey 已归一化，身份不变
+ */
+ async setItemEnabled(registryKey: string, enabled: boolean): Promise<{ newRegistryKey?: string }> {
+ try {
+ if (this.isShellExtKey(registryKey)) {
+ const script = this.ps.buildShellExtToggleScript(registryKey, enabled);
+ await this.ps.executeElevated<{ ok: boolean }>(script);
+ return {};
+ } else {
+ const script = this.ps.buildSetEnabledScript(registryKey, enabled);
+ await this.ps.executeElevated<{ ok: boolean }>(script);
+ return {};
+ }
+ } catch (e) {
+ if (this.inTransaction) {
+ await this.rollback();
+ }
+ throw new Error(
+ `修改菜单项状态失败 [${registryKey}]: ${(e as Error).message}`
+ );
+ }
+ }
+
+ /**
+ * 创建回滚点（事务开始前调用）
+ */
+ createRollbackPoint(items: Array<{ registryKey: string; isEnabled: boolean }>): void {
+ this.rollbackData.clear();
+ for (const item of items) {
+ this.rollbackData.set(item.registryKey, item.isEnabled);
+ }
+ this.inTransaction = true;
+ }
+
+ /**
+ * 回滚到之前保存的状态
+ */
+ async rollback(): Promise<void> {
+ if (!this.inTransaction) return;
+ log.warn('Rolling back registry changes...');
+ try {
+ for (const [key, wasEnabled] of this.rollbackData) {
+ await this.setItemEnabledInternal(key, wasEnabled);
+ }
+ } finally {
+ this.inTransaction = false;
+ this.rollbackData.clear();
+ }
+ }
+
+ /**
+ * 提交事务（清空回滚数据）
+ */
+ commitTransaction(): void {
+ this.inTransaction = false;
+ this.rollbackData.clear();
+ }
+
+ /**
+ * 获取场景对应的完整注册表路径（用于 UI 显示）
+ */
+ getFullRegistryPath(scene: MenuScene): string {
+ return `${HKCR_PREFIX}\\${SCENE_REGISTRY_PATHS[scene]}`;
+ }
+
+ private async setItemEnabledInternal(registryKey: string, enabled: boolean): Promise<void> {
+ if (this.isShellExtKey(registryKey)) {
+ const script = this.ps.buildShellExtToggleScript(registryKey, enabled);
+ await this.ps.executeElevated<{ ok: boolean }>(script);
+ } else {
+ const script = this.ps.buildSetEnabledScript(registryKey, enabled);
+ await this.ps.executeElevated<{ ok: boolean }>(script);
+ }
+ }
+
+ /** Shell 扩展的 registryKey 包含 'shellex' 和 'ContextMenuHandlers' 路径段 */
+ private isShellExtKey(registryKey: string): boolean {
+ return registryKey.includes('shellex') && registryKey.includes('ContextMenuHandlers');
+ }
+
+private inferSource(subKeyName: string): string {
+ return subKeyName || '';
+ }
+
+ private determineType(itemType?: string): MenuItemType {
+ Iif (itemType === 'ShellExt') return MenuItemType.ShellExt;
+ return MenuItemType.System;
+ }
+}
+ |
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| File | ++ | Statements | ++ | Branches | ++ | Functions | ++ | Lines | ++ |
|---|---|---|---|---|---|---|---|---|---|
| BackupService.ts | +
+
+ |
+ 18.98% | +15/79 | +2.85% | +1/35 | +36.36% | +4/11 | +22.38% | +15/67 | +
| MenuManagerService.ts | +
+
+ |
+ 100% | +46/46 | +85.71% | +12/14 | +100% | +9/9 | +100% | +38/38 | +
| PowerShellBridge.ts | +
+
+ |
+ 14.81% | +8/54 | +14.28% | +3/21 | +33.33% | +2/6 | +15.38% | +8/52 | +
| RegistryService.ts | +
+
+ |
+ 47.45% | +28/59 | +16.66% | +4/24 | +50% | +6/12 | +48.21% | +27/56 | +
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 +25 +26 +27 +28 +29 +30 +31 +32 +33 +34 +35 +36 +37 +38 +39 +40 +41 +42 +43 +44 +45 +46 +47 +48 +49 +50 +51 +52 +53 +54 | + + + + +9x + + + + + + + +8x +7x +4x +4x + + + + + + + +4x + +1x + +4x + + + + + + + +3x +3x + +3x +3x + + + +2x +1x + + + + + +3x + + | import { execFileSync, execFile } from 'child_process';
+import { app } from 'electron';
+import log from './logger';
+
+// 进程级缓存，避免每次写操作都 spawn 子进程检测
+let _adminCache: boolean | null = null;
+
+/**
+ * 检查当前进程是否以管理员身份运行（进程令牌已提权）
+ * 使用 Windows Security Principal API，比 net session 更可靠：
+ * net session 在域环境或特定组策略下可能误报 true
+ */
+export function isAdmin(): boolean {
+ if (process.platform !== 'win32') return true;
+ if (_adminCache !== null) return _adminCache;
+ try {
+ const out = execFileSync(
+ 'powershell.exe',
+ [
+ '-NonInteractive', '-NoProfile', '-Command',
+ '([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)',
+ ],
+ { stdio: 'pipe' }
+ ).toString().trim();
+ _adminCache = out === 'True';
+ } catch {
+ _adminCache = false;
+ }
+ return _adminCache;
+}
+
+/**
+ * 以管理员身份重启应用（UAC 提权）
+ * 使用 PowerShell Start-Process -Verb RunAs
+ */
+export function restartAsAdmin(): void {
+ const exePath = app.getPath('exe');
+ log.info(`Restarting as admin: ${exePath}`);
+
+ const script = `Start-Process -FilePath "${exePath}" -Verb RunAs`;
+ execFile(
+ 'powershell.exe',
+ ['-NonInteractive', '-NoProfile', '-Command', script],
+ (err) => {
+ if (err) {
+ log.error('Failed to restart as admin:', err);
+ }
+ }
+ );
+
+ // 延迟退出，确保 Start-Process 已发出
+ setTimeout(() => app.quit(), 500);
+}
+ |
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| File | ++ | Statements | ++ | Branches | ++ | Functions | ++ | Lines | ++ |
|---|---|---|---|---|---|---|---|---|---|
| AdminHelper.ts | +
+
+ |
+ 100% | +18/18 | +100% | +6/6 | +100% | +4/4 | +100% | +15/15 | +
| ipcWrapper.ts | +
+
+ |
+ 100% | +7/7 | +100% | +2/2 | +100% | +2/2 | +100% | +7/7 | +
| logger.ts | +
+
+ |
+ 0% | +0/6 | +100% | +0/0 | +0% | +0/3 | +0% | +0/6 | +
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 | + + + + + + + + + + + +8x +8x +8x +1x + +7x +7x +7x + + + + | import { IpcResult } from '../../shared/types';
+import log from 'electron-log';
+
+/**
+ * 统一 IPC handler 包装：捕获异常，返回 IpcResult<T>
+ * renderer 层无需 try-catch
+ */
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export function wrapHandler<T>(
+ fn: (...args: any[]) => T | Promise<T>
+): (...args: any[]) => Promise<IpcResult<T>> {
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
+ return async (...args: any[]): Promise<IpcResult<T>> => {
+ try {
+ const data = await fn(...args);
+ return { success: true, data };
+ } catch (e: unknown) {
+ const msg = e instanceof Error ? e.message : String(e);
+ log.error('[IPC Error]', msg, e);
+ return { success: false, error: msg };
+ }
+ };
+}
+ |
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 | + + + + + + + + + + + + + + + + + | import log from 'electron-log';
+import path from 'path';
+import { app } from 'electron';
+
+export function initLogger(): void {
+ log.transports.file.resolvePathFn = () =>
+ path.join(app.getPath('userData'), 'logs', 'main.log');
+ log.transports.file.level = 'info';
+ log.transports.file.maxDays = 7;
+ log.transports.console.level = 'debug';
+}
+
+export function getLogDir(): string {
+ return path.join(app.getPath('userData'), 'logs');
+}
+
+export default log;
+ |
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 +25 +26 +27 +28 +29 +30 | 6x +6x +6x +6x +6x +6x +6x + + +6x +6x +6x +6x + + +6x +6x +6x +6x +6x +6x +6x +6x + + +6x +6x +6x + + | export enum MenuScene {
+ Desktop = 'Desktop',
+ File = 'File',
+ Folder = 'Folder',
+ Drive = 'Drive',
+ DirectoryBackground = 'DirectoryBackground',
+ RecycleBin = 'RecycleBin',
+}
+
+export enum MenuItemType {
+ System = 'System',
+ Custom = 'Custom',
+ ShellExt = 'ShellExt',
+}
+
+export enum OperationType {
+ Create = 'Create',
+ Update = 'Update',
+ Delete = 'Delete',
+ Enable = 'Enable',
+ Disable = 'Disable',
+ Backup = 'Backup',
+ Restore = 'Restore',
+}
+
+export enum BackupType {
+ Auto = 'Auto',
+ Manual = 'Manual',
+}
+ |
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| File | ++ | Statements | ++ | Branches | ++ | Functions | ++ | Lines | ++ |
|---|---|---|---|---|---|---|---|---|---|
| enums.ts | +
+
+ |
+ 100% | +22/22 | +100% | +0/0 | +100% | +4/4 | +100% | +22/22 | +
| ipc-channels.ts | +
+
+ |
+ 100% | +1/1 | +100% | +0/0 | +100% | +0/0 | +100% | +1/1 | +
+ Press n or j to go to the next uncovered block, b, p or k for the previous block. +
+ +| 1 +2 +3 +4 +5 +6 +7 +8 +9 +10 +11 +12 +13 +14 +15 +16 +17 +18 +19 +20 +21 +22 +23 +24 +25 +26 +27 +28 +29 +30 +31 +32 +33 +34 | +1x + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + | // IPC 频道常量 — renderer 与 main 共用
+export const IPC = {
+ // 菜单管理
+ REGISTRY_GET_ITEMS: 'registry:getItems',
+ REGISTRY_TOGGLE: 'registry:toggle',
+ REGISTRY_BATCH: 'registry:batch',
+
+ // 操作历史
+ HISTORY_GET_ALL: 'history:getAll',
+ HISTORY_UNDO: 'history:undo',
+ HISTORY_CLEAR: 'history:clear',
+
+ // 备份管理
+ BACKUP_GET_ALL: 'backup:getAll',
+ BACKUP_CREATE: 'backup:create',
+ BACKUP_RESTORE: 'backup:restore',
+ BACKUP_DELETE: 'backup:delete',
+ BACKUP_EXPORT: 'backup:export',
+ BACKUP_IMPORT: 'backup:import',
+ BACKUP_PREVIEW_DIFF: 'backup:previewDiff',
+
+ // 系统/窗口
+ SYS_IS_ADMIN: 'sys:isAdmin',
+ SYS_RESTART_AS_ADMIN: 'sys:restartAsAdmin',
+ SYS_OPEN_REGEDIT: 'sys:openRegedit',
+ SYS_OPEN_LOG_DIR: 'sys:openLogDir',
+ SYS_COPY_CLIPBOARD: 'sys:copyClipboard',
+ SYS_OPEN_EXTERNAL: 'sys:openExternal',
+ WIN_MINIMIZE: 'win:minimize',
+ WIN_MAXIMIZE: 'win:maximize',
+ WIN_CLOSE: 'win:close',
+ WIN_IS_MAXIMIZED: 'win:isMaximized',
+} as const;
+ |