portainer/app/react/edge/edge-devices/WaitingRoomView/queries.ts

55 lines
1.8 KiB
TypeScript

import { useMutation, useQueryClient } from 'react-query';
import { EdgeTypes, EnvironmentId } from '@/react/portainer/environments/types';
import axios, { parseAxiosError } from '@/portainer/services/axios';
import { promiseSequence } from '@/portainer/helpers/promise-utils';
import { useIntegratedLicenseInfo } from '@/react/portainer/licenses/use-license.service';
import { useEnvironmentList } from '@/react/portainer/environments/queries';
import {
mutationOptions,
withError,
withInvalidate,
} from '@/react-tools/react-query';
import { queryKey as nodesCountQueryKey } from '@/react/portainer/system/useNodesCount';
import { LicenseType } from '@/react/portainer/licenses/types';
import { queryKeys } from '@/react/portainer/environments/queries/query-keys';
export function useAssociateDeviceMutation() {
const queryClient = useQueryClient();
return useMutation(
(ids: EnvironmentId[]) =>
promiseSequence(ids.map((id) => () => associateDevice(id))),
mutationOptions(
withError('Failed to associate devices'),
withInvalidate(queryClient, [queryKeys.base(), nodesCountQueryKey])
)
);
}
async function associateDevice(environmentId: EnvironmentId) {
try {
await axios.post(`/endpoints/${environmentId}/edge/trust`);
} catch (e) {
throw parseAxiosError(e as Error, 'Failed to associate device');
}
}
export function useLicenseOverused(moreNodes: number) {
const integratedInfo = useIntegratedLicenseInfo();
return (
!!integratedInfo &&
integratedInfo.licenseInfo.type === LicenseType.Essentials &&
integratedInfo.usedNodes + moreNodes > integratedInfo.licenseInfo.nodes
);
}
export function useUntrustedCount() {
const query = useEnvironmentList({
edgeDeviceUntrusted: true,
types: EdgeTypes,
});
return query.totalCount;
}