memstreams/src/buffer-reader.ts

Memory Streams library for JavaScript based on readable-stream

import {Readable} from 'readable-stream';
import {any2buf} from './utils';
import {MemReadable, MemReaderOptions} from './types';

export class BufferReader extends Readable implements MemReadable {
  it?: Iterator<any>;

  constructor(
    source?: Iterable<any> | Array<any>,
    options: MemReaderOptions = {},
  ) {
    options.objectMode = false;
    super(options);
    this.init(source, options);
  }

  init(source?: Iterable<any> | Array<any>, options: MemReaderOptions = {}) {
    this.it = source?.[Symbol.iterator]();
  }

  _read() {
    if (this.it) {
      const next = this.it.next();
      if (next.done) {
        this.push(null);
      } else {
        return this.push(
          any2buf(
            next.value,
            <BufferEncoding>this._readableState.encoding || 'utf8',
          ),
        );
      }
    }
  }
}