forked from QuantumNous/new-api
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathuse-status.ts
More file actions
82 lines (73 loc) · 2.44 KB
/
Copy pathuse-status.ts
File metadata and controls
82 lines (73 loc) · 2.44 KB
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
/*
Copyright (C) 2023-2026 QuantumNous
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
For commercial licensing, please contact support@quantumnous.com
*/
import { useQuery } from '@tanstack/react-query'
import type { SystemStatus } from '@/features/auth/types'
import { getStatus } from '@/lib/api'
import { useSystemConfigStore } from '@/stores/system-config-store'
import { mapStatusDataToConfig } from './use-system-config'
// Get initial cache from localStorage
function getInitialStatus(): SystemStatus | undefined {
try {
if (typeof window !== 'undefined') {
const saved = window.localStorage.getItem('status')
return saved ? (JSON.parse(saved) as SystemStatus) : undefined
}
} catch {
/* empty */
}
return undefined
}
export function useStatus() {
const { data, isLoading, error } = useQuery({
queryKey: ['status'],
queryFn: async () => {
const status = await getStatus()
try {
if (status) {
const { setConfig } = useSystemConfigStore.getState()
setConfig(mapStatusDataToConfig(status))
}
} catch (err) {
if (import.meta.env.DEV) {
// eslint-disable-next-line no-console
console.warn(
'[useStatus] Failed to sync status to system config',
err
)
}
}
// Save to localStorage
try {
if (typeof window !== 'undefined' && status) {
window.localStorage.setItem('status', JSON.stringify(status))
}
} catch {
/* empty */
}
return status as SystemStatus | null
},
// Use localStorage data as initial data
placeholderData: getInitialStatus(),
// Data becomes stale after 5 minutes
staleTime: 5 * 60 * 1000,
// Cache expires after 30 minutes
gcTime: 30 * 60 * 1000,
})
return {
status: data ?? null,
loading: isLoading,
error,
}
}