]>
Commit | Line | Data |
---|---|---|
8c3ae071 JZ |
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'), | |
a90a633f JZ |
9 | config = require('./config.js'), |
10 | readlineSync = require('readline-sync'), | |
8c3ae071 JZ |
11 | async = require('async'), |
12 | fs = require('fs'), | |
7bb99aff | 13 | request = require('request'), |
94176733 | 14 | url = require('url'), |
8c3ae071 JZ |
15 | path = require('path'); |
16 | ||
17 | require('colors'); | |
18 | ||
19 | var API = '/api/files/'; | |
20 | ||
a90a633f JZ |
21 | var gQuery = {}; |
22 | ||
8c3ae071 | 23 | function checkConfig() { |
a90a633f | 24 | if (!config.server() || !config.username() || !config.password()) { |
8c3ae071 JZ |
25 | console.log('You have run "login" first'); |
26 | process.exit(1); | |
27 | } | |
c10e69bf | 28 | |
a90a633f JZ |
29 | gQuery = { username: config.username(), password: config.password() }; |
30 | ||
7bb99aff | 31 | console.error('Using server %s', config.server().yellow); |
8c3ae071 JZ |
32 | } |
33 | ||
34 | function collectFiles(filesOrFolders) { | |
35 | var tmp = []; | |
36 | ||
37 | filesOrFolders.forEach(function (filePath) { | |
38 | var stat = fs.statSync(filePath); | |
39 | ||
40 | if (stat.isFile()) { | |
41 | tmp.push(filePath); | |
42 | } else if (stat.isDirectory()) { | |
43 | var files = fs.readdirSync(filePath).map(function (file) { return path.join(filePath, file); }); | |
44 | tmp = tmp.concat(collectFiles(files)); | |
45 | } else { | |
46 | console.log('Skipping %s', filePath.cyan); | |
47 | } | |
48 | }); | |
49 | ||
50 | return tmp; | |
51 | } | |
52 | ||
94176733 JZ |
53 | function login(uri) { |
54 | var tmp = url.parse(uri); | |
55 | if (!tmp.host) tmp = url.parse('https://' + uri); | |
56 | ||
b9ad5d91 | 57 | var server = tmp.protocol + '//' + tmp.host; |
591ad40c | 58 | |
b9ad5d91 | 59 | console.log('Using server', server.bold); |
a90a633f JZ |
60 | |
61 | var username = readlineSync.question('Username: ', { hideEchoBack: false }); | |
62 | var password = readlineSync.question('Password: ', { hideEchoBack: true }); | |
63 | ||
64 | superagent.get(server + API + '/').query({ username: username, password: password }).end(function (error, result) { | |
94176733 | 65 | if (error && error.code === 'ENOTFOUND') { |
b9ad5d91 JZ |
66 | console.log('No such server %s'.red, server.bold); |
67 | process.exit(1); | |
68 | } | |
69 | if (error.code) { | |
70 | console.log('Failed to connect to server %s'.red, server.bold, error.code); | |
94176733 JZ |
71 | process.exit(1); |
72 | } | |
a90a633f | 73 | if (result.status === 401) { |
b9ad5d91 | 74 | console.log('Login failed.'.red); |
a90a633f JZ |
75 | process.exit(1); |
76 | } | |
77 | ||
78 | config.set('server', server); | |
79 | config.set('username', username); | |
80 | ||
81 | // TODO this is clearly bad and needs fixing | |
82 | config.set('password', password); | |
83 | ||
84 | gQuery = { username: username, password: password }; | |
c7576037 | 85 | |
b9ad5d91 | 86 | console.log('Ok'.green); |
a90a633f | 87 | }); |
8c3ae071 JZ |
88 | } |
89 | ||
eb83e4da | 90 | function put(filePath, otherFilePaths, options) { |
8c3ae071 JZ |
91 | checkConfig(); |
92 | ||
93 | var files = collectFiles([ filePath ].concat(otherFilePaths)); | |
94 | ||
95 | async.eachSeries(files, function (file, callback) { | |
96 | var relativeFilePath = path.resolve(file).slice(process.cwd().length + 1); | |
97 | ||
eb83e4da | 98 | console.log('Uploading file %s -> %s', relativeFilePath.cyan, ((options.destination ? options.destination : '') + '/' + relativeFilePath).cyan); |
8c3ae071 | 99 | |
a90a633f | 100 | superagent.put(config.server() + API + relativeFilePath).query(gQuery).attach('file', file).end(callback); |
8c3ae071 JZ |
101 | }, function (error) { |
102 | if (error) { | |
103 | console.log('Failed to put file.', error); | |
104 | process.exit(1); | |
105 | } | |
106 | ||
107 | console.log('Done'); | |
108 | }); | |
109 | } | |
110 | ||
111 | function get(filePath) { | |
112 | checkConfig(); | |
113 | ||
7bb99aff JZ |
114 | // if no argument provided, fetch root |
115 | filePath = filePath || '/'; | |
116 | ||
117 | request.get(config.server() + API + filePath, { qs: gQuery }, function (error, result, body) { | |
118 | if (error) return console.error(error); | |
119 | if (result.statusCode === 401) return console.log('Login failed'); | |
120 | if (result.statusCode === 404) return console.log('No such file or directory'); | |
8c3ae071 | 121 | |
7bb99aff JZ |
122 | // 222 indicates directory listing |
123 | if (result.statusCode === 222) { | |
8c3ae071 | 124 | console.log('Files:'); |
7bb99aff | 125 | JSON.parse(body).entries.forEach(function (entry) { |
8c3ae071 JZ |
126 | console.log('\t %s', entry); |
127 | }); | |
128 | } else { | |
7bb99aff | 129 | console.log(body); |
8c3ae071 JZ |
130 | } |
131 | }); | |
7bb99aff JZ |
132 | // var req = superagent.get(config.server() + API + filePath); |
133 | // req.query(gQuery); | |
134 | // req.end(function (error, result) { | |
135 | // if (error && error.status === 401) return console.log('Login failed'); | |
136 | // if (error && error.status === 404) return console.log('No such file or directory'); | |
137 | // if (error) return console.log('Failed', result ? result.body : error); | |
138 | ||
139 | // if (result.body && result.body.entries) { | |
140 | // console.log('Files:'); | |
141 | // result.body.entries.forEach(function (entry) { | |
142 | // console.log('\t %s', entry); | |
143 | // }); | |
144 | // } else { | |
145 | // req.pipe(process.stdout); | |
146 | // } | |
147 | // }); | |
8c3ae071 JZ |
148 | } |
149 | ||
150 | function del(filePath) { | |
151 | checkConfig(); | |
152 | ||
153 | var relativeFilePath = path.resolve(filePath).slice(process.cwd().length + 1); | |
a90a633f JZ |
154 | superagent.del(config.server() + API + relativeFilePath).query(gQuery).end(function (error, result) { |
155 | if (error && error.status === 401) return console.log('Login failed'); | |
156 | if (error && error.status === 404) return console.log('No such file or directory'); | |
8c3ae071 JZ |
157 | if (error) return console.log('Failed', result ? result.body : error); |
158 | console.log('Success', result.body); | |
159 | }); | |
160 | } |