feat(File): support COS (#272)

Author: leeyeh

package.json

@@ -11,6 +11,7 @@
     "test": "gulp test"
   },
   "dependencies": {
+    "form-data": "^1.0.0-rc4",
     "localstorage-memory": "^1.0.1",
     "md5": "^2.0.0",
     "qiniu": "6.1.3",
@@ -49,13 +50,15 @@
   "browser": {
     "react-native": false,
     "./src/browserify-wrapper/ajax.js": "./src/browserify-wrapper/ajax-browser.js",
-    "./src/browserify-wrapper/upload.js": "./src/browserify-wrapper/upload-browser.js",
+    "./src/uploader/qiniu.js": "./src/uploader/qiniu-browser.js",
     "./src/browserify-wrapper/localStorage.js": "./src/browserify-wrapper/localstorage-browser.js",
     "./src/browserify-wrapper/parse-base64.js": "./src/browserify-wrapper/parse-base64-browser.js",
+    "./src/uploader/cos.js": "./src/uploader/cos-browser.js",
     "./dist/node/browserify-wrapper/ajax.js": "./dist/node/browserify-wrapper/ajax-browser.js",
-    "./dist/node/browserify-wrapper/upload.js": "./dist/node/browserify-wrapper/upload-browser.js",
+    "./dist/node/uploader/qiniu.js": "./dist/node/uploader/qiniu-browser.js",
     "./dist/node/browserify-wrapper/localStorage.js": "./dist/node/browserify-wrapper/localstorage-browser.js",
-    "./dist/node/browserify-wrapper/parse-base64.js": "./dist/node/browserify-wrapper/parse-base64-browser.js"
+    "./dist/node/browserify-wrapper/parse-base64.js": "./dist/node/browserify-wrapper/parse-base64-browser.js",
+    "./dist/node/uploader/cos.js": "./dist/node/uploader/cos-browser.js"
   },
   "eslintConfig": {
     "env": {

src/browserify-wrapper/ajax-browser.js

@@ -8,13 +8,18 @@
 const AVPromise = require('../promise');
 const debug = require('debug')('ajax');
 
-const ajax = (method, url, data, headers = {}) => {
+const ajax = (method, url, data, headers = {}, onprogress) => {
   debug(method, url, data, headers);
 
   const promise = new AVPromise();
 
   let handled = false;
   const xhr = new global.XMLHttpRequest();
+
+  if (onprogress && xhr.upload) {
+    xhr.upload.onprogress = onprogress;
+  }
+
   xhr.onreadystatechange = () => {
     if (xhr.readyState === 4) {
       if (handled) {
@@ -48,7 +53,7 @@ const ajax = (method, url, data, headers = {}) => {
     xhr.setRequestHeader(name, headers[name]);
   }
 
-  xhr.send(JSON.stringify(data));
+  xhr.send(data);
   return promise;
 };
 

src/browserify-wrapper/ajax.js

@@ -64,6 +64,6 @@ module.exports = function _ajax(method, resourceUrl, data, headers = {}) {
     promise.reject(err);
   });
 
-  req.end(JSON.stringify(data));
+  req.end(data);
   return promise;
 };

src/browserify-wrapper/upload-browser.js

@@ -6,6 +6,8 @@
 'use strict';
 
 const _ = require('underscore');
+const cos = require('./uploader/cos');
+const qiniu = require('./uploader/qiniu');
 
 module.exports = function(AV) {
 
@@ -617,7 +619,7 @@ module.exports = function(AV) {
      * @return {AV.Promise} Resolved with the response
      * @private
      */
-    _qiniuToken: function(type) {
+    _fileToken: function(type, route = 'fileTokens') {
       const name = this.attributes.name;
       //Create 16-bits uuid as qiniu key.
       const extName = extname(name);
@@ -636,7 +638,7 @@ module.exports = function(AV) {
         this.attributes.metaData.mime_type = type;
       }
       this._qiniu_key = key;
-      return AV._request("qiniu", null, null, 'POST', data);
+      return AV._request(route, null, null, 'POST', data);
     },
 
     /**
@@ -670,8 +672,22 @@ module.exports = function(AV) {
         var isCnNodeFlag = isCnNode();
         if (this._source && isCnNodeFlag) {
           // 通过国内 CDN 服务商上传
-          var upload = require('./browserify-wrapper/upload');
-          upload(this, AV, saveOptions);
+          this._previousSave = this._source.then((data, type) => {
+            return this._fileToken(type).catch(() => this._fileToken(type, 'qiniu'))
+              .then(uploadInfo => {
+                let uploadPromise;
+                if (uploadInfo.provider === 'qcloud') {
+                  uploadPromise = cos(uploadInfo, data, this, saveOptions);
+                } else {
+                  uploadPromise = qiniu(uploadInfo, data, this, saveOptions);
+                }
+                return uploadPromise.catch(err => {
+                  //destroy this file object when upload fails.
+                  this.destroy();
+                  throw err;
+                });
+              });
+          });
         } else if (this.attributes.url && this.attributes.metaData.__source === 'external') {
           //external link file.
           var data = {

src/browserify-wrapper/upload.js

@@ -0,0 +1,16 @@
+'use strict';
+
+const ajax = require('../browserify-wrapper/ajax.js');
+
+module.exports =function upload(uploadInfo, data, file, saveOptions) {
+  file.attributes.url = uploadInfo.url;
+  file._bucket = uploadInfo.bucket;
+  file.id = uploadInfo.objectId;
+  const uploadUrl = uploadInfo.upload_url + "?sign=" + encodeURIComponent(uploadInfo.token);
+  const formData = new FormData();
+  formData.append('fileContent', data);
+  formData.append('op', 'upload');
+
+  return ajax('POST', uploadUrl, formData, undefined, saveOptions.onProgress)
+    .then(() => file);
+};

src/file.js

@@ -0,0 +1,31 @@
+'use strict';
+const FormData = require('form-data');
+
+const ajax = require('../browserify-wrapper/ajax.js');
+const Promise = require('../promise');
+const debug = require('debug')('cos');
+
+module.exports =function upload(uploadInfo, data, file) {
+  file.attributes.url = uploadInfo.url;
+  file._bucket = uploadInfo.bucket;
+  file.id = uploadInfo.objectId;
+  const uploadUrl = uploadInfo.upload_url;
+  const body = new Buffer(data, 'base64');
+  debug(uploadUrl, data);
+  const formData = new FormData();
+  formData.append('fileContent', body);
+  formData.append('op', 'upload');
+
+  const promise = new Promise();
+
+  const request = formData.submit(uploadUrl, function(err, res) {
+    debug(err, res.statusCode);
+    if (err) {
+      promise.reject(err);
+    } else {
+      promise.resolve(file);
+    }
+  });
+  request.setHeader('Authorization', uploadInfo.token);
+  return promise;
+};

src/uploader/cos-browser.js

@@ -0,0 +1,24 @@
+/**
+ * 每位工程师都有保持代码优雅的义务
+ * Each engineer has a duty to keep the code elegant
+**/
+
+'use strict';
+
+const ajax = require('../browserify-wrapper/ajax.js');
+
+module.exports = function upload(uploadInfo, data, file, saveOptions) {
+    file.attributes.url = uploadInfo.url;
+    file._bucket = uploadInfo.bucket;
+    file.id = uploadInfo.objectId;
+    //Get the uptoken to upload files to qiniu.
+    const uptoken = uploadInfo.token;
+    const formData = new FormData();
+    formData.append('file', data);
+    formData.append('name', file.attributes.name);
+    formData.append('key', file._qiniu_key);
+    formData.append('token', uptoken);
+
+    return ajax('POST', 'https://up.qbox.me', formData, undefined, saveOptions.onProgress)
+      .then(() => file);
+};

src/uploader/cos.js

@@ -0,0 +1,33 @@
+/**
+ * 每位工程师都有保持代码优雅的义务
+ * Each engineer has a duty to keep the code elegant
+ **/
+
+'use strict';
+
+//Use qiniu sdk to upload files to qiniu.
+const qiniu = require('qiniu');
+const Promise = require('../promise');
+
+module.exports =function upload(uploadInfo, data, file) {
+    file.attributes.url = uploadInfo.url;
+    file._bucket = uploadInfo.bucket;
+    file.id = uploadInfo.objectId;
+    //Get the uptoken to upload files to qiniu.
+    const uptoken = uploadInfo.token;
+    const promise = new Promise();
+    const extra = new qiniu.io.PutExtra();
+    if(file.attributes.metaData.mime_type)
+      extra.mimeType = file.attributes.metaData.mime_type;
+    const body = new Buffer(data, 'base64');
+    qiniu.io.put(uptoken, file._qiniu_key, body, extra, function(err, ret) {
+      delete file._qiniu_key;
+      delete file.attributes.base64;
+      if(!err) {
+        promise.resolve(file);
+      } else {
+        promise.reject(err);
+      }
+    });
+    return promise;
+};