lsofrs 4.1.0

//! Aggregate FD summary/statistics mode

use std::collections::HashMap;
use std::io::{self, Write};

use crossterm::event::KeyEvent;
use ratatui::buffer::Buffer;
use ratatui::layout::Rect;
use ratatui::style::{Modifier, Style};

use crate::filter::Filter;
use crate::theme::LsofTheme;
use crate::tui_app::{TuiMode, TuiState, set_str};
use crate::types::*;

// Keep the old Theme import for single-shot print_summary
use crate::output::Theme;

const TOP_N: usize = 10;
const BAR_MAX: usize = 20;

struct TypeStats {
    type_name: String,
    count: usize,
}

struct ProcStats {
    pid: i32,
    command: String,
    uid: u32,
    fd_count: usize,
}

struct UserStats {
    uid: u32,
    username: String,
    proc_count: usize,
    file_count: usize,
}

pub fn print_summary(procs: &[Process], theme: &Theme, json_output: bool) {
    let (type_stats, proc_stats, user_stats, total_files) = compute_stats(procs);

    if json_output {
        print_summary_json(
            procs.len(),
            total_files,
            &type_stats,
            &proc_stats,
            &user_stats,
        );
    } else {
        print_summary_text(
            procs.len(),
            total_files,
            &type_stats,
            &proc_stats,
            &user_stats,
            theme,
        );
    }
}

fn compute_stats(procs: &[Process]) -> (Vec<TypeStats>, Vec<ProcStats>, Vec<UserStats>, usize) {
    let mut type_map: HashMap<String, usize> = HashMap::new();
    let mut user_map: HashMap<u32, (String, usize, usize)> = HashMap::new();
    let mut proc_stats: Vec<ProcStats> = Vec::new();
    let mut total_files = 0usize;

    for p in procs {
        let fd_count = p.files.len();
        total_files += fd_count;

        for f in &p.files {
            *type_map
                .entry(f.file_type.as_str().to_string())
                .or_insert(0) += 1;
        }

        let entry = user_map
            .entry(p.uid)
            .or_insert_with(|| (p.username(), 0, 0));
        entry.1 += 1;
        entry.2 += fd_count;

        proc_stats.push(ProcStats {
            pid: p.pid,
            command: p.command.clone(),
            uid: p.uid,
            fd_count,
        });
    }

    proc_stats.sort_by(|a, b| b.fd_count.cmp(&a.fd_count).then(a.pid.cmp(&b.pid)));

    let mut type_stats: Vec<TypeStats> = type_map
        .into_iter()
        .map(|(type_name, count)| TypeStats { type_name, count })
        .collect();
    type_stats.sort_by(|a, b| b.count.cmp(&a.count).then(a.type_name.cmp(&b.type_name)));

    let mut user_stats: Vec<UserStats> = user_map
        .into_iter()
        .map(|(uid, (username, proc_count, file_count))| UserStats {
            uid,
            username,
            proc_count,
            file_count,
        })
        .collect();
    user_stats.sort_by(|a, b| b.file_count.cmp(&a.file_count).then(a.uid.cmp(&b.uid)));

    (type_stats, proc_stats, user_stats, total_files)
}

