]>
Commit | Line | Data |
---|---|---|
ca2d3b4d JZ |
1 | 'use strict'; |
2 | ||
3 | var fs = require('fs'), | |
4 | path = require('path'), | |
5 | ejs = require('ejs'), | |
6 | rimraf = require('rimraf'), | |
eaa62184 | 7 | mkdirp = require('mkdirp'), |
ca2d3b4d JZ |
8 | HttpError = require('connect-lastmile').HttpError, |
9 | HttpSuccess = require('connect-lastmile').HttpSuccess; | |
10 | ||
11 | exports = module.exports = { | |
12 | get: get, | |
13 | put: put, | |
14 | del: del | |
15 | }; | |
16 | ||
17 | var FILE_BASE = path.resolve(__dirname, '../files'); | |
18 | ||
19 | // http://stackoverflow.com/questions/11293857/fastest-way-to-copy-file-in-node-js | |
20 | function copyFile(source, target, cb) { | |
21 | var cbCalled = false; | |
22 | ||
eaa62184 JZ |
23 | // ensure directory |
24 | mkdirp(path.dirname(target), function (error) { | |
25 | if (error) return cb(error); | |
ca2d3b4d | 26 | |
eaa62184 JZ |
27 | var rd = fs.createReadStream(source); |
28 | rd.on("error", function(err) { | |
29 | done(err); | |
30 | }); | |
ca2d3b4d | 31 | |
eaa62184 JZ |
32 | var wr = fs.createWriteStream(target); |
33 | wr.on("error", function(err) { | |
34 | done(err); | |
35 | }); | |
ca2d3b4d | 36 | |
eaa62184 JZ |
37 | wr.on("close", function(ex) { |
38 | done(); | |
39 | }); | |
40 | ||
41 | rd.pipe(wr); | |
ca2d3b4d | 42 | |
eaa62184 JZ |
43 | function done(err) { |
44 | if (!cbCalled) { | |
45 | cb(err); | |
46 | cbCalled = true; | |
47 | } | |
ca2d3b4d | 48 | } |
eaa62184 | 49 | }); |
ca2d3b4d JZ |
50 | } |
51 | ||
52 | function render(view, options) { | |
53 | return ejs.render(fs.readFileSync(view, 'utf8'), options); | |
54 | } | |
55 | ||
56 | function getAbsolutePath(filePath) { | |
57 | var absoluteFilePath = path.resolve(FILE_BASE, filePath); | |
58 | ||
59 | if (absoluteFilePath.indexOf(FILE_BASE) !== 0) return null; | |
60 | return absoluteFilePath; | |
61 | } | |
62 | ||
63 | function get(req, res, next) { | |
64 | var filePath = req.params[0]; | |
65 | var absoluteFilePath = getAbsolutePath(filePath); | |
66 | if (!absoluteFilePath) return next(new HttpError(403, 'Path not allowed')); | |
67 | ||
68 | fs.stat(absoluteFilePath, function (error, result) { | |
69 | if (error) return next(new HttpError(404, error)); | |
70 | ||
a7f450d7 | 71 | console.log('get', absoluteFilePath); |
ca2d3b4d JZ |
72 | |
73 | if (result.isFile()) return res.sendfile(absoluteFilePath); | |
74 | if (result.isDirectory()) return res.status(200).send({ entries: fs.readdirSync(absoluteFilePath) }); | |
75 | ||
76 | return next(new HttpError(500, 'unsupported type')); | |
77 | }); | |
78 | } | |
79 | ||
80 | function put(req, res, next) { | |
81 | var filePath = req.params[0]; | |
82 | ||
83 | if (!req.files.file) return next(new HttpError(400, 'missing file')); | |
84 | ||
85 | var absoluteFilePath = getAbsolutePath(filePath); | |
86 | if (!absoluteFilePath) return next(new HttpError(403, 'Path not allowed')); | |
87 | ||
88 | fs.stat(absoluteFilePath, function (error, result) { | |
89 | if (error && error.code !== 'ENOENT') return next(new HttpError(500, error)); | |
90 | ||
a7f450d7 | 91 | console.log('put', absoluteFilePath, req.files.file); |
ca2d3b4d JZ |
92 | |
93 | if (result && result.isDirectory()) return next(new HttpError(409, 'cannot put on directories')); | |
94 | if (!result || result.isFile()) { | |
95 | return copyFile(req.files.file.path, absoluteFilePath, function (error) { | |
96 | if (error) return next(new HttpError(500, error)); | |
97 | next(new HttpSuccess(201, {})); | |
98 | }); | |
99 | } | |
100 | ||
101 | return next(new HttpError(500, 'unsupported type')); | |
102 | }); | |
103 | } | |
104 | ||
105 | function del(req, res, next) { | |
106 | var filePath = req.params[0]; | |
107 | var absoluteFilePath = getAbsolutePath(filePath); | |
eaa62184 JZ |
108 | if (!absoluteFilePath) return next(new HttpError(404, 'Not found')); |
109 | if (absoluteFilePath.slice(FILE_BASE.length) === '') return next(new HttpError(403, 'Forbidden')); | |
ca2d3b4d JZ |
110 | |
111 | fs.stat(absoluteFilePath, function (error, result) { | |
112 | if (error) return next(new HttpError(404, error)); | |
113 | ||
114 | rimraf(absoluteFilePath, function (error) { | |
115 | if (error) return next(new HttpError(500, 'Unable to remove')); | |
a7f450d7 | 116 | next(new HttpSuccess(200, {})); |
ca2d3b4d JZ |
117 | }); |
118 | }); | |
119 | } |