grunt-qiniu-qupload/tasks/qiniu_qupload.js

a grunt plugin which can upload assets to qiniu.

/*
 * grunt-qiniu-upload
 * https://github.com/wjs/grunt-qiniu-upload
 *
 * Copyright (c) 2015 wjs
 * Licensed under the MIT license.
 */

'use strict';

module.exports = function(grunt) {

  var fs = require('fs');
  var path = require('path');
  var qiniu = require('qiniu');


  // Please see the Grunt documentation for more information regarding task
  // creation: http://gruntjs.com/creating-tasks

  grunt.registerMultiTask('qiniu_qupload', 'a grunt plugin which can upload assets to qiniu.', function() {

    function uploadFile(localFile, prefix, options) {

      var uptoken = (new qiniu.rs.PutPolicy(options.bucket)).token();
      var key = prefix + path.basename(localFile);
      var extra = new qiniu.io.PutExtra();

      if (options.overwrite || options.removeExistOnly) {
        var client = new qiniu.rs.Client();
        client.remove(options.bucket, key, function(err, ret) {
          if (!err) {
            // ok
            console.log('Delete old file success! >>> ', key);
          } else {
            console.log('Delete old file error! >>> ', key, err);
            // http://developer.qiniu.com/docs/v6/api/reference/codes.html
          }
        });
        if (options.removeExistOnly) {
          return;
        }
      }

      qiniu.io.putFile(uptoken, key, localFile, extra, function(err, ret) {
        if (!err) {
          // 上传成功， 处理返回值
          console.log('upload success! >>> ', ret.key);
          // ret.key & ret.hash
        } else {
          // 上传失败， 处理返回代码
          console.log(err);
          // http://developer.qiniu.com/docs/v6/api/reference/codes.html
        }
      });

    }

    function uploadFileOrDir(filePath, prefix, skip, options) {
      if (grunt.file.exists(filePath)) {
        if (grunt.file.isFile(filePath)) {
          uploadFile(filePath, prefix, options);
        } else if (grunt.file.isDir(filePath)) {
          grunt.file.recurse(filePath, function(abspath, rootdir, subdir, filename) {
            if (skip && skip.indexOf(filename) > -1) {} else {
              uploadFile(abspath, prefix, options);
            }
          });
        } else {
          console.error('[ERROR] >>> ', filePath, 'is not a file or directory!');
        }
      } else {
        console.error('[ERROR] >>> ', filePath, 'is not exists!');
      }
    }


    // Merge task-specific and/or target-specific options with these defaults.
    var options = this.options({
      cwd: '.',
      ak: null,
      sk: null,
      bucket: null,
      removeExistOnly: false,
      overwrite: false,
      assets: null
    });

    // qiniu.io is async
    this.async();

    if (options && options.ak && options.sk) {
      qiniu.conf.ACCESS_KEY = options.ak;
      qiniu.conf.SECRET_KEY = options.sk;

      if (options.cwd && options.bucket && options.assets && Array.isArray(options.assets)) {
        for (var i = options.assets.length - 1; i >= 0; i--) {
          uploadFileOrDir(path.resolve(options.cwd, options.assets[i].src), options.assets[i].prefix, options.assets[i].skip, options);
        }
      }
    }

  });

};