93 lines
2.9 KiB
TypeScript
93 lines
2.9 KiB
TypeScript
import { useQuery, useQueryClient } from "@tanstack/react-query";
|
|
import { useCallback, useEffect } from "react";
|
|
import { UnistylesRuntime } from "react-native-unistyles";
|
|
import { useSessionStore } from "@/stores/session-store";
|
|
import { usePanelStore } from "@/stores/panel-store";
|
|
import type { CheckoutDiffResponse } from "@server/shared/messages";
|
|
|
|
const CHECKOUT_DIFF_STALE_TIME = 30_000;
|
|
|
|
function checkoutDiffQueryKey(
|
|
serverId: string,
|
|
cwd: string,
|
|
mode: "uncommitted" | "base",
|
|
baseRef?: string
|
|
) {
|
|
return ["checkoutDiff", serverId, cwd, mode, baseRef ?? ""] as const;
|
|
}
|
|
|
|
interface UseCheckoutDiffQueryOptions {
|
|
serverId: string;
|
|
cwd: string;
|
|
mode: "uncommitted" | "base";
|
|
baseRef?: string;
|
|
enabled?: boolean;
|
|
}
|
|
|
|
export type ParsedDiffFile = CheckoutDiffResponse["payload"]["files"][number];
|
|
export type DiffHunk = ParsedDiffFile["hunks"][number];
|
|
export type DiffLine = DiffHunk["lines"][number];
|
|
export type HighlightToken = NonNullable<DiffLine["tokens"]>[number];
|
|
|
|
export function useCheckoutDiffQuery({
|
|
serverId,
|
|
cwd,
|
|
mode,
|
|
baseRef,
|
|
enabled = true,
|
|
}: UseCheckoutDiffQueryOptions) {
|
|
const queryClient = useQueryClient();
|
|
const client = useSessionStore(
|
|
(state) => state.sessions[serverId]?.client ?? null
|
|
);
|
|
const isConnected = useSessionStore(
|
|
(state) => state.sessions[serverId]?.connection.isConnected ?? false
|
|
);
|
|
const isMobile =
|
|
UnistylesRuntime.breakpoint === "xs" || UnistylesRuntime.breakpoint === "sm";
|
|
const mobileView = usePanelStore((state) => state.mobileView);
|
|
const desktopFileExplorerOpen = usePanelStore((state) => state.desktop.fileExplorerOpen);
|
|
const explorerTab = usePanelStore((state) => state.explorerTab);
|
|
const isOpen = isMobile ? mobileView === "file-explorer" : desktopFileExplorerOpen;
|
|
|
|
const query = useQuery({
|
|
queryKey: checkoutDiffQueryKey(serverId, cwd, mode, baseRef),
|
|
queryFn: async () => {
|
|
if (!client) {
|
|
throw new Error("Daemon client not available");
|
|
}
|
|
return await client.getCheckoutDiff(cwd, { mode, baseRef });
|
|
},
|
|
enabled: !!client && isConnected && !!cwd && enabled,
|
|
staleTime: CHECKOUT_DIFF_STALE_TIME,
|
|
refetchInterval: 10_000,
|
|
});
|
|
|
|
// Revalidate when sidebar opens with "changes" tab active
|
|
useEffect(() => {
|
|
if (!isOpen || explorerTab !== "changes" || !cwd) {
|
|
return;
|
|
}
|
|
queryClient.invalidateQueries({
|
|
queryKey: checkoutDiffQueryKey(serverId, cwd, mode, baseRef),
|
|
});
|
|
}, [isOpen, explorerTab, serverId, cwd, mode, baseRef, queryClient]);
|
|
|
|
const refresh = useCallback(() => {
|
|
return query.refetch();
|
|
}, [query]);
|
|
|
|
const payload = query.data ?? null;
|
|
const payloadError = payload?.error ?? null;
|
|
|
|
return {
|
|
files: payload?.files ?? [],
|
|
payloadError,
|
|
isLoading: query.isLoading,
|
|
isFetching: query.isFetching,
|
|
isError: query.isError || Boolean(payloadError),
|
|
error: query.error,
|
|
refresh,
|
|
};
|
|
}
|