archy/core/archipelago/src/api/rpc/system.rs

use super::RpcHandler;
use anyhow::{Context, Result};
use tracing::debug;

impl RpcHandler {
    /// system.stats — CPU usage, RAM used/total, disk used/total, uptime, load average
    pub(super) async fn handle_system_stats(&self) -> Result<serde_json::Value> {
        debug!("Getting system stats");

        let uptime = read_uptime().await.unwrap_or(0.0);
        let load = read_loadavg().await.unwrap_or((0.0, 0.0, 0.0));
        let cpu = read_cpu_usage().await.unwrap_or(0.0);
        let (mem_used, mem_total) = read_meminfo().await.unwrap_or((0, 0));
        let (disk_used, disk_total) = read_disk_usage().await.unwrap_or((0, 0));

        Ok(serde_json::json!({
            "uptime_secs": uptime as u64,
            "load_avg_1": load.0,
            "load_avg_5": load.1,
            "load_avg_15": load.2,
            "cpu_usage_percent": cpu,
            "mem_used_bytes": mem_used,
            "mem_total_bytes": mem_total,
            "disk_used_bytes": disk_used,
            "disk_total_bytes": disk_total,
        }))
    }

    /// system.processes — top 10 processes by CPU
    pub(super) async fn handle_system_processes(&self) -> Result<serde_json::Value> {
        debug!("Getting top processes");

        let procs = read_top_processes().await.unwrap_or_default();

        Ok(serde_json::json!({ "processes": procs }))
    }

    /// system.temperature — thermal zone readings
    pub(super) async fn handle_system_temperature(&self) -> Result<serde_json::Value> {
        debug!("Getting system temperature");

        let temps = read_temperatures().await.unwrap_or_default();

        Ok(serde_json::json!({ "temperatures": temps }))
    }
}

/// Read system uptime from /proc/uptime (seconds since boot).
async fn read_uptime() -> Result<f64> {
    let content = tokio::fs::read_to_string("/proc/uptime")
        .await
        .context("Failed to read /proc/uptime")?;
    let uptime: f64 = content
        .split_whitespace()
        .next()
        .ok_or_else(|| anyhow::anyhow!("Empty /proc/uptime"))?
        .parse()
        .context("Failed to parse uptime")?;
    Ok(uptime)
}

/// Read load averages from /proc/loadavg.
async fn read_loadavg() -> Result<(f64, f64, f64)> {
    let content = tokio::fs::read_to_string("/proc/loadavg")
        .await
        .context("Failed to read /proc/loadavg")?;
    let mut parts = content.split_whitespace();
    let l1: f64 = parts
        .next()
        .ok_or_else(|| anyhow::anyhow!("Missing load1"))?
        .parse()
        .context("parse load1")?;
    let l5: f64 = parts
        .next()
        .ok_or_else(|| anyhow::anyhow!("Missing load5"))?
        .parse()
        .context("parse load5")?;
    let l15: f64 = parts
        .next()
        .ok_or_else(|| anyhow::anyhow!("Missing load15"))?
        .parse()
        .context("parse load15")?;
    Ok((l1, l5, l15))
}

/// Compute CPU usage by sampling /proc/stat twice with a 250ms gap.
async fn read_cpu_usage() -> Result<f64> {
    let snap1 = read_cpu_jiffies().await?;
    tokio::time::sleep(std::time::Duration::from_millis(250)).await;
    let snap2 = read_cpu_jiffies().await?;

    let total_delta = snap2.total.saturating_sub(snap1.total);
    let idle_delta = snap2.idle.saturating_sub(snap1.idle);

    if total_delta == 0 {
        return Ok(0.0);
    }

    let usage = 100.0 * (1.0 - (idle_delta as f64 / total_delta as f64));
    Ok((usage * 10.0).round() / 10.0) // one decimal
}

struct CpuJiffies {
    total: u64,
    idle: u64,
}

async fn read_cpu_jiffies() -> Result<CpuJiffies> {
    let content = tokio::fs::read_to_string("/proc/stat")
        .await
        .context("Failed to read /proc/stat")?;
    let cpu_line = content
        .lines()
        .next()
        .ok_or_else(|| anyhow::anyhow!("Empty /proc/stat"))?;
    // cpu  user nice system idle iowait irq softirq steal guest guest_nice
    let vals: Vec<u64> = cpu_line
        .split_whitespace()
        .skip(1) // skip "cpu"
        .filter_map(|v| v.parse().ok())
        .collect();
    if vals.len() < 4 {
        anyhow::bail!("Not enough fields in /proc/stat cpu line");
    }
    let idle = vals[3]; // idle column
    let total: u64 = vals.iter().sum();
    Ok(CpuJiffies { total, idle })
}

