mod cli;
mod config;
mod error;
mod platforms;
mod server;
mod session;
use clap::Parser;
use cli::*;
use color_eyre::eyre::{Result, WrapErr};
use config::handle_server_arg;
fn main() -> Result<()> {
color_eyre::install()?;
let args = Cli::parse();
let config_dir = config::get_directory()?;
let mut config = config::load_or_create(config_dir)?;
match args.command {
Commands::Attach { server } => {
session::attach(handle_server_arg(server, &config)?, &config)
.wrap_err("Failed to attach to session session")?
}
Commands::Config => {
println!("{:#?}", config)
}
Commands::Default { action } => match action {
DefaultCommands::Get => {
if let Some(default_server) = &config.default_server {
println!("\"{default_server}\"")
} else {
println!("None")
}
}
DefaultCommands::Set { server } => config.default_server = Some(server),
},
Commands::DeleteAllSessions { force } => if force {
session::delete_all()
} else {
session::delete_all_confirmed()
}
.wrap_err("Failed to delete all sessions")?,
Commands::DeleteSession { session, force } => {
session::delete_server_session(handle_server_arg(session, &config)?, force)
.wrap_err("Failed to delete session")?
}
Commands::Deploy { server } => {
let server = handle_server_arg(server, &config)?;
session::new_server(
&server,
Some(server::get_command(&server, &config)?),
&config,
)?;
}
Commands::Execute { server, commands } => {
let session_name = session::get_name(handle_server_arg(server, &config)?);
for command in commands {
session::write_line(&session_name, command)?;
}
}
Commands::List {
active,
inactive,
dead,
} => {
let mut servers = vec![];
server::for_each(|s| servers.push(server::ServerObject::new(s)), &config)
.wrap_err("Failed to get servers")?;
if active {
server::retain_active(&mut servers).wrap_err("Failed to retain active servers")?;
} else if inactive {
server::retain_and_tag_inactive(&mut servers, &config)
.wrap_err("Failed to retain inactive servers")?;
if dead {
server::tag_dead(&mut servers).wrap_err("Failed to tag dead servers")?;
}
} else if dead {
server::retain_and_tag_dead(&mut servers, &config)
.wrap_err("Failed to retain dead servers")?;
} else {
server::fully_tag_servers(&mut servers, &config)
.wrap_err("Failed to tag active servers")?;
}
for server in servers {
println!("{server}");
}
}
Commands::Rcon { server, commands } => {
server::rcon(handle_server_arg(server, &config)?, commands, &config)
.wrap_err("Failed to run rcon command")?
}
Commands::New {
platform,
version,
name,
} => server::create_new(platform, version, name, &config)
.wrap_err(format!("Failed to create {platform} server"))?,
Commands::Remove { servers, force } => if force {
server::remove_servers(servers, &config)
} else {
server::remove_servers_with_confirmation(servers, &config)
}
.wrap_err("Failed to remove server")?,
Commands::Restart => server::restart(&config).wrap_err("Failed to restart server")?,
Commands::Stop { server } => {
let server = handle_server_arg(server, &config)?;
server::rcon(&server, vec!["stop"], &config)
.wrap_err_with(|| format!("Failed to stop server {}", server))?;
}
Commands::Template { action } => match action {
TemplateCommands::New { server } => server::new_template(&server, &config)
.wrap_err_with(|| format!("Failed to create template with server {server}"))?,
TemplateCommands::From { template, server } => {
server::from_template(&template, server.as_deref(), &config)
.wrap_err_with(|| format!("Failed to use template {template}"))?
}
},
Commands::Reinstall {
git,
commit,
path,
from_crate,
} => {
if let Some(path) = path {
server::reinstall_with_path(&path)
.wrap_err(format!("Failed to update package with {}", path.display()))?
} else if git {
server::reinstall_with_git(commit)
.wrap_err("Failed to update package with git repo")?
} else if from_crate {
server::reinstall_with_crate().wrap_err("Failed to update package with crate")?
} else {
unreachable!("Clap ensures git or a path is provided")
}
}
Commands::Update {
server,
platform,
version,
} => server::update_existing(server, platform, version, &config)
.wrap_err("Failed to update server")?,
};
Ok(())
}