Function wrap

packages/core/sdk/src/executor.test.ts:960–979

(inner: FumaDb)

  record: <A>(key: string, run: () => Promise<A>) => Promise<A>,
): FumaDb => {
  const wrap = (inner: FumaDb): FumaDb =>
    new Proxy(inner, {
      get(target, prop) {
        if (prop === "withContext") {
          return (context: unknown) =>
            wrap((target.withContext as (c: unknown) => FumaDb)(context));
        }
        if (prop === "findFirst" || prop === "findMany") {
          return (table: unknown, query: unknown) =>
            record(`${String(table)}.${prop}`, () =>
              (Reflect.get(target, prop) as (t: unknown, q: unknown) => Promise<unknown>).call(
                target,
                table,
                query,
              ),
            );
        }
        return Reflect.get(target, prop);
      },
    });
  return wrap(db);
};