import type { Callable } from "@emprespresso/utils";
import {
  type ITraceableMapper,
  type ITraceableTuple,
  TraceableImpl,
  TraceableLogger,
} from "./mod.ts";

export class Traceable<T> extends TraceableImpl<T, TraceableLogger> {
  static from<T>(t: T) {
    return new Traceable(t, new TraceableLogger());
  }

  static withFunctionTrace<F extends Callable, T>(
    f: F,
  ): ITraceableMapper<T, TraceableLogger, ITraceableTuple<T>> {
    return (t) => [t.item, f.name];
  }

  static withClassTrace<C extends object, T>(
    c: C,
  ): ITraceableMapper<T, TraceableLogger, ITraceableTuple<T>> {
    return (t) => [t.item, c.constructor.name];
  }

  static promiseify<T, U>(
    mapper: ITraceableMapper<T, TraceableLogger, U>,
  ): ITraceableMapper<Promise<T>, TraceableLogger, Promise<U>> {
    return (traceablePromise) =>
      traceablePromise.flatMapAsync(async (t) =>
        t.move(await t.item).map(mapper)
      ).item;
  }
}