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.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
  98         console.log('Uploading file %s -> %s', relativeFilePath.cyan, ((options.destination ? options.destination : '') + '/' + relativeFilePath).cyan);
  99
 100         superagent.put(config.server() + API + relativeFilePath).query(gQuery).attach('file', file).end(callback);
 101     }, function (error) {
 102         if (error) {
 103             console.log('Failed to put file.', error);
 104             process.exit(1);
 105         }
 106
 107         console.log('Done');
 108     });
 109 }
 110
 111 function get(filePath) {
 112     checkConfig();
 113
 114     // if no argument provided, fetch root
 115     filePath = filePath || '/';
 116
 117     request.get(config.server() + API + filePath, { qs: gQuery }, function (error, result, body) {
 118         if (error) return console.error(error);
 119         if (result.statusCode === 401) return console.log('Login failed');
 120         if (result.statusCode === 404) return console.log('No such file or directory');
 121
 122         // 222 indicates directory listing
 123         if (result.statusCode === 222) {
 124             console.log('Files:');
 125             JSON.parse(body).entries.forEach(function (entry) {
 126                 console.log('\t %s', entry);
 127             });
 128         } else {
 129             console.log(body);
 130         }
 131     });
 132     // var req = superagent.get(config.server() + API + filePath);
 133     // req.query(gQuery);
 134     // req.end(function (error, result) {
 135     //     if (error && error.status === 401) return console.log('Login failed');
 136     //     if (error && error.status === 404) return console.log('No such file or directory');
 137     //     if (error) return console.log('Failed', result ? result.body : error);
 138
 139     //     if (result.body && result.body.entries) {
 140     //         console.log('Files:');
 141     //         result.body.entries.forEach(function (entry) {
 142     //             console.log('\t %s', entry);
 143     //         });
 144     //     } else {
 145     //         req.pipe(process.stdout);
 146     //     }
 147     // });
 148 }
 149
 150 function del(filePath) {
 151     checkConfig();
 152
 153     var relativeFilePath = path.resolve(filePath).slice(process.cwd().length + 1);
 154     superagent.del(config.server() + API + relativeFilePath).query(gQuery).end(function (error, result) {
 155         if (error && error.status === 401) return console.log('Login failed');
 156         if (error && error.status === 404) return console.log('No such file or directory');
 157         if (error) return console.log('Failed', result ? result.body : error);
 158         console.log('Success', result.body);
 159     });
 160 }