pkgs/by-name/ya/yambar-modules/src/main.rs


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

// nixos-config - My current NixOS configuration
//
// Copyright (C) 2025 Benedikt Peetz <benedikt.peetz@b-peetz.de>
// SPDX-License-Identifier: GPL-3.0-or-later
//
// This file is part of my nixos-config.
//
// You should have received a copy of the License along with this program.
// If not, see <https://www.gnu.org/licenses/gpl-3.0.txt>.

use std::{env::args, process};

mod cpu;
mod memory;

fn main() {
    let args: Vec<String> = args().collect();

    if args.len() != 2 {
        eprintln!("Usage: yambar-modules cpu|memory");
        process::exit(1);
    }

    match args[1].as_str() {
        "cpu" => {
            cpu::cpu();
        }
        "memory" => {
            memory::memory();
        }
        other => {
            eprintln!("'{other}' is not a valid command. Only 'cpu' or 'memory'.");
            process::exit(1);
        }
    }
}