Blame view

node_modules/needle/lib/multipart.js 2.83 KB
7820380e   “wangming”   1
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
  var readFile = require('fs').readFile,
      basename = require('path').basename;
  
  exports.build = function(data, boundary, callback) {
  
    if (typeof data != 'object' || typeof data.pipe == 'function')
      return callback(new Error('Multipart builder expects data as key/val object.'));
  
    var body   = '',
        object = flatten(data),
        count  = Object.keys(object).length;
  
    if (count === 0)
      return callback(new Error('Empty multipart body. Invalid data.'))
  
    function done(err, section) {
      if (err) return callback(err);
      if (section) body += section;
      --count || callback(null, body + '--' + boundary + '--');
    };
  
    for (var key in object) {
      var value = object[key];
      if (value === null || typeof value == 'undefined') {
        done();
      } else if (Buffer.isBuffer(value)) {
        var part = { buffer: value, content_type: 'application/octet-stream' };
        generate_part(key, part, boundary, done);
      } else {
        var part = (value.buffer || value.file || value.content_type) ? value : { value: value };
        generate_part(key, part, boundary, done);
      }
    }
  
  }
  
  function generate_part(name, part, boundary, callback) {
  
    var return_part = '--' + boundary + '\r\n';
    return_part += 'Content-Disposition: form-data; name="' + name + '"';
  
    function append(data, filename) {
  
      if (data) {
        var binary = part.content_type.indexOf('text') == -1;
        return_part += '; filename="' + encodeURIComponent(filename) + '"\r\n';
        if (binary) return_part += 'Content-Transfer-Encoding: binary\r\n';
        return_part += 'Content-Type: ' + part.content_type + '\r\n\r\n';
        return_part += binary ? data.toString('binary') : data.toString('utf8');
      }
  
      callback(null, return_part + '\r\n');
    };
  
    if ((part.file || part.buffer) && part.content_type) {
  
      var filename = part.filename ? part.filename : part.file ? basename(part.file) : name;
      if (part.buffer) return append(part.buffer, filename);
  
      readFile(part.file, function(err, data) {
        if (err) return callback(err);
        append(data, filename);
      });
  
    } else {
  
      if (typeof part.value == 'object')
        return callback(new Error('Object received for ' + name + ', expected string.'))
  
      if (part.content_type) {
        return_part += '\r\n';
        return_part += 'Content-Type: ' + part.content_type;
      }
  
      return_part += '\r\n\r\n';
      return_part += Buffer.from(String(part.value), 'utf8').toString('binary');
      append();
  
    }
  
  }
  
  // flattens nested objects for multipart body
  function flatten(object, into, prefix) {
    into = into || {};
  
    for(var key in object) {
      var prefix_key = prefix ? prefix + '[' + key + ']' : key;
      var prop = object[key];
  
      if (prop && typeof prop === 'object' && !(prop.buffer || prop.file || prop.content_type))
        flatten(prop, into, prefix_key)
      else
        into[prefix_key] = prop;
    }
  
    return into;
  }