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