refactor(app): centralize checkout query keys (#844)

2026-05-09 22:40:26 +08:00
parent f17fb77013
commit a4bcc7b169
12 changed files with 164 additions and 110 deletions
@@ -0,0 +1,82 @@
+import type { Query, QueryClient } from "@tanstack/react-query";
+
+interface CheckoutQueryIdentity {
+  serverId: string;
+  cwd: string;
+}
+
+type CheckoutQueryKey = readonly unknown[];
+
+export function checkoutStatusQueryKey(serverId: string, cwd: string) {
+  return ["checkoutStatus", serverId, cwd] as const;
+}
+
+export function checkoutDiffQueryKey(
+  serverId: string,
+  cwd: string,
+  mode: "uncommitted" | "base",
+  baseRef?: string,
+  ignoreWhitespace?: boolean,
+) {
+  return ["checkoutDiff", serverId, cwd, mode, baseRef ?? "", ignoreWhitespace === true] as const;
+}
+
+export function checkoutPrStatusQueryKey(serverId: string, cwd: string) {
+  return ["checkoutPrStatus", serverId, cwd] as const;
+}
+
+export function prPaneTimelineQueryKey({
+  serverId,
+  cwd,
+  prNumber,
+}: {
+  serverId: string;
+  cwd: string;
+  prNumber: number | null;
+}) {
+  return ["prPaneTimeline", serverId, cwd, prNumber] as const;
+}
+
+export async function invalidateCheckoutGitQueriesForClient(
+  queryClient: QueryClient,
+  identity: CheckoutQueryIdentity,
+) {
+  await Promise.all([
+    queryClient.invalidateQueries({
+      queryKey: checkoutStatusQueryKey(identity.serverId, identity.cwd),
+    }),
+    queryClient.invalidateQueries({
+      predicate: checkoutQueryPredicate("checkoutDiff", identity),
+    }),
+    queryClient.invalidateQueries({
+      predicate: checkoutQueryPredicate("checkoutPrStatus", identity),
+    }),
+    queryClient.invalidateQueries({
+      predicate: checkoutQueryPredicate("prPaneTimeline", identity),
+    }),
+  ]);
+}
+
+function checkoutQueryPredicate(
+  queryKind: CheckoutQueryKey[0],
+  identity: CheckoutQueryIdentity,
+): (query: Query) => boolean {
+  return (query) => {
+    const key = query.queryKey;
+    return (
+      isCheckoutQueryKey(key) &&
+      key[0] === queryKind &&
+      key[1] === identity.serverId &&
+      key[2] === identity.cwd
+    );
+  };
+}
+
+function isCheckoutQueryKey(key: readonly unknown[]): key is CheckoutQueryKey {
+  return (
+    key.length >= 3 &&
+    typeof key[0] === "string" &&
+    typeof key[1] === "string" &&
+    typeof key[2] === "string"
+  );
+}