24 lines
876 B
TypeScript
24 lines
876 B
TypeScript
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;
|
|
}
|