92 lines
2.6 KiB
TypeScript
92 lines
2.6 KiB
TypeScript
import { useEffect, useState } from 'react'
|
|
|
|
import { ExtensionTypeEnum, MonitoringExtension } from '@janhq/core'
|
|
|
|
import { useSetAtom } from 'jotai'
|
|
|
|
import { extensionManager } from '@/extension/ExtensionManager'
|
|
import {
|
|
availableRamAtom,
|
|
cpuUsageAtom,
|
|
totalRamAtom,
|
|
usedRamAtom,
|
|
nvidiaTotalVramAtom,
|
|
} from '@/helpers/atoms/SystemBar.atom'
|
|
|
|
export default function useGetSystemResources() {
|
|
const [ram, setRam] = useState<number>(0)
|
|
const [cpu, setCPU] = useState<number>(0)
|
|
|
|
const [gpus, setGPUs] = useState<Record<string, never>[]>([])
|
|
const setTotalRam = useSetAtom(totalRamAtom)
|
|
const setUsedRam = useSetAtom(usedRamAtom)
|
|
const setAvailableRam = useSetAtom(availableRamAtom)
|
|
const setCpuUsage = useSetAtom(cpuUsageAtom)
|
|
const setTotalNvidiaVram = useSetAtom(nvidiaTotalVramAtom)
|
|
|
|
const getSystemResources = async () => {
|
|
if (
|
|
!extensionManager.get<MonitoringExtension>(
|
|
ExtensionTypeEnum.SystemMonitoring
|
|
)
|
|
) {
|
|
return
|
|
}
|
|
const monitoring = extensionManager.get<MonitoringExtension>(
|
|
ExtensionTypeEnum.SystemMonitoring
|
|
)
|
|
const resourceInfor = await monitoring?.getResourcesInfo()
|
|
const currentLoadInfor = await monitoring?.getCurrentLoad()
|
|
|
|
const ram =
|
|
(resourceInfor?.mem?.usedMemory ?? 0) /
|
|
(resourceInfor?.mem?.totalMemory ?? 1)
|
|
if (resourceInfor?.mem?.usedMemory) setUsedRam(resourceInfor.mem.usedMemory)
|
|
if (resourceInfor?.mem?.totalMemory)
|
|
setTotalRam(resourceInfor.mem.totalMemory)
|
|
|
|
setRam(Math.round(ram * 100))
|
|
if (resourceInfor.mem.totalMemory && resourceInfor.mem.usedMemory)
|
|
setAvailableRam(
|
|
resourceInfor.mem.totalMemory - resourceInfor.mem.usedMemory
|
|
)
|
|
setCPU(Math.round(currentLoadInfor?.cpu?.usage ?? 0))
|
|
setCpuUsage(Math.round(currentLoadInfor?.cpu?.usage ?? 0))
|
|
|
|
const gpus = currentLoadInfor?.gpu ?? []
|
|
setGPUs(gpus)
|
|
|
|
let totalNvidiaVram = 0
|
|
if (gpus.length > 0) {
|
|
totalNvidiaVram = gpus.reduce(
|
|
(total: number, gpu: { memoryTotal: string }) =>
|
|
total + Number(gpu.memoryTotal),
|
|
0
|
|
)
|
|
}
|
|
setTotalNvidiaVram(totalNvidiaVram)
|
|
}
|
|
|
|
useEffect(() => {
|
|
getSystemResources()
|
|
|
|
// Fetch interval - every 2s
|
|
// TODO: Will we really need this?
|
|
// There is a possibility that this will be removed and replaced by the process event hook?
|
|
const intervalId = setInterval(() => {
|
|
getSystemResources()
|
|
}, 5000)
|
|
|
|
// clean up interval
|
|
return () => clearInterval(intervalId)
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
|
}, [])
|
|
|
|
return {
|
|
totalRamAtom,
|
|
ram,
|
|
cpu,
|
|
gpus,
|
|
}
|
|
}
|