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