/// Render summary text into a String buffer (for single-shot non-live mode)
#[allow(clippy::too_many_arguments)]
fn render_summary_text(
    total_procs: usize,
    total_files: usize,
    types: &[TypeStats],
    procs: &[ProcStats],
    users: &[UserStats],
    theme: &Theme,
    iteration: Option<u64>,
    interval: Option<u64>,
) -> String {
    use std::fmt::Write as FmtWrite;
    let mut buf = String::with_capacity(2048);
    let r = theme.reset();

    if let (Some(iter), Some(int)) = (iteration, interval) {
        let _ = writeln!(
            buf,
            "{bold}{hdr} lsofrs summary -- refresh {int}s -- #{iter} -- q to quit {r}",
            bold = theme.bold(),
            hdr = theme.hdr_bg(),
        );
    } else {
        let _ = writeln!(buf, "{bold}=== lsofrs summary ==={r}", bold = theme.bold());
    }
    let _ = writeln!(buf);
    let _ = writeln!(
        buf,
        "  {cyan}Processes:{r} {:<10}  {yellow}Open files:{r} {}",
        fmt_num(total_procs),
        fmt_num(total_files),
        cyan = theme.cyan(),
        yellow = theme.yellow(),
    );

    let _ = writeln!(
        buf,
        "\n{bold}-- File type breakdown --{r}\n",
        bold = theme.bold()
    );

    let max_count = types.first().map(|t| t.count).unwrap_or(1);
    for (i, ts) in types.iter().enumerate() {
        if i >= 15 {
            let other: usize = types[i..].iter().map(|t| t.count).sum();
            let pct = if total_files > 0 {
                other as f64 / total_files as f64 * 100.0
            } else {
                0.0
            };
            let bar_len = (other as f64 / max_count as f64 * BAR_MAX as f64) as usize;
            let _ = writeln!(
                buf,
                "  {dim}(other){r}  {:>8}  {:>5.1}%  {green}{}{r}",
                fmt_num(other),
                pct,
                "█".repeat(bar_len),
                dim = theme.dim(),
                green = theme.green(),
            );
            break;
        }
        let pct = if total_files > 0 {
            ts.count as f64 / total_files as f64 * 100.0
        } else {
            0.0
        };
        let bar_len = (ts.count as f64 / max_count as f64 * BAR_MAX as f64) as usize;
        let _ = writeln!(
            buf,
            "  {blue}{:<6}{r}  {:>8}  {:>5.1}%  {green}{}{r}",
            ts.type_name,
            fmt_num(ts.count),
            pct,
            "█".repeat(bar_len),
            blue = theme.blue(),
            green = theme.green(),
        );
    }

    let _ = writeln!(
        buf,
        "\n{bold}-- Top {TOP_N} processes by FD count --{r}\n",
        bold = theme.bold()
    );
    let _ = writeln!(
        buf,
        "  {hdr}{bold}{:>7}  {:<15}  {:<8}  {:>8}{r}",
        "PID",
        "COMMAND",
        "USER",
        "FDs",
        hdr = theme.hdr_bg(),
        bold = theme.bold(),
    );
    for ps in procs.iter().take(TOP_N) {
        let username = users::get_user_by_uid(ps.uid)
            .map(|u| u.name().to_string_lossy().into_owned())
            .unwrap_or_else(|| ps.uid.to_string());
        let cmd = if ps.command.len() > 15 {
            &ps.command[..15]
        } else {
            &ps.command
        };
        let user = if username.len() > 8 {
            &username[..8]
        } else {
            &username
        };
        let _ = writeln!(
            buf,
            "  {mag}{:>7}{r}  {cyan}{:<15}{r}  {yellow}{:<8}{r}  {:>8}",
            ps.pid,
            cmd,
            user,
            fmt_num(ps.fd_count),
            mag = theme.magenta(),
            cyan = theme.cyan(),
            yellow = theme.yellow(),
        );
    }

    let _ = writeln!(
        buf,
        "\n{bold}-- Per-user totals --{r}\n",
        bold = theme.bold()
    );
    let _ = writeln!(
        buf,
        "  {hdr}{bold}{:<10}  {:>8}  {:>8}{r}",
        "USER",
        "PROCS",
        "FILES",
        hdr = theme.hdr_bg(),
        bold = theme.bold(),
    );
    for us in users.iter().take(20) {
        let uname = if us.username.len() > 10 {
            &us.username[..10]
        } else {
            &us.username
        };
        let _ = writeln!(
            buf,
            "  {yellow}{:<10}{r}  {:>8}  {:>8}",
            uname,
            fmt_num(us.proc_count),
            fmt_num(us.file_count),
            yellow = theme.yellow(),
        );
    }

    let _ = writeln!(buf);
    buf
}

