DankMaterialShell/Services/SystemMonitorService.qml

import QtQuick
import Quickshell
import Quickshell.Io
pragma Singleton
pragma ComponentBehavior: Bound

Singleton {
    id: root
    
    property real cpuUsage: 0.0
    property int cpuCores: 1
    property string cpuModel: ""
    property real cpuFrequency: 0.0
    
    property var prevCpuStats: [0, 0, 0, 0, 0, 0, 0, 0]
    
    property real memoryUsage: 0.0
    property real totalMemory: 0.0
    property real usedMemory: 0.0
    property real freeMemory: 0.0
    property real availableMemory: 0.0
    property real bufferMemory: 0.0
    property real cacheMemory: 0.0
    
    property real cpuTemperature: 0.0
    
    property string kernelVersion: ""
    property string distribution: ""
    property string hostname: ""
    property string uptime: ""
    property string scheduler: ""
    property string architecture: ""
    property string loadAverage: ""
    property int processCount: 0
    property int threadCount: 0
    property string bootTime: ""
    property string motherboard: ""
    property string biosVersion: ""
    property var diskMounts: []
    property string diskUsage: ""
    
    property int cpuUpdateInterval: 3000
    property int memoryUpdateInterval: 5000
    property int temperatureUpdateInterval: 10000
    property int systemInfoUpdateInterval: 30000
    
    property bool enabledForTopBar: true
    property bool enabledForDetailedView: false

    Component.onCompleted: {
        console.log("SystemMonitorService: Starting initialization...")
        getCpuInfo()
        updateSystemStats()
        updateSystemInfo()
        console.log("SystemMonitorService: Initialization complete")
    }
    
    Process {
        id: cpuInfoProcess
        command: ["bash", "-c", "lscpu | grep -E 'Model name|CPU\\(s\\):' | head -2"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                const lines = text.trim().split('\n')
                for (const line of lines) {
                    if (line.includes("Model name")) {
                        root.cpuModel = line.split(":")[1].trim()
                    } else if (line.includes("CPU(s):")) {
                        root.cpuCores = parseInt(line.split(":")[1].trim())
                    }
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("CPU info check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: cpuUsageProcess
        command: ["bash", "-c", "head -1 /proc/stat | awk '{print $2,$3,$4,$5,$6,$7,$8,$9}'"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    const stats = text.trim().split(" ").map(x => parseInt(x))
                    if (root.prevCpuStats[0] > 0) {
                        let diffs = []
                        for (let i = 0; i < 8; i++) {
                            diffs[i] = stats[i] - root.prevCpuStats[i]
                        }
                        
                        const totalTime = diffs.reduce((a, b) => a + b, 0)
                        const idleTime = diffs[3] + diffs[4]
                        
                        if (totalTime > 0) {
                            root.cpuUsage = Math.max(0, Math.min(100, ((totalTime - idleTime) / totalTime) * 100))
                        }
                    }
                    root.prevCpuStats = stats
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("CPU usage check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: memoryUsageProcess
        command: ["bash", "-c", "free -m | awk 'NR==2{printf \"%.1f %.1f %.1f %.1f\", $3*100/$2, $2, $3, $7}'"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    const parts = text.trim().split(" ")
                    root.memoryUsage = parseFloat(parts[0])
                    root.totalMemory = parseFloat(parts[1])
                    root.usedMemory = parseFloat(parts[2])
                    root.availableMemory = parseFloat(parts[3])
                    root.freeMemory = root.totalMemory - root.usedMemory
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Memory usage check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: cpuFrequencyProcess
        command: ["bash", "-c", "cat /proc/cpuinfo | grep 'cpu MHz' | head -1 | awk '{print $4}'"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.cpuFrequency = parseFloat(text.trim())
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("CPU frequency check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: temperatureProcess
        command: ["bash", "-c", "if [ -f /sys/class/thermal/thermal_zone0/temp ]; then cat /sys/class/thermal/thermal_zone0/temp | awk '{print $1/1000}'; else sensors 2>/dev/null | grep 'Core 0' | awk '{print $3}' | sed 's/+//g;s/°C//g' | head -1; fi"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.cpuTemperature = parseFloat(text.trim())
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("CPU temperature check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: kernelInfoProcess
        command: ["bash", "-c", "uname -r"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.kernelVersion = text.trim()
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Kernel info check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: distributionProcess
        command: ["bash", "-c", "grep PRETTY_NAME /etc/os-release 2>/dev/null | cut -d'=' -f2 | tr -d '\"' || echo 'Unknown'"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.distribution = text.trim()
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Distribution check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: hostnameProcess
        command: ["bash", "-c", "hostname"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.hostname = text.trim()
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Hostname check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: uptimeProcess
        command: ["bash", "-c", "uptime -p | sed 's/up //'"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.uptime = text.trim()
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Uptime check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: schedulerProcess
        command: ["bash", "-c", "cat /sys/block/sda/queue/scheduler 2>/dev/null | grep -o '\\[.*\\]' | tr -d '[]' || echo 'Unknown'"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.scheduler = text.trim()
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Scheduler check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: architectureProcess
        command: ["bash", "-c", "uname -m"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.architecture = text.trim()
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Architecture check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: loadAverageProcess
        command: ["bash", "-c", "cat /proc/loadavg | cut -d' ' -f1,2,3"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.loadAverage = text.trim()
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Load average check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: processCountProcess
        command: ["bash", "-c", "ps aux | wc -l"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.processCount = parseInt(text.trim()) - 1
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Process count check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: threadCountProcess
        command: ["bash", "-c", "cat /proc/stat | grep processes | awk '{print $2}'"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.threadCount = parseInt(text.trim())
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Thread count check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: bootTimeProcess
        command: ["bash", "-c", "who -b | awk '{print $3, $4}' || stat -c %w /proc/1 2>/dev/null | cut -d' ' -f1,2 || echo 'Unknown'"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.bootTime = text.trim()
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Boot time check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: motherboardProcess
        command: ["bash", "-c", "if [ -r /sys/devices/virtual/dmi/id/board_vendor ] && [ -r /sys/devices/virtual/dmi/id/board_name ]; then echo \"$(cat /sys/devices/virtual/dmi/id/board_vendor 2>/dev/null) $(cat /sys/devices/virtual/dmi/id/board_name 2>/dev/null)\"; else echo 'Unknown'; fi"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.motherboard = text.trim()
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Motherboard check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: biosProcess
        command: ["bash", "-c", "if [ -r /sys/devices/virtual/dmi/id/bios_version ] && [ -r /sys/devices/virtual/dmi/id/bios_date ]; then echo \"$(cat /sys/devices/virtual/dmi/id/bios_version 2>/dev/null) $(cat /sys/devices/virtual/dmi/id/bios_date 2>/dev/null)\"; else echo 'Unknown'; fi"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    root.biosVersion = text.trim()
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("BIOS check failed with exit code:", exitCode)
            }
        }
    }
    
    Process {
        id: diskMountsProcess
        command: ["bash", "-c", "df -h --output=source,target,fstype,size,used,avail,pcent | tail -n +2 | grep -v tmpfs | grep -v devtmpfs | head -10"]
        running: false
        
        stdout: StdioCollector {
            onStreamFinished: {
                if (text.trim()) {
                    let mounts = []
                    const lines = text.trim().split('\n')
                    for (const line of lines) {
                        const parts = line.split(/\s+/)
                        if (parts.length >= 7) {
                            mounts.push({
                                device: parts[0],
                                mount: parts[1],
                                fstype: parts[2],
                                size: parts[3],
                                used: parts[4],
                                avail: parts[5],
                                percent: parts[6]
                            })
                        }
                    }
                    root.diskMounts = mounts
                }
            }
        }
        
        onExited: (exitCode) => {
            if (exitCode !== 0) {
                console.warn("Disk mounts check failed with exit code:", exitCode)
            }
        }
    }
    
    Timer {
        id: cpuTimer
        interval: root.cpuUpdateInterval
        running: root.enabledForTopBar || root.enabledForDetailedView
        repeat: true
        
        onTriggered: {
            if (root.enabledForTopBar || root.enabledForDetailedView) {
                cpuUsageProcess.running = true
                cpuFrequencyProcess.running = true
            }
        }
    }
    
    Timer {
        id: memoryTimer
        interval: root.memoryUpdateInterval
        running: root.enabledForTopBar || root.enabledForDetailedView
        repeat: true
        
        onTriggered: {
            if (root.enabledForTopBar || root.enabledForDetailedView) {
                memoryUsageProcess.running = true
            }
        }
    }
    
    Timer {
        id: temperatureTimer
        interval: root.temperatureUpdateInterval
        running: root.enabledForDetailedView
        repeat: true
        
        onTriggered: {
            if (root.enabledForDetailedView) {
                temperatureProcess.running = true
            }
        }
    }
    
    Timer {
        id: systemInfoTimer
        interval: root.systemInfoUpdateInterval
        running: root.enabledForDetailedView
        repeat: true
        
        onTriggered: {
            if (root.enabledForDetailedView) {
                updateSystemInfo()
            }
        }
    }
    
    function getCpuInfo() {
        cpuInfoProcess.running = true
    }
    
    function updateSystemStats() {
        if (root.enabledForTopBar || root.enabledForDetailedView) {
            cpuUsageProcess.running = true
            memoryUsageProcess.running = true
            cpuFrequencyProcess.running = true
            if (root.enabledForDetailedView) {
                temperatureProcess.running = true
            }
        }
    }
    
    function updateSystemInfo() {
        kernelInfoProcess.running = true
        distributionProcess.running = true
        hostnameProcess.running = true
        uptimeProcess.running = true
        schedulerProcess.running = true
        architectureProcess.running = true
        loadAverageProcess.running = true
        processCountProcess.running = true
        threadCountProcess.running = true
        bootTimeProcess.running = true
        motherboardProcess.running = true
        biosProcess.running = true
        diskMountsProcess.running = true
    }
    
    function enableTopBarMonitoring(enabled) {
        root.enabledForTopBar = enabled
    }
    
    function enableDetailedMonitoring(enabled) {
        root.enabledForDetailedView = enabled
    }
    
    function getCpuUsageColor() {
        if (cpuUsage > 80) return "#e74c3c"
        if (cpuUsage > 60) return "#f39c12"
        return "#27ae60"
    }
    
    function getMemoryUsageColor() {
        if (memoryUsage > 90) return "#e74c3c"
        if (memoryUsage > 75) return "#f39c12"
        return "#3498db"
    }
    
    function formatMemory(mb) {
        if (mb >= 1024) {
            return (mb / 1024).toFixed(1) + " GB"
        }
        return mb.toFixed(0) + " MB"
    }
    
    function getTemperatureColor() {
        if (cpuTemperature > 80) return "#e74c3c"
        if (cpuTemperature > 65) return "#f39c12"
        return "#27ae60"
    }
}