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     url = require('url'),
  14     path = require('path');
  15
  16 require('colors');
  17
  18 var API = '/api/files/';
  19
  20 var gQuery = {};
  21
  22 function checkConfig() {
  23     if (!config.server() || !config.username() || !config.password()) {
  24         console.log('You have run "login" first');
  25         process.exit(1);
  26     }
  27
  28     gQuery = { username: config.username(), password: config.password() };
  29
  30     console.log('Using server %s', config.server().yellow);
  31 }
  32
  33 function collectFiles(filesOrFolders) {
  34     var tmp = [];
  35
  36     filesOrFolders.forEach(function (filePath) {
  37         var stat = fs.statSync(filePath);
  38
  39         if (stat.isFile()) {
  40             tmp.push(filePath);
  41         } else if (stat.isDirectory()) {
  42             var files = fs.readdirSync(filePath).map(function (file) { return path.join(filePath, file); });
  43             tmp = tmp.concat(collectFiles(files));
  44         } else {
  45             console.log('Skipping %s', filePath.cyan);
  46         }
  47     });
  48
  49     return tmp;
  50 }
  51
  52 function login(uri) {
  53     var tmp = url.parse(uri);
  54     if (!tmp.host) tmp = url.parse('https://' + uri);
  55
  56     var server = tmp.protocol + '//' + tmp.host;
  57
  58     console.log('Using server', server.bold);
  59
  60     var username = readlineSync.question('Username: ', { hideEchoBack: false });
  61     var password = readlineSync.question('Password: ', { hideEchoBack: true });
  62
  63     superagent.get(server + API + '/').query({ username: username, password: password }).end(function (error, result) {
  64         if (error && error.code === 'ENOTFOUND') {
  65             console.log('No such server %s'.red, server.bold);
  66             process.exit(1);
  67         }
  68         if (error.code) {
  69             console.log('Failed to connect to server %s'.red, server.bold, error.code);
  70             process.exit(1);
  71         }
  72         if (result.status === 401) {
  73             console.log('Login failed.'.red);
  74             process.exit(1);
  75         }
  76
  77         config.set('server', server);
  78         config.set('username', username);
  79
  80         // TODO this is clearly bad and needs fixing
  81         config.set('password', password);
  82
  83         gQuery = { username: username, password: password };
  84
  85         console.log('Ok'.green);
  86     });
  87 }
  88
  89 function put(filePath, otherFilePaths, options) {
  90     checkConfig();
  91
  92     var files = collectFiles([ filePath ].concat(otherFilePaths));
  93
  94     async.eachSeries(files, function (file, callback) {
  95         var relativeFilePath = path.resolve(file).slice(process.cwd().length + 1);
  96
  97         console.log('Uploading file %s -> %s', relativeFilePath.cyan, ((options.destination ? options.destination : '') + '/' + relativeFilePath).cyan);
  98
  99         superagent.put(config.server() + API + relativeFilePath).query(gQuery).attach('file', file).end(callback);
 100     }, function (error) {
 101         if (error) {
 102             console.log('Failed to put file.', error);
 103             process.exit(1);
 104         }
 105
 106         console.log('Done');
 107     });
 108 }
 109
 110 function get(filePath) {
 111     checkConfig();
 112
 113     superagent.get(config.server() + API + filePath).query(gQuery).end(function (error, result) {
 114         if (error && error.status === 401) return console.log('Login failed');
 115         if (error && error.status === 404) return console.log('No such file or directory');
 116         if (error) return console.log('Failed', result ? result.body : error);
 117
 118         if (result.body && result.body.entries) {
 119             console.log('Files:');
 120             result.body.entries.forEach(function (entry) {
 121                 console.log('\t %s', entry);
 122             });
 123         } else {
 124             console.log(result.text);
 125         }
 126     });
 127 }
 128
 129 function del(filePath) {
 130     checkConfig();
 131
 132     var relativeFilePath = path.resolve(filePath).slice(process.cwd().length + 1);
 133     superagent.del(config.server() + API + relativeFilePath).query(gQuery).end(function (error, result) {
 134         if (error && error.status === 401) return console.log('Login failed');
 135         if (error && error.status === 404) return console.log('No such file or directory');
 136         if (error) return console.log('Failed', result ? result.body : error);
 137         console.log('Success', result.body);
 138     });
 139 }