wlan-cloud-ucentralgw-ui/src/hooks/Network/System.ts

import { useToast } from '@chakra-ui/react';
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
import * as axios from 'axios';
import { useTranslation } from 'react-i18next';
import { AxiosError } from 'models/Axios';

type System = {
  UI?: string;
  certificates?: { expiresOn: number; filename: string }[];
  hostname: string;
  os: string;
  processors: 16;
  start: number;
  uptime: number;
  version: string;
};

const axiosInstance = axios.default.create();

axiosInstance.defaults.timeout = 120000;
axiosInstance.defaults.headers.get.Accept = 'application/json';
axiosInstance.defaults.headers.post.Accept = 'application/json';

export const useGetSystemInfo = ({ endpoint, name, token }: { endpoint: string; name: string; token: string }) =>
  useQuery(
    ['get-system-info', name, endpoint],
    () =>
      axiosInstance
        .get(`${endpoint}/api/v1/system?command=info`, {
          headers: {
            Authorization: `Bearer ${token}`,
          },
        })
        .then(({ data }: { data: System }) => data),
    {
      staleTime: 60000,
    },
  );

export const useGetSubsystems = ({
  endpoint,
  enabled,
  name,
  token,
}: {
  endpoint: string;
  enabled: boolean;
  name: string;
  token: string;
}) =>
  useQuery(
    ['get-subsystems', name, endpoint],
    () =>
      axiosInstance
        .post(
          `${endpoint}/api/v1/system`,
          { command: 'getsubsystemnames' },
          {
            headers: {
              Authorization: `Bearer ${token}`,
            },
          },
        )
        .then(({ data }: { data: { list: string[] } }) => data.list ?? []),
    {
      staleTime: 60000,
      enabled,
    },
  );

export const useReloadSubsystems = ({
  endpoint,
  resetSubs,
  token,
}: {
  endpoint: string;
  resetSubs: () => void;
  token: string;
}) => {
  const { t } = useTranslation();
  const toast = useToast();

  return useMutation(
    (subsToReload: string[]) =>
      axiosInstance.post(
        `${endpoint}/api/v1/system`,
        { command: 'reload', subsystems: subsToReload },
        {
          headers: {
            Authorization: `Bearer ${token}`,
          },
        },
      ),
    {
      onSuccess: () => {
        toast({
          id: 'system-fetching-success',
          title: t('common.success'),
          description: t('system.success_reload'),
          status: 'success',
          duration: 5000,
          isClosable: true,
          position: 'top-right',
        });
        resetSubs();
      },
      onError: (e: AxiosError) => {
        toast({
          id: 'system-fetching-error',
          title: t('common.error'),
          description: t('crud.error_fetching_obj', {
            e: e?.response?.data?.ErrorDescription,
            obj: t('system.title'),
          }),
          status: 'error',
          duration: 5000,
          isClosable: true,
          position: 'top-right',
        });
      },
    },
  );
};

export const useGetSystemLogLevels = ({
  endpoint,
  enabled,
  token,
}: {
  endpoint: string;
  enabled: boolean;
  token: string;
}) =>
  useQuery(
    ['get-log-levels', endpoint],
    () =>
      axiosInstance
        .post(
          `${endpoint}/api/v1/system`,
          { command: 'getloglevels' },
          {
            headers: {
              Authorization: `Bearer ${token}`,
            },
          },
        )
        .then(({ data }: { data: { tagList: { tag: string; value: string }[] } }) => data.tagList ?? []),
    {
      staleTime: 60000,
      enabled,
    },
  );

export const useGetSystemLogLevelNames = ({
  endpoint,
  enabled,
  token,
}: {
  endpoint: string;
  enabled: boolean;
  token: string;
}) =>
  useQuery(
    ['get-log-level-names', endpoint],
    () =>
      axiosInstance
        .post(
          `${endpoint}/api/v1/system`,
          { command: 'getloglevelnames' },
          {
            headers: {
              Authorization: `Bearer ${token}`,
            },
          },
        )
        .then(({ data }: { data: { list: string[] } }) => data.list ?? []),
    {
      staleTime: 60000,
      enabled,
    },
  );

const changeLogLevel = (endpoint: string, token: string) => async (subsystems: { tag: string; value: string }[]) =>
  axiosInstance.post(
    `${endpoint}/api/v1/system`,
    { command: 'setloglevel', subsystems },
    {
      headers: {
        Authorization: `Bearer ${token}`,
      },
    },
  );
export const useUpdateSystemLogLevels = ({ endpoint, token }: { endpoint: string; token: string }) => {
  const queryClient = useQueryClient();

  return useMutation(changeLogLevel(endpoint, token), {
    onSuccess: () => {
      queryClient.invalidateQueries(['get-log-levels', endpoint]);
    },
  });
};