feat: with lockfile

Author: subframe7536

src/manager/base.ts

@@ -16,7 +16,7 @@ export abstract class BaseFileManager implements FileManager {
   ) {
     if (!this.hasBakFile) {
       cpSync(this.srcPath, this.bakPath)
-      logger.info('Create', this.bakPath)
+      logger.info(`Create backup file [${this.bakPath}]`)
     }
   }
 
@@ -26,7 +26,7 @@ export abstract class BaseFileManager implements FileManager {
 
   async reload() {
     if (!this.hasBakFile) {
-      logger.warn(`bak file [${this.bakPath}] does not exist, skip reload`)
+      logger.warn(`Backup file [${this.bakPath}] does not exist, skip reload`)
     } else {
       writeFileSync(this.srcPath, await this.patch(readFileSync(this.bakPath, 'utf-8')))
       logger.info(`Config reload [${this.srcPath}]`)
@@ -35,7 +35,7 @@ export abstract class BaseFileManager implements FileManager {
 
   async rollback() {
     if (!this.hasBakFile) {
-      logger.warn(`bak file [${this.bakPath}] does not exist, skip rollback`)
+      logger.warn(`Backup file [${this.bakPath}] does not exist, skip rollback`)
     } else {
       const originJS = readFileSync(this.bakPath, 'utf-8')
       writeFileSync(this.srcPath, originJS)

src/manager/css.ts

@@ -10,12 +10,7 @@ const banner = '/* Custom UI Style Start */'
 const footer = '/* Custom UI Style End */'
 
 function generateBackgroundCSS() {
-  const plt = process?.platform
-
-  const url = (
-    config[`backgroundUrl${captialize(plt)}` as keyof ConfigShorthandMap] || config.backgroundUrl
-  ) as string
-
+  const url = config.backgroundUrl
   if (!url) {
     return ''
   }

src/utils.ts

@@ -1,15 +1,44 @@
+import type { Promisable } from '@subframe7536/type-utils'
+import { existsSync, rmSync } from 'node:fs'
+import { readFileSync, writeFileSync } from 'atomically'
 import { useLogger } from 'reactive-vscode'
 import { commands, window } from 'vscode'
-import { displayName } from './generated/meta'
+import { displayName, name } from './generated/meta'
 
 export const logger = useLogger(displayName)
 
-export async function runAndRestart(message: string, action: () => Promise<any>) {
-  await action()
-  const item = await window.showInformationMessage(message, { title: 'Restart vscode' })
-  if (item) {
-    commands.executeCommand('workbench.action.reloadWindow')
+const lockFileName = `__${name}__.lock`
+
+async function runWithLock(fn: () => Promisable<void>) {
+  let count = 5
+  const check = () => existsSync(lockFileName)
+  while (check() && count--) {
+    await new Promise(resolve => setTimeout(resolve, 1000))
+  }
+  if (!count) {
+    // If exists and expire time exceed 10 minutes, just remove it
+    if (check() && Number(readFileSync(lockFileName, 'utf-8')) - Date.now() > 6e5) {
+      rmSync(lockFileName)
+    } else {
+      await showMessage('File locked, cancel operation')
+      return
+    }
   }
+  writeFileSync(lockFileName, String(Date.now()))
+  try {
+    await fn()
+  } finally {
+    rmSync(lockFileName)
+  }
+}
+export async function runAndRestart(message: string, action: () => Promise<any>) {
+  await runWithLock(async () => {
+    await action()
+    const item = await window.showInformationMessage(message, { title: 'Restart vscode' })
+    if (item) {
+      commands.executeCommand('workbench.action.reloadWindow')
+    }
+  })
 }
 
 export async function showMessage(content: string) {