@orpc/shared 0.0.0-next.4a7a06a → 0.0.0-next.4b3a4c5

package/dist/index.mjs

@@ -12,6 +12,24 @@ function splitInHalf(arr) {
   return [arr.slice(0, half), arr.slice(half)];
 }
 
+function readAsBuffer(source) {
+  if (typeof source.bytes === "function") {
+    return source.bytes();
+  }
+  return source.arrayBuffer();
+}
+
+const ORPC_NAME = "orpc";
+const ORPC_SHARED_PACKAGE_NAME = "@orpc/shared";
+const ORPC_SHARED_PACKAGE_VERSION = "0.0.0-next.4b3a4c5";
+
+class AbortError extends Error {
+  constructor(...rest) {
+    super(...rest);
+    this.name = "AbortError";
+  }
+}
+
 function once(fn) {
   let cached;
   return () => {
@@ -32,15 +50,420 @@ function sequential(fn) {
     });
   };
 }
+function defer(callback) {
+  if (typeof setTimeout === "function") {
+    setTimeout(callback, 0);
+  } else {
+    Promise.resolve().then(() => Promise.resolve().then(() => Promise.resolve().then(callback)));
+  }
+}
+
+const SPAN_ERROR_STATUS = 2;
+const GLOBAL_OTEL_CONFIG_KEY = `__${ORPC_SHARED_PACKAGE_NAME}@${ORPC_SHARED_PACKAGE_VERSION}/otel/config__`;
+function setGlobalOtelConfig(config) {
+  globalThis[GLOBAL_OTEL_CONFIG_KEY] = config;
+}
+function getGlobalOtelConfig() {
+  return globalThis[GLOBAL_OTEL_CONFIG_KEY];
+}
+function startSpan(name, options = {}, context) {
+  const tracer = getGlobalOtelConfig()?.tracer;
+  return tracer?.startSpan(name, options, context);
+}
+function setSpanError(span, error, options = {}) {
+  if (!span) {
+    return;
+  }
+  const exception = toOtelException(error);
+  span.recordException(exception);
+  if (!options.signal?.aborted || options.signal.reason !== error) {
+    span.setStatus({
+      code: SPAN_ERROR_STATUS,
+      message: exception.message
+    });
+  }
+}
+function setSpanAttribute(span, key, value) {
+  if (!span || value === void 0) {
+    return;
+  }
+  span.setAttribute(key, value);
+}
+function toOtelException(error) {
+  if (error instanceof Error) {
+    const exception = {
+      message: error.message,
+      name: error.name,
+      stack: error.stack
+    };
+    if ("code" in error && (typeof error.code === "string" || typeof error.code === "number")) {
+      exception.code = error.code;
+    }
+    return exception;
+  }
+  return { message: String(error) };
+}
+function toSpanAttributeValue(data) {
+  if (data === void 0) {
+    return "undefined";
+  }
+  try {
+    return JSON.stringify(data, (_, value) => {
+      if (typeof value === "bigint") {
+        return value.toString();
+      }
+      if (value instanceof Map || value instanceof Set) {
+        return Array.from(value);
+      }
+      return value;
+    });
+  } catch {
+    return String(data);
+  }
+}
+async function runWithSpan({ name, context, ...options }, fn) {
+  const tracer = getGlobalOtelConfig()?.tracer;
+  if (!tracer) {
+    return fn();
+  }
+  const callback = async (span) => {
+    try {
+      return await fn(span);
+    } catch (e) {
+      setSpanError(span, e, options);
+      throw e;
+    } finally {
+      span.end();
+    }
+  };
+  if (context) {
+    return tracer.startActiveSpan(name, options, context, callback);
+  } else {
+    return tracer.startActiveSpan(name, options, callback);
+  }
+}
+async function runInSpanContext(span, fn) {
+  const otelConfig = getGlobalOtelConfig();
+  if (!span || !otelConfig) {
+    return fn();
+  }
+  const ctx = otelConfig.trace.setSpan(otelConfig.context.active(), span);
+  return otelConfig.context.with(ctx, fn);
+}
+
+class AsyncIdQueue {
+  openIds = /* @__PURE__ */ new Set();
+  queues = /* @__PURE__ */ new Map();
+  waiters = /* @__PURE__ */ new Map();
+  get length() {
+    return this.openIds.size;
+  }
+  get waiterIds() {
+    return Array.from(this.waiters.keys());
+  }
+  hasBufferedItems(id) {
+    return Boolean(this.queues.get(id)?.length);
+  }
+  open(id) {
+    this.openIds.add(id);
+  }
+  isOpen(id) {
+    return this.openIds.has(id);
+  }
+  push(id, item) {
+    this.assertOpen(id);
+    const pending = this.waiters.get(id);
+    if (pending?.length) {
+      pending.shift()[0](item);
+      if (pending.length === 0) {
+        this.waiters.delete(id);
+      }
+    } else {
+      const items = this.queues.get(id);
+      if (items) {
+        items.push(item);
+      } else {
+        this.queues.set(id, [item]);
+      }
+    }
+  }
+  async pull(id) {
+    this.assertOpen(id);
+    const items = this.queues.get(id);
+    if (items?.length) {
+      const item = items.shift();
+      if (items.length === 0) {
+        this.queues.delete(id);
+      }
+      return item;
+    }
+    return new Promise((resolve, reject) => {
+      const waitingPulls = this.waiters.get(id);
+      const pending = [resolve, reject];
+      if (waitingPulls) {
+        waitingPulls.push(pending);
+      } else {
+        this.waiters.set(id, [pending]);
+      }
+    });
+  }
+  close({ id, reason } = {}) {
+    if (id === void 0) {
+      this.waiters.forEach((pendingPulls, id2) => {
+        pendingPulls.forEach(([, reject]) => {
+          reject(reason ?? new Error(`[AsyncIdQueue] Queue[${id2}] was closed or aborted while waiting for pulling.`));
+        });
+      });
+      this.waiters.clear();
+      this.openIds.clear();
+      this.queues.clear();
+      return;
+    }
+    this.waiters.get(id)?.forEach(([, reject]) => {
+      reject(reason ?? new Error(`[AsyncIdQueue] Queue[${id}] was closed or aborted while waiting for pulling.`));
+    });
+    this.waiters.delete(id);
+    this.openIds.delete(id);
+    this.queues.delete(id);
+  }
+  assertOpen(id) {
+    if (!this.isOpen(id)) {
+      throw new Error(`[AsyncIdQueue] Cannot access queue[${id}] because it is not open or aborted.`);
+    }
+  }
+}
+
+function isAsyncIteratorObject(maybe) {
+  if (!maybe || typeof maybe !== "object") {
+    return false;
+  }
+  return "next" in maybe && typeof maybe.next === "function" && Symbol.asyncIterator in maybe && typeof maybe[Symbol.asyncIterator] === "function";
+}
+const fallbackAsyncDisposeSymbol = Symbol.for("asyncDispose");
+const asyncDisposeSymbol = Symbol.asyncDispose ?? fallbackAsyncDisposeSymbol;
+class AsyncIteratorClass {
+  #isDone = false;
+  #isExecuteComplete = false;
+  #cleanup;
+  #next;
+  constructor(next, cleanup) {
+    this.#cleanup = cleanup;
+    this.#next = sequential(async () => {
+      if (this.#isDone) {
+        return { done: true, value: void 0 };
+      }
+      try {
+        const result = await next();
+        if (result.done) {
+          this.#isDone = true;
+        }
+        return result;
+      } catch (err) {
+        this.#isDone = true;
+        throw err;
+      } finally {
+        if (this.#isDone && !this.#isExecuteComplete) {
+          this.#isExecuteComplete = true;
+          await this.#cleanup("next");
+        }
+      }
+    });
+  }
+  next() {
+    return this.#next();
+  }
+  async return(value) {
+    this.#isDone = true;
+    if (!this.#isExecuteComplete) {
+      this.#isExecuteComplete = true;
+      await this.#cleanup("return");
+    }
+    return { done: true, value };
+  }
+  async throw(err) {
+    this.#isDone = true;
+    if (!this.#isExecuteComplete) {
+      this.#isExecuteComplete = true;
+      await this.#cleanup("throw");
+    }
+    throw err;
+  }
+  /**
+   * asyncDispose symbol only available in esnext, we should fallback to Symbol.for('asyncDispose')
+   */
+  async [asyncDisposeSymbol]() {
+    this.#isDone = true;
+    if (!this.#isExecuteComplete) {
+      this.#isExecuteComplete = true;
+      await this.#cleanup("dispose");
+    }
+  }
+  [Symbol.asyncIterator]() {
+    return this;
+  }
+}
+function replicateAsyncIterator(source, count) {
+  const queue = new AsyncIdQueue();
+  const ids = Array.from({ length: count }, (_, i) => i.toString());
+  let isSourceFinished = false;
+  const start = once(async () => {
+    try {
+      while (true) {
+        const item = await source.next();
+        ids.forEach((id) => {
+          if (queue.isOpen(id)) {
+            queue.push(id, { next: item });
+          }
+        });
+        if (item.done) {
+          break;
+        }
+      }
+    } catch (error) {
+      ids.forEach((id) => {
+        if (queue.isOpen(id)) {
+          queue.push(id, { error });
+        }
+      });
+    } finally {
+      isSourceFinished = true;
+    }
+  });
+  const replicated = ids.map((id) => {
+    queue.open(id);
+    return new AsyncIteratorClass(
+      async () => {
+        start();
+        const item = await queue.pull(id);
+        if (item.next) {
+          return item.next;
+        }
+        throw item.error;
+      },
+      async (reason) => {
+        queue.close({ id });
+        if (reason !== "next" && !queue.length && !isSourceFinished) {
+          isSourceFinished = true;
+          await source?.return?.();
+        }
+      }
+    );
+  });
+  return replicated;
+}
+function asyncIteratorWithSpan({ name, ...options }, iterator) {
+  let span;
+  return new AsyncIteratorClass(
+    async () => {
+      span ??= startSpan(name);
+      try {
+        const result = await runInSpanContext(span, () => iterator.next());
+        span?.addEvent(result.done ? "completed" : "yielded");
+        return result;
+      } catch (err) {
+        setSpanError(span, err, options);
+        throw err;
+      }
+    },
+    async (reason) => {
+      try {
+        if (reason !== "next") {
+          await runInSpanContext(span, () => iterator.return?.());
+        }
+      } catch (err) {
+        setSpanError(span, err, options);
+        throw err;
+      } finally {
+        span?.end();
+      }
+    }
+  );
+}
+
+class EventPublisher {
+  #listenersMap = /* @__PURE__ */ new Map();
+  #maxBufferedEvents;
+  constructor(options = {}) {
+    this.#maxBufferedEvents = options.maxBufferedEvents ?? 100;
+  }
+  get size() {
+    return this.#listenersMap.size;
+  }
+  /**
+   * Emits an event and delivers the payload to all subscribed listeners.
+   */
+  publish(event, payload) {
+    const listeners = this.#listenersMap.get(event);
+    if (!listeners) {
+      return;
+    }
+    for (const listener of listeners) {
+      listener(payload);
+    }
+  }
+  subscribe(event, listenerOrOptions) {
+    if (typeof listenerOrOptions === "function") {
+      let listeners = this.#listenersMap.get(event);
+      if (!listeners) {
+        this.#listenersMap.set(event, listeners = /* @__PURE__ */ new Set());
+      }
+      listeners.add(listenerOrOptions);
+      return () => {
+        listeners.delete(listenerOrOptions);
+        if (listeners.size === 0) {
+          this.#listenersMap.delete(event);
+        }
+      };
+    }
+    const signal = listenerOrOptions?.signal;
+    const maxBufferedEvents = listenerOrOptions?.maxBufferedEvents ?? this.#maxBufferedEvents;
+    signal?.throwIfAborted();
+    const bufferedEvents = [];
+    const pullResolvers = [];
+    const unsubscribe = this.subscribe(event, (payload) => {
+      const resolver = pullResolvers.shift();
+      if (resolver) {
+        resolver[0]({ done: false, value: payload });
+      } else {
+        bufferedEvents.push(payload);
+        if (bufferedEvents.length > maxBufferedEvents) {
+          bufferedEvents.shift();
+        }
+      }
+    });
+    const abortListener = (event2) => {
+      unsubscribe();
+      pullResolvers.forEach((resolver) => resolver[1](event2.target.reason));
+      pullResolvers.length = 0;
+      bufferedEvents.length = 0;
+    };
+    signal?.addEventListener("abort", abortListener, { once: true });
+    return new AsyncIteratorClass(async () => {
+      if (signal?.aborted) {
+        throw signal.reason;
+      }
+      if (bufferedEvents.length > 0) {
+        return { done: false, value: bufferedEvents.shift() };
+      }
+      return new Promise((resolve, reject) => {
+        pullResolvers.push([resolve, reject]);
+      });
+    }, async () => {
+      unsubscribe();
+      signal?.removeEventListener("abort", abortListener);
+      pullResolvers.forEach((resolver) => resolver[0]({ done: true, value: void 0 }));
+      pullResolvers.length = 0;
+      bufferedEvents.length = 0;
+    });
+  }
+}
 
 class SequentialIdGenerator {
-  nextId = 0;
+  index = BigInt(0);
   generate() {
-    if (this.nextId === Number.MAX_SAFE_INTEGER) {
-      this.nextId = 0;
-      return Number.MAX_SAFE_INTEGER;
-    }
-    return this.nextId++;
+    const id = this.index.toString(32);
+    this.index++;
+    return id;
   }
 }
 
