meipian-qiniu/index.js

const path = require('path')
const fs = require('fs')
const chalk = require('chalk')

const inquiry = require('./lib/inquiry')
const strike = require('./lib/strike')
const upload = require('./lib/upload')
const refresh = require('./lib/refresh')

const home = require('user-home')
const keyDir = path.resolve(home, './.qiniu')
const keyFile = path.resolve(home, './.qiniu/key.json')

function readFiles(srcDir, rootDir, ext) {
  let files = []
  let names = []
  let loop = (function loop(src) {
    let dirs = fs.readdirSync(src)

    for (let temp in dirs) {
      let dir = path.join(src, dirs[temp])
      let stat = fs.statSync(dir);

      if (stat.isFile() &&
        (!(new RegExp((ext || '\\.\\*') + '$').test(dirs[temp])))) {
        files.push(dir = dir.replace(/\\/g, '\/'))
        names.push(dir.replace(new RegExp('^.*/' + rootDir + '/'), '/'))
      } else if (stat.isDirectory()) {
        loop(dir)
      }
    }
  })
  loop(srcDir)
  return {
    files: files,
    names: names
  }
}

module.exports = function(options) {
  let args = arguments
  let lastParam = args[args.length - 1]
  let callback = lastParam && (typeof lastParam === 'function') ? lastParam : ''
  let clone = {
    bucket: options.bucket,
    src: options.src,
    fix: options.fix,
    ext: options.ext,
    urls: options.urls || [],
    rootDir: options.rootDir || '/'
  }
  let bucket = clone.bucket
  let src = clone.src
  let fix = clone.fix
  let ext = clone.ext
  let answer = null

  let saveFile = function() {

    if ((!fs.existsSync(keyFile)) && answer.save) {
      fs.existsSync(keyDir) ? '' : fs.mkdirSync(path.join(home, '.qiniu'))
      fs.createWriteStream(keyFile).write(JSON.stringify(answer));
    }
  }

  {
    if (fs.existsSync(keyFile)) {

      update.apply(answer = require(keyFile), [callback])
    } else {

      inquiry(function(respAnswer) {
        update.apply(answer = respAnswer, [callback])
      })
    }
  }

  function update(fn) {
    let infos = readFiles(src, clone.rootDir, ext)
    let tempFiles = infos.files.slice(0)
    let tempNames = infos.names.slice(0)

    strike.apply(answer, [bucket, fix, function() {
      (function inner() {
        upload.apply(answer, [bucket, path.resolve(tempFiles[0]), fix + tempNames[0], function() {
          tempFiles.splice(0, 1)
          tempNames.splice(0, 1)

          if ((!tempFiles.length)) {
            saveFile.call()
            fn && fn.apply(null, [answer])

            if (clone.urls.length) {
              referUrl(fn)
            }
          } else {
            inner()
          }
        }])
      })()
    }])

    console.log('  ' + chalk.bgRed('  Upload number: ' + infos.files.length + ' files  '))
  }

  function referUrl(fn) {
    refresh.apply(answer, [clone.urls, function() {
      fn && fn.apply(null, [answer])
    }])
  }
}