gitscope
/
← marzneshin / app/dashboard/src/stores/nodes.store.ts / blame
Files Commits 446 Branches 1 Tags 43 Markers 4 README License
https://dev.kronoz.win/r/marzneshin.git
app/dashboard/src/stores/nodes.store.ts · v0.1.2
Source
1 bf7c131 Mardin 2y ago import { useQuery } from 'react-query';
2 bf7c131 Mardin 2y ago import { fetch } from 'service/http';
3 bf7c131 Mardin 2y ago import { create } from 'zustand';
4 5bb2712 Mardin 2y ago import { z } from 'zod';
5 89f8f93 Mardin 2y ago import { FilterUsageType, NodesFilterType } from 'types/filter';
6 6aab6f0 Mrdn 2y ago import { pageSizeManagers } from 'utils/userPreferenceStorage';
7 89f8f93 Mardin 2y ago import { useDashboard } from './dashboard.store';
8 a61ba53 Mrdn 2y ago import { queryClient } from 'service/react-query';
9 a61ba53 Mrdn 2y ago import { queryIds } from 'constants/query-ids';
10 a47eb4c Dawsh 2y ago
11 a47eb4c Dawsh 2y ago export const NodeSchema = z.object({
12 a47eb4c Dawsh 2y ago name: z.string().min(1),
13 a47eb4c Dawsh 2y ago address: z.string().min(1),
14 a47eb4c Dawsh 2y ago port: z
15 a47eb4c Dawsh 2y ago .number()
16 a47eb4c Dawsh 2y ago .min(1)
17 a47eb4c Dawsh 2y ago .or(z.string().transform((v) => parseFloat(v))),
18 a47eb4c Dawsh 2y ago xray_version: z.string().nullable().optional(),
19 a47eb4c Dawsh 2y ago id: z.number().nullable().optional(),
20 a47eb4c Dawsh 2y ago status: z
21 5bb2712 Mardin 2y ago .enum(['healthy', 'unhealthy', 'disabled']),
22 a47eb4c Dawsh 2y ago message: z.string().nullable().optional(),
23 5bb2712 Mardin 2y ago usage_coefficient: z.number().default(1),
24 5bb2712 Mardin 2y ago add_as_new_host: z.boolean(),
25 5bb2712 Mardin 2y ago });
26 5bb2712 Mardin 2y ago
27 5bb2712 Mardin 2y ago export const NodeCreateSchema = z.object({
28 5bb2712 Mardin 2y ago name: z.string().min(1),
29 5bb2712 Mardin 2y ago address: z.string().min(1),
30 5bb2712 Mardin 2y ago port: z
31 5bb2712 Mardin 2y ago .number()
32 5bb2712 Mardin 2y ago .min(1)
33 5bb2712 Mardin 2y ago .or(z.string().transform((v) => parseFloat(v))),
34 5bb2712 Mardin 2y ago id: z.number().nullable().optional(),
35 5bb2712 Mardin 2y ago status: z
36 5bb2712 Mardin 2y ago .enum(['none', 'disabled']),
37 5bb2712 Mardin 2y ago usage_coefficient: z.number().default(1),
38 a47eb4c Dawsh 2y ago add_as_new_host: z.boolean().optional(),
39 a47eb4c Dawsh 2y ago });
40 a47eb4c Dawsh 2y ago
41 a47eb4c Dawsh 2y ago export type NodeType = z.infer<typeof NodeSchema>;
42 5bb2712 Mardin 2y ago export type NodeCreate = z.infer<typeof NodeCreateSchema>;
43 a47eb4c Dawsh 2y ago
44 a47eb4c Dawsh 2y ago export const getNodeDefaultValues = (): NodeType => ({
45 bf7c131 Mardin 2y ago name: '',
46 bf7c131 Mardin 2y ago address: '',
47 5bb2712 Mardin 2y ago status: 'unhealthy',
48 a47eb4c Dawsh 2y ago port: 62050,
49 bf7c131 Mardin 2y ago xray_version: '',
50 5bb2712 Mardin 2y ago add_as_new_host: false,
51 5bb2712 Mardin 2y ago usage_coefficient: 1,
52 5bb2712 Mardin 2y ago });
53 5bb2712 Mardin 2y ago
54 5bb2712 Mardin 2y ago export const getNodeCreateDefaultValues = (): NodeCreate => ({
55 5bb2712 Mardin 2y ago name: '',
56 5bb2712 Mardin 2y ago address: '',
57 5bb2712 Mardin 2y ago status: 'none',
58 5bb2712 Mardin 2y ago port: 62050,
59 5bb2712 Mardin 2y ago add_as_new_host: false,
60 5bb2712 Mardin 2y ago usage_coefficient: 1,
61 5bb2712 Mardin 2y ago id: 0,
62 a47eb4c Dawsh 2y ago });
63 a47eb4c Dawsh 2y ago
64 bf7c131 Mardin 2y ago export const FetchNodesQueryKey = 'fetch-nodes-query-key';
65 a47eb4c Dawsh 2y ago
66 a61ba53 Mrdn 2y ago export function fetchNodes(): Promise<NodeType[]> {
67 a61ba53 Mrdn 2y ago useDashboard.setState({ loading: true })
68 a61ba53 Mrdn 2y ago return fetch('/nodes').then((nodes) => {
69 a61ba53 Mrdn 2y ago return nodes;
70 a61ba53 Mrdn 2y ago }).finally(() => {
71 a61ba53 Mrdn 2y ago useDashboard.setState({ loading: false })
72 a61ba53 Mrdn 2y ago });
73 5bb2712 Mardin 2y ago }
74 5bb2712 Mardin 2y ago
75 a47eb4c Dawsh 2y ago export type NodeStore = {
76 5bb2712 Mardin 2y ago certificate: string,
77 5bb2712 Mardin 2y ago deletingNode?: NodeCreate | null;
78 5bb2712 Mardin 2y ago editingNode: NodeCreate | null;
79 5bb2712 Mardin 2y ago isAddingNode: boolean;
80 4205669 Mardin 2y ago isShowingNodesUsage: boolean;
81 5bb2712 Mardin 2y ago isEditingNode: boolean;
82 4205669 Mardin 2y ago isEditingCore: boolean;
83 5bb2712 Mardin 2y ago nodesFilters: NodesFilterType;
84 5bb2712 Mardin 2y ago refetchNodes: () => void;
85 5bb2712 Mardin 2y ago refetchCertificate: () => void;
86 a47eb4c Dawsh 2y ago fetchNodesUsage: (query: FilterUsageType) => Promise<void>;
87 5bb2712 Mardin 2y ago addNode: (node: NodeCreate) => Promise<unknown>;
88 5bb2712 Mardin 2y ago deleteNode: (node: NodeCreate) => Promise<unknown>;
89 5bb2712 Mardin 2y ago updateNode: (node: NodeCreate) => Promise<unknown>;
90 a47eb4c Dawsh 2y ago reconnectNode: (node: NodeType) => Promise<unknown>;
91 5bb2712 Mardin 2y ago onDeletingNode: (node: NodeCreate | undefined) => void;
92 5bb2712 Mardin 2y ago onEditingNode: (node: NodeType | undefined) => void;
93 5bb2712 Mardin 2y ago onAddingNode: (isAddingNode: boolean) => void;
94 5bb2712 Mardin 2y ago onFilterChange: (filters: Partial<NodesFilterType>) => void;
95 4205669 Mardin 2y ago onShowingNodesUsage: (isShowingNodesUsage: boolean) => void;
96 a47eb4c Dawsh 2y ago };
97 a47eb4c Dawsh 2y ago
98 a47eb4c Dawsh 2y ago export const useNodesQuery = () => {
99 a47eb4c Dawsh 2y ago return useQuery({
100 a47eb4c Dawsh 2y ago queryKey: FetchNodesQueryKey,
101 5bb2712 Mardin 2y ago queryFn: fetchNodes,
102 5bb2712 Mardin 2y ago refetchInterval: useNodes.getState().editingNode ? 3000 : undefined,
103 a47eb4c Dawsh 2y ago refetchOnWindowFocus: false,
104 a47eb4c Dawsh 2y ago });
105 a47eb4c Dawsh 2y ago };
106 a47eb4c Dawsh 2y ago
107 a47eb4c Dawsh 2y ago export const useNodes = create<NodeStore>((set, get) => ({
108 5bb2712 Mardin 2y ago certificate: '',
109 5bb2712 Mardin 2y ago editingNode: null,
110 5bb2712 Mardin 2y ago deletingNode: null,
111 5bb2712 Mardin 2y ago isAddingNode: false,
112 5bb2712 Mardin 2y ago isEditingNode: false,
113 4205669 Mardin 2y ago isShowingNodesUsage: false,
114 4205669 Mardin 2y ago isEditingCore: false,
115 5bb2712 Mardin 2y ago nodesFilters: {
116 5bb2712 Mardin 2y ago name: '',
117 6aab6f0 Mrdn 2y ago limit: pageSizeManagers.nodes.getPageSize(),
118 5bb2712 Mardin 2y ago sort: '-created_at',
119 a47eb4c Dawsh 2y ago },
120 5bb2712 Mardin 2y ago onFilterChange: (filters) => {
121 5bb2712 Mardin 2y ago set({
122 5bb2712 Mardin 2y ago nodesFilters: {
123 5bb2712 Mardin 2y ago ...get().nodesFilters,
124 5bb2712 Mardin 2y ago ...filters,
125 5bb2712 Mardin 2y ago },
126 5bb2712 Mardin 2y ago });
127 5bb2712 Mardin 2y ago get().refetchNodes();
128 5bb2712 Mardin 2y ago },
129 5bb2712 Mardin 2y ago addNode: async (body) => {
130 5bb2712 Mardin 2y ago return fetch('/nodes', { method: 'POST', body }).then(() => { get().refetchCertificate() });
131 5bb2712 Mardin 2y ago },
132 5bb2712 Mardin 2y ago refetchNodes: () => {
133 a61ba53 Mrdn 2y ago queryClient.invalidateQueries(queryIds.nodes);
134 a47eb4c Dawsh 2y ago },
135 5bb2712 Mardin 2y ago refetchCertificate: () => {
136 5bb2712 Mardin 2y ago fetch('/nodes/settings').then((res) => {
137 5bb2712 Mardin 2y ago set({ certificate: res.certificate });
138 5bb2712 Mardin 2y ago })
139 5bb2712 Mardin 2y ago },
140 5bb2712 Mardin 2y ago fetchNodesUsage: (query: FilterUsageType) => {
141 bf7c131 Mardin 2y ago return fetch('/nodes/usage', { query });
142 a47eb4c Dawsh 2y ago },
143 5bb2712 Mardin 2y ago updateNode: (body) => {
144 5bb2712 Mardin 2y ago console.log('update: ' + body.id)
145 5bb2712 Mardin 2y ago return fetch(`/nodes/${body.id}`, {
146 bf7c131 Mardin 2y ago method: 'PUT',
147 a47eb4c Dawsh 2y ago body,
148 a47eb4c Dawsh 2y ago });
149 a47eb4c Dawsh 2y ago },
150 5bb2712 Mardin 2y ago onDeletingNode: (node: NodeCreate | undefined) => {
151 5bb2712 Mardin 2y ago if (node !== undefined) {
152 5bb2712 Mardin 2y ago set({ deletingNode: node });
153 5bb2712 Mardin 2y ago }
154 5bb2712 Mardin 2y ago else {
155 5bb2712 Mardin 2y ago set({ deletingNode: undefined });
156 5bb2712 Mardin 2y ago }
157 a47eb4c Dawsh 2y ago },
158 5bb2712 Mardin 2y ago reconnectNode: (body) => {
159 5bb2712 Mardin 2y ago return fetch(`/nodes/${body.id}/reconnect`, {
160 bf7c131 Mardin 2y ago method: 'POST',
161 a47eb4c Dawsh 2y ago });
162 a47eb4c Dawsh 2y ago },
163 a47eb4c Dawsh 2y ago deleteNode: () => {
164 5bb2712 Mardin 2y ago return fetch(`/nodes/${get().deletingNode?.id}`, {
165 bf7c131 Mardin 2y ago method: 'DELETE',
166 a47eb4c Dawsh 2y ago });
167 a47eb4c Dawsh 2y ago },
168 5bb2712 Mardin 2y ago onEditingNode: (editingNode: NodeType | undefined) => {
169 5bb2712 Mardin 2y ago if (editingNode !== undefined) {
170 5bb2712 Mardin 2y ago const { status, ...rest } = editingNode;
171 5bb2712 Mardin 2y ago set({ editingNode: { ...rest, status: (status === 'healthy' || status === 'unhealthy') ? 'none' : 'disabled' }, isEditingNode: true });
172 5bb2712 Mardin 2y ago }
173 5bb2712 Mardin 2y ago else {
174 5bb2712 Mardin 2y ago set({ editingNode: undefined, isEditingNode: false });
175 5bb2712 Mardin 2y ago
176 5bb2712 Mardin 2y ago }
177 5bb2712 Mardin 2y ago },
178 5bb2712 Mardin 2y ago onAddingNode: (isAddingNode: boolean) => {
179 5bb2712 Mardin 2y ago set({ isAddingNode });
180 4205669 Mardin 2y ago },
181 4205669 Mardin 2y ago onShowingNodesUsage: (isShowingNodesUsage: boolean) => {
182 4205669 Mardin 2y ago set({ isShowingNodesUsage });
183 4205669 Mardin 2y ago },
184 a47eb4c Dawsh 2y ago }));