@@ -82,13 +505,13 @@ function onFinish(callback) {
     }
   };
 }
-async function intercept(interceptors, options, main) {
-  const next = async (options2, index) => {
+function intercept(interceptors, options, main) {
+  const next = (options2, index) => {
     const interceptor = interceptors[index];
     if (!interceptor) {
-      return await main(options2);
+      return main(options2);
     }
-    return await interceptor({
+    return interceptor({
       ...options2,
       next: (newOptions = options2) => next(newOptions, index + 1)
     });
@@ -96,69 +519,6 @@ async function intercept(interceptors, options, main) {
   return next(options, 0);
 }
 
-function isAsyncIteratorObject(maybe) {
-  if (!maybe || typeof maybe !== "object") {
-    return false;
-  }
-  return Symbol.asyncIterator in maybe && typeof maybe[Symbol.asyncIterator] === "function";
-}
-function createAsyncIteratorObject(next, cleanup) {
-  let isExecuteComplete = false;
-  let isDone = false;
-  const iterator = {
-    next: sequential(async () => {
-      if (isDone) {
-        return { done: true, value: void 0 };
-      }
-      try {
-        const result = await next();
-        if (result.done) {
-          isDone = true;
-        }
-        return result;
-      } catch (err) {
-        isDone = true;
-        throw err;
-      } finally {
-        if (isDone && !isExecuteComplete) {
-          isExecuteComplete = true;
-          await cleanup("next");
-        }
-      }
-    }),
-    async return(value) {
-      isDone = true;
-      if (!isExecuteComplete) {
-        isExecuteComplete = true;
-        await cleanup("return");
-      }
-      return { done: true, value };
-    },
-    async throw(err) {
-      isDone = true;
-      if (!isExecuteComplete) {
-        isExecuteComplete = true;
-        await cleanup("throw");
-      }
-      throw err;
-    },
-    /**
-     * asyncDispose symbol only available in esnext, we should fallback to Symbol.for('asyncDispose')
-     */
-    async [Symbol.asyncDispose ?? Symbol.for("asyncDispose")]() {
-      isDone = true;
-      if (!isExecuteComplete) {
-        isExecuteComplete = true;
-        await cleanup("dispose");
-      }
-    },
-    [Symbol.asyncIterator]() {
-      return iterator;
-    }
-  };
-  return iterator;
-}
-
 function parseEmptyableJSON(text) {
   if (!text) {
     return void 0;
@@ -217,80 +577,82 @@ function get(object, path) {
   }
   return current;
 }
+function isPropertyKey(value) {
+  const type = typeof value;
+  return type === "string" || type === "number" || type === "symbol";
+}
+const NullProtoObj = /* @__PURE__ */ (() => {
+  const e = function() {
+  };
+  e.prototype = /* @__PURE__ */ Object.create(null);
+  Object.freeze(e.prototype);
+  return e;
+})();
 
-class AsyncIdQueue {
-  openIds = /* @__PURE__ */ new Set();
-  items = /* @__PURE__ */ new Map();
-  pendingPulls = /* @__PURE__ */ new Map();
-  get length() {
-    return this.openIds.size;
-  }
-  open(id) {
-    this.openIds.add(id);
-  }
-  isOpen(id) {
-    return this.openIds.has(id);
-  }
-  push(id, item) {
-    this.assertOpen(id);
-    const pending = this.pendingPulls.get(id);
-    if (pending?.length) {
-      pending.shift()[0](item);
-      if (pending.length === 0) {
-        this.pendingPulls.delete(id);
-      }
-    } else {
-      const items = this.items.get(id);
-      if (items) {
-        items.push(item);
-      } else {
-        this.items.set(id, [item]);
+function preventNativeAwait(target) {
+  return new Proxy(target, {
+    get(target2, prop, receiver) {
+      const value = Reflect.get(target2, prop, receiver);
+      if (prop !== "then" || typeof value !== "function") {
+        return value;
       }
+      return new Proxy(value, {
+        apply(targetFn, thisArg, args) {
+          if (args.length !== 2 || args.some((arg) => !isNativeFunction(arg))) {
+            return Reflect.apply(targetFn, thisArg, args);
+          }
+          let shouldOmit = true;
+          args[0].call(thisArg, preventNativeAwait(new Proxy(target2, {
+            get: (target3, prop2, receiver2) => {
+              if (shouldOmit && prop2 === "then") {
+                shouldOmit = false;
+                return void 0;
+              }
+              return Reflect.get(target3, prop2, receiver2);
+            }
+          })));
+        }
+      });
     }
-  }
-  async pull(id) {
-    this.assertOpen(id);
-    const items = this.items.get(id);
-    if (items?.length) {
-      const item = items.shift();
-      if (items.length === 0) {
-        this.items.delete(id);
-      }
-      return item;
+  });
+}
+const NATIVE_FUNCTION_REGEX = /^\s*function\s*\(\)\s*\{\s*\[native code\]\s*\}\s*$/;
+function isNativeFunction(fn) {
+  return typeof fn === "function" && NATIVE_FUNCTION_REGEX.test(fn.toString());
+}
+
+function streamToAsyncIteratorClass(stream) {
+  const reader = stream.getReader();
+  return new AsyncIteratorClass(
+    async () => {
+      return reader.read();
+    },
+    async () => {
+      await reader.cancel();
     }
-    return new Promise((resolve, reject) => {
-      const waitingPulls = this.pendingPulls.get(id);
-      const pending = [resolve, reject];
-      if (waitingPulls) {
-        waitingPulls.push(pending);
+  );
+}
+function asyncIteratorToStream(iterator) {
+  return new ReadableStream({
+    async pull(controller) {
+      const { done, value } = await iterator.next();
+      if (done) {
+        controller.close();
       } else {
-        this.pendingPulls.set(id, [pending]);
+        controller.enqueue(value);
       }
-    });
-  }
-  close({ id, reason } = {}) {
-    if (id === void 0) {
-      this.pendingPulls.forEach((pendingPulls, id2) => {
-        pendingPulls.forEach(([, reject]) => {
-          reject(reason ?? new Error(`[AsyncIdQueue] Queue[${id2}] was closed or aborted while waiting for pulling.`));
-        });
-      });
-      this.pendingPulls.clear();
-      this.openIds.clear();
-      this.items.clear();
-      return;
-    }
-    this.pendingPulls.get(id)?.forEach(([, reject]) => {
-      reject(reason ?? new Error(`[AsyncIdQueue] Queue[${id}] was closed or aborted while waiting for pulling.`));
-    });
-    this.pendingPulls.delete(id);
-    this.openIds.delete(id);
-    this.items.delete(id);
-  }
-  assertOpen(id) {
-    if (!this.isOpen(id)) {
-      throw new Error(`[AsyncIdQueue] Cannot access queue[${id}] because it is not open or aborted.`);
+    },
+    async cancel() {
+      await iterator.return?.();
     }
+  });
+}
+
+function tryDecodeURIComponent(value) {
+  try {
+    return decodeURIComponent(value);
+  } catch {
+    return value;
   }
 }
 
@@ -301,4 +663,4 @@ function value(value2, ...args) {
   return value2;
 }
 
-export { AsyncIdQueue, SequentialIdGenerator, clone, createAsyncIteratorObject, findDeepMatches, get, intercept, isAsyncIteratorObject, isObject, isTypescriptObject, onError, onFinish, onStart, onSuccess, once, parseEmptyableJSON, resolveMaybeOptionalOptions, sequential, splitInHalf, stringifyJSON, toArray, value };
+export { AbortError, AsyncIdQueue, AsyncIteratorClass, EventPublisher, NullProtoObj, ORPC_NAME, ORPC_SHARED_PACKAGE_NAME, ORPC_SHARED_PACKAGE_VERSION, SequentialIdGenerator, asyncIteratorToStream, asyncIteratorWithSpan, clone, defer, findDeepMatches, get, getGlobalOtelConfig, intercept, isAsyncIteratorObject, isObject, isPropertyKey, isTypescriptObject, onError, onFinish, onStart, onSuccess, once, parseEmptyableJSON, preventNativeAwait, readAsBuffer, replicateAsyncIterator, resolveMaybeOptionalOptions, runInSpanContext, runWithSpan, sequential, setGlobalOtelConfig, setSpanAttribute, setSpanError, splitInHalf, startSpan, streamToAsyncIteratorClass, stringifyJSON, toArray, toOtelException, toSpanAttributeValue, tryDecodeURIComponent, value };