Spaces:
Running
Running
File size: 3,107 Bytes
b034029 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 | import { create } from 'zustand';
import { ApiError, fetchUsageOverview } from '@/lib/api';
import type { UsageOverviewResponse, UsageTimeRange } from '@/lib/types';
export const USAGE_STATS_STALE_TIME_MS = 60_000;
interface LoadUsageStatsOptions {
force?: boolean;
staleTimeMs?: number;
range?: UsageTimeRange;
start?: string;
end?: string;
}
interface UsageStatsState {
usage: UsageOverviewResponse | null;
loading: boolean;
error: string;
lastRefreshedAt: number | null;
lastQueryKey: string | null;
loadUsageStats: (options?: LoadUsageStatsOptions) => Promise<void>;
clearUsageStats: () => void;
}
let activeRequest: Promise<void> | null = null;
let activeRequestKey: string | null = null;
let activeRequestController: AbortController | null = null;
const buildQueryKey = (range: UsageTimeRange, start?: string, end?: string): string =>
`${range}:${start ?? ''}:${end ?? ''}`;
export const useUsageStatsStore = create<UsageStatsState>((set, get) => ({
usage: null,
loading: false,
error: '',
lastRefreshedAt: null,
lastQueryKey: null,
loadUsageStats: async (options = {}) => {
const {
force = false,
staleTimeMs = USAGE_STATS_STALE_TIME_MS,
range = 'all',
start,
end,
} = options;
const { lastRefreshedAt, loading, usage, lastQueryKey } = get();
const now = Date.now();
const queryKey = buildQueryKey(range, start, end);
if (!force && usage && lastRefreshedAt && lastQueryKey === queryKey && now - lastRefreshedAt < staleTimeMs) {
return;
}
if (loading && activeRequest) {
if (activeRequestKey === queryKey) {
return activeRequest;
}
activeRequestController?.abort();
}
const controller = new AbortController();
activeRequestController = controller;
activeRequestKey = queryKey;
set({ loading: true, error: '' });
activeRequest = (async () => {
try {
const response = await fetchUsageOverview(range, start, end, controller.signal);
if (activeRequestController !== controller) {
return;
}
set({
usage: response,
loading: false,
error: '',
lastRefreshedAt: Date.now(),
lastQueryKey: queryKey,
});
} catch (error) {
if (controller.signal.aborted) {
return;
}
const message = error instanceof ApiError && error.status === 401
? 'AUTH_REQUIRED'
: error instanceof Error
? error.message
: 'Failed to load usage overview'
if (activeRequestController === controller) {
set({
loading: false,
error: message
});
}
throw error;
} finally {
if (activeRequestController === controller) {
activeRequest = null;
activeRequestKey = null;
activeRequestController = null;
}
}
})();
return activeRequest;
},
clearUsageStats: () => set({ usage: null, error: '', loading: false, lastRefreshedAt: null, lastQueryKey: null })
}));
|