70 lines
2.6 KiB
Rust
70 lines
2.6 KiB
Rust
use chrono::Utc;
|
|
use clap::{ArgMatches, Command, arg};
|
|
use crate::cli::env_settings;
|
|
use crate::status::ServerStatus;
|
|
use crate::utils::{ validate_name, extract_archive, move_all_elements};
|
|
|
|
use diesel::prelude::*;
|
|
use crate::{DbPool, models}; use crate::schema;
|
|
|
|
use tokio::{fs::{File, create_dir}, io::AsyncWriteExt};
|
|
|
|
use std::path::Path;
|
|
|
|
|
|
pub fn create_subcommand() -> Command {
|
|
let mut arguments = vec![
|
|
arg!(<NAME> "Name of the server"),
|
|
arg!([SERVER_ARCHIVE] "initial server archive (supported : prism zip)"),
|
|
];
|
|
|
|
arguments.extend(env_settings::get_args());
|
|
|
|
Command::new("add")
|
|
.about("Add a server")
|
|
.args(arguments.as_slice())
|
|
}
|
|
|
|
|
|
pub async fn subcommand(arg: &ArgMatches, pool: DbPool) -> Result<(), String>{
|
|
use schema::servers::dsl::*;
|
|
let config = crate::Config::load();
|
|
let server_name = arg.get_one::<String>("NAME").unwrap();
|
|
|
|
|
|
if !validate_name(server_name) {
|
|
return Err("Invalid name provided".to_string());
|
|
}
|
|
|
|
let conn = &mut pool.get().unwrap();
|
|
|
|
if let Ok(_) = schema::servers::table.filter(schema::servers::name.eq(server_name)).select(models::Servers::as_select()).first(conn) {
|
|
return Err("Server with this name already exists".to_string())
|
|
}
|
|
|
|
let path = config.config_path + "/" + server_name + "/";
|
|
|
|
create_dir(&path).await.map_err(|e| format!("failed to create the config folder, reason : {}", e))?;
|
|
|
|
let mut file = File::create(path.to_string() + "env").await.map_err(|e| format!("failed to create .env file, reason {}", e))?;
|
|
file.write_all(env_settings::args_to_env(arg, true).as_bytes()).await.map_err(|e| format!("Failed to write to env file, reason : {}", e))?;
|
|
|
|
create_dir(path.to_string() + "server").await.map_err(|e| format!("failed to create server subfolder, reason: {}", e))?;
|
|
|
|
if let Some(archive) = arg.get_one::<String>("SERVER_ARCHIVE") {
|
|
extract_archive(archive.to_owned(), path.to_string() + "tmp").await?;
|
|
move_all_elements(Path::new(&(path.to_string() + "tmp/minecraft")), Path::new(&(path.to_string() + "server"))).map_err(|e| format!("failed to move the elements, reason: {}", e))?;
|
|
std::fs::remove_dir_all(path.to_string() + "tmp").map_err(|e| format!("failed to remove the tmp dir, reason : {}", e))?;
|
|
}
|
|
|
|
let new_server = models::CreateServer {
|
|
name: server_name,
|
|
last_login: Some(Utc::now().naive_utc()),
|
|
status: ServerStatus::Stopped,
|
|
redirect_ip: None
|
|
};
|
|
|
|
diesel::insert_into(servers).values(&new_server).execute(conn).map_err(|e| format!("Failed to insert in db, error : {:?}", e))?;
|
|
Ok(())
|
|
}
|