cli/actions.js

   1 'use strict';
   2
   3 exports.login = login;
   4 exports.logout = logout;
   5 exports.put = put;
   6 exports.get = get;
   7 exports.del = del;
   8
   9 var superagent = require('superagent'),
  10     config = require('./config.js'),
  11     readlineSync = require('readline-sync'),
  12     safe = require('safetydance'),
  13     async = require('async'),
  14     fs = require('fs'),
  15     request = require('request'),
  16     url = require('url'),
  17     path = require('path');
  18
  19 require('colors');
  20
  21 var API = '/api/files/';
  22
  23 var gQuery = {};
  24
  25 function checkConfig() {
  26     if (!config.server() || !config.accessToken()) {
  27         console.log('You have run "login" first');
  28         process.exit(1);
  29     }
  30
  31     gQuery = { access_token: config.accessToken() };
  32
  33     console.error('Using server %s', config.server().cyan);
  34 }
  35
  36 function collectFiles(filesOrFolders, options) {
  37     var tmp = [];
  38
  39     filesOrFolders.forEach(function (filePath) {
  40         var baseName = path.basename(filePath);
  41         if (!options.all && baseName[0] === '.' && baseName.length > 1) return;
  42
  43         var stat = fs.statSync(filePath);
  44
  45         if (stat.isFile()) {
  46             tmp.push(filePath);
  47         } else if (stat.isDirectory()) {
  48             var files = fs.readdirSync(filePath).map(function (file) { return path.join(filePath, file); });
  49             tmp = tmp.concat(collectFiles(files, options));
  50         } else {
  51             console.log('Skipping %s', filePath.cyan);
  52         }
  53     });
  54
  55     return tmp;
  56 }
  57
  58 function login(uri) {
  59     var tmp = url.parse(uri);
  60     if (!tmp.slashes) tmp = url.parse('https://' + uri);
  61
  62     var server = tmp.protocol + '//' + tmp.host;
  63
  64     console.log('Using server', server.cyan);
  65
  66     var username = readlineSync.question('Username: ');
  67     var password = readlineSync.question('Password: ', { hideEchoBack: true, mask: '' });
  68
  69     superagent.post(server + '/api/login').send({ username: username, password: password }).end(function (error, result) {
  70         if (error && error.code === 'ENOTFOUND') {
  71             console.log('Server %s not found.'.red, server.bold);
  72             process.exit(1);
  73         }
  74         if (error && error.code) {
  75             console.log('Failed to connect to server %s'.red, server.bold, error.code);
  76             process.exit(1);
  77         }
  78         if (result.status !== 201) {
  79             console.log('Login failed.\n'.red);
  80             return login(uri);
  81         }
  82
  83         // TODO remove at some point, this is just to clear the previous old version values
  84         config.set('username', '');
  85         config.set('password', '');
  86
  87         config.set('server', server);
  88         config.set('accessToken', result.body.accessToken);
  89
  90         gQuery = { access_token: result.body.accessToken };
  91
  92         console.log('Login successful'.green);
  93     });
  94 }
  95
  96 function logout() {
  97     checkConfig();
  98
  99     superagent.post(config.server() + '/api/logout').query(gQuery).end(function (error, result) {
 100         if (result && result.statusCode !== 200) console.log('Failed to logout: ' + result.statusCode);
 101         if (error) console.log(error);
 102
 103         // TODO remove at some point, this is just to clear the previous old version values
 104         config.set('username', '');
 105         config.set('password', '');
 106         config.set('server', '');
 107         config.set('accessToken', '');
 108
 109         console.log('Done'.green);
 110     });
 111 }
 112
 113 function put(filePath, otherFilePaths, options) {
 114     checkConfig();
 115
 116     var destination = '';
 117
 118     // take the last argument as destination
 119     if (otherFilePaths.length > 0) {
 120         destination = otherFilePaths.pop();
 121         if (otherFilePaths.length > 0 && destination[destination.length-1] !== '/') destination += '/';
 122     }
 123
 124     var files = collectFiles([ filePath ].concat(otherFilePaths), options);
 125
 126     async.eachSeries(files, function (file, callback) {
 127         var relativeFilePath;
 128
 129         if (path.isAbsolute(file)) {
 130             relativeFilePath = path.basename(file);
 131         } else if (path.resolve(file).indexOf(process.cwd()) === 0) { // relative to current dir
 132             relativeFilePath = path.resolve(file).slice(process.cwd().length + 1);
 133         } else { // relative but somewhere else
 134             relativeFilePath = path.basename(file);
 135         }
 136
 137         var destinationPath = (destination ? '/' + destination : '') + '/' + relativeFilePath;
 138         console.log('Uploading file %s -> %s', relativeFilePath.cyan, destinationPath.cyan);
 139
 140         superagent.post(config.server() + API + destinationPath).query(gQuery).attach('file', file).end(function (error, result) {
 141             if (result && result.statusCode === 403) return callback(new Error('Upload destination ' + destinationPath + ' not allowed'));
 142             if (result && result.statusCode !== 201) return callback(new Error('Error uploading file: ' + result.statusCode));
 143             if (error) return callback(error);
 144
 145             console.log('Uploaded to ' + config.server() + destinationPath);
 146
 147             callback(null);
 148         });
 149     }, function (error) {
 150         if (error) {
 151             console.log('Failed to put file.', error.message.red);
 152             process.exit(1);
 153         }
 154
 155         console.log('Done');
 156     });
 157 }
 158
 159 function get(filePath) {
 160     checkConfig();
 161
 162     // if no argument provided, fetch root
 163     filePath = filePath || '/';
 164
 165     request.get(config.server() + API + filePath, { qs: gQuery }, function (error, result, body) {
 166         if (result && result.statusCode === 401) return console.log('Login failed');
 167         if (result && result.statusCode === 404) return console.log('No such file or directory %s', filePath.yellow);
 168         if (error) return console.error(error);
 169
 170         // 222 indicates directory listing
 171         if (result.statusCode === 222) {
 172             var files = safe.JSON.parse(body);
 173             if (!files || files.entries.length === 0) {
 174                 console.log('No files on the server. Use %s to upload some.', 'surfer put <file>'.yellow);
 175             } else {
 176                 console.log('Entries:');
 177                 files.entries.forEach(function (entry) {
 178                     console.log('\t %s', entry.isDirectory ? entry.filePath + '/' : entry.filePath);
 179                 });
 180             }
 181         } else {
 182             process.stdout.write(body);
 183         }
 184     });
 185     // var req = superagent.get(config.server() + API + filePath);
 186     // req.query(gQuery);
 187     // req.end(function (error, result) {
 188     //     if (error && error.status === 401) return console.log('Login failed');
 189     //     if (error && error.status === 404) return console.log('No such file or directory');
 190     //     if (error) return console.log('Failed', result ? result.body : error);
 191
 192     //     if (result.body && result.body.entries) {
 193     //         console.log('Files:');
 194     //         result.body.entries.forEach(function (entry) {
 195     //             console.log('\t %s', entry);
 196     //         });
 197     //     } else {
 198     //         req.pipe(process.stdout);
 199     //     }
 200     // });
 201 }
 202
 203 function del(filePath, options) {
 204     checkConfig();
 205
 206     var query = safe.JSON.parse(safe.JSON.stringify(gQuery));
 207     query.recursive = options.recursive;
 208     query.dryRun = options.dryRun;
 209
 210     var relativeFilePath = path.resolve(filePath).slice(process.cwd().length + 1);
 211     superagent.del(config.server() + API + relativeFilePath).query(query).end(function (error, result) {
 212         if (error && error.status === 401) return console.log('Login failed'.red);
 213         if (error && error.status === 404) return console.log('No such file or directory');
 214         if (error && error.status === 403) return console.log('Failed. Target is a directory. Use %s to delete directories.', '--recursive'.yellow);
 215         if (error) return console.log('Failed', result ? result.body : error);
 216
 217         if (options.dryRun) {
 218             console.log('This would remove %s files:', result.body.entries.length);
 219             result.body.entries.forEach(function (entry) {
 220                 console.log('\t %s', entry);
 221             });
 222         } else {
 223             console.log('Success. Removed %s files.', result.body.entries.length);
 224         }
 225     });
 226 }