@cldmv/slothlet 2.11.0 → 3.0.1

package/dist/lib/helpers/als-eventemitter.mjs

@@ -1,256 +0,0 @@
-/*
-	Copyright 2026 CLDMV/Shinrai
-
-	Licensed under the Apache License, Version 2.0 (the "License");
-	you may not use this file except in compliance with the License.
-	You may obtain a copy of the License at
-
-		http://www.apache.org/licenses/LICENSE-2.0
-
-	Unless required by applicable law or agreed to in writing, software
-	distributed under the License is distributed on an "AS IS" BASIS,
-	WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-	See the License for the specific language governing permissions and
-	limitations under the License.
-*/
-
-
-
-
-
-import { AsyncResource, AsyncLocalStorage } from "node:async_hooks";
-import { EventEmitter } from "node:events";
-
-
-const defaultALS = new AsyncLocalStorage();
-
-
-let originalMethods = null;
-
-
-const globalResourceSet = new Set();
-
-
-
-const globalListenerTracker = new WeakMap(); 
-const allPatchedListeners = new Set(); 
-
-
-export function enableAlsForEventEmitters(als = defaultALS) {
-	
-	const kPatched = Symbol.for("slothlet.als.patched");
-
-	if (EventEmitter.prototype[kPatched]) return;
-	EventEmitter.prototype[kPatched] = true;
-
-	
-	const kMap = Symbol("slothlet.als.listenerMap");
-
-	
-	function runtime_ensureMap(emitter) {
-		if (!emitter[kMap]) emitter[kMap] = new WeakMap();
-		return emitter[kMap];
-	}
-
-	
-	function runtime_wrapListener(listener) {
-		
-		const store = als.getStore();
-		if (!store) return listener;
-
-		
-		const resource = new AsyncResource("slothlet-als-listener");
-
-		
-		globalResourceSet.add(resource);
-
-		
-		const runtime_wrappedListener = function (...args) {
-			return resource.runInAsyncScope(
-				() => {
-					return listener.apply(this, args);
-				},
-				this,
-				...args
-			);
-		};
-
-		
-		runtime_wrappedListener._slothletResource = resource;
-
-		return runtime_wrappedListener;
-	}
-
-	const proto = EventEmitter.prototype;
-
-	
-	const origOn = proto.on;
-	const origOnce = proto.once;
-	const origAdd = proto.addListener;
-	const origPre = proto.prependListener;
-	const origPreO = proto.prependOnceListener;
-	const origOff = proto.off ?? proto.removeListener;
-	const origRem = proto.removeListener;
-
-	
-	function runtime_installWrapper(addFnName, orig) {
-		proto[addFnName] = function (event, listener) {
-			const map = runtime_ensureMap(this);
-			const wrapped = runtime_wrapListener(listener);
-
-			
-			
-			if (!globalListenerTracker.has(this)) {
-				globalListenerTracker.set(this, new Set());
-			}
-			const listenerInfo = {
-				emitter: this,
-				event,
-				originalListener: listener,
-				wrappedListener: wrapped,
-				addMethod: addFnName
-			};
-			globalListenerTracker.get(this).add(listenerInfo);
-			allPatchedListeners.add(listenerInfo);
-
-			if (wrapped !== listener) map.set(listener, wrapped);
-			return orig.call(this, event, wrapped);
-		};
-	}
-
-	
-	runtime_installWrapper("on", origOn);
-	runtime_installWrapper("once", origOnce);
-	runtime_installWrapper("addListener", origAdd);
-	if (origPre) runtime_installWrapper("prependListener", origPre);
-	if (origPreO) runtime_installWrapper("prependOnceListener", origPreO);
-
-	
-	function runtime_createRemoveWrapper(method) {
-		
-		const runtime_removeWrapper = function (event, listener) {
-			const map = runtime_ensureMap(this);
-			const wrapped = map.get(listener) || listener;
-
-			
-			if (globalListenerTracker.has(this)) {
-				const emitterListeners = globalListenerTracker.get(this);
-				for (const info of emitterListeners) {
-					if (info.originalListener === listener || info.wrappedListener === wrapped) {
-						emitterListeners.delete(info);
-						allPatchedListeners.delete(info);
-						break;
-					}
-				}
-			}
-
-			
-			if (wrapped && wrapped._slothletResource) {
-				const resource = wrapped._slothletResource;
-				globalResourceSet.delete(resource);
-				try {
-					resource.emitDestroy();
-				} catch (_) {
-					
-				}
-			}
-
-			map.delete(listener);
-			return method.call(this, event, wrapped);
-		};
-
-		return runtime_removeWrapper;
-	}
-
-	
-	if (proto.off) proto.off = runtime_createRemoveWrapper(origOff);
-	proto.removeListener = runtime_createRemoveWrapper(origRem);
-
-	
-	const origRemoveAll = proto.removeAllListeners;
-	proto.removeAllListeners = function (event) {
-		const res = origRemoveAll.call(this, event);
-		if (this[kMap]) this[kMap] = new WeakMap();
-		return res;
-	};
-
-	
-	if (!originalMethods) {
-		originalMethods = {
-			on: origOn,
-			once: origOnce,
-			addListener: origAdd,
-			prependListener: origPre,
-			prependOnceListener: origPreO,
-			off: origOff,
-			removeListener: origRem,
-			removeAllListeners: origRemoveAll
-		};
-	}
-}
-
-
-
-export function cleanupAllSlothletListeners() {
-	let cleanedCount = 0;
-	let errorCount = 0;
-
-	
-	for (const listenerInfo of allPatchedListeners) {
-		try {
-			const { emitter, event, wrappedListener } = listenerInfo;
-			if (emitter && typeof emitter.removeListener === "function") {
-				emitter.removeListener(event, wrappedListener);
-				cleanedCount++;
-			}
-		} catch (_) {
-			errorCount++;
-			
-		}
-	}
-
-	
-	allPatchedListeners.clear();
-	
-
-	if (process.env.SLOTHLET_DEBUG === "1" || process.env.SLOTHLET_DEBUG === "true") {
-		console.log(`[slothlet] Cleaned up ${cleanedCount} listeners (${errorCount} errors)`);
-	}
-}
-
-export function disableAlsForEventEmitters() {
-	const kPatched = Symbol.for("slothlet.als.patched");
-
-	if (!EventEmitter.prototype[kPatched] || !originalMethods) return;
-
-	
-	cleanupAllSlothletListeners();
-
-	
-	for (const resource of globalResourceSet) {
-		try {
-			resource.emitDestroy();
-		} catch (err) {
-			
-			console.warn("[slothlet] AsyncResource cleanup warning:", err.message);
-		}
-	}
-	globalResourceSet.clear();
-
-	
-	const proto = EventEmitter.prototype;
-	proto.on = originalMethods.on;
-	proto.once = originalMethods.once;
-	proto.addListener = originalMethods.addListener;
-	if (originalMethods.prependListener) proto.prependListener = originalMethods.prependListener;
-	if (originalMethods.prependOnceListener) proto.prependOnceListener = originalMethods.prependOnceListener;
-	if (originalMethods.off) proto.off = originalMethods.off;
-	proto.removeListener = originalMethods.removeListener;
-	proto.removeAllListeners = originalMethods.removeAllListeners;
-
-	
-	delete EventEmitter.prototype[kPatched];
-
-	
-	originalMethods = null;
-}