@orpc/shared 0.0.0-next.75f1e0e → 0.0.0-next.7605f6c

package/dist/index.mjs

@@ -1,4 +1,4 @@
-export { group, guard, mapEntries, mapValues, omit } from 'radash';
+export { group, guard, mapEntries, mapValues, omit, retry, sleep } from 'radash';
 
 function resolveMaybeOptionalOptions(rest) {
   return rest[0] ?? {};
@@ -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.7605f6c";
+
+class AbortError extends Error {
+  constructor(...rest) {
+    super(...rest);
+    this.name = "AbortError";
+  }
+}
+
 function once(fn) {
   let cached;
   return () => {
@@ -40,13 +58,112 @@ function defer(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();
-  items = /* @__PURE__ */ new Map();
-  pendingPulls = /* @__PURE__ */ new Map();
+  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);
   }
@@ -55,59 +172,57 @@ class AsyncIdQueue {
   }
   push(id, item) {
     this.assertOpen(id);
-    const pending = this.pendingPulls.get(id);
+    const pending = this.waiters.get(id);
     if (pending?.length) {
       pending.shift()[0](item);
       if (pending.length === 0) {
-        this.pendingPulls.delete(id);
+        this.waiters.delete(id);
       }
     } else {
-      const items = this.items.get(id);
+      const items = this.queues.get(id);
       if (items) {
         items.push(item);
       } else {
-        this.items.set(id, [item]);
+        this.queues.set(id, [item]);
       }
     }
   }
   async pull(id) {
     this.assertOpen(id);
-    const items = this.items.get(id);
+    const items = this.queues.get(id);
     if (items?.length) {
       const item = items.shift();
       if (items.length === 0) {
-        this.items.delete(id);
+        this.queues.delete(id);
       }
       return item;
     }
     return new Promise((resolve, reject) => {
-      const waitingPulls = this.pendingPulls.get(id);
+      const waitingPulls = this.waiters.get(id);
       const pending = [resolve, reject];
       if (waitingPulls) {
         waitingPulls.push(pending);
       } else {
-        this.pendingPulls.set(id, [pending]);
+        this.waiters.set(id, [pending]);
       }
     });
   }
   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.waiters.forEach((pendingPulls, id2) => {
+        const error2 = reason ?? new AbortError(`[AsyncIdQueue] Queue[${id2}] was closed or aborted while waiting for pulling.`);
+        pendingPulls.forEach(([, reject]) => reject(error2));
       });
-      this.pendingPulls.clear();
+      this.waiters.clear();
       this.openIds.clear();
-      this.items.clear();
+      this.queues.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);
+    const error = reason ?? new AbortError(`[AsyncIdQueue] Queue[${id}] was closed or aborted while waiting for pulling.`);
+    this.waiters.get(id)?.forEach(([, reject]) => reject(error));
+    this.waiters.delete(id);
     this.openIds.delete(id);
-    this.items.delete(id);
+    this.queues.delete(id);
   }
   assertOpen(id) {
     if (!this.isOpen(id)) {
@@ -120,111 +235,148 @@ function isAsyncIteratorObject(maybe) {
   if (!maybe || typeof maybe !== "object") {
     return false;
   }
-  return Symbol.asyncIterator in maybe && typeof maybe[Symbol.asyncIterator] === "function";
+  return "next" in maybe && typeof maybe.next === "function" && 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) {
+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) {
-          isDone = true;
+          this.#isDone = true;
         }
         return result;
       } catch (err) {
-        isDone = true;
+        this.#isDone = true;
         throw err;
       } finally {
-        if (isDone && !isExecuteComplete) {
-          isExecuteComplete = true;
-          await cleanup("next");
+        if (this.#isDone && !this.#isExecuteComplete) {
+          this.#isExecuteComplete = true;
+          await this.#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;
+    });
+  }
+  next() {
+    return this.#next();
+  }
+  async return(value) {
+    this.#isDone = true;
+    if (!this.#isExecuteComplete) {
+      this.#isExecuteComplete = true;
+      await this.#cleanup("return");
     }
-  };
-  return iterator;
+    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 replicated = [];
-  let error;
+  const ids = Array.from({ length: count }, (_, i) => i.toString());
+  let isSourceFinished = false;
   const start = once(async () => {
     try {
       while (true) {
         const item = await source.next();
-        for (let id = 0; id < count; id++) {
+        ids.forEach((id) => {
           if (queue.isOpen(id)) {
-            queue.push(id, item);
+            queue.push(id, { next: item });
           }
-        }
+        });
         if (item.done) {
           break;
         }
       }
-    } catch (e) {
-      error = { value: e };
+    } catch (error) {
+      ids.forEach((id) => {
+        if (queue.isOpen(id)) {
+          queue.push(id, { error });
+        }
+      });
+    } finally {
+      isSourceFinished = true;
     }
   });
-  for (let id = 0; id < count; id++) {
+  const replicated = ids.map((id) => {
     queue.open(id);
-    replicated.push(createAsyncIteratorObject(
-      () => {
+    return new AsyncIteratorClass(
+      async () => {
         start();
-        return new Promise((resolve, reject) => {
-          queue.pull(id).then(resolve).catch(reject);
-          defer(() => {
-            if (error) {
-              reject(error.value);
-            }
-          });
-        });
+        const item = await queue.pull(id);
+        if (item.next) {
+          return item.next;
+        }
+        throw item.error;
       },
       async (reason) => {
         queue.close({ id });
-        if (reason !== "next") {
-          if (replicated.every((_, id2) => !queue.isOpen(id2))) {
-            await source?.return?.();
-          }
+        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();
@@ -251,18 +403,19 @@ class EventPublisher {
     if (typeof listenerOrOptions === "function") {
       let listeners = this.#listenersMap.get(event);
       if (!listeners) {
-        this.#listenersMap.set(event, listeners = /* @__PURE__ */ new Set());
+        this.#listenersMap.set(event, listeners = []);
       }
-      listeners.add(listenerOrOptions);
-      return () => {
-        listeners.delete(listenerOrOptions);
-        if (listeners.size === 0) {
+      listeners.push(listenerOrOptions);
+      return once(() => {
+        listeners.splice(listeners.indexOf(listenerOrOptions), 1);
+        if (listeners.length === 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) => {
@@ -283,7 +436,7 @@ class EventPublisher {
       bufferedEvents.length = 0;
     };
     signal?.addEventListener("abort", abortListener, { once: true });
-    return createAsyncIteratorObject(async () => {
+    return new AsyncIteratorClass(async () => {
       if (signal?.aborted) {
         throw signal.reason;
       }
@@ -304,14 +457,18 @@ class EventPublisher {
 }
 
 class SequentialIdGenerator {
-  nextId = 0;
+  index = BigInt(1);
   generate() {
-    if (this.nextId === Number.MAX_SAFE_INTEGER) {
-      this.nextId = 0;
-      return Number.MAX_SAFE_INTEGER;
-    }
-    return this.nextId++;
+    const id = this.index.toString(36);
+    this.index++;
+    return id;
+  }
+}
+function compareSequentialIds(a, b) {
+  if (a.length !== b.length) {
+    return a.length - b.length;
   }
+  return a < b ? -1 : a > b ? 1 : 0;
 }
 
 function onStart(callback) {
@@ -391,6 +548,12 @@ function findDeepMatches(check, payload, segments = [], maps = [], values = [])
   }
   return { maps, values };
 }
+function getConstructor(value) {
+  if (!isTypescriptObject(value)) {
+    return null;
+  }
+  return Object.getPrototypeOf(value)?.constructor;
+}
 function isObject(value) {
   if (!value || typeof value !== "object") {
     return false;
@@ -410,6 +573,9 @@ function clone(value) {
     for (const key in value) {
       result[key] = clone(value[key]);
     }
+    for (const sym of Object.getOwnPropertySymbols(value)) {
+      result[sym] = clone(value[sym]);
+    }
     return result;
   }
   return value;
@@ -442,5 +608,104 @@ function value(value2, ...args) {
   }
   return value2;
 }
+function fallback(value2, fallback2) {
+  return value2 === void 0 ? fallback2 : value2;
+}
+
+function preventNativeAwait(target) {
+  return new Proxy(target, {
+    get(target2, prop, receiver) {
+      const value2 = Reflect.get(target2, prop, receiver);
+      if (prop !== "then" || typeof value2 !== "function") {
+        return value2;
+      }
+      return new Proxy(value2, {
+        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);
+            }
+          })));
+        }
+      });
+    }
+  });
+}
+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 overlayProxy(target, partial) {
+  const proxy = new Proxy(typeof target === "function" ? partial : target, {
+    get(_, prop) {
+      const targetValue = prop in partial ? partial : value(target);
+      const v = Reflect.get(targetValue, prop);
+      return typeof v === "function" ? v.bind(targetValue) : v;
+    },
+    has(_, prop) {
+      return Reflect.has(partial, prop) || Reflect.has(value(target), prop);
+    }
+  });
+  return proxy;
+}
+
+function streamToAsyncIteratorClass(stream) {
+  const reader = stream.getReader();
+  return new AsyncIteratorClass(
+    async () => {
+      return reader.read();
+    },
+    async () => {
+      await reader.cancel();
+    }
+  );
+}
+function asyncIteratorToStream(iterator) {
+  return new ReadableStream({
+    async pull(controller) {
+      const { done, value } = await iterator.next();
+      if (done) {
+        controller.close();
+      } else {
+        controller.enqueue(value);
+      }
+    },
+    async cancel() {
+      await iterator.return?.();
+    }
+  });
+}
+function asyncIteratorToUnproxiedDataStream(iterator) {
+  return new ReadableStream({
+    async pull(controller) {
+      const { done, value } = await iterator.next();
+      if (done) {
+        controller.close();
+      } else {
+        const unproxied = isObject(value) ? { ...value } : Array.isArray(value) ? value.map((i) => i) : value;
+        controller.enqueue(unproxied);
+      }
+    },
+    async cancel() {
+      await iterator.return?.();
+    }
+  });
+}
+
+function tryDecodeURIComponent(value) {
+  try {
+    return decodeURIComponent(value);
+  } catch {
+    return value;
+  }
+}
 
-export { AsyncIdQueue, EventPublisher, NullProtoObj, SequentialIdGenerator, clone, createAsyncIteratorObject, defer, findDeepMatches, get, intercept, isAsyncIteratorObject, isObject, isPropertyKey, isTypescriptObject, onError, onFinish, onStart, onSuccess, once, parseEmptyableJSON, replicateAsyncIterator, resolveMaybeOptionalOptions, sequential, splitInHalf, stringifyJSON, toArray, value };
+export { AbortError, AsyncIdQueue, AsyncIteratorClass, EventPublisher, NullProtoObj, ORPC_NAME, ORPC_SHARED_PACKAGE_NAME, ORPC_SHARED_PACKAGE_VERSION, SequentialIdGenerator, asyncIteratorToStream, asyncIteratorToUnproxiedDataStream, asyncIteratorWithSpan, clone, compareSequentialIds, defer, fallback, findDeepMatches, get, getConstructor, getGlobalOtelConfig, intercept, isAsyncIteratorObject, isObject, isPropertyKey, isTypescriptObject, onError, onFinish, onStart, onSuccess, once, overlayProxy, parseEmptyableJSON, preventNativeAwait, readAsBuffer, replicateAsyncIterator, resolveMaybeOptionalOptions, runInSpanContext, runWithSpan, sequential, setGlobalOtelConfig, setSpanAttribute, setSpanError, splitInHalf, startSpan, streamToAsyncIteratorClass, stringifyJSON, toArray, toOtelException, toSpanAttributeValue, tryDecodeURIComponent, value };

package/package.json

@@ -1,18 +1,18 @@
 {
   "name": "@orpc/shared",
   "type": "module",
-  "version": "0.0.0-next.75f1e0e",
+  "version": "0.0.0-next.7605f6c",
   "license": "MIT",
-  "homepage": "https://orpc.unnoq.com",
+  "homepage": "https://orpc.dev",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/unnoq/orpc.git",
+    "url": "git+https://github.com/middleapi/orpc.git",
     "directory": "packages/shared"
   },
   "keywords": [
-    "unnoq",
     "orpc"
   ],
+  "sideEffects": false,
   "exports": {
     ".": {
       "types": "./dist/index.d.mts",
@@ -23,14 +23,23 @@
   "files": [
     "dist"
   ],
+  "peerDependencies": {
+    "@opentelemetry/api": ">=1.9.0"
+  },
+  "peerDependenciesMeta": {
+    "@opentelemetry/api": {
+      "optional": true
+    }
+  },
   "dependencies": {
-    "radash": "^12.1.0",
-    "type-fest": "^4.39.1"
+    "radash": "^12.1.1",
+    "type-fest": "^5.4.4"
   },
   "devDependencies": {
-    "arktype": "2.1.20",
-    "valibot": "^1.1.0",
-    "zod": "^3.25.49"
+    "@opentelemetry/api": "^1.9.0",
+    "arktype": "2.2.0",
+    "valibot": "^1.2.0",
+    "zod": "^4.3.6"
   },
   "scripts": {
     "build": "unbuild",