42 lines
1.4 KiB
TypeScript
42 lines
1.4 KiB
TypeScript
import { platform } from "@yaakapp-internal/platform";
|
|
import type { ModelPayload } from "@yaakapp-internal/models";
|
|
import { atom, useAtomValue } from "jotai";
|
|
import { generateId } from "../lib/generateId";
|
|
import { jotaiStore } from "../lib/jotai";
|
|
|
|
const requestUpdateKeyAtom = atom<Record<string, string>>({});
|
|
|
|
platform.listen<ModelPayload[]>("model_writes", (payloads) => {
|
|
const changedIds: string[] = [];
|
|
for (const payload of payloads) {
|
|
if (payload.change.type !== "upsert") continue;
|
|
|
|
if (
|
|
(payload.model.model === "http_request" ||
|
|
payload.model.model === "grpc_request" ||
|
|
payload.model.model === "websocket_request") &&
|
|
((payload.updateSource.type === "window" &&
|
|
payload.updateSource.label !== platform.window.label) ||
|
|
payload.updateSource.type !== "window")
|
|
) {
|
|
changedIds.push(payload.model.id);
|
|
}
|
|
}
|
|
if (changedIds.length > 0) wasUpdatedExternally(changedIds);
|
|
});
|
|
|
|
export function wasUpdatedExternally(changedRequestIds: string | string[]) {
|
|
const ids = Array.isArray(changedRequestIds) ? changedRequestIds : [changedRequestIds];
|
|
jotaiStore.set(requestUpdateKeyAtom, (m) => {
|
|
const next = { ...m };
|
|
for (const id of ids) next[id] = generateId();
|
|
return next;
|
|
});
|
|
}
|
|
|
|
export function useRequestUpdateKey(requestId: string | null) {
|
|
const keys = useAtomValue(requestUpdateKeyAtom);
|
|
const key = keys[requestId ?? "n/a"];
|
|
return `${requestId}::${key ?? "default"}`;
|
|
}
|