]>
Commit | Line | Data |
---|---|---|
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 | } |