import getServer, { Server } from '@/api/server/getServer'; import { action, Action, computed, Computed, createContextStore, thunk, Thunk } from 'easy-peasy'; import socket, { SocketStore } from './socket'; import files, { ServerFileStore } from '@/state/server/files'; import subusers, { ServerSubuserStore } from '@/state/server/subusers'; import { composeWithDevTools } from 'redux-devtools-extension'; import schedules, { ServerScheduleStore } from '@/state/server/schedules'; import databases, { ServerDatabaseStore } from '@/state/server/databases'; import isEqual from 'react-fast-compare'; export type ServerStatus = 'offline' | 'starting' | 'stopping' | 'running' | null; interface ServerDataStore { data?: Server; inConflictState: Computed; isInstalling: Computed; permissions: string[]; getServer: Thunk, ServerStore, Promise>; setServer: Action; setServerFromState: Action Server>; setPermissions: Action; } const server: ServerDataStore = { permissions: [], inConflictState: computed((state) => { if (!state.data) { return false; } return state.data.status !== null || state.data.isTransferring || state.data.isNodeUnderMaintenance; }), isInstalling: computed((state) => { return state.data?.status === 'installing' || state.data?.status === 'install_failed'; }), getServer: thunk(async (actions, payload) => { const [server, permissions] = await getServer(payload); actions.setServer(server); actions.setPermissions(permissions); }), setServer: action((state, payload) => { if (!isEqual(payload, state.data)) { state.data = payload; } }), setServerFromState: action((state, payload) => { const output = payload(state.data!); if (!isEqual(output, state.data)) { state.data = output; } }), setPermissions: action((state, payload) => { if (!isEqual(payload, state.permissions)) { state.permissions = payload; } }), }; interface ServerStatusStore { value: ServerStatus; setServerStatus: Action; } const status: ServerStatusStore = { value: null, setServerStatus: action((state, payload) => { state.value = payload; }), }; export interface ServerStore { server: ServerDataStore; subusers: ServerSubuserStore; databases: ServerDatabaseStore; files: ServerFileStore; schedules: ServerScheduleStore; socket: SocketStore; status: ServerStatusStore; clearServerState: Action; } export const ServerContext = createContextStore( { server, socket, status, databases, files, subusers, schedules, clearServerState: action((state) => { state.server.data = undefined; state.server.permissions = []; state.databases.data = []; state.subusers.data = []; state.files.directory = '/'; state.files.selectedFiles = []; state.schedules.data = []; if (state.socket.instance) { state.socket.instance.removeAllListeners(); state.socket.instance.close(); } state.socket.instance = null; state.socket.connected = false; }), }, { compose: composeWithDevTools({ name: 'ServerStore', trace: true, }), } );