args-cli 0.1.1

Command-line interface for args
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

use std::process::Command;

use super::service::{Service, ServiceManager};
use crate::{config::runner::SYSTEM_USER, util::run_command};

pub fn install_service() -> anyhow::Result<()> {
    ensure_args_user()?;

    let manager = ServiceManager::new()?;

    println!("Removing any existing runner service...");
    manager.uninstall()?;

    println!("Installing runner service...");
    manager.install()?;
    println!("Runner installed.");

    Ok(())
}

fn user_exists(username: &str) -> bool {
    Command::new("id")
        .arg(username)
        .stdout(std::process::Stdio::null())
        .stderr(std::process::Stdio::null())
        .status()
        .map(|s| s.success())
        .unwrap_or(false)
}

fn ensure_args_user() -> anyhow::Result<()> {
    if user_exists(SYSTEM_USER) {
        return Ok(());
    }

    println!("Creating system user '{SYSTEM_USER}'...");

    #[cfg(target_os = "macos")]
    {
        let uid = find_free_uid_macos()?;
        let uid_str = uid.to_string();
        let user_path = format!("/Users/{SYSTEM_USER}");
        let cmds: &[&[&str]] = &[
            &["dscl", ".", "-create", &user_path],
            &["dscl", ".", "-create", &user_path, "UniqueID", &uid_str],
            &["dscl", ".", "-create", &user_path, "PrimaryGroupID", "20"],
            &[
                "dscl",
                ".",
                "-create",
                &user_path,
                "UserShell",
                "/usr/bin/false",
            ],
            &[
                "dscl",
                ".",
                "-create",
                &user_path,
                "RealName",
                "Args Runner",
            ],
            &[
                "dscl",
                ".",
                "-create",
                &user_path,
                "NFSHomeDirectory",
                "/var/empty",
            ],
        ];
        for cmd in cmds {
            run_command("sudo", cmd)?;
        }
    }

    #[cfg(not(target_os = "macos"))]
    {
        run_command(
            "sudo",
            &[
                "useradd",
                "--system",
                "--shell",
                "/usr/sbin/nologin",
                "--no-create-home",
                SYSTEM_USER,
            ],
        )?;
    }

    println!("System user '{SYSTEM_USER}' created.");
    Ok(())
}

#[cfg(target_os = "macos")]
fn find_free_uid_macos() -> anyhow::Result<u32> {
    for uid in 200u32..=400 {
        let out = Command::new("dscl")
            .args([".", "-search", "/Users", "UniqueID", &uid.to_string()])
            .output()?;
        if String::from_utf8_lossy(&out.stdout).trim().is_empty() {
            return Ok(uid);
        }
    }
    anyhow::bail!("No free UID found in range 200-400 for args user");
}