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, options) {
  35     var tmp = [];
  36
  37     filesOrFolders.forEach(function (filePath) {
  38         var baseName = path.basename(filePath);
  39         if (!options.all && baseName[0] === '.' && baseName.length > 1) return;
  40
  41         var stat = fs.statSync(filePath);
  42
  43         if (stat.isFile()) {
  44             tmp.push(filePath);
  45         } else if (stat.isDirectory()) {
  46             var files = fs.readdirSync(filePath).map(function (file) { return path.join(filePath, file); });
  47             tmp = tmp.concat(collectFiles(files, options));
  48         } else {
  49             console.log('Skipping %s', filePath.cyan);
  50         }
  51     });
  52
  53     return tmp;
  54 }
  55
  56 function login(uri) {
  57     var tmp = url.parse(uri);
  58     if (!tmp.host) tmp = url.parse('https://' + uri);
  59
  60     var server = tmp.protocol + '//' + tmp.host;
  61
  62     console.log('Using server', server.bold);
  63
  64     var username = readlineSync.question('Username: ', { hideEchoBack: false });
  65     var password = readlineSync.question('Password: ', { hideEchoBack: true });
  66
  67     superagent.get(server + API + '/').query({ username: username, password: password }).end(function (error, result) {
  68         if (error && error.code === 'ENOTFOUND') {
  69             console.log('No such server %s'.red, server.bold);
  70             process.exit(1);
  71         }
  72         if (error && error.code) {
  73             console.log('Failed to connect to server %s'.red, server.bold, error.code);
  74             process.exit(1);
  75         }
  76         if (result.status === 401) {
  77             console.log('Login failed.'.red);
  78             process.exit(1);
  79         }
  80
  81         config.set('server', server);
  82         config.set('username', username);
  83
  84         // TODO this is clearly bad and needs fixing
  85         config.set('password', password);
  86
  87         gQuery = { username: username, password: password };
  88
  89         console.log('Ok'.green);
  90     });
  91 }
  92
  93 function put(filePath, otherFilePaths, options) {
  94     checkConfig();
  95
  96     var files = collectFiles([ filePath ].concat(otherFilePaths), options);
  97
  98     async.eachSeries(files, function (file, callback) {
  99         var relativeFilePath;
 100
 101         if (path.isAbsolute(file)) {
 102             relativeFilePath = path.basename(file);
 103         } else if (path.resolve(file).indexOf(process.cwd()) === 0) { // relative to current dir
 104             relativeFilePath = path.resolve(file).slice(process.cwd().length + 1);
 105         } else { // relative but somewhere else
 106             relativeFilePath = path.basename(file);
 107         }
 108
 109         var destinationPath = (options.destination ? '/' + options.destination : '') + '/' + relativeFilePath;
 110         console.log('Uploading file %s -> %s', relativeFilePath.cyan, destinationPath.cyan);
 111
 112         superagent.put(config.server() + API + destinationPath).query(gQuery).attach('file', file).end(function (error, result) {
 113             if (error) return callback(error);
 114             if (result.statusCode !== 201) return callback(new Error('Error uploading file: ' + result.statusCode));
 115
 116             console.log('Uploaded to ' + config.server() + destinationPath);
 117
 118             callback(null);
 119         });
 120     }, function (error) {
 121         if (error) {
 122             console.log('Failed to put file.', error);
 123             process.exit(1);
 124         }
 125
 126         console.log('Done');
 127     });
 128 }
 129
 130 function get(filePath) {
 131     checkConfig();
 132
 133     // if no argument provided, fetch root
 134     filePath = filePath || '/';
 135
 136     request.get(config.server() + API + filePath, { qs: gQuery }, function (error, result, body) {
 137         if (error) return console.error(error);
 138         if (result.statusCode === 401) return console.log('Login failed');
 139         if (result.statusCode === 404) return console.log('No such file or directory');
 140
 141         // 222 indicates directory listing
 142         if (result.statusCode === 222) {
 143             console.log('Files:');
 144             JSON.parse(body).entries.forEach(function (entry) {
 145                 console.log('\t %s', entry);
 146             });
 147         } else {
 148             process.stdout.write(body);
 149         }
 150     });
 151     // var req = superagent.get(config.server() + API + filePath);
 152     // req.query(gQuery);
 153     // req.end(function (error, result) {
 154     //     if (error && error.status === 401) return console.log('Login failed');
 155     //     if (error && error.status === 404) return console.log('No such file or directory');
 156     //     if (error) return console.log('Failed', result ? result.body : error);
 157
 158     //     if (result.body && result.body.entries) {
 159     //         console.log('Files:');
 160     //         result.body.entries.forEach(function (entry) {
 161     //             console.log('\t %s', entry);
 162     //         });
 163     //     } else {
 164     //         req.pipe(process.stdout);
 165     //     }
 166     // });
 167 }
 168
 169 function del(filePath) {
 170     checkConfig();
 171
 172     var relativeFilePath = path.resolve(filePath).slice(process.cwd().length + 1);
 173     superagent.del(config.server() + API + relativeFilePath).query(gQuery).end(function (error, result) {
 174         if (error && error.status === 401) return console.log('Login failed');
 175         if (error && error.status === 404) return console.log('No such file or directory');
 176         if (error) return console.log('Failed', result ? result.body : error);
 177         console.log('Success', result.body);
 178     });
 179 }