Added: upyun support

src/main/index.js

@@ -23,6 +23,14 @@ const settingWinURL = process.env.NODE_ENV === 'development'
   ? `http://localhost:9080/#setting/upload`
   : `file://${__dirname}/index.html#setting/upload`
 
+const uploadFailed = () => {
+  const notification = new Notification({
+    title: '上传失败',
+    body: '请检查你的图床配置！'
+  })
+  notification.show()
+}
+
 function createTray () {
   tray = new Tray(`${__static}/menubar.png`)
   const contextMenu = Menu.buildFromTemplate([
@@ -68,6 +76,15 @@ function createTray () {
             db.read().set('picBed.current', 'tcyun')
               .write()
           }
+        },
+        {
+          label: '又拍云图床',
+          type: 'radio',
+          checked: db.read().get('picBed.current').value() === 'upyun',
+          click () {
+            db.read().set('picBed.current', 'upyun')
+              .write()
+          }
         }
       ]
     },
@@ -109,18 +126,22 @@ function createTray () {
   tray.on('drop-files', async (event, files) => {
     const pasteStyle = db.read().get('picBed.pasteStyle').value() || 'markdown'
     const imgs = await uploader(files, 'imgFromPath', window.webContents)
-    for (let i in imgs) {
-      clipboard.writeText(pasteTemplate(pasteStyle, imgs[i].imgUrl))
-      const notification = new Notification({
-        title: '上传成功',
-        body: imgs[i].imgUrl,
-        icon: files[i]
-      })
-      setTimeout(() => {
-        notification.show()
-      }, i * 100)
+    if (imgs !== false) {
+      for (let i in imgs) {
+        clipboard.writeText(pasteTemplate(pasteStyle, imgs[i].imgUrl))
+        const notification = new Notification({
+          title: '上传成功',
+          body: imgs[i].imgUrl,
+          icon: files[i]
+        })
+        setTimeout(() => {
+          notification.show()
+        }, i * 100)
+      }
+      window.webContents.send('dragFiles', imgs)
+    } else {
+      uploadFailed()
     }
-    window.webContents.send('dragFiles', imgs)
   })
   // toggleWindow()
 }
@@ -232,35 +253,43 @@ const showWindow = () => {
 
 ipcMain.on('uploadClipboardFiles', async (evt, file) => {
   const img = await uploader(file, 'imgFromClipboard', window.webContents)
-  const pasteStyle = db.read().get('picBed.pasteStyle').value() || 'markdown'
-  clipboard.writeText(pasteTemplate(pasteStyle, img[0].imgUrl))
-  const notification = new Notification({
-    title: '上传成功',
-    body: img[0].imgUrl,
-    icon: file[0]
-  })
-  notification.show()
-  window.webContents.send('clipboardFiles', [])
-  window.webContents.send('uploadFiles', img)
+  if (img !== false) {
+    const pasteStyle = db.read().get('picBed.pasteStyle').value() || 'markdown'
+    clipboard.writeText(pasteTemplate(pasteStyle, img[0].imgUrl))
+    const notification = new Notification({
+      title: '上传成功',
+      body: img[0].imgUrl,
+      icon: file[0]
+    })
+    notification.show()
+    window.webContents.send('clipboardFiles', [])
+    window.webContents.send('uploadFiles', img)
+  } else {
+    uploadFailed()
+  }
 })
 
 ipcMain.on('uploadChoosedFiles', async (evt, files) => {
   const imgs = await uploader(files, 'imgFromUploader', settingWindow.webContents)
-  const pasteStyle = db.read().get('picBed.pasteStyle').value() || 'markdown'
-  let pasteText = ''
-  for (let i in imgs) {
-    pasteText += pasteTemplate(pasteStyle, imgs[i].imgUrl) + '\r\n'
-    const notification = new Notification({
-      title: '上传成功',
-      body: imgs[i].imgUrl,
-      icon: files[i].path
-    })
-    setTimeout(() => {
-      notification.show()
-    }, i * 100)
+  if (imgs !== false) {
+    const pasteStyle = db.read().get('picBed.pasteStyle').value() || 'markdown'
+    let pasteText = ''
+    for (let i in imgs) {
+      pasteText += pasteTemplate(pasteStyle, imgs[i].imgUrl) + '\r\n'
+      const notification = new Notification({
+        title: '上传成功',
+        body: imgs[i].imgUrl,
+        icon: files[i].path
+      })
+      setTimeout(() => {
+        notification.show()
+      }, i * 100)
+    }
+    clipboard.writeText(pasteText)
+    window.webContents.send('uploadFiles', imgs)
+  } else {
+    uploadFailed()
   }
-  clipboard.writeText(pasteText)
-  window.webContents.send('uploadFiles', imgs)
 })
 
 app.on('ready', () => {

src/main/utils/upYunUpload.js

@@ -0,0 +1,80 @@
+import request from 'request-promise'
+import * as img2Base64 from './img2base64'
+import db from '../../datastore/index'
+import { Notification, clipboard } from 'electron'
+import crypto from 'crypto'
+import MD5 from 'md5'
+
+// generate COS signature string
+const generateSignature = (fileName) => {
+  const options = db.read().get('picBed.upyun').value()
+  // const apiKey = options.apiKey
+  const operator = options.operator
+  const password = options.password
+  const md5Password = MD5(password)
+  const date = new Date().toGMTString()
+  const uri = `/${options.bucket}/${fileName}`
+  const value = `PUT&${uri}&${date}`
+  const sign = crypto.createHmac('sha1', md5Password).update(value).digest('base64')
+  return `UPYUN ${operator}:${sign}`
+}
+
+const postOptions = (fileName, signature, imgBase64) => {
+  const options = db.read().get('picBed.upyun').value()
+  const bucket = options.bucket
+  return {
+    method: 'PUT',
+    url: `https://v0.api.upyun.com/${bucket}/${fileName}`,
+    headers: {
+      Authorization: signature,
+      Date: new Date().toGMTString()
+    },
+    body: Buffer.from(imgBase64, 'base64'),
+    resolveWithFullResponse: true
+  }
+}
+
+const upYunUpload = async (img, type, webContents) => {
+  try {
+    webContents.send('uploadProgress', 0)
+    const imgList = await img2Base64[type](img)
+    webContents.send('uploadProgress', 30)
+    const length = imgList.length
+    const upyunOptions = db.read().get('picBed.upyun').value()
+    for (let i in imgList) {
+      const singature = generateSignature(imgList[i].fileName)
+      const options = postOptions(imgList[i].fileName, singature, imgList[i].base64Image)
+      const body = await request(options)
+      if (body.statusCode === 200) {
+        delete imgList[i].base64Image
+        imgList[i]['imgUrl'] = `${upyunOptions.url}/${imgList[i].fileName}${upyunOptions.options}`
+        imgList[i]['type'] = 'upyun'
+        if (i - length === -1) {
+          webContents.send('uploadProgress', 60)
+        }
+      } else {
+        webContents.send('uploadProgress', -1)
+        const notification = new Notification({
+          title: '上传失败！',
+          body: '服务端出错，请稍后再试'
+        })
+        notification.show()
+      }
+    }
+    webContents.send('uploadProgress', 100)
+    return imgList
+  } catch (err) {
+    console.log(err)
+    const body = JSON.parse(err.error)
+    webContents.send('uploadProgress', -1)
+    const notification = new Notification({
+      title: '上传失败！',
+      body: `错误码：${body.code}，请打开浏览器粘贴地址查看相关原因。`
+    })
+    notification.show()
+    clipboard.writeText('http://docs.upyun.com/api/errno/')
+    // throw new Error(err)
+  }
+}
+
+export default upYunUpload

src/main/utils/uploader.js

@@ -1,16 +1,33 @@
 import weiboUpload from './weiboUpload'
 import qiniuUpload from './qiniuUpload'
 import tcYunUpload from './tcYunUpload'
+import upYunUpload from './upYunUpload'
 import db from '../../datastore/index'
+
+const checkUploader = (type) => {
+  const currentUploader = db.read().get(`picBed.${type}`).value()
+  if (currentUploader) {
+    return true
+  } else {
+    return false
+  }
+}
+
 const uploader = (img, type, webContents) => {
   const uploadType = db.read().get('picBed.current').value()
-  switch (uploadType) {
-    case 'weibo':
-      return weiboUpload(img, type, webContents)
-    case 'qiniu':
-      return qiniuUpload(img, type, webContents)
-    case 'tcyun':
-      return tcYunUpload(img, type, webContents)
+  if (checkUploader(uploadType)) {
+    switch (uploadType) {
+      case 'weibo':
+        return weiboUpload(img, type, webContents)
+      case 'qiniu':
+        return qiniuUpload(img, type, webContents)
+      case 'tcyun':
+        return tcYunUpload(img, type, webContents)
+      case 'upyun':
+        return upYunUpload(img, type, webContents)
+    }
+  } else {
+    return false
   }
 }