cli/actions.js

   1 'use strict';
   2
   3 exports.login = login;
   4 exports.put = put;
   5 exports.get = get;
   6 exports.del = del;
   7
   8 var superagent = require('superagent'),
   9     config = require('./config.js'),
  10     readlineSync = require('readline-sync'),
  11     async = require('async'),
  12     fs = require('fs'),
  13     request = require('request'),
  14     url = require('url'),
  15     path = require('path');
  16
  17 require('colors');
  18
  19 var API = '/api/files/';
  20
  21 var gQuery = {};
  22
  23 function checkConfig() {
  24     if (!config.server() || !config.username() || !config.password()) {
  25         console.log('You have run "login" first');
  26         process.exit(1);
  27     }
  28
  29     gQuery = { username: config.username(), password: config.password() };
  30
  31     console.error('Using server %s', config.server().yellow);
  32 }
  33
  34 function collectFiles(filesOrFolders) {
  35     var tmp = [];
  36
  37     filesOrFolders.forEach(function (filePath) {
  38         var stat = fs.statSync(filePath);
  39
  40         if (stat.isFile()) {
  41             tmp.push(filePath);
  42         } else if (stat.isDirectory()) {
  43             var files = fs.readdirSync(filePath).map(function (file) { return path.join(filePath, file); });
  44             tmp = tmp.concat(collectFiles(files));
  45         } else {
  46             console.log('Skipping %s', filePath.cyan);
  47         }
  48     });
  49
  50     return tmp;
  51 }
  52
  53 function login(uri) {
  54     var tmp = url.parse(uri);
  55     if (!tmp.host) tmp = url.parse('https://' + uri);
  56
  57     var server = tmp.protocol + '//' + tmp.host;
  58
  59     console.log('Using server', server.bold);
  60
  61     var username = readlineSync.question('Username: ', { hideEchoBack: false });
  62     var password = readlineSync.question('Password: ', { hideEchoBack: true });
  63
  64     superagent.get(server + API + '/').query({ username: username, password: password }).end(function (error, result) {
  65         if (error && error.code === 'ENOTFOUND') {
  66             console.log('No such server %s'.red, server.bold);
  67             process.exit(1);
  68         }
  69         if (error && error.code) {
  70             console.log('Failed to connect to server %s'.red, server.bold, error.code);
  71             process.exit(1);
  72         }
  73         if (result.status === 401) {
  74             console.log('Login failed.'.red);
  75             process.exit(1);
  76         }
  77
  78         config.set('server', server);
  79         config.set('username', username);
  80
  81         // TODO this is clearly bad and needs fixing
  82         config.set('password', password);
  83
  84         gQuery = { username: username, password: password };
  85
  86         console.log('Ok'.green);
  87     });
  88 }
  89
  90 function put(filePath, otherFilePaths, options) {
  91     checkConfig();
  92
  93     var files = collectFiles([ filePath ].concat(otherFilePaths));
  94
  95     async.eachSeries(files, function (file, callback) {
  96         var relativeFilePath = path.resolve(file).slice(process.cwd().length + 1);
  97         var destinationPath = (options.destination ? '/' + options.destination : '') + '/' + relativeFilePath;
  98         console.log('Uploading file %s -> %s', relativeFilePath.cyan, destinationPath.cyan);
  99
 100         superagent.put(config.server() + API + relativeFilePath).query(gQuery).attach('file', file).end(function (error, result) {
 101             if (error) return callback(error);
 102             if (result.statusCode !== 201) return callback(new Error('Error uploading file: ' + result.statusCode));
 103
 104             console.log('Uploaded to ' + config.server() + destinationPath);
 105         });
 106     }, function (error) {
 107         if (error) {
 108             console.log('Failed to put file.', error);
 109             process.exit(1);
 110         }
 111
 112         console.log('Done');
 113     });
 114 }
 115
 116 function get(filePath) {
 117     checkConfig();
 118
 119     // if no argument provided, fetch root
 120     filePath = filePath || '/';
 121
 122     request.get(config.server() + API + filePath, { qs: gQuery }, function (error, result, body) {
 123         if (error) return console.error(error);
 124         if (result.statusCode === 401) return console.log('Login failed');
 125         if (result.statusCode === 404) return console.log('No such file or directory');
 126
 127         // 222 indicates directory listing
 128         if (result.statusCode === 222) {
 129             console.log('Files:');
 130             JSON.parse(body).entries.forEach(function (entry) {
 131                 console.log('\t %s', entry);
 132             });
 133         } else {
 134             console.log(body);
 135         }
 136     });
 137     // var req = superagent.get(config.server() + API + filePath);
 138     // req.query(gQuery);
 139     // req.end(function (error, result) {
 140     //     if (error && error.status === 401) return console.log('Login failed');
 141     //     if (error && error.status === 404) return console.log('No such file or directory');
 142     //     if (error) return console.log('Failed', result ? result.body : error);
 143
 144     //     if (result.body && result.body.entries) {
 145     //         console.log('Files:');
 146     //         result.body.entries.forEach(function (entry) {
 147     //             console.log('\t %s', entry);
 148     //         });
 149     //     } else {
 150     //         req.pipe(process.stdout);
 151     //     }
 152     // });
 153 }
 154
 155 function del(filePath) {
 156     checkConfig();
 157
 158     var relativeFilePath = path.resolve(filePath).slice(process.cwd().length + 1);
 159     superagent.del(config.server() + API + relativeFilePath).query(gQuery).end(function (error, result) {
 160         if (error && error.status === 401) return console.log('Login failed');
 161         if (error && error.status === 404) return console.log('No such file or directory');
 162         if (error) return console.log('Failed', result ? result.body : error);
 163         console.log('Success', result.body);
 164     });
 165 }