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