/// Read memory info from /proc/meminfo.
/// Returns (used_bytes, total_bytes).
async fn read_meminfo() -> Result<(u64, u64)> {
    let content = tokio::fs::read_to_string("/proc/meminfo")
        .await
        .context("Failed to read /proc/meminfo")?;

    let mut total_kb: u64 = 0;
    let mut available_kb: u64 = 0;

    for line in content.lines() {
        if let Some(val) = line.strip_prefix("MemTotal:") {
            total_kb = parse_meminfo_kb(val)?;
        } else if let Some(val) = line.strip_prefix("MemAvailable:") {
            available_kb = parse_meminfo_kb(val)?;
        }
    }

    let used_bytes = total_kb.saturating_sub(available_kb) * 1024;
    let total_bytes = total_kb * 1024;
    Ok((used_bytes, total_bytes))
}

fn parse_meminfo_kb(val: &str) -> Result<u64> {
    val.trim()
        .trim_end_matches("kB")
        .trim()
        .parse::<u64>()
        .context("parse meminfo value")
}

/// Read disk usage via `df` for the root filesystem.
/// Returns (used_bytes, total_bytes).
async fn read_disk_usage() -> Result<(u64, u64)> {
    let output = tokio::process::Command::new("df")
        .args(["--block-size=1", "--output=used,size", "/"])
        .output()
        .await
        .context("Failed to run df")?;

    if !output.status.success() {
        anyhow::bail!("df failed: {}", String::from_utf8_lossy(&output.stderr));
    }

    let stdout = String::from_utf8(output.stdout).context("df output not utf8")?;
    // Skip header line
    let data_line = stdout
        .lines()
        .nth(1)
        .ok_or_else(|| anyhow::anyhow!("No data line from df"))?;
    let mut parts = data_line.split_whitespace();
    let used: u64 = parts
        .next()
        .ok_or_else(|| anyhow::anyhow!("Missing used"))?
        .parse()
        .context("parse df used")?;
    let total: u64 = parts
        .next()
        .ok_or_else(|| anyhow::anyhow!("Missing total"))?
        .parse()
        .context("parse df total")?;

    Ok((used, total))
}

/// Read top 10 processes by CPU from `ps`.
async fn read_top_processes() -> Result<Vec<serde_json::Value>> {
    let output = tokio::process::Command::new("ps")
        .args(["--no-headers", "-eo", "pid,%cpu,%mem,comm", "--sort=-%cpu"])
        .output()
        .await
        .context("Failed to run ps")?;

    if !output.status.success() {
        anyhow::bail!("ps failed: {}", String::from_utf8_lossy(&output.stderr));
    }

    let stdout = String::from_utf8(output.stdout).context("ps output not utf8")?;
    let procs: Vec<serde_json::Value> = stdout
        .lines()
        .take(10)
        .filter_map(|line| {
            let mut parts = line.split_whitespace();
            let pid = parts.next()?.parse::<u32>().ok()?;
            let cpu: f64 = parts.next()?.parse().ok()?;
            let mem: f64 = parts.next()?.parse().ok()?;
            let name = parts.collect::<Vec<_>>().join(" ");
            Some(serde_json::json!({
                "pid": pid,
                "cpu_percent": cpu,
                "mem_percent": mem,
                "name": name,
            }))
        })
        .collect();

    Ok(procs)
}

/// Read temperatures from /sys/class/thermal/thermal_zone*/temp.
async fn read_temperatures() -> Result<Vec<serde_json::Value>> {
    let mut temps = Vec::new();
    let thermal_dir = std::path::Path::new("/sys/class/thermal");
    if !thermal_dir.exists() {
        return Ok(temps);
    }

    let mut entries = tokio::fs::read_dir(thermal_dir)
        .await
        .context("Failed to read /sys/class/thermal")?;

    while let Some(entry) = entries.next_entry().await? {
        let name = entry.file_name();
        let name_str = name.to_string_lossy();
        if !name_str.starts_with("thermal_zone") {
            continue;
        }

        let temp_path = entry.path().join("temp");
        let type_path = entry.path().join("type");

        let millideg = match tokio::fs::read_to_string(&temp_path).await {
            Ok(s) => s.trim().parse::<i64>().unwrap_or(0),
            Err(_) => continue,
        };

        let zone_type = tokio::fs::read_to_string(&type_path)
            .await
            .map(|s| s.trim().to_string())
            .unwrap_or_else(|_| name_str.to_string());

        temps.push(serde_json::json!({
            "zone": zone_type,
            "temp_celsius": millideg as f64 / 1000.0,
        }));
    }

    Ok(temps)
}