catfood-bar 0.3.0

A system bar component of the catfood utility suite
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

use ratatui::{prelude::Stylize, style::Color, text::Span};
use serde::Deserialize;
use std::process::Command;

use crate::logging;

#[derive(Deserialize, Debug)]
struct Workspace {
    id: i32,
}

#[derive(Debug, Default, Clone)]
pub struct Workspaces {
    pub workspaces: Vec<String>,
    pub active_workspace: String,
}

impl Workspaces {
    pub fn new() -> Self {
        Self {
            workspaces: get_workspaces().unwrap_or_default(),
            active_workspace: get_active_workspace().unwrap_or_default(),
        }
    }

    pub fn update(&mut self) {
        self.workspaces = get_workspaces().unwrap_or_default();
        self.active_workspace = get_active_workspace().unwrap_or_default();
    }

    pub fn render_as_spans(&self, colorize: bool) -> Vec<Span<'_>> {
        let rainbow_colors = [
            Color::Red,      // 1
            Color::Yellow,   // 2
            Color::Green,    // 3
            Color::Cyan,     // 4
            Color::Blue,     // 5
            Color::Magenta,  // 6
            Color::LightRed, // 7
        ];

        self.workspaces
            .iter()
            .map(|w| {
                if w == &self.active_workspace {
                    if colorize {
                        if let Ok(workspace_num) = w.parse::<usize>() {
                            let color_index = (workspace_num - 1) % rainbow_colors.len();
                            let bg_color = rainbow_colors[color_index];
                            // Use black text for better readability on all colored backgrounds
                            Span::raw(format!(" {} ", w)).bg(bg_color).fg(Color::Black)
                        } else {
                            // Fallback for non-numeric workspace names
                            Span::raw(format!(" {} ", w))
                                .bg(Color::White)
                                .fg(Color::Black)
                        }
                    } else {
                        // Non-colorized mode: black text on white background for active workspace
                        Span::raw(format!(" {} ", w))
                            .bg(Color::White)
                            .fg(Color::Black)
                    }
                } else if colorize {
                    if let Ok(workspace_num) = w.parse::<usize>() {
                        let color_index = (workspace_num - 1) % rainbow_colors.len();
                        let color = rainbow_colors[color_index];
                        Span::raw(format!(" {} ", w)).fg(color)
                    } else {
                        Span::raw(format!(" {} ", w))
                    }
                } else {
                    // Non-colorized mode: white text for non-active workspaces
                    Span::raw(format!(" {} ", w)).fg(Color::White)
                }
            })
            .collect::<Vec<Span>>()
    }
}

fn get_workspaces() -> Option<Vec<String>> {
    let output = Command::new("hyprctl")
        .args(["workspaces", "-j"])
        .output()
        .expect("failed to get workspaces");

    if output.status.success() {
        let stdout = str::from_utf8(&output.stdout).unwrap();
        let json: Vec<Workspace> =
            serde_json::from_str(stdout).expect("failed to parse workspaces");

        let mut workspaces: Vec<String> = json.iter().map(|j| j.id.to_string()).collect();
        workspaces.sort_by_key(|w| w.parse::<i32>().unwrap_or(0));
        return Some(workspaces);
    } else {
        logging::log_component_error(
            "WORKSPACES",
            str::from_utf8(&output.stderr).unwrap_or("unknown error"),
        );
    }

    None
}

fn get_active_workspace() -> Option<String> {
    let output = Command::new("hyprctl")
        .args(["activeworkspace", "-j"])
        .output()
        .expect("failed to get active workspace");

    if output.status.success() {
        let stdout = str::from_utf8(&output.stdout).unwrap();
        let json: Workspace =
            serde_json::from_str(stdout).expect("failed to parse active workspace");

        return Some(json.id.clone().to_string());
    } else {
        logging::log_component_error(
            "WORKSPACES",
            str::from_utf8(&output.stderr).unwrap_or("unknown error"),
        );
    }

    None
}