forked from twilson63/express-couchFs
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
74 lines (64 loc) · 1.95 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
var fs = require('fs');
var nano = require('nano');
var express = require('express');
var mime = require('mime');
module.exports = function(config) {
var app = express();
var server, db;
if (config.couch) {
db = nano(config.couch);
} else {
server = nano(config.url);
db = config.database_parameter_name || 'COUCH_DB';
}
function getDb(req) {
if (typeof db === 'object') {
return db;
} else {
return server.use(req[db]);
}
}
app.get('/api/file/:name', function(req, res) {
getDb(req).get(req.params.name, function(e, doc) {
var headers = {'Content-Type': doc.mime};
if (!/png|jpg|gif/.test(doc.mime)) {
headers['Content-Disposition'] = 'attachment; filename="' + doc.name + '"';
}
// file type -- as mime type
//Content-Disposition: attachment; filename="fname.ext"
getDb(req).attachment.get(req.params.name, 'file', function(err, body) {
res.writeHead(200, headers);
res.end(body);
});
});
});
// handle file upload
app.post('/api/file', function(req, res) {
var meta = {
name: req.files.uploadFile.name,
type: 'file',
mime: req.files.uploadFile.type,
size: req.files.uploadFile.size
};
var filename = req.files.uploadFile.name;
getDb(req).insert(meta, attachFile);
function attachFile(err, body) {
if (err) { return res.send(500, err); }
fs.readFile(req.files.uploadFile.path, function(err, data) {
if (err) { return res.send(500, err); }
getDb(req).attachment.insert(body.id, 'file', data, meta.type,
{ rev: body.rev }, function(e,b) {
if (e) { return res.send(500, e); }
res.send(b);
})
});
}
});
// app del file
app.del('/api/file/:name', function(req, res) {
getDb(req).get(req.params.name, function(err, body) {
getDb(req).destroy(req.params.name, body._rev).pipe(res);
});
});
return app;
};