summaryrefslogtreecommitdiff
path: root/cli/js/web/streams/readable_stream_async_iterator.ts
blob: c6b9759a5de7c910f3be84be6cbaa4ce93ecdc37 (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
// 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);