jan/web-app/src/routes/settings/hardware.tsx

410 lines
13 KiB
TypeScript

import { createFileRoute } from '@tanstack/react-router'
import { route } from '@/constants/routes'
import SettingsMenu from '@/containers/SettingsMenu'
import HeaderPage from '@/containers/HeaderPage'
import { Card, CardItem } from '@/containers/Card'
import { Switch } from '@/components/ui/switch'
import { Progress } from '@/components/ui/progress'
import { useTranslation } from 'react-i18next'
import { useHardware } from '@/hooks/useHardware'
import { useVulkan } from '@/hooks/useVulkan'
import type { GPU, HardwareData } from '@/hooks/useHardware'
import { useEffect } from 'react'
import {
DndContext,
closestCenter,
KeyboardSensor,
PointerSensor,
useSensor,
useSensors,
DragEndEvent,
} from '@dnd-kit/core'
import {
SortableContext,
verticalListSortingStrategy,
useSortable,
} from '@dnd-kit/sortable'
import { CSS } from '@dnd-kit/utilities'
import {
IconGripVertical,
IconDeviceDesktopAnalytics,
} from '@tabler/icons-react'
import { getHardwareInfo } from '@/services/hardware'
import { WebviewWindow } from '@tauri-apps/api/webviewWindow'
import { formatMegaBytes } from '@/lib/utils'
import { windowKey } from '@/constants/windows'
// eslint-disable-next-line @typescript-eslint/no-explicit-any
export const Route = createFileRoute(route.settings.hardware as any)({
component: Hardware,
})
function SortableGPUItem({ gpu, index }: { gpu: GPU; index: number }) {
const {
attributes,
listeners,
setNodeRef,
transform,
transition,
isDragging,
} = useSortable({ id: gpu.id || index })
const { toggleGPUActivation, gpuLoading } = useHardware()
const style = {
transform: CSS.Transform.toString(transform),
transition,
opacity: isDragging ? 0.5 : 1,
position: 'relative' as const,
zIndex: isDragging ? 1 : 0,
}
return (
<div ref={setNodeRef} style={style} className="mb-4 last:mb-0">
<CardItem
title={
<div className="flex items-center gap-2">
<div
{...attributes}
{...listeners}
className="size-6 cursor-move flex items-center justify-center rounded hover:bg-main-view-fg/10 transition-all duration-200 ease-in-out"
>
<IconGripVertical size={18} className="text-main-view-fg/60" />
</div>
<span className="text-main-view-fg/80">{gpu.name}</span>
</div>
}
actions={
<div className="flex items-center gap-4">
<Switch
checked={gpu.activated}
disabled={!!gpuLoading[index]}
onCheckedChange={() => toggleGPUActivation(index)}
/>
</div>
}
/>
<div className="ml-8 mt-3">
<CardItem
title="VRAM"
actions={
<span className="text-main-view-fg/80">
{formatMegaBytes(gpu.free_vram)} free of{' '}
{formatMegaBytes(gpu.total_vram)}
</span>
}
/>
<CardItem
title="Driver Version"
actions={
<span className="text-main-view-fg/80">
{gpu.additional_information?.driver_version || '-'}
</span>
}
/>
<CardItem
title="Compute Capability"
actions={
<span className="text-main-view-fg/80">
{gpu.additional_information?.compute_cap || '-'}
</span>
}
/>
</div>
</div>
)
}
function Hardware() {
const { t } = useTranslation()
const {
hardwareData,
setHardwareData,
updateCPUUsage,
updateRAMAvailable,
reorderGPUs,
pollingPaused,
} = useHardware()
const { vulkanEnabled, setVulkanEnabled } = useVulkan()
useEffect(() => {
getHardwareInfo().then((data) =>
setHardwareData(data as unknown as HardwareData)
)
}, [setHardwareData])
// Set up DnD sensors
const sensors = useSensors(
useSensor(PointerSensor),
useSensor(KeyboardSensor)
)
// Handle drag end event
const handleDragEnd = (event: DragEndEvent) => {
const { active, over } = event
if (over && active.id !== over.id) {
// Find the indices of the dragged item and the drop target
const oldIndex = hardwareData.gpus.findIndex(
(gpu) => gpu.id === active.id
)
const newIndex = hardwareData.gpus.findIndex((gpu) => gpu.id === over.id)
if (oldIndex !== -1 && newIndex !== -1) {
reorderGPUs(oldIndex, newIndex)
}
}
}
useEffect(() => {
if (pollingPaused) return
const intervalId = setInterval(() => {
getHardwareInfo().then((data) => {
updateCPUUsage(data.cpu.usage)
updateRAMAvailable(data.ram.available)
})
}, 5000)
return () => clearInterval(intervalId)
}, [setHardwareData, updateCPUUsage, updateRAMAvailable, pollingPaused])
const handleClickSystemMonitor = async () => {
try {
// Check if system monitor window already exists
const existingWindow = await WebviewWindow.getByLabel(
windowKey.systemMonitorWindow
)
if (existingWindow) {
// If window exists, focus it
await existingWindow.setFocus()
console.log('Focused existing system monitor window')
} else {
// Create a new system monitor window
const monitorWindow = new WebviewWindow(windowKey.systemMonitorWindow, {
url: route.systemMonitor,
title: 'System Monitor - Jan',
width: 900,
height: 600,
resizable: true,
center: true,
})
// Listen for window creation
monitorWindow.once('tauri://created', () => {
console.log('System monitor window created')
})
// Listen for window errors
monitorWindow.once('tauri://error', (e) => {
console.error('Error creating system monitor window:', e)
})
}
} catch (error) {
console.error('Failed to open system monitor window:', error)
}
}
return (
<div className="flex flex-col h-full">
<HeaderPage>
<div className="flex items-center gap-2 justify-between w-full pr-3">
<h1 className="font-medium">{t('common.settings')}</h1>
<div
className="flex items-center gap-1 hover:bg-main-view-fg/8 px-1.5 py-0.5 rounded relative z-10 cursor-pointer"
onClick={handleClickSystemMonitor}
>
<IconDeviceDesktopAnalytics className="text-main-view-fg/50 size-5" />
<p>System monitor</p>
</div>
</div>
</HeaderPage>
<div className="flex h-full w-full">
<SettingsMenu />
<div className="p-4 w-full h-[calc(100%-32px)] overflow-y-auto">
<div className="flex flex-col justify-between gap-4 gap-y-3 w-full">
{/* OS Information */}
<Card title="Operating System">
<CardItem
title="Name"
actions={
<span className="text-main-view-fg/80">
{hardwareData.os?.name}
</span>
}
/>
<CardItem
title="Version"
actions={
<span className="text-main-view-fg/80">
{hardwareData.os?.version}
</span>
}
/>
</Card>
{/* CPU Information */}
<Card title="CPU">
<CardItem
title="Model"
actions={
<span className="text-main-view-fg/80">
{hardwareData.cpu?.model}
</span>
}
/>
<CardItem
title="Architecture"
actions={
<span className="text-main-view-fg/80">
{hardwareData.cpu?.arch}
</span>
}
/>
<CardItem
title="Cores"
actions={
<span className="text-main-view-fg/80">
{hardwareData.cpu?.cores}
</span>
}
/>
{hardwareData.cpu?.instructions.join(', ').length > 0 && (
<CardItem
title="Instructions"
column={hardwareData.cpu?.instructions.length > 6}
actions={
<span className="text-main-view-fg/80 break-words">
{hardwareData.cpu?.instructions?.join(', ')}
</span>
}
/>
)}
<CardItem
title="Usage"
actions={
<div className="flex items-center gap-2">
{hardwareData.cpu?.usage > 0 && (
<>
<Progress
value={hardwareData.cpu?.usage}
className="h-2 w-10"
/>
<span className="text-main-view-fg/80">
{hardwareData.cpu?.usage?.toFixed(2)}%
</span>
</>
)}
</div>
}
/>
</Card>
{/* RAM Information */}
<Card title="Memory">
<CardItem
title="Total RAM"
actions={
<span className="text-main-view-fg/80">
{formatMegaBytes(hardwareData.ram.total)}
</span>
}
/>
<CardItem
title="Available RAM"
actions={
<span className="text-main-view-fg/80">
{formatMegaBytes(hardwareData.ram?.available)}
</span>
}
/>
<CardItem
title="Usage"
actions={
<div className="flex items-center gap-2">
{hardwareData.ram?.total > 0 && (
<>
<Progress
value={
((hardwareData.ram?.total -
hardwareData.ram?.available) /
hardwareData.ram?.total) *
100
}
className="h-2 w-10"
/>
<span className="text-main-view-fg/80">
{(
((hardwareData.ram?.total -
hardwareData.ram?.available) /
hardwareData.ram?.total) *
100
).toFixed(2)}
%
</span>
</>
)}
</div>
}
/>
</Card>
{/* Vulkan Settings */}
{hardwareData.gpus.length > 0 && (
<Card title="Vulkan">
<CardItem
title="Enable Vulkan"
description="Use Vulkan API for GPU acceleration. Do not enable Vulkan if you have an NVIDIA GPU as it may cause compatibility issues."
actions={
<div className="flex items-center gap-4">
<Switch
checked={vulkanEnabled}
onCheckedChange={(checked) => {
setVulkanEnabled(checked)
setTimeout(() => {
window.location.reload()
}, 500) // Reload after 500ms to apply changes
}}
/>
</div>
}
/>
</Card>
)}
{/* GPU Information */}
{!IS_MACOS ? (
<Card title="GPUs">
{hardwareData.gpus.length > 0 ? (
<DndContext
sensors={sensors}
collisionDetection={closestCenter}
onDragEnd={handleDragEnd}
>
<SortableContext
items={hardwareData.gpus.map((gpu) => gpu.id)}
strategy={verticalListSortingStrategy}
>
{hardwareData.gpus.map((gpu, index) => (
<SortableGPUItem
key={gpu.id || index}
gpu={gpu}
index={index}
/>
))}
</SortableContext>
</DndContext>
) : (
<CardItem title="No GPUs detected" actions={<></>} />
)}
</Card>
) : (
<></>
)}
</div>
</div>
</div>
</div>
)
}