forked from expressjs/connect-multiparty
-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
112 lines (92 loc) · 2.4 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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/*!
* Connect - multipart
* Copyright(c) 2010 Sencha Inc.
* Copyright(c) 2011 TJ Holowaychuk
* Copyright(c) 2013 Andrew Kelley
* MIT Licensed
*/
/**
* Module dependencies.
*/
var multiparty = require('multiparty');
var onFinished = require('on-finished');
var qs = require('qs');
var typeis = require('type-is');
/**
* Multipart:
*
* Parse multipart/form-data request bodies,
* providing the parsed object as `req.body`
* and `req.files`.
*
* Configuration:
*
* The options passed are merged with [multiparty](https://github.com/andrewrk/node-multiparty)'s
* `Form` object, allowing you to configure the upload directory,
* size limits, etc. For example if you wish to change the upload dir do the following.
*
* app.use(connect.multipart({ uploadDir: path }));
*
* @param {Object} options
* @return {Function}
* @api public
*/
exports = module.exports = function(options){
options = options || {};
return function multipart(req, res, next) {
if (req._body) return next();
req.body = req.body || {};
req.files = req.files || {};
// ignore GET
if ('GET' === req.method || 'HEAD' === req.method) return next();
// check Content-Type
if (!typeis(req, 'multipart/form-data')) return next();
// flag as parsed
req._body = true;
// parse
var form = new multiparty.Form(options);
var data = {};
var files = {};
var done = false;
function ondata(name, val, data){
if (Array.isArray(data[name])) {
data[name].push(val);
} else if (data[name]) {
data[name] = [data[name], val];
} else {
data[name] = val;
}
}
form.on('field', function(name, val){
ondata(name, val, data);
});
form.on('file', function(name, val){
val.name = val.originalFilename;
val.type = val.headers['content-type'] || null;
ondata(name, val, files);
});
form.on('error', function(err){
if (done) return;
done = true;
err.status = 400;
if (!req.readable) return next(err);
req.resume();
onFinished(req, function(){
next(err);
});
});
form.on('close', function() {
if (done) return;
done = true;
try {
req.body = qs.parse(data);
req.files = qs.parse(files);
next();
} catch (err) {
err.status = 400;
next(err);
}
});
form.parse(req);
}
};