✨ Feature: add delete local file after uploading, fix compatibility with auto-delete

ISSUES CLOSED: #40

src/main/apis/app/system/index.ts

@@ -23,6 +23,7 @@ import clipboardListener from 'clipboard-event'
 import clipboardPoll from '~/main/utils/clipboardPoll'
 import picgo from '../../core/picgo'
 import { uploadClipboardFiles } from '../uploader/apis'
+import { cloneDeep } from 'lodash'
 let contextMenu: Menu | null
 let tray: Tray | null
 
@@ -394,13 +395,18 @@ export function createTray () {
     // so the tray window must be available
     tray.on('drop-files', async (event: Event, files: string[]) => {
       const pasteStyle = db.get('settings.pasteStyle') || 'markdown'
+      const rawInput = cloneDeep(files)
       const trayWindow = windowManager.get(IWindowList.TRAY_WINDOW)!
       const imgs = await uploader
         .setWebContents(trayWindow.webContents)
         .upload(files)
+      const deleteLocalFile = db.get('settings.deleteLocalFile') || false
       if (imgs !== false) {
         const pasteText: string[] = []
         for (let i = 0; i < imgs.length; i++) {
+          if (deleteLocalFile) {
+            await fs.remove(rawInput[i])
+          }
           pasteText.push(await (pasteTemplate(pasteStyle, imgs[i], db.get('settings.customLink'))))
           const notification = new Notification({
             title: T('UPLOAD_SUCCEED'),

src/main/apis/app/uploader/apis.ts

@@ -13,6 +13,8 @@ import { T } from '~/main/i18n/index'
 import ALLApi from '@/apis/allApi'
 import picgo from '@core/picgo'
 import GuiApi from '../../gui'
+import fs from 'fs-extra'
+import { cloneDeep } from 'lodash'
 
 const handleClipboardUploading = async (): Promise<false | ImgInfo[]> => {
   const useBuiltinClipboard = db.get('settings.useBuiltinClipboard') === undefined ? true : !!db.get('settings.useBuiltinClipboard')
@@ -71,12 +73,21 @@ export const uploadClipboardFiles = async (): Promise<IStringKeyMap> => {
 
 export const uploadChoosedFiles = async (webContents: WebContents, files: IFileWithPath[]): Promise<IStringKeyMap[]> => {
   const input = files.map(item => item.path)
+  const rawInput = cloneDeep(input)
   const imgs = await uploader.setWebContents(webContents).upload(input)
   const result = []
   if (imgs !== false) {
     const pasteStyle = db.get('settings.pasteStyle') || 'markdown'
+    const deleteLocalFile = db.get('settings.deleteLocalFile') || false
     const pasteText: string[] = []
     for (let i = 0; i < imgs.length; i++) {
+      if (deleteLocalFile) {
+        fs.remove(rawInput[i]).then(() => {
+          picgo.log.info(`delete local file: ${rawInput[i]}`)
+        }).catch((err: Error) => {
+          picgo.log.error(err)
+        })
+      }
       pasteText.push(await (pasteTemplate(pasteStyle, imgs[i], db.get('settings.customLink'))))
       const notification = new Notification({
         title: T('UPLOAD_SUCCEED'),

src/main/apis/gui/index.ts

@@ -18,6 +18,8 @@ import {
 } from '~/universal/events/constants'
 import { DBStore } from '@picgo/store'
 import { T } from '~/main/i18n'
+import fs from 'fs-extra'
+import { cloneDeep } from 'lodash'
 
 // Cross-process support may be required in the future
 class GuiApi implements IGuiApi {
@@ -76,11 +78,16 @@ class GuiApi implements IGuiApi {
   async upload (input: IUploadOption) {
     this.windowId = await getWindowId()
     const webContents = this.getWebcontentsByWindowId(this.windowId)
+    const rawInput = cloneDeep(input)
     const imgs = await uploader.setWebContents(webContents!).upload(input)
     if (imgs !== false) {
       const pasteStyle = db.get('settings.pasteStyle') || 'markdown'
+      const deleteLocalFile = db.get('settings.deleteLocalFile') || false
       const pasteText: string[] = []
       for (let i = 0; i < imgs.length; i++) {
+        if (deleteLocalFile) {
+          await fs.remove(rawInput[i])
+        }
         pasteText.push(await (pasteTemplate(pasteStyle, imgs[i], db.get('settings.customLink'))))
         const notification = new Notification({
           title: T('UPLOAD_SUCCEED'),