From bba250b0d9fe1a961c055f59d63164654d23d6a6 Mon Sep 17 00:00:00 2001
From: Molunerfinn <marksz@teamsz.xyz>
Date: Wed, 18 Jul 2018 15:34:35 +0800
Subject: [PATCH] Fixed: #80 || mime-type in ali-oss

---
 src/main/utils/aliYunUpload.js | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/main/utils/aliYunUpload.js b/src/main/utils/aliYunUpload.js
index b80ddfa..67ae531 100644
--- a/src/main/utils/aliYunUpload.js
+++ b/src/main/utils/aliYunUpload.js
@@ -3,12 +3,13 @@ import * as img2Base64 from './img2base64'
 import db from '../../datastore/index'
 import { Notification, clipboard } from 'electron'
 import crypto from 'crypto'
+import mime from 'mime-types'
 
 // generate OSS signature
 const generateSignature = (fileName) => {
   const options = db.read().get('picBed.aliyun').value()
   const date = new Date().toGMTString()
-  const signString = `PUT\n\n\n${date}\n/${options.bucket}/${options.path}${fileName}`
+  const signString = `PUT\n\n${mime.lookup(fileName)}\n${date}\n/${options.bucket}/${options.path}${fileName}`
 
   const signature = crypto.createHmac('sha1', options.accessKeySecret).update(signString).digest('base64')
   return `OSS ${options.accessKeyId}:${signature}`
@@ -22,7 +23,8 @@ const postOptions = (fileName, signature, imgBase64) => {
     headers: {
       Host: `${options.bucket}.${options.area}.aliyuncs.com`,
       Authorization: signature,
-      Date: new Date().toGMTString()
+      Date: new Date().toGMTString(),
+      'content-type': mime.lookup(fileName)
     },
     body: Buffer.from(imgBase64, 'base64'),
     resolveWithFullResponse: true