fn print_summary_text(
    total_procs: usize,
    total_files: usize,
    types: &[TypeStats],
    procs: &[ProcStats],
    users: &[UserStats],
    theme: &Theme,
) {
    let buf = render_summary_text(
        total_procs,
        total_files,
        types,
        procs,
        users,
        theme,
        None,
        None,
    );
    let out = io::stdout();
    let mut out = out.lock();
    let _ = out.write_all(buf.as_bytes());
    let _ = out.flush();
}

/// Live summary mode struct implementing TuiMode
pub struct SummaryLiveMode {
    type_stats: Vec<TypeStats>,
    proc_stats: Vec<ProcStats>,
    user_stats: Vec<UserStats>,
    total_procs: usize,
    total_files: usize,
}

impl SummaryLiveMode {
    pub fn new() -> Self {
        Self {
            type_stats: Vec::new(),
            proc_stats: Vec::new(),
            user_stats: Vec::new(),
            total_procs: 0,
            total_files: 0,
        }
    }

    /// Build tooltip for a row in the summary view.
    /// Layout: row 0 = summary line, row 1 = blank, row 2 = section header "type breakdown",
    /// row 3 = blank, rows 4..4+types = type rows, then blank, section header "top procs",
    /// blank, header row, proc rows, blank, section header "per-user", blank, header, user rows.
    pub fn get_tooltip_lines(&self, content_row: usize) -> Vec<(String, String)> {
        // Row 0: summary line (Processes: N, Open files: N)
        // Row 1: blank
        // Row 2: "-- File type breakdown --"
        // Row 3: blank
        // Rows 4..4+type_count: type entries
        let type_count = self.type_stats.len().min(15);
        let type_start = 4;
        let type_end = type_start + type_count;

        if content_row >= type_start && content_row < type_end {
            let idx = content_row - type_start;
            if let Some(ts) = self.type_stats.get(idx) {
                let pct = if self.total_files > 0 {
                    ts.count as f64 / self.total_files as f64 * 100.0
                } else {
                    0.0
                };
                return vec![
                    ("Type".into(), ts.type_name.clone()),
                    ("Count".into(), ts.count.to_string()),
                    ("Percentage".into(), format!("{pct:.1}%")),
                ];
            }
        }

        // After types: blank, section header, blank, header row, then proc rows
        let proc_section_start = type_end + 1; // blank
        // proc_section_start: blank line
        // proc_section_start + 1: section header
        // proc_section_start + 2: blank
        // proc_section_start + 3: column header row
        let proc_data_start = proc_section_start + 3;
        let proc_count = self.proc_stats.len().min(TOP_N);
        let proc_end = proc_data_start + proc_count;

        if content_row >= proc_data_start && content_row < proc_end {
            let idx = content_row - proc_data_start;
            if let Some(ps) = self.proc_stats.get(idx) {
                let username = users::get_user_by_uid(ps.uid)
                    .map(|u| u.name().to_string_lossy().into_owned())
                    .unwrap_or_else(|| ps.uid.to_string());
                return vec![
                    ("PID".into(), ps.pid.to_string()),
                    ("Command".into(), ps.command.clone()),
                    ("User".into(), username),
                    ("FD Count".into(), ps.fd_count.to_string()),
                ];
            }
        }

        // After procs: blank, section header, blank, header row, user rows
        let user_section_start = proc_end + 1; // blank
        let user_data_start = user_section_start + 3;
        let user_count = self.user_stats.len().min(20);
        let user_end = user_data_start + user_count;

        if content_row >= user_data_start && content_row < user_end {
            let idx = content_row - user_data_start;
            if let Some(us) = self.user_stats.get(idx) {
                return vec![
                    ("User".into(), us.username.clone()),
                    ("UID".into(), us.uid.to_string()),
                    ("Processes".into(), us.proc_count.to_string()),
                    ("Files".into(), us.file_count.to_string()),
                ];
            }
        }

        vec![]
    }

