@travetto/context/src/decorator.ts

Async-aware state management, maintaining context across asynchronous calls.

import { AsyncMethodDescriptor } from '@travetto/runtime';

import { AsyncContext } from './service.ts';

/**
 * Allows running a function while providing an async context
 */
export function WithAsyncContext() {
  return function <T extends { context: AsyncContext }>(
    target: T,
    prop: string,
    descriptor: AsyncMethodDescriptor<T>
  ): typeof descriptor {
    const og = descriptor.value!;
    descriptor.value = function (...args: unknown[]): ReturnType<typeof og> {
      return this.context.run(og.bind(this, ...args));
    };

    Object.defineProperty(descriptor.value, 'name', { value: og.name });

    return descriptor;
  };
}