Blame view

node_modules/iconv-lite/lib/streams.js 3.3 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
99
100
101
102
103
104
105
106
107
108
109
  "use strict";
  
  var Buffer = require("safer-buffer").Buffer;
  
  // NOTE: Due to 'stream' module being pretty large (~100Kb, significant in browser environments), 
  // we opt to dependency-inject it instead of creating a hard dependency.
  module.exports = function(stream_module) {
      var Transform = stream_module.Transform;
  
      // == Encoder stream =======================================================
  
      function IconvLiteEncoderStream(conv, options) {
          this.conv = conv;
          options = options || {};
          options.decodeStrings = false; // We accept only strings, so we don't need to decode them.
          Transform.call(this, options);
      }
  
      IconvLiteEncoderStream.prototype = Object.create(Transform.prototype, {
          constructor: { value: IconvLiteEncoderStream }
      });
  
      IconvLiteEncoderStream.prototype._transform = function(chunk, encoding, done) {
          if (typeof chunk != 'string')
              return done(new Error("Iconv encoding stream needs strings as its input."));
          try {
              var res = this.conv.write(chunk);
              if (res && res.length) this.push(res);
              done();
          }
          catch (e) {
              done(e);
          }
      }
  
      IconvLiteEncoderStream.prototype._flush = function(done) {
          try {
              var res = this.conv.end();
              if (res && res.length) this.push(res);
              done();
          }
          catch (e) {
              done(e);
          }
      }
  
      IconvLiteEncoderStream.prototype.collect = function(cb) {
          var chunks = [];
          this.on('error', cb);
          this.on('data', function(chunk) { chunks.push(chunk); });
          this.on('end', function() {
              cb(null, Buffer.concat(chunks));
          });
          return this;
      }
  
  
      // == Decoder stream =======================================================
  
      function IconvLiteDecoderStream(conv, options) {
          this.conv = conv;
          options = options || {};
          options.encoding = this.encoding = 'utf8'; // We output strings.
          Transform.call(this, options);
      }
  
      IconvLiteDecoderStream.prototype = Object.create(Transform.prototype, {
          constructor: { value: IconvLiteDecoderStream }
      });
  
      IconvLiteDecoderStream.prototype._transform = function(chunk, encoding, done) {
          if (!Buffer.isBuffer(chunk) && !(chunk instanceof Uint8Array))
              return done(new Error("Iconv decoding stream needs buffers as its input."));
          try {
              var res = this.conv.write(chunk);
              if (res && res.length) this.push(res, this.encoding);
              done();
          }
          catch (e) {
              done(e);
          }
      }
  
      IconvLiteDecoderStream.prototype._flush = function(done) {
          try {
              var res = this.conv.end();
              if (res && res.length) this.push(res, this.encoding);                
              done();
          }
          catch (e) {
              done(e);
          }
      }
  
      IconvLiteDecoderStream.prototype.collect = function(cb) {
          var res = '';
          this.on('error', cb);
          this.on('data', function(chunk) { res += chunk; });
          this.on('end', function() {
              cb(null, res);
          });
          return this;
      }
  
      return {
          IconvLiteEncoderStream: IconvLiteEncoderStream,
          IconvLiteDecoderStream: IconvLiteDecoderStream,
      };
  };