    /// Total data row count for display purposes.
    pub fn data_row_count(&self) -> usize {
        self.type_stats.len().min(15)
            + self.proc_stats.len().min(TOP_N)
            + self.user_stats.len().min(20)
    }
}

impl TuiMode for SummaryLiveMode {
    fn update(&mut self, filter: &Filter) {
        let mut procs = crate::gather_processes();
        procs.retain(|p| filter.matches_process(p));
        for p in &mut procs {
            p.files.retain(|f| filter.matches_file(f));
        }

        let (type_stats, proc_stats, user_stats, total_files) = compute_stats(&procs);
        self.type_stats = type_stats;
        self.proc_stats = proc_stats;
        self.user_stats = user_stats;
        self.total_procs = procs.len();
        self.total_files = total_files;
    }

    fn render(&self, buf: &mut Buffer, area: Rect, theme: &LsofTheme, _state: &TuiState) {
        render_summary_ratatui(
            buf,
            area,
            theme,
            self.total_procs,
            self.total_files,
            &self.type_stats,
            &self.proc_stats,
            &self.user_stats,
        );
    }

    fn handle_key(&mut self, _key: KeyEvent, _state: &mut TuiState) -> bool {
        false
    }

    fn title(&self) -> &str {
        "summary"
    }

    fn help_keys(&self) -> Vec<(&str, &str)> {
        vec![]
    }
}

/// Live summary mode — auto-refresh via TUI framework
pub fn run_summary_live(filter: &Filter, interval: u64, theme: &LsofTheme) {
    let mut mode = SummaryLiveMode::new();
    crate::tui_app::run_tui(&mut mode, filter, interval, theme);
}

