ao-cli 0.1.12

A unified administration tool for Linux systems
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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142

use super::arch::Pacman;
use super::debian::Apt;
use super::fedora::Dnf;
use super::linux_generic::{
    StandardBoot, StandardDev, StandardDisk, StandardDistro, StandardGroup, StandardGui,
    StandardLog, StandardMonitor, StandardNet, StandardSec, StandardSelf, StandardSys,
    StandardUser, StandardVirt, Systemd,
};
use super::{
    BootManager, DevManager, DiskManager, DistroManager, Domain, GroupManager, GuiManager,
    LogManager, MonitorManager, NetManager, PackageManager, SecManager, SelfManager,
    ServiceManager, SysManager, UserManager, VirtManager,
};
use anyhow::{Result, bail};
use std::fs::File;
use std::io::{BufRead, BufReader};

pub struct DetectedSystem {
    pub pkg: Box<dyn PackageManager>,
    pub svc: Box<dyn ServiceManager>,
    pub net: Box<dyn NetManager>,
    pub dev: Box<dyn DevManager>,
    pub virt: Box<dyn VirtManager>,
    pub sec: Box<dyn SecManager>,
    pub boot: Box<dyn BootManager>,
    pub gui: Box<dyn GuiManager>,
    pub user: Box<dyn UserManager>,
    pub group: Box<dyn GroupManager>,
    pub disk: Box<dyn DiskManager>,
    pub sys: Box<dyn SysManager>,
    pub log: Box<dyn LogManager>,
    pub distro: Box<dyn DistroManager>,
    pub monitor: Box<dyn MonitorManager>,
    pub self_manager: Box<dyn SelfManager>,
}

impl DetectedSystem {
    pub fn domains(&self) -> Vec<&dyn Domain> {
        vec![
            self.pkg.as_ref(),
            self.svc.as_ref(),
            self.net.as_ref(),
            self.dev.as_ref(),
            self.virt.as_ref(),
            self.sec.as_ref(),
            self.boot.as_ref(),
            self.gui.as_ref(),
            self.user.as_ref(),
            self.group.as_ref(),
            self.disk.as_ref(),
            self.sys.as_ref(),
            self.log.as_ref(),
            self.distro.as_ref(),
            self.monitor.as_ref(),
            self.self_manager.as_ref(),
        ]
    }
}

pub fn detect_system() -> Result<DetectedSystem> {
    // Read /etc/os-release to determine the distribution
    let mut os_release = String::new();
    if let Ok(file) = File::open("/etc/os-release") {
        let reader = BufReader::new(file);
        for l in reader.lines().map_while(Result::ok) {
            os_release.push_str(&l);
            os_release.push('\n');
        }
    }

    let is_debian_based = os_release.contains("ID=ubuntu")
        || os_release.contains("ID=debian")
        || os_release.contains("ID_LIKE=debian");
    let is_arch_based = os_release.contains("ID=arch") || os_release.contains("ID_LIKE=arch");
    let is_fedora_based = os_release.contains("ID=fedora") || os_release.contains("ID_LIKE=fedora");

    if is_debian_based {
        return Ok(DetectedSystem {
            pkg: Box::new(Apt),
            svc: Box::new(Systemd),
            net: Box::new(StandardNet),
            dev: Box::new(StandardDev),
            virt: Box::new(StandardVirt),
            sec: Box::new(StandardSec),
            boot: Box::new(StandardBoot),
            gui: Box::new(StandardGui),
            user: Box::new(StandardUser),
            group: Box::new(StandardGroup),
            disk: Box::new(StandardDisk),
            sys: Box::new(StandardSys),
            log: Box::new(StandardLog),
            distro: Box::new(StandardDistro),
            monitor: Box::new(StandardMonitor),
            self_manager: Box::new(StandardSelf),
        });
    }

    if is_arch_based {
        return Ok(DetectedSystem {
            pkg: Box::new(Pacman),
            svc: Box::new(Systemd),
            net: Box::new(StandardNet),
            dev: Box::new(StandardDev),
            virt: Box::new(StandardVirt),
            sec: Box::new(StandardSec),
            boot: Box::new(StandardBoot),
            gui: Box::new(StandardGui),
            user: Box::new(StandardUser),
            group: Box::new(StandardGroup),
            disk: Box::new(StandardDisk),
            sys: Box::new(StandardSys),
            log: Box::new(StandardLog),
            distro: Box::new(StandardDistro),
            monitor: Box::new(StandardMonitor),
            self_manager: Box::new(StandardSelf),
        });
    }

    if is_fedora_based {
        return Ok(DetectedSystem {
            pkg: Box::new(Dnf),
            svc: Box::new(Systemd),
            net: Box::new(StandardNet),
            dev: Box::new(StandardDev),
            virt: Box::new(StandardVirt),
            sec: Box::new(StandardSec),
            boot: Box::new(StandardBoot),
            gui: Box::new(StandardGui),
            user: Box::new(StandardUser),
            group: Box::new(StandardGroup),
            disk: Box::new(StandardDisk),
            sys: Box::new(StandardSys),
            log: Box::new(StandardLog),
            distro: Box::new(StandardDistro),
            monitor: Box::new(StandardMonitor),
            self_manager: Box::new(StandardSelf),
        });
    }

    // Default or panic if unsupported
    bail!("Unsupported operating system. Detected: \n{}", os_release)
}