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.2.1
Source
1 16e5ea1 Mrdn 2y ago import { useQuery } from '@tanstack/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 512d183 Mrdn 2y ago export async 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 5bb2712 Mardin 2y ago nodesFilters: NodesFilterType;
83 5bb2712 Mardin 2y ago refetchNodes: () => void;
84 5bb2712 Mardin 2y ago refetchCertificate: () => void;
85 a47eb4c Dawsh 2y ago fetchNodesUsage: (query: FilterUsageType) => Promise<void>;
86 5bb2712 Mardin 2y ago addNode: (node: NodeCreate) => Promise<unknown>;
87 5bb2712 Mardin 2y ago deleteNode: (node: NodeCreate) => Promise<unknown>;
88 5bb2712 Mardin 2y ago updateNode: (node: NodeCreate) => Promise<unknown>;
89 a47eb4c Dawsh 2y ago reconnectNode: (node: NodeType) => Promise<unknown>;
90 5bb2712 Mardin 2y ago onDeletingNode: (node: NodeCreate | undefined) => void;
91 5bb2712 Mardin 2y ago onEditingNode: (node: NodeType | undefined) => void;
92 5bb2712 Mardin 2y ago onAddingNode: (isAddingNode: boolean) => void;
93 5bb2712 Mardin 2y ago onFilterChange: (filters: Partial<NodesFilterType>) => void;
94 4205669 Mardin 2y ago onShowingNodesUsage: (isShowingNodesUsage: boolean) => void;
95 a47eb4c Dawsh 2y ago };
96 a47eb4c Dawsh 2y ago
97 16e5ea1 Mrdn 2y ago // TODO: Refactor all useQuery for inbounds, hosts,
98 16e5ea1 Mrdn 2y ago // services and uses to be a custom hook like this.
99 a47eb4c Dawsh 2y ago export const useNodesQuery = () => {
100 a47eb4c Dawsh 2y ago return useQuery({
101 16e5ea1 Mrdn 2y ago queryKey: [FetchNodesQueryKey],
102 5bb2712 Mardin 2y ago queryFn: fetchNodes,
103 5bb2712 Mardin 2y ago refetchInterval: useNodes.getState().editingNode ? 3000 : undefined,
104 a47eb4c Dawsh 2y ago refetchOnWindowFocus: false,
105 a47eb4c Dawsh 2y ago });
106 a47eb4c Dawsh 2y ago };
107 a47eb4c Dawsh 2y ago
108 a47eb4c Dawsh 2y ago export const useNodes = create<NodeStore>((set, get) => ({
109 5bb2712 Mardin 2y ago certificate: '',
110 5bb2712 Mardin 2y ago editingNode: null,
111 5bb2712 Mardin 2y ago deletingNode: null,
112 5bb2712 Mardin 2y ago isAddingNode: false,
113 5bb2712 Mardin 2y ago isEditingNode: false,
114 4205669 Mardin 2y ago isShowingNodesUsage: false,
115 4205669 Mardin 2y ago isEditingCore: false,
116 5bb2712 Mardin 2y ago nodesFilters: {
117 5bb2712 Mardin 2y ago name: '',
118 6aab6f0 Mrdn 2y ago limit: pageSizeManagers.nodes.getPageSize(),
119 5bb2712 Mardin 2y ago sort: '-created_at',
120 a47eb4c Dawsh 2y ago },
121 5bb2712 Mardin 2y ago onFilterChange: (filters) => {
122 5bb2712 Mardin 2y ago set({
123 5bb2712 Mardin 2y ago nodesFilters: {
124 5bb2712 Mardin 2y ago ...get().nodesFilters,
125 5bb2712 Mardin 2y ago ...filters,
126 5bb2712 Mardin 2y ago },
127 5bb2712 Mardin 2y ago });
128 5bb2712 Mardin 2y ago get().refetchNodes();
129 5bb2712 Mardin 2y ago },
130 5bb2712 Mardin 2y ago addNode: async (body) => {
131 5bb2712 Mardin 2y ago return fetch('/nodes', { method: 'POST', body }).then(() => { get().refetchCertificate() });
132 5bb2712 Mardin 2y ago },
133 5bb2712 Mardin 2y ago refetchNodes: () => {
134 16e5ea1 Mrdn 2y ago queryClient.invalidateQueries({ queryKey: [queryIds.nodes] });
135 a47eb4c Dawsh 2y ago },
136 5bb2712 Mardin 2y ago refetchCertificate: () => {
137 5bb2712 Mardin 2y ago fetch('/nodes/settings').then((res) => {
138 5bb2712 Mardin 2y ago set({ certificate: res.certificate });
139 5bb2712 Mardin 2y ago })
140 5bb2712 Mardin 2y ago },
141 5bb2712 Mardin 2y ago fetchNodesUsage: (query: FilterUsageType) => {
142 bf7c131 Mardin 2y ago return fetch('/nodes/usage', { query });
143 a47eb4c Dawsh 2y ago },
144 5bb2712 Mardin 2y ago updateNode: (body) => {
145 5bb2712 Mardin 2y ago console.log('update: ' + body.id)
146 5bb2712 Mardin 2y ago return fetch(`/nodes/${body.id}`, {
147 bf7c131 Mardin 2y ago method: 'PUT',
148 a47eb4c Dawsh 2y ago body,
149 a47eb4c Dawsh 2y ago });
150 a47eb4c Dawsh 2y ago },
151 5bb2712 Mardin 2y ago onDeletingNode: (node: NodeCreate | undefined) => {
152 5bb2712 Mardin 2y ago if (node !== undefined) {
153 5bb2712 Mardin 2y ago set({ deletingNode: node });
154 5bb2712 Mardin 2y ago }
155 5bb2712 Mardin 2y ago else {
156 5bb2712 Mardin 2y ago set({ deletingNode: undefined });
157 5bb2712 Mardin 2y ago }
158 a47eb4c Dawsh 2y ago },
159 5bb2712 Mardin 2y ago reconnectNode: (body) => {
160 5bb2712 Mardin 2y ago return fetch(`/nodes/${body.id}/reconnect`, {
161 bf7c131 Mardin 2y ago method: 'POST',
162 a47eb4c Dawsh 2y ago });
163 a47eb4c Dawsh 2y ago },
164 a47eb4c Dawsh 2y ago deleteNode: () => {
165 5bb2712 Mardin 2y ago return fetch(`/nodes/${get().deletingNode?.id}`, {
166 bf7c131 Mardin 2y ago method: 'DELETE',
167 a47eb4c Dawsh 2y ago });
168 a47eb4c Dawsh 2y ago },
169 5bb2712 Mardin 2y ago onEditingNode: (editingNode: NodeType | undefined) => {
170 5bb2712 Mardin 2y ago if (editingNode !== undefined) {
171 5bb2712 Mardin 2y ago const { status, ...rest } = editingNode;
172 5bb2712 Mardin 2y ago set({ editingNode: { ...rest, status: (status === 'healthy' || status === 'unhealthy') ? 'none' : 'disabled' }, isEditingNode: true });
173 5bb2712 Mardin 2y ago }
174 5bb2712 Mardin 2y ago else {
175 5bb2712 Mardin 2y ago set({ editingNode: undefined, isEditingNode: false });
176 5bb2712 Mardin 2y ago
177 5bb2712 Mardin 2y ago }
178 5bb2712 Mardin 2y ago },
179 5bb2712 Mardin 2y ago onAddingNode: (isAddingNode: boolean) => {
180 5bb2712 Mardin 2y ago set({ isAddingNode });
181 4205669 Mardin 2y ago },
182 4205669 Mardin 2y ago onShowingNodesUsage: (isShowingNodesUsage: boolean) => {
183 4205669 Mardin 2y ago set({ isShowingNodesUsage });
184 4205669 Mardin 2y ago },
185 a47eb4c Dawsh 2y ago }));