/// Render summary directly to a ratatui buffer using theme styles.
#[allow(clippy::too_many_arguments)]
fn render_summary_ratatui(
    buf: &mut Buffer,
    area: Rect,
    theme: &LsofTheme,
    total_procs: usize,
    total_files: usize,
    types: &[TypeStats],
    procs: &[ProcStats],
    users: &[UserStats],
) {
    let bold_s = Style::default()
        .fg(theme.bold_fg)
        .add_modifier(Modifier::BOLD);
    let dim_s = Style::default().fg(theme.dim_fg);
    let cmd_s = Style::default().fg(theme.cmd_fg);
    let pid_s = Style::default().fg(theme.pid_fg);
    let user_s = Style::default().fg(theme.user_fg);
    let type_s = Style::default().fg(theme.type_fg);
    let fd_s = Style::default().fg(theme.fd_fg);
    let hdr_s = Style::default()
        .fg(theme.header_fg)
        .bg(theme.header_bg)
        .add_modifier(Modifier::BOLD);
    let section_s = Style::default()
        .fg(theme.section_fg)
        .add_modifier(Modifier::BOLD);

    let mut row = area.y;
    let cx = area.x + 2;
    let w = area.width;

    // Summary line
    if row < area.y + area.height {
        let info = format!(
            "Processes: {}    Open files: {}",
            fmt_num(total_procs),
            fmt_num(total_files),
        );
        set_str(buf, cx, row, &info, bold_s, w);
        row += 2;
    }

    // Type breakdown
    if row < area.y + area.height {
        set_str(buf, cx, row, "-- File type breakdown --", section_s, w);
        row += 2;
    }

    let max_count = types.first().map(|t| t.count).unwrap_or(1);
    for (i, ts) in types.iter().enumerate() {
        if row >= area.y + area.height || i >= 15 {
            break;
        }
        let pct = if total_files > 0 {
            ts.count as f64 / total_files as f64 * 100.0
        } else {
            0.0
        };
        let bar_len = (ts.count as f64 / max_count as f64 * BAR_MAX as f64) as usize;
        set_str(buf, cx, row, &ts.type_name, type_s, 6);
        let nums = format!("  {:>8}  {:>5.1}%  ", fmt_num(ts.count), pct);
        set_str(buf, cx + 6, row, &nums, dim_s, 20);
        let bar = "█".repeat(bar_len);
        set_str(buf, cx + 26, row, &bar, fd_s, bar_len as u16);
        row += 1;
    }

    row += 1;

    // Top processes
    if row < area.y + area.height {
        let title = format!("-- Top {} processes by FD count --", TOP_N);
        set_str(buf, cx, row, &title, section_s, w);
        row += 2;
    }

    if row < area.y + area.height {
        for x in area.x..area.x + area.width {
            let c = &mut buf[(x.min(area.x + area.width - 1), row)];
            c.set_style(hdr_s);
        }
        let hdr = format!(
            "{:>7}  {:<15}  {:<8}  {:>8}",
            "PID", "COMMAND", "USER", "FDs"
        );
        set_str(buf, cx, row, &hdr, hdr_s, w);
        row += 1;
    }

    for ps in procs.iter().take(TOP_N) {
        if row >= area.y + area.height {
            break;
        }
        let username = users::get_user_by_uid(ps.uid)
            .map(|u| u.name().to_string_lossy().into_owned())
            .unwrap_or_else(|| ps.uid.to_string());
        let cmd = if ps.command.len() > 15 {
            &ps.command[..15]
        } else {
            &ps.command
        };
        let user = if username.len() > 8 {
            &username[..8]
        } else {
            &username
        };

        let pid_str = format!("{:>7}", ps.pid);
        set_str(buf, cx, row, &pid_str, pid_s, 7);
        let cmd_str = format!("  {:<15}", cmd);
        set_str(buf, cx + 7, row, &cmd_str, cmd_s, 17);
        let user_str = format!("  {:<8}", user);
        set_str(buf, cx + 24, row, &user_str, user_s, 10);
        let fd_str = format!("  {:>8}", fmt_num(ps.fd_count));
        set_str(buf, cx + 34, row, &fd_str, bold_s, 10);
        row += 1;
    }

    row += 1;

    // Per-user totals
    if row < area.y + area.height {
        set_str(buf, cx, row, "-- Per-user totals --", section_s, w);
        row += 2;
    }

    if row < area.y + area.height {
        for x in area.x..area.x + area.width {
            let c = &mut buf[(x.min(area.x + area.width - 1), row)];
            c.set_style(hdr_s);
        }
        let hdr = format!("{:<10}  {:>8}  {:>8}", "USER", "PROCS", "FILES");
        set_str(buf, cx, row, &hdr, hdr_s, w);
        row += 1;
    }

    for us in users.iter().take(20) {
        if row >= area.y + area.height {
            break;
        }
        let uname = if us.username.len() > 10 {
            &us.username[..10]
        } else {
            &us.username
        };
        let user_str = format!("{:<10}", uname);
        set_str(buf, cx, row, &user_str, user_s, 10);
        let nums = format!(
            "  {:>8}  {:>8}",
            fmt_num(us.proc_count),
            fmt_num(us.file_count)
        );
        set_str(buf, cx + 10, row, &nums, dim_s, 20);
        row += 1;
    }
}

