fix: correct uninstall script

Author: subframe7536

src/cache.ts

@@ -0,0 +1,27 @@
+import fs from 'node:fs'
+import os from 'node:os'
+
+export const cacheFilePath = `${os.tmpdir()}/__CUS_CACHE__`
+
+type ConfigCacheItem = [src: string, bak: string]
+export type ConfigCache = {
+  builtin: ConfigCacheItem[]
+  extension: ConfigCacheItem[]
+}
+
+let _cache: ConfigCache = {
+  builtin: [],
+  extension: [],
+}
+export function addBuiltinConfigCache(src: string, bak: string) {
+  _cache.builtin.push([src, bak])
+}
+
+export function addExtensionConfigCache(src: string, bak: string) {
+  _cache.extension.push([src, bak])
+}
+
+export function flushCache() {
+  const data = JSON.stringify(_cache, null, 2)
+  fs.writeFileSync(cacheFilePath, data, 'utf-8')
+}

src/manager/base.ts

@@ -4,6 +4,7 @@ import fs from 'node:fs'
 
 import { readFileSync, writeFileSync } from 'atomically'
 
+import { addBuiltinConfigCache, addExtensionConfigCache } from '../cache'
 import { log } from '../logger'
 import { promptWarn } from '../utils'
 
@@ -19,7 +20,14 @@ export abstract class BaseFileManager implements FileManager {
   constructor(
     public srcPath: string,
     public bakPath: string,
-  ) { }
+    isExtension?: boolean,
+  ) {
+    if (isExtension) {
+      addExtensionConfigCache(srcPath, bakPath)
+    } else {
+      addBuiltinConfigCache(srcPath, bakPath)
+    }
+  }
 
   get hasBakFile() {
     return fs.existsSync(this.bakPath)

src/manager/extension.ts

@@ -4,6 +4,7 @@ import path from 'node:path'
 
 import { extensions } from 'vscode'
 
+import { flushCache } from '../cache'
 import { config } from '../config'
 import { log } from '../logger'
 import { promptWarn } from '../utils'
@@ -53,7 +54,7 @@ function tryParseRegex(str: string): RegExp | string {
 
 class ExtensionFileManager extends BaseFileManager {
   constructor(private config: ExtensionPatchConfig) {
-    super(config.filePath, getBackupPath(config.filePath))
+    super(config.filePath, getBackupPath(config.filePath), true)
   }
 
   patch(content: string): Promisable<string> {
@@ -98,5 +99,6 @@ export function createExtensionFileManagers(skipWarnIfExtensionNotExist = false)
       )
     }
   }
+  flushCache()
   return managers
 }

src/manager/index.ts

@@ -2,6 +2,7 @@ import type { FileManager } from './base'
 
 import { version } from 'vscode'
 
+import { flushCache } from '../cache'
 import { config } from '../config'
 import { runAndRestart } from '../utils'
 import { CssFileManager } from './css'
@@ -29,6 +30,7 @@ export function createFileManagers() {
     new WebViewFileManager(),
     new JsonFileManager(), // MUST be the end of built-in file managers
   ]
+  flushCache()
 
   return {
     hasBakFile: () => builtinManagers.every(m => m.hasBakFile),

src/uninstall.ts

@@ -1,46 +1,19 @@
+import type { ConfigCache } from './cache'
+
 import fs from 'node:fs'
 
-import { createExtensionFileManagers } from './manager/extension'
-import * as paths from './path'
+import { cacheFilePath } from './cache'
 
 function uninstall(srcPath: string, bakPath: string) {
   if (fs.existsSync(srcPath)) {
     fs.renameSync(bakPath, srcPath)
   }
 }
 
-uninstall(
-  paths.cssPath,
-  paths.cssBakPath,
-)
-
-uninstall(
-  paths.mainPath,
-  paths.mainBakPath,
-)
-
-uninstall(
-  paths.rendererPath,
-  paths.rendererBakPath,
-)
-
-uninstall(
-  paths.webviewHTMLPath,
-  paths.webviewHTMLBakPath,
-)
-
-if (paths.htmlPath) {
-  uninstall(
-    paths.htmlPath,
-    paths.htmlBakPath,
+try {
+  const cache: ConfigCache = JSON.parse(fs.readFileSync(cacheFilePath, 'utf-8'))
+  Promise.all(
+    [...cache.builtin, ...cache.extension]
+      .map(async ([src, bak]) => uninstall(src, bak)),
   )
-}
-
-uninstall(
-  paths.productJSONPath,
-  paths.productJSONBakPath,
-)
-
-for (const manager of createExtensionFileManagers(true)) {
-  uninstall(manager.srcPath, manager.bakPath)
-}
+} catch { }