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