import type { ConnectionPoolClient } from "./factories/createConnectionPool.js";
import { UnexpectedStateError } from "@slonik/errors";
import type { DeferredPromise } from "@slonik/utilities";

type PoolClientState = {
  activeQuery?: DeferredPromise<null>;
  connectionId: string;
  poolId: string;
  terminated: Error | null;
  transactionDepth: null | number;
  transactionId: null | string;
};

export const poolClientStateMap = new WeakMap<ConnectionPoolClient, PoolClientState>();

export const getPoolClientState = (poolClient: ConnectionPoolClient): PoolClientState => {
  const poolClientState = poolClientStateMap.get(poolClient);

  if (!poolClientState) {
    throw new UnexpectedStateError("Pool Client state is unavailable.");
  }

  return poolClientState;
};