✨ Feature: add upload-clipboard-image from electron' clipboard

ISSUES CLOSED: #822

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

@@ -10,9 +10,19 @@ import db, { GalleryDB } from '~/main/apis/core/datastore'
 import { handleCopyUrl } from '~/main/utils/common'
 import { handleUrlEncode } from '#/utils/common'
 import { T } from '#/i18n/index'
-export const uploadClipboardFiles = async (): Promise<string> => {
+// import dayjs from 'dayjs'
+
+const handleClipboardUploading = async (): Promise<false | ImgInfo[]> => {
+  const useBuiltinClipboard = !!db.get('settings.useBuiltinClipboard')
   const win = windowManager.getAvailableWindow()
-  const img = await uploader.setWebContents(win!.webContents).upload()
+  if (useBuiltinClipboard) {
+    return await uploader.setWebContents(win!.webContents).uploadWithBuildInClipboard()
+  }
+  return await uploader.setWebContents(win!.webContents).upload()
+}
+
+export const uploadClipboardFiles = async (): Promise<string> => {
+  const img = await handleClipboardUploading()
   if (img !== false) {
     if (img.length > 0) {
       const trayWindow = windowManager.get(IWindowList.TRAY_WINDOW)

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

@@ -2,7 +2,8 @@ import {
   Notification,
   BrowserWindow,
   ipcMain,
-  WebContents
+  WebContents,
+  clipboard
 } from 'electron'
 import dayjs from 'dayjs'
 import picgo from '@core/picgo'
@@ -15,6 +16,8 @@ import { showNotification, calcDurationRange } from '~/main/utils/common'
 import { RENAME_FILE_NAME, TALKING_DATA_EVENT } from '~/universal/events/constants'
 import logger from '@core/picgo/logger'
 import { T } from '~/universal/i18n'
+import fse from 'fs-extra'
+import path from 'path'
 
 const waitForShow = (webcontent: WebContents) => {
   return new Promise<void>((resolve) => {
@@ -109,6 +112,32 @@ class Uploader {
     return this
   }
 
+  /**
+   * use electron's clipboard image to upload
+   */
+  async uploadWithBuildInClipboard (): Promise<ImgInfo[]|false> {
+    let filePath = ''
+    try {
+      const nativeImage = clipboard.readImage()
+      if (nativeImage.isEmpty()) {
+        return false
+      }
+      const buffer = nativeImage.toPNG()
+      const baseDir = picgo.baseDir
+      const fileName = `${dayjs().format('YYYYMMDDHHmmSSS')}.png`
+      filePath = path.join(baseDir, fileName)
+      await fse.writeFile(filePath, buffer)
+      return await this.upload([filePath])
+    } catch (e: any) {
+      logger.error(e)
+      return false
+    } finally {
+      if (filePath) {
+        fse.unlink(filePath)
+      }
+    }
+  }
+
   async upload (img?: IUploadOption): Promise<ImgInfo[]|false> {
     try {
       const startTime = Date.now()