fn print_summary_json(
    total_procs: usize,
    total_files: usize,
    types: &[TypeStats],
    procs: &[ProcStats],
    users: &[UserStats],
) {
    let out = io::stdout();
    let mut out = out.lock();

    let types_obj: serde_json::Value = types
        .iter()
        .map(|t| (t.type_name.clone(), serde_json::json!(t.count)))
        .collect::<serde_json::Map<String, serde_json::Value>>()
        .into();

    let top_procs: Vec<serde_json::Value> = procs
        .iter()
        .take(TOP_N)
        .map(|p| {
            serde_json::json!({
                "pid": p.pid,
                "command": p.command,
                "uid": p.uid,
                "fd_count": p.fd_count,
            })
        })
        .collect();

    let users_arr: Vec<serde_json::Value> = users
        .iter()
        .map(|u| {
            serde_json::json!({
                "uid": u.uid,
                "username": u.username,
                "proc_count": u.proc_count,
                "file_count": u.file_count,
            })
        })
        .collect();

    let summary = serde_json::json!({
        "summary": {
            "total_files": total_files,
            "total_processes": total_procs,
            "types": types_obj,
            "top_processes": top_procs,
            "users": users_arr,
        }
    });

    let _ = serde_json::to_writer_pretty(&mut out, &summary);
    let _ = writeln!(out);
}

fn fmt_num(n: usize) -> String {
    let s = n.to_string();
    let mut result = String::new();
    for (i, c) in s.chars().rev().enumerate() {
        if i > 0 && i % 3 == 0 {
            result.push(',');
        }
        result.push(c);
    }
    result.chars().rev().collect()
}

#[cfg(test)]
mod tests {
    use super::*;

    fn make_proc(pid: i32, cmd: &str, uid: u32, files: Vec<OpenFile>) -> Process {
        Process {
            pid,
            ppid: 1,
            pgid: pid,
            uid,
            command: cmd.to_string(),
            files,
            sel_flags: 0,
            sel_state: 0,
        }
    }

    fn make_file(ft: FileType) -> OpenFile {
        OpenFile {
            fd: FdName::Number(3),
            access: Access::Read,
            file_type: ft,
            name: "/tmp/x".to_string(),
            ..Default::default()
        }
    }

    #[test]
    fn print_summary_empty() {
        let theme = Theme::new(false);
        print_summary(&[], &theme, false);
    }

    #[test]
    fn print_summary_json_empty() {
        let theme = Theme::new(false);
        print_summary(&[], &theme, true);
    }

    #[test]
    fn print_summary_with_data() {
        let theme = Theme::new(false);
        let procs = vec![
            make_proc(
                1,
                "init",
                0,
                vec![make_file(FileType::Reg), make_file(FileType::Dir)],
            ),
            make_proc(
                2,
                "bash",
                501,
                vec![make_file(FileType::Reg), make_file(FileType::Pipe)],
            ),
        ];
        print_summary(&procs, &theme, false);
    }

    #[test]
    fn print_summary_json_with_data() {
        let theme = Theme::new(false);
        let procs = vec![
            make_proc(1, "init", 0, vec![make_file(FileType::IPv4)]),
            make_proc(
                2,
                "nginx",
                33,
                vec![make_file(FileType::IPv4), make_file(FileType::IPv4)],
            ),
        ];
        print_summary(&procs, &theme, true);
    }

    #[test]
    fn fmt_num_small() {
        assert_eq!(fmt_num(0), "0");
        assert_eq!(fmt_num(42), "42");
        assert_eq!(fmt_num(999), "999");
    }

    #[test]
    fn fmt_num_thousands() {
        assert_eq!(fmt_num(1000), "1,000");
        assert_eq!(fmt_num(12345), "12,345");
        assert_eq!(fmt_num(1234567), "1,234,567");
    }

    #[test]
    fn render_summary_ratatui_empty() {
        let theme = LsofTheme::from_name(crate::theme::ThemeName::NeonSprawl);
        let area = Rect::new(0, 0, 100, 40);
        let mut buf = Buffer::empty(area);
        render_summary_ratatui(&mut buf, area, &theme, 0, 0, &[], &[], &[]);
    }

    #[test]
    fn summary_live_mode_title() {
        let mode = SummaryLiveMode::new();
        assert_eq!(mode.title(), "summary");
    }

    #[test]
    fn summary_live_mode_help_keys_empty() {
        let mode = SummaryLiveMode::new();
        assert!(mode.help_keys().is_empty());
    }
}