import type { GrpcRequest, HttpRequest, WebsocketRequest } from "@yaakapp-internal/models";
import { atom, useAtomValue } from "jotai";
import { activeRequestIdAtom } from "./useActiveRequestId";
import { allRequestsAtom } from "./useAllRequests";

export const activeRequestAtom = atom((get) => {
  const activeRequestId = get(activeRequestIdAtom);
  const requests = get(allRequestsAtom);
  return requests.find((r) => r.id === activeRequestId) ?? null;
});

interface TypeMap {
  http_request: HttpRequest;
  grpc_request: GrpcRequest;
  websocket_request: WebsocketRequest;
}

export function useActiveRequest<T extends keyof TypeMap>(model?: T): TypeMap[T] | null {
  const activeRequest = useAtomValue(activeRequestAtom);
  if (model == null) return activeRequest as TypeMap[T];
  if (activeRequest?.model === model) return activeRequest as TypeMap[T];
  return null;
}