🔨 Refactor(custom): refactor short url ggenerationg func

src/main/utils/common.ts

@@ -59,25 +59,15 @@ export const showMessageBox = (options: any) => {
 }
 
 export const calcDurationRange = (duration: number) => {
-  if (duration < 1000) {
+  if (duration < 1000) return 500
-    return 500
+  if (duration < 1500) return 1000
-  } else if (duration < 1500) {
+  if (duration < 3000) return 2000
-    return 1000
+  if (duration < 5000) return 3000
-  } else if (duration < 3000) {
+  if (duration < 7000) return 5000
-    return 2000
+  if (duration < 10000) return 8000
-  } else if (duration < 5000) {
+  if (duration < 12000) return 10000
-    return 3000
+  if (duration < 20000) return 15000
-  } else if (duration < 7000) {
+  if (duration < 30000) return 20000
-    return 5000
-  } else if (duration < 10000) {
-    return 8000
-  } else if (duration < 12000) {
-    return 10000
-  } else if (duration < 20000) {
-    return 15000
-  } else if (duration < 30000) {
-    return 20000
-  }
   // max range
   return 100000
 }
@@ -105,22 +95,19 @@ export const ensureFilePath = (filePath: string, prefix = 'file://'): string =>
 export const getClipboardFilePath = (): string => {
   // TODO: linux support
   const img = clipboard.readImage()
-  if (img.isEmpty()) {
+  const platform = process.platform
-    if (process.platform === 'win32') {
+
-      const imgPath = clipboard.readBuffer('FileNameW')?.toString('ucs2')?.replace(RegExp(String.fromCharCode(0), 'g'), '')
+  if (!img.isEmpty() && platform === 'darwin') {
-      if (imgPath) {
-        return imgPath
-      }
-    }
-  } else {
-    if (process.platform === 'darwin') {
     let imgPath = clipboard.read('public.file-url') // will get file://xxx/xxx
     imgPath = ensureFilePath(imgPath)
-      if (imgPath) {
+    return imgPath ? imgPath.replace('file://', '') : ''
-        return imgPath.replace('file://', '')
-      }
   }
+
+  if (img.isEmpty() && platform === 'win32') {
+    const imgPath = clipboard.readBuffer('FileNameW')?.toString('ucs2')?.replace(RegExp(String.fromCharCode(0), 'g'), '')
+    return imgPath || ''
   }
+
   return ''
 }
 
@@ -128,9 +115,7 @@ export const handleUrlEncodeWithSetting = (url: string) => db.get(configPaths.se
 
 const c1nApi = 'https://c1n.cn/link/short'
 
-export const generateShortUrl = async (url: string) => {
+const generateC1NShortUrl = async (url: string) => {
-  const server = db.get(configPaths.settings.shortUrlServer) || IShortUrlServer.C1N
-  if (server === IShortUrlServer.C1N) {
   const form = new FormData()
   form.append('url', url)
   const c1nToken = db.get(configPaths.settings.c1nToken) || ''
@@ -150,7 +135,10 @@ export const generateShortUrl = async (url: string) => {
   } catch (e: any) {
     logger.error(e)
   }
-  } else if (server === IShortUrlServer.YOURLS) {
+  return url
+}
+
+const generateYOURLSShortUrl = async (url: string) => {
   let domain = db.get(configPaths.settings.yourlsDomain) || ''
   const signature = db.get(configPaths.settings.yourlsSignature) || ''
   if (domain && signature) {
@@ -159,11 +147,11 @@ export const generateShortUrl = async (url: string) => {
     }
     try {
       const res = await axios.get(`${domain}/yourls-api.php?signature=${signature}&action=shorturl&format=json&url=${url}`)
-        if (res.data.shorturl) {
+      if (res.data?.shorturl) {
         return res.data.shorturl
       }
     } catch (e: any) {
-        if (e.response.data.message.indexOf('already exists in database') !== -1) {
+      if (e.response?.data?.message?.indexOf('already exists in database') !== -1) {
         return e.response.data.shorturl
       }
       logger.error(e)
@@ -171,7 +159,10 @@ export const generateShortUrl = async (url: string) => {
   } else {
     logger.warn('Yourls server or signature is not set')
   }
-  } else if (server === IShortUrlServer.CFWORKER) {
+  return url
+}
+
+const generateCFWORKERShortUrl = async (url: string) => {
   let cfWorkerHost = db.get(configPaths.settings.cfWorkerHost) || ''
   cfWorkerHost = cfWorkerHost.replace(/\/$/, '')
   if (cfWorkerHost) {
@@ -179,7 +170,7 @@ export const generateShortUrl = async (url: string) => {
       const res = await axios.post(cfWorkerHost, {
         url
       })
-        if (res.data.status === 200 && res.data.key.startsWith('/')) {
+      if (res.data?.status === 200 && res.data?.key?.startsWith('/')) {
         return `${cfWorkerHost}${res.data.key}`
       }
     } catch (e: any) {
@@ -188,6 +179,19 @@ export const generateShortUrl = async (url: string) => {
   } else {
     logger.warn('CF Worker host is not set')
   }
-  }
   return url
 }
+
+export const generateShortUrl = async (url: string) => {
+  const server = db.get(configPaths.settings.shortUrlServer) || IShortUrlServer.C1N
+  switch (server) {
+    case IShortUrlServer.C1N:
+      return generateC1NShortUrl(url)
+    case IShortUrlServer.YOURLS:
+      return generateYOURLSShortUrl(url)
+    case IShortUrlServer.CFWORKER:
+      return generateCFWORKERShortUrl(url)
+    default:
+      return url
+  }
+}