ptth/crates/ptth_server/src/bin/ptth_server.rs

use std::{
	error::Error,
	path::PathBuf,
};

use structopt::StructOpt;
use tokio::runtime;

use ptth_server::{
	ConfigFile,
	load_toml,
	run_server,
};

#[derive (Debug, StructOpt)]
struct Opt {
	//#[structopt (long)]
	//file_server_root: Option <PathBuf>,
	
	#[structopt (long)]
	config_path: Option <PathBuf>,
	
	#[structopt (long)]
	asset_root: Option <PathBuf>,
	
	#[structopt (long)]
	print_tripcode: bool,
}

fn main () -> Result <(), Box <dyn Error>> {
	let opt = Opt::from_args ();
	
	tracing_subscriber::fmt::init ();
	let path = opt.config_path.clone ().unwrap_or_else (|| PathBuf::from ("./config/ptth_server.toml"));
	let config_file: ConfigFile = load_toml::load (&path);
	
	if opt.print_tripcode {
		println! (r#""{}" = "{}""#, config_file.name, config_file.tripcode ());
		return Ok (());
	}
	
	let mut rt = runtime::Runtime::new ()?;
	
	rt.block_on (async move {
		run_server (
			config_file, 
			ptth_core::graceful_shutdown::init (),
			Some (path),
			opt.asset_root
		).await
	})?;
	
	Ok (())
}