nstdlib-nightly/lib/internal/streams/add-abort-signal.mjs

Node.js standard library converted to runtime-agnostic ES modules.

// Source: https://github.com/nodejs/node/blob/65eff1eb/lib/internal/streams/add-abort-signal.js

import { AbortError, codes as __codes__ } from "nstdlib/lib/internal/errors";
import {
  isNodeStream,
  isWebStream,
  kControllerErrorFunction,
} from "nstdlib/lib/internal/streams/utils";
import * as eos from "nstdlib/lib/internal/streams/end-of-stream";
import * as __hoisted_internal_events_abort_listener__ from "nstdlib/lib/internal/events/abort_listener";

const { ERR_INVALID_ARG_TYPE } = __codes__;

let addAbortListener;

// This method is inlined here for readable-stream
// It also does not allow for signal to not exist on the stream
// https://github.com/nodejs/node/pull/36061#discussion_r533718029
const validateAbortSignal = (signal, name) => {
  if (typeof signal !== "object" || !("aborted" in signal)) {
    throw new ERR_INVALID_ARG_TYPE(name, "AbortSignal", signal);
  }
};

const _export_addAbortSignal_ = function addAbortSignal(signal, stream) {
  validateAbortSignal(signal, "signal");
  if (!isNodeStream(stream) && !isWebStream(stream)) {
    throw new ERR_INVALID_ARG_TYPE(
      "stream",
      ["ReadableStream", "WritableStream", "Stream"],
      stream,
    );
  }
  return module.exports.addAbortSignalNoValidate(signal, stream);
};
export { _export_addAbortSignal_ as addAbortSignal };

const _export_addAbortSignalNoValidate_ = function (signal, stream) {
  if (typeof signal !== "object" || !("aborted" in signal)) {
    return stream;
  }
  const onAbort = isNodeStream(stream)
    ? () => {
        stream.destroy(new AbortError(undefined, { cause: signal.reason }));
      }
    : () => {
        stream[kControllerErrorFunction](
          new AbortError(undefined, { cause: signal.reason }),
        );
      };
  if (signal.aborted) {
    onAbort();
  } else {
    addAbortListener ??=
      __hoisted_internal_events_abort_listener__.addAbortListener;
    const disposable = addAbortListener(signal, onAbort);
    eos(stream, disposable[Symbol.for("nodejs.dispose")]);
  }
  return stream;
};
export { _export_addAbortSignalNoValidate_ as addAbortSignalNoValidate };