source: trip-planner-front/node_modules/needle/lib/multipart.js@ b738035

Last change on this file since b738035 was 6a3a178, checked in by Ema <ema_spirova@…>, 3 years ago

initial commit

  • Property mode set to 100644
File size: 2.8 KB
Line 
1var readFile = require('fs').readFile,
2 basename = require('path').basename;
3
4exports.build = function(data, boundary, callback) {
5
6 if (typeof data != 'object' || typeof data.pipe == 'function')
7 return callback(new Error('Multipart builder expects data as key/val object.'));
8
9 var body = '',
10 object = flatten(data),
11 count = Object.keys(object).length;
12
13 if (count === 0)
14 return callback(new Error('Empty multipart body. Invalid data.'))
15
16 function done(err, section) {
17 if (err) return callback(err);
18 if (section) body += section;
19 --count || callback(null, body + '--' + boundary + '--');
20 };
21
22 for (var key in object) {
23 var value = object[key];
24 if (value === null || typeof value == 'undefined') {
25 done();
26 } else if (Buffer.isBuffer(value)) {
27 var part = { buffer: value, content_type: 'application/octet-stream' };
28 generate_part(key, part, boundary, done);
29 } else {
30 var part = (value.buffer || value.file || value.content_type) ? value : { value: value };
31 generate_part(key, part, boundary, done);
32 }
33 }
34
35}
36
37function generate_part(name, part, boundary, callback) {
38
39 var return_part = '--' + boundary + '\r\n';
40 return_part += 'Content-Disposition: form-data; name="' + name + '"';
41
42 function append(data, filename) {
43
44 if (data) {
45 var binary = part.content_type.indexOf('text') == -1;
46 return_part += '; filename="' + encodeURIComponent(filename) + '"\r\n';
47 if (binary) return_part += 'Content-Transfer-Encoding: binary\r\n';
48 return_part += 'Content-Type: ' + part.content_type + '\r\n\r\n';
49 return_part += binary ? data.toString('binary') : data.toString('utf8');
50 }
51
52 callback(null, return_part + '\r\n');
53 };
54
55 if ((part.file || part.buffer) && part.content_type) {
56
57 var filename = part.filename ? part.filename : part.file ? basename(part.file) : name;
58 if (part.buffer) return append(part.buffer, filename);
59
60 readFile(part.file, function(err, data) {
61 if (err) return callback(err);
62 append(data, filename);
63 });
64
65 } else {
66
67 if (typeof part.value == 'object')
68 return callback(new Error('Object received for ' + name + ', expected string.'))
69
70 if (part.content_type) {
71 return_part += '\r\n';
72 return_part += 'Content-Type: ' + part.content_type;
73 }
74
75 return_part += '\r\n\r\n';
76 return_part += Buffer.from(String(part.value), 'utf8').toString('binary');
77 append();
78
79 }
80
81}
82
83// flattens nested objects for multipart body
84function flatten(object, into, prefix) {
85 into = into || {};
86
87 for(var key in object) {
88 var prefix_key = prefix ? prefix + '[' + key + ']' : key;
89 var prop = object[key];
90
91 if (prop && typeof prop === 'object' && !(prop.buffer || prop.file || prop.content_type))
92 flatten(prop, into, prefix_key)
93 else
94 into[prefix_key] = prop;
95 }
96
97 return into;
98}
Note: See TracBrowser for help on using the repository browser.