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('Run %s first', 'surfer login'.yellow);
  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, options) {
  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 = options.username || readlineSync.question('Username: ');
  67     var password = options.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
  81             // remove the password to avoid a login loop
  82             delete options.password;
  83
  84             return login(uri, options);
  85         }
  86
  87         // TODO remove at some point, this is just to clear the previous old version values
  88         config.set('username', '');
  89         config.set('password', '');
  90
  91         config.set('server', server);
  92         config.set('accessToken', result.body.accessToken);
  93
  94         console.log('Login successful'.green);
  95     });
  96 }
  97
  98 function logout() {
  99     if (!config.accessToken()) return console.log('Done'.green);
 100
 101     superagent.post(config.server() + '/api/logout').query({ access_token: config.accessToken() }).end(function (error, result) {
 102         if (result && result.statusCode !== 200) console.log('Failed to logout: ' + result.statusCode);
 103         if (error) console.log(error);
 104
 105         // TODO remove at some point, this is just to clear the previous old version values
 106         config.set('username', '');
 107         config.set('password', '');
 108         config.set('server', '');
 109         config.set('accessToken', '');
 110
 111         console.log('Done'.green);
 112     });
 113 }
 114
 115 function put(filePath, otherFilePaths, options) {
 116     checkConfig();
 117
 118     var destination = '';
 119
 120     // take the last argument as destination
 121     if (otherFilePaths.length > 0) {
 122         destination = otherFilePaths.pop();
 123         if (otherFilePaths.length > 0 && destination[destination.length-1] !== '/') destination += '/';
 124     }
 125
 126     var files = collectFiles([ filePath ].concat(otherFilePaths), options);
 127
 128     async.eachSeries(files, function (file, callback) {
 129         var relativeFilePath;
 130
 131         if (path.isAbsolute(file)) {
 132             relativeFilePath = path.basename(file);
 133         } else if (path.resolve(file).indexOf(process.cwd()) === 0) { // relative to current dir
 134             relativeFilePath = path.resolve(file).slice(process.cwd().length + 1);
 135         } else { // relative but somewhere else
 136             relativeFilePath = path.basename(file);
 137         }
 138
 139         var destinationPath = (destination ? '/' + destination : '') + '/' + relativeFilePath;
 140         console.log('Uploading file %s -> %s', relativeFilePath.cyan, destinationPath.cyan);
 141
 142         superagent.post(config.server() + API + destinationPath).query(gQuery).attach('file', file).end(function (error, result) {
 143             if (result && result.statusCode === 403) return callback(new Error('Upload destination ' + destinationPath + ' not allowed'));
 144             if (result && result.statusCode !== 201) return callback(new Error('Error uploading file: ' + result.statusCode));
 145             if (error) return callback(error);
 146
 147             console.log('Uploaded to ' + config.server() + destinationPath);
 148
 149             callback(null);
 150         });
 151     }, function (error) {
 152         if (error) {
 153             console.log('Failed to put file.', error.message.red);
 154             process.exit(1);
 155         }
 156
 157         console.log('Done');
 158     });
 159 }
 160
 161 function get(filePath) {
 162     checkConfig();
 163
 164     // if no argument provided, fetch root
 165     filePath = filePath || '/';
 166
 167     request.get(config.server() + API + filePath, { qs: gQuery }, function (error, result, body) {
 168         if (result && result.statusCode === 401) return console.log('Login failed');
 169         if (result && result.statusCode === 404) return console.log('No such file or directory %s', filePath.yellow);
 170         if (error) return console.error(error);
 171
 172         // 222 indicates directory listing
 173         if (result.statusCode === 222) {
 174             var files = safe.JSON.parse(body);
 175             if (!files || files.entries.length === 0) {
 176                 console.log('No files on the server. Use %s to upload some.', 'surfer put <file>'.yellow);
 177             } else {
 178                 console.log('Entries:');
 179                 files.entries.forEach(function (entry) {
 180                     console.log('\t %s', entry.isDirectory ? entry.filePath + '/' : entry.filePath);
 181                 });
 182             }
 183         } else {
 184             process.stdout.write(body);
 185         }
 186     });
 187     // var req = superagent.get(config.server() + API + filePath);
 188     // req.query(gQuery);
 189     // req.end(function (error, result) {
 190     //     if (error && error.status === 401) return console.log('Login failed');
 191     //     if (error && error.status === 404) return console.log('No such file or directory');
 192     //     if (error) return console.log('Failed', result ? result.body : error);
 193
 194     //     if (result.body && result.body.entries) {
 195     //         console.log('Files:');
 196     //         result.body.entries.forEach(function (entry) {
 197     //             console.log('\t %s', entry);
 198     //         });
 199     //     } else {
 200     //         req.pipe(process.stdout);
 201     //     }
 202     // });
 203 }
 204
 205 function del(filePath, options) {
 206     checkConfig();
 207
 208     var query = safe.JSON.parse(safe.JSON.stringify(gQuery));
 209     query.recursive = options.recursive;
 210     query.dryRun = options.dryRun;
 211
 212     var relativeFilePath = path.resolve(filePath).slice(process.cwd().length + 1);
 213     superagent.del(config.server() + API + relativeFilePath).query(query).end(function (error, result) {
 214         if (error && error.status === 401) return console.log('Login failed'.red);
 215         if (error && error.status === 404) return console.log('No such file or directory');
 216         if (error && error.status === 403) return console.log('Failed. Target is a directory. Use %s to delete directories.', '--recursive'.yellow);
 217         if (error) return console.log('Failed', result ? result.body : error);
 218
 219         if (options.dryRun) {
 220             console.log('This would remove %s files:', result.body.entries.length);
 221             result.body.entries.forEach(function (entry) {
 222                 console.log('\t %s', entry);
 223             });
 224         } else {
 225             console.log('Success. Removed %s files.', result.body.entries.length);
 226         }
 227     });
 228 }