ComputerSurge/komodo
3
0
Fork 0
forked from github-starred/komodo
Code Pull Requests 1 Actions Packages Releases Activity

cli 0.1.8 start periphery with systemd

Browse Source
This commit is contained in:
mbecker20
2023-01-11 22:08:16 +00:00
parent 34822d6d19
commit 572560d5b2
4 changed files with 78 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cli/Cargo.toml
View File

@@ -1,6 +1,6 @@
[package]
name = "monitor_cli"
version = "0.1.7"
version = "0.1.8"
edition = "2021"
authors = ["MoghTech"]
description = "monitor cli | tools to setup monitor system"

84
cli/src/helpers.rs
View File

@@ -307,6 +307,63 @@ pub fn gen_periphery_config(sub_matches: &ArgMatches) {
);
}
pub fn start_periphery_systemd(sub_matches: &ArgMatches) {
let config_path = sub_matches
.get_one::<String>("config-path")
.map(|p| p.as_str())
.unwrap_or("~/.monitor/periphery.config.toml")
.to_string();
println!(
"\n========================\n {} \n========================\n",
"periphery config".bold()
);
println!("{}: {config_path}", "config path".dimmed());
println!(
"\npress {} to start {}. {}",
"ENTER".green().bold(),
"monitor periphery".bold(),
"(ctrl-c to cancel)".dimmed()
);
let buffer = &mut [0u8];
let res = std::io::stdin().read_exact(buffer);
if res.is_err() {
println!("pressed another button, exiting");
}
println!("\ninstalling periphery binary...\n");
let install_output = run_command_pipe_to_terminal(&format!("cargo install {PERIPHERY_CRATE}"));
if install_output.success() {
println!("\ninstallation finished, starting monitor periphery...\n")
} else {
eprintln!(
"\n❌ there was some {} during periphery installation ❌\n",
"error".red()
);
return;
}
gen_periphery_service_file(&config_path);
let command = format!("systemctl --user daemon-reload && systemctl --user enable --now periphery");
let output = run_command_pipe_to_terminal(&command);
if output.success() {
println!(
"\n✅ {} has been started up ✅\n",
"monitor periphery".bold()
)
} else {
eprintln!("\n❌ there was some {} on startup ❌\n", "error".red())
}
}
pub fn start_periphery_daemon(sub_matches: &ArgMatches) {
let config_path = sub_matches
.get_one::<String>("config-path")
@@ -452,22 +509,13 @@ pub fn start_periphery_container(sub_matches: &ArgMatches) {
}
}
pub fn gen_periphery_service_file(sub_matches: &ArgMatches) {
let mut file = File::create("/etc/systemd/system/periphery.service")
.expect("failed to create config file. user must be root for this operation.");
file.write_all(generate_periphery_unit_file().as_bytes())
.expect("failed to write config file. user must be root for this operation");
let output = run_command_pipe_to_terminal("systemctl daemon-reload");
if output.success() {
println!("\n✅ successfully added service to systemd ✅\n")
} else {
eprintln!(
"\n❌ there was some {} adding service to systemd ❌\n",
"error".red()
)
}
pub fn gen_periphery_service_file(config_path: &str) {
let home = env::var("HOME").expect("failed to find $HOME env var");
let _ = std::fs::create_dir_all(format!("{home}/.config/systemd/user"));
let mut file = File::create(format!("{home}/.config/systemd/user/periphery.service"))
.expect("failed to create user systemd unit file");
file.write_all(periphery_unit_file(config_path).as_bytes())
.expect("failed to write config file");
}
fn write_to_toml(path: &str, toml: impl Serialize) {
@@ -495,7 +543,7 @@ fn generate_secret(length: usize) -> String {
.collect()
}
fn generate_periphery_unit_file() -> String {
fn periphery_unit_file(config_path: &str) -> String {
let home = env::var("HOME").expect("failed to find $HOME env var");
let user = env::var("USER").expect("failed to find $USER env var");
format!("[Unit]
@@ -506,7 +554,7 @@ After=network.target
Type=simple
User={user}
WorkingDirectory={home}
ExecStart=/bin/bash --login -c 'source {home}/.bashrc; $HOME/.cargo/bin/periphery --config-path ~/.monitor/periphery.config.toml --home-dir $HOME'
ExecStart=/bin/bash --login -c 'source {home}/.bashrc; $HOME/.cargo/bin/periphery --config-path {config_path} --home-dir $HOME'
TimeoutStartSec=0
[Install]

15
cli/src/main.rs
View File

@@ -135,13 +135,18 @@ fn cli() -> Command {
arg!(--repo-dir <PATH> "if running in container, this should be '/repos'. default is ~/.monitor/repos").required(false)
)
)
// .subcommand(
// Command::new("gen-service-file")
// .about("generate a periphery systemd service file")
// )
.subcommand(
Command::new("start")
.about("tools to start periphery as daemon or container")
.subcommand(
Command::new("systemd")
.about("manage periphery with systemd running under current user")
.arg(
arg!(--config-path <PATH> "specify the file path to use for config. default is ~/.monitor/periphery.config.toml")
.required(false)
)
)
.subcommand(
Command::new("daemon")
.about("start up monitor periphery daemon")
@@ -206,10 +211,10 @@ fn main() {
);
match periphery_command {
("gen-config", sub_matches) => gen_periphery_config(sub_matches),
// ("gen-service-file", sub_matches) => gen_periphery_service_file(sub_matches),
("start", sub_matches) => {
let periphery_start_command = sub_matches.subcommand().expect("\n❌ invalid call, should be 'monitor periphery start <daemon, container> <flags>' ❌\n");
match periphery_start_command {
("systemd", sub_matches) => start_periphery_systemd(sub_matches),
("daemon", sub_matches) => start_periphery_daemon(sub_matches),
("container", sub_matches) => start_periphery_container(sub_matches),
_ => println!("\n❌ invalid call, should be 'monitor periphery start <daemon, container> <flags>' ❌\n")