Function wrapDbClient

src/server/db/index.ts:1314–1346

(
  rawDb: T,
  customTransaction?: <R>(fn: (tx: any) => Promise<R> | R) => Promise<R>,
)

}

function wrapDbClient<T extends object>(
  rawDb: T,
  customTransaction?: <R>(fn: (tx: any) => Promise<R> | R) => Promise<R>,
) {
  return new Proxy(rawDb as Record<string, unknown>, {
    get(target, prop, receiver) {
      if (prop === 'transaction') {
        if (customTransaction) return customTransaction;

        const originalTransaction = target.transaction;
        if (typeof originalTransaction !== 'function') return undefined;
        return async <R>(fn: (tx: any) => Promise<R> | R) => {
          return await (originalTransaction as (handler: (tx: unknown) => Promise<R> | R) => Promise<R>).call(target, async (tx: unknown) => {
            return await fn(wrapDbClient(tx as object));
          });
        };
      }

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

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

function initSqliteDb() {
  const sqlitePath = resolveSqlitePath();