111 lines
2.5 KiB
Rust
111 lines
2.5 KiB
Rust
#![warn (clippy::pedantic)]
|
|
|
|
use std::{
|
|
fs::File,
|
|
io::Write,
|
|
path::PathBuf,
|
|
os::unix::fs::PermissionsExt,
|
|
};
|
|
|
|
use structopt::StructOpt;
|
|
|
|
use ptth_server::{
|
|
load_toml,
|
|
run_server,
|
|
};
|
|
|
|
#[derive (Debug, StructOpt)]
|
|
struct Opt {
|
|
#[structopt (long)]
|
|
auto_gen_key: bool,
|
|
|
|
#[structopt (long)]
|
|
simulate_slow_upload: bool,
|
|
|
|
#[structopt (long)]
|
|
file_server_root: Option <PathBuf>,
|
|
|
|
#[structopt (long)]
|
|
asset_root: Option <PathBuf>,
|
|
|
|
#[structopt (long)]
|
|
config_path: Option <PathBuf>,
|
|
|
|
#[structopt (long)]
|
|
name: Option <String>,
|
|
|
|
#[structopt (long)]
|
|
print_tripcode: bool,
|
|
|
|
#[structopt (long)]
|
|
relay_url: Option <String>,
|
|
}
|
|
|
|
#[derive (Default, serde::Deserialize)]
|
|
pub struct ConfigFile {
|
|
pub name: Option <String>,
|
|
pub api_key: String,
|
|
pub relay_url: Option <String>,
|
|
pub file_server_root: Option <PathBuf>,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main () -> Result <(), anyhow::Error> {
|
|
tracing_subscriber::fmt::init ();
|
|
|
|
let opt = Opt::from_args ();
|
|
let asset_root = opt.asset_root;
|
|
|
|
let path = opt.config_path.clone ().unwrap_or_else (|| PathBuf::from ("./config/ptth_server.toml"));
|
|
|
|
let config_file: ConfigFile = match load_toml::load (&path) {
|
|
Err (ptth_server::errors::LoadTomlError::Io (_)) => if opt.auto_gen_key {
|
|
use rand::RngCore;
|
|
let mut buffer = vec! [0_u8; 64];
|
|
rand::thread_rng ().fill_bytes (&mut buffer);
|
|
|
|
let api_key = base64::encode (&buffer);
|
|
|
|
{
|
|
let mut f = File::create (&path)?;
|
|
let metadata = f.metadata ()?;
|
|
let mut permissions = metadata.permissions ();
|
|
permissions.set_mode (0o600);
|
|
f.set_permissions (permissions)?;
|
|
|
|
f.write_all (format! ("api_key = \"{}\"\n", api_key).as_bytes ())?;
|
|
}
|
|
|
|
load_toml::load (&path)?
|
|
}
|
|
else {
|
|
panic! ("API key not provided in config file and auto-gen-key not provided");
|
|
},
|
|
Ok (x) => x,
|
|
Err (e) => return Err (e.into ()),
|
|
};
|
|
|
|
let config_file = ptth_server::ConfigFile {
|
|
name: opt.name.or (config_file.name).expect ("`name` must be provided in command line or config file"),
|
|
api_key: config_file.api_key,
|
|
relay_url: opt.relay_url.or (config_file.relay_url).expect ("`relay_url` must be provided in command line or config file"),
|
|
file_server_root: opt.file_server_root.or (config_file.file_server_root),
|
|
simulate_slow_upload: opt.simulate_slow_upload,
|
|
};
|
|
|
|
if opt.print_tripcode {
|
|
println! (r#"name = "{}""#, config_file.name);
|
|
println! (r#"tripcode = "{}""#, config_file.tripcode ());
|
|
return Ok (());
|
|
}
|
|
|
|
run_server (
|
|
config_file,
|
|
ptth_core::graceful_shutdown::init (),
|
|
Some (path),
|
|
asset_root
|
|
).await?;
|
|
|
|
Ok (())
|
|
}
|