chromium/third_party/google-closure-library/closure/goog/streams/full_native_impl.js

/**
 * @license
 * Copyright The Closure Library Authors.
 * SPDX-License-Identifier: Apache-2.0
 */

/**
 * @fileoverview A native implementation of the ponyfill.
 */
goog.module('goog.streams.fullNativeImpl');

const fullTypes = goog.require('goog.streams.fullTypes');
const liteNativeImpl = goog.require('goog.streams.liteNativeImpl');

/**
 * The implemenation of ReadableStream.
 * @template T
 * @implements {fullTypes.ReadableStream<T>}
 */
class NativeReadableStream extends liteNativeImpl.NativeReadableStream {
  /**
   * @param {!ReadableStream} stream
   */
  constructor(stream) {
    super(stream);

    /**
     * Returns an AsyncIterator over the ReadableStream.
     * https://streams.spec.whatwg.org/#rs-asynciterator
     * @return {!AsyncIterator<!IIterableResult<T>>}
     */
    this[Symbol.asyncIterator] = this.getIterator;
  }

  /**
   * @return {!NativeReadableStreamDefaultReader}
   * @override
   */
  getReader() {
    return new NativeReadableStreamDefaultReader(
        /** @type {!ReadableStreamDefaultReader} */ (this.stream.getReader()));
  }

  /** @override */
  cancel(reason) {
    return this.stream.cancel(reason);
  }

  /** @override */
  getIterator(options = undefined) {
    return this.stream.getIterator(options);
  }

  /** @override */
  tee() {
    return this.stream.tee().map((stream) => new NativeReadableStream(stream));
  }
}

/**
 * @param {!fullTypes.ReadableStreamUnderlyingSource<T>=} underlyingSource
 * @param {!fullTypes.ReadableStreamStrategy<T>=} strategy
 * @return {!NativeReadableStream<T>}
 * @suppress {strictMissingProperties}
 * @template T
 */
function newReadableStream(underlyingSource = {}, strategy = {}) {
  /** @const {!ReadableStreamSource} */
  const source = {};
  if (underlyingSource.start) {
    source.start = (controller) => {
      return underlyingSource.start(
          new NativeReadableStreamDefaultController(controller));
    };
  }
  if (underlyingSource.pull) {
    source.pull = (controller) => {
      return underlyingSource.pull(
          new NativeReadableStreamDefaultController(controller));
    };
  }
  if (underlyingSource.cancel) {
    source.cancel = underlyingSource.cancel;
  }
  const stream = new ReadableStream(source, strategy);
  return new NativeReadableStream(stream);
}

/**
 * The DefaultReader for a ReadableStream.
 * @template T
 * @implements {fullTypes.ReadableStreamDefaultReader<T>}
 */
class NativeReadableStreamDefaultReader extends
    liteNativeImpl.NativeReadableStreamDefaultReader {
  /** @override */
  cancel(reason) {
    return /** @type {!Promise<undefined>} */ (this.reader.cancel(reason));
  }
}

/**
 * @template T
 * @implements {fullTypes.ReadableStreamAsyncIterator<T>}
 */
class NativeReadableStreamAsyncIterator {
  /**
   * @param {!AsyncIterator<T>} iterator
   */
  constructor(iterator) {
    /** @private @const {!AsyncIterator<T>} */
    this.iterator_ = iterator;
  }

  /** @override */
  next() {
    return this.iterator_.next();
  }

  /** @override */
  return(value) {
    return /** @type {{return: function(*): !Promise<!IIterableResult<T>>}} */ (
               this.iterator_)
        .return(value);
  }
}

/**
 * The controller for a ReadableStream. Adds cancellation and backpressure.
 * @template T
 * @implements {fullTypes.ReadableStreamDefaultController<T>}
 */
class NativeReadableStreamDefaultController extends
    liteNativeImpl.NativeReadableStreamDefaultController {
  /** @override */
  get desiredSize() {
    return this.controller.desiredSize;
  }
}

exports = {
  NativeReadableStream,
  NativeReadableStreamAsyncIterator,
  NativeReadableStreamDefaultController,
  NativeReadableStreamDefaultReader,
  newReadableStream,
};