summaryrefslogtreecommitdiff
path: root/cli/js/web/streams/readable_stream_async_iterator.ts
blob: cd656e73d3c922636978b9bd8f2a31f548884e5e (plain)
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
// Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.

import * as sym from "./symbols.ts";
import {
  isReadableStreamAsyncIterator,
  ReadableStreamAsyncIterator,
  readableStreamCreateReadResult,
  readableStreamReaderGenericCancel,
  readableStreamReaderGenericRelease,
  readableStreamDefaultReaderRead,
} from "./internals.ts";
import { assert } from "../../util.ts";

// eslint-disable-next-line @typescript-eslint/no-explicit-any
const AsyncIteratorPrototype: AsyncIterableIterator<any> = Object.getPrototypeOf(
  Object.getPrototypeOf(async function* () {}).prototype
);

export const ReadableStreamAsyncIteratorPrototype: ReadableStreamAsyncIterator = Object.setPrototypeOf(
  {
    next(
      this: ReadableStreamAsyncIterator
      // eslint-disable-next-line @typescript-eslint/no-explicit-any
    ): Promise<ReadableStreamReadResult<any>> {
      if (!isReadableStreamAsyncIterator(this)) {
        return Promise.reject(
          new TypeError("invalid ReadableStreamAsyncIterator.")
        );
      }
      const reader = this[sym.asyncIteratorReader];
      if (!reader[sym.ownerReadableStream]) {
        return Promise.reject(
          new TypeError("reader owner ReadableStream is undefined.")
        );
      }
      return readableStreamDefaultReaderRead(reader).then((result) => {
        assert(typeof result === "object");
        const { done } = result;
        assert(typeof done === "boolean");
        if (done) {
          readableStreamReaderGenericRelease(reader);
        }
        const { value } = result;
        return readableStreamCreateReadResult(value, done, true);
      });
    },
    return(
      this: ReadableStreamAsyncIterator,
      // eslint-disable-next-line @typescript-eslint/no-explicit-any
      value?: any | PromiseLike<any>
      // eslint-disable-next-line @typescript-eslint/no-explicit-any
    ): Promise<ReadableStreamReadResult<any>> {
      if (!isReadableStreamAsyncIterator(this)) {
        return Promise.reject(
          new TypeError("invalid ReadableStreamAsyncIterator.")
        );
      }
      const reader = this[sym.asyncIteratorReader];
      if (!reader[sym.ownerReadableStream]) {
        return Promise.reject(
          new TypeError("reader owner ReadableStream is undefined.")
        );
      }
      if (reader[sym.readRequests].length) {
        return Promise.reject(
          new TypeError("reader has outstanding read requests.")
        );
      }
      if (!this[sym.preventCancel]) {
        const result = readableStreamReaderGenericCancel(reader, value);
        readableStreamReaderGenericRelease(reader);
        return result.then(() =>
          readableStreamCreateReadResult(value, true, true)
        );
      }
      readableStreamReaderGenericRelease(reader);
      return Promise.resolve(readableStreamCreateReadResult(value, true, true));
    },
  },
  AsyncIteratorPrototype
);