将uploader在commandx中实现

Author: alinode-oss

.gitignore

@@ -0,0 +1 @@
+node_modules

package.json

@@ -18,5 +18,8 @@
     "url": "https://github.com/aliyun-node/commands/issues"
   },
   "homepage": "https://github.com/aliyun-node/commands#readme",
-  "dependencies": {}
+  "dependencies": {
+    "formstream": "^1.0.0",
+    "urllib": "^2.2.2"
+  }
 }

upload_file

@@ -1,20 +1,24 @@
 #!/bin/bash
 
-UPLOADER=`which uploader`
+export ENABLE_NODE_LOG=NO
+
+NODE=`which node`
 
 if [ -f "$HOME/.nodepath" ]; then
-  UPLOADER=`cat $HOME/.nodepath`/uploader
+  NODE=`cat $HOME/.nodepath`/node;
 fi
 
 FULL_PATH=$2
 TARGET_PATH=$2
 
 GZIP=`which gzip`
 
-if [ $GZIP ]; then
+if [ $GZIP ] && [ -f "$FULL_PATH" ]; then
   $GZIP -c $FULL_PATH > $FULL_PATH".gz"
   TARGET_PATH=$FULL_PATH".gz"
 fi
 
+UPLOADER=`pwd`/uploader.js
+
 #uploader server filepath token id
-ENABLE_NODE_LOG=NO $UPLOADER $1 $TARGET_PATH $3 $4
+$NODE UPLOADER $1 $TARGET_PATH $3 $4

uploader.js

@@ -0,0 +1,85 @@
+'use strict';
+
+// ./uploader server filepath token id
+var fs = require('fs');
+var os = require('os');
+var crypto = require('crypto');
+var urllib = require('urllib');
+var formstream = require('formstream');
+var tunnel = require('tunnel-agent');
+
+var argv = process.argv.slice(2);
+
+var server = argv[0];
+var filepath = argv[1];
+var token = argv[2];
+var id = argv[3];
+
+// check args
+if (!server || !filepath || !token || !id) {
+  console.log('参数错误');
+  console.log('\t usage: ./uploader server filepath token id');
+  process.exit(1);
+}
+
+// check filepath
+fs.stat(filepath, function (err, stat) {
+  if (err) {
+    console.log('文件不存在');
+    console.log(err.message);
+    process.exit(1);
+    return;
+  }
+
+  if (stat.size <= 0) {
+    console.log('空文件');
+    process.exit(1);
+    return;
+  }
+
+  var form = formstream();
+  form.file('file', filepath, filepath, stat.size);
+
+  var nonce = '' + parseInt((Math.random() * 100000000000), 10);
+  // get signature
+  var shasum = crypto.createHash('sha1');
+  shasum.update([os.hostname(), token, nonce, id].join(''));
+  var sign = shasum.digest('hex');
+
+  var url = 'http://' + server + '/files/' + id + '?nonce=' + nonce + '&sign=' + sign;
+
+  var gateway = process.env.GATEWAY;
+  if (gateway) {
+    // upload to gateway
+    url = 'http://' + gateway + '/file?target=' + encodeURIComponent(url);
+  }
+
+  var agent = false;
+  if (process.env.http_proxy) {
+    var parts = process.env.http_proxy.split(':');
+    agent = tunnel.httpOverHttp({
+      proxy: {
+        host: parts[0],
+        port: parts[1]
+      }
+    });
+  }
+
+  var opts = {
+    dataType: 'json',
+    type: 'POST',
+    timeout: 60000 * 20, // 20分钟超时
+    headers: form.headers(),
+    stream: form,
+    agent: agent
+  };
+
+  urllib.request(url, opts, function (err, data, res) {
+    if (err) {
+      throw err;
+    }
+
+    console.log(JSON.stringify(data));
+    process.exit(0);
+  });
+});