Function wrapQueryLike

src/server/db/index.ts:1267–1312

(value: T)

}

function wrapQueryLike<T>(value: T): T {
  if (!shouldWrapObject(value)) return value;
  const target = value as unknown as object;
  if (wrappedObjects.has(target)) {
    return wrappedObjects.get(target) as T;
  }

  const proxy = new Proxy(target as Record<string, unknown>, {
    get(innerTarget, prop, receiver) {
      if (prop === 'then' && typeof innerTarget.then === 'function') {
        return innerTarget.then.bind(innerTarget);
      }

      if (prop === 'all' && typeof innerTarget.all !== 'function' && typeof innerTarget.execute === 'function') {
        return async (...args: unknown[]) => normalizeAllResult(await (innerTarget.execute as (...a: unknown[]) => Promise<unknown>)(...args));
      }

      if (prop === 'get' && typeof innerTarget.get !== 'function' && typeof innerTarget.execute === 'function') {
        return async (...args: unknown[]) => {
          const rows = normalizeAllResult(await (innerTarget.execute as (...a: unknown[]) => Promise<unknown>)(...args));
          return rows[0] ?? undefined;
        };
      }

      if (prop === 'run' && typeof innerTarget.run !== 'function' && typeof innerTarget.execute === 'function') {
        return async (...args: unknown[]) => normalizeRunResult(await (innerTarget.execute as (...a: unknown[]) => Promise<unknown>)(...args));
      }

      const original = Reflect.get(innerTarget, prop, receiver);
      if (typeof original !== 'function') {
        return original;
      }

      return (...args: unknown[]) => {
        const result = original.apply(innerTarget, args);
        if (shouldWrapObject(result)) {
          return wrapQueryLike(result);
        }
        return result;
      };
    },
  });

  wrappedObjects.set(target, proxy);
  return proxy as unknown as T;
}

function wrapDbClient<T extends object>(
  rawDb: T,