summaryrefslogtreecommitdiff
path: root/ext/node/polyfills/internal/streams/destroy.mjs
diff options
context:
space:
mode:
authorBartek IwaƄczuk <biwanczuk@gmail.com>2023-02-14 17:38:45 +0100
committerGitHub <noreply@github.com>2023-02-14 17:38:45 +0100
commitd47147fb6ad229b1c039aff9d0959b6e281f4df5 (patch)
tree6e9e790f2b9bc71b5f0c9c7e64b95cae31579d58 /ext/node/polyfills/internal/streams/destroy.mjs
parent1d00bbe47e2ca14e2d2151518e02b2324461a065 (diff)
feat(ext/node): embed std/node into the snapshot (#17724)
This commit moves "deno_std/node" in "ext/node" crate. The code is transpiled and snapshotted during the build process. During the first pass a minimal amount of work was done to create the snapshot, a lot of code in "ext/node" depends on presence of "Deno" global. This code will be gradually fixed in the follow up PRs to migrate it to import relevant APIs from "internal:" modules. Currently the code from snapshot is not used in any way, and all Node/npm compatibility still uses code from "https://deno.land/std/node" (or from the location specified by "DENO_NODE_COMPAT_URL"). This will also be handled in a follow up PRs. --------- Co-authored-by: crowlkats <crowlkats@toaxl.com> Co-authored-by: Divy Srivastava <dj.srivastava23@gmail.com> Co-authored-by: Yoshiya Hinosawa <stibium121@gmail.com>
Diffstat (limited to 'ext/node/polyfills/internal/streams/destroy.mjs')
-rw-r--r--ext/node/polyfills/internal/streams/destroy.mjs320
1 files changed, 320 insertions, 0 deletions
diff --git a/ext/node/polyfills/internal/streams/destroy.mjs b/ext/node/polyfills/internal/streams/destroy.mjs
new file mode 100644
index 000000000..b065f2119
--- /dev/null
+++ b/ext/node/polyfills/internal/streams/destroy.mjs
@@ -0,0 +1,320 @@
+// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.
+// Copyright Joyent and Node contributors. All rights reserved. MIT license.
+// deno-lint-ignore-file
+
+import { aggregateTwoErrors, ERR_MULTIPLE_CALLBACK } from "internal:deno_node/polyfills/internal/errors.ts";
+import * as process from "internal:deno_node/polyfills/_process/process.ts";
+
+const kDestroy = Symbol("kDestroy");
+const kConstruct = Symbol("kConstruct");
+
+function checkError(err, w, r) {
+ if (err) {
+ // Avoid V8 leak, https://github.com/nodejs/node/pull/34103#issuecomment-652002364
+ err.stack; // eslint-disable-line no-unused-expressions
+
+ if (w && !w.errored) {
+ w.errored = err;
+ }
+ if (r && !r.errored) {
+ r.errored = err;
+ }
+ }
+}
+
+// Backwards compat. cb() is undocumented and unused in core but
+// unfortunately might be used by modules.
+function destroy(err, cb) {
+ const r = this._readableState;
+ const w = this._writableState;
+ // With duplex streams we use the writable side for state.
+ const s = w || r;
+
+ if ((w && w.destroyed) || (r && r.destroyed)) {
+ if (typeof cb === "function") {
+ cb();
+ }
+
+ return this;
+ }
+
+ // We set destroyed to true before firing error callbacks in order
+ // to make it re-entrance safe in case destroy() is called within callbacks
+ checkError(err, w, r);
+
+ if (w) {
+ w.destroyed = true;
+ }
+ if (r) {
+ r.destroyed = true;
+ }
+
+ // If still constructing then defer calling _destroy.
+ if (!s.constructed) {
+ this.once(kDestroy, function (er) {
+ _destroy(this, aggregateTwoErrors(er, err), cb);
+ });
+ } else {
+ _destroy(this, err, cb);
+ }
+
+ return this;
+}
+
+function _destroy(self, err, cb) {
+ let called = false;
+
+ function onDestroy(err) {
+ if (called) {
+ return;
+ }
+ called = true;
+
+ const r = self._readableState;
+ const w = self._writableState;
+
+ checkError(err, w, r);
+
+ if (w) {
+ w.closed = true;
+ }
+ if (r) {
+ r.closed = true;
+ }
+
+ if (typeof cb === "function") {
+ cb(err);
+ }
+
+ if (err) {
+ process.nextTick(emitErrorCloseNT, self, err);
+ } else {
+ process.nextTick(emitCloseNT, self);
+ }
+ }
+ try {
+ const result = self._destroy(err || null, onDestroy);
+ if (result != null) {
+ const then = result.then;
+ if (typeof then === "function") {
+ then.call(
+ result,
+ function () {
+ process.nextTick(onDestroy, null);
+ },
+ function (err) {
+ process.nextTick(onDestroy, err);
+ },
+ );
+ }
+ }
+ } catch (err) {
+ onDestroy(err);
+ }
+}
+
+function emitErrorCloseNT(self, err) {
+ emitErrorNT(self, err);
+ emitCloseNT(self);
+}
+
+function emitCloseNT(self) {
+ const r = self._readableState;
+ const w = self._writableState;
+
+ if (w) {
+ w.closeEmitted = true;
+ }
+ if (r) {
+ r.closeEmitted = true;
+ }
+
+ if ((w && w.emitClose) || (r && r.emitClose)) {
+ self.emit("close");
+ }
+}
+
+function emitErrorNT(self, err) {
+ const r = self._readableState;
+ const w = self._writableState;
+
+ if ((w && w.errorEmitted) || (r && r.errorEmitted)) {
+ return;
+ }
+
+ if (w) {
+ w.errorEmitted = true;
+ }
+ if (r) {
+ r.errorEmitted = true;
+ }
+
+ self.emit("error", err);
+}
+
+function undestroy() {
+ const r = this._readableState;
+ const w = this._writableState;
+
+ if (r) {
+ r.constructed = true;
+ r.closed = false;
+ r.closeEmitted = false;
+ r.destroyed = false;
+ r.errored = null;
+ r.errorEmitted = false;
+ r.reading = false;
+ r.ended = false;
+ r.endEmitted = false;
+ }
+
+ if (w) {
+ w.constructed = true;
+ w.destroyed = false;
+ w.closed = false;
+ w.closeEmitted = false;
+ w.errored = null;
+ w.errorEmitted = false;
+ w.ended = false;
+ w.ending = false;
+ w.finalCalled = false;
+ w.prefinished = false;
+ w.finished = false;
+ }
+}
+
+function errorOrDestroy(stream, err, sync) {
+ // We have tests that rely on errors being emitted
+ // in the same tick, so changing this is semver major.
+ // For now when you opt-in to autoDestroy we allow
+ // the error to be emitted nextTick. In a future
+ // semver major update we should change the default to this.
+
+ const r = stream._readableState;
+ const w = stream._writableState;
+
+ if ((w && w.destroyed) || (r && r.destroyed)) {
+ return this;
+ }
+
+ if ((r && r.autoDestroy) || (w && w.autoDestroy)) {
+ stream.destroy(err);
+ } else if (err) {
+ // Avoid V8 leak, https://github.com/nodejs/node/pull/34103#issuecomment-652002364
+ err.stack; // eslint-disable-line no-unused-expressions
+
+ if (w && !w.errored) {
+ w.errored = err;
+ }
+ if (r && !r.errored) {
+ r.errored = err;
+ }
+ if (sync) {
+ process.nextTick(emitErrorNT, stream, err);
+ } else {
+ emitErrorNT(stream, err);
+ }
+ }
+}
+
+function construct(stream, cb) {
+ if (typeof stream._construct !== "function") {
+ return;
+ }
+
+ const r = stream._readableState;
+ const w = stream._writableState;
+
+ if (r) {
+ r.constructed = false;
+ }
+ if (w) {
+ w.constructed = false;
+ }
+
+ stream.once(kConstruct, cb);
+
+ if (stream.listenerCount(kConstruct) > 1) {
+ // Duplex
+ return;
+ }
+
+ process.nextTick(constructNT, stream);
+}
+
+function constructNT(stream) {
+ let called = false;
+
+ function onConstruct(err) {
+ if (called) {
+ errorOrDestroy(stream, err ?? new ERR_MULTIPLE_CALLBACK());
+ return;
+ }
+ called = true;
+
+ const r = stream._readableState;
+ const w = stream._writableState;
+ const s = w || r;
+
+ if (r) {
+ r.constructed = true;
+ }
+ if (w) {
+ w.constructed = true;
+ }
+
+ if (s.destroyed) {
+ stream.emit(kDestroy, err);
+ } else if (err) {
+ errorOrDestroy(stream, err, true);
+ } else {
+ process.nextTick(emitConstructNT, stream);
+ }
+ }
+
+ try {
+ const result = stream._construct(onConstruct);
+ if (result != null) {
+ const then = result.then;
+ if (typeof then === "function") {
+ then.call(
+ result,
+ function () {
+ process.nextTick(onConstruct, null);
+ },
+ function (err) {
+ process.nextTick(onConstruct, err);
+ },
+ );
+ }
+ }
+ } catch (err) {
+ onConstruct(err);
+ }
+}
+
+function emitConstructNT(stream) {
+ stream.emit(kConstruct);
+}
+
+function isRequest(stream) {
+ return stream && stream.setHeader && typeof stream.abort === "function";
+}
+
+// Normalize destroy for legacy.
+function destroyer(stream, err) {
+ if (!stream) return;
+ if (isRequest(stream)) return stream.abort();
+ if (isRequest(stream.req)) return stream.req.abort();
+ if (typeof stream.destroy === "function") return stream.destroy(err);
+ if (typeof stream.close === "function") return stream.close();
+}
+
+export default {
+ construct,
+ destroyer,
+ destroy,
+ undestroy,
+ errorOrDestroy,
+};
+export { construct, destroy, destroyer, errorOrDestroy, undestroy };