91 lines
2.2 KiB
Rust
91 lines
2.2 KiB
Rust
mod config;
|
|
mod servers;
|
|
mod update;
|
|
mod upstreams;
|
|
|
|
use crate::config::ConfigV1;
|
|
use crate::servers::Server;
|
|
|
|
use log::{debug, error, info};
|
|
use pico_args::Arguments;
|
|
use std::path::PathBuf;
|
|
|
|
fn main() {
|
|
let mut args = Arguments::from_env();
|
|
|
|
match args.subcommand().expect("Unexpected error").as_deref() {
|
|
Some("serve") => serve(),
|
|
Some("update") => update::update(),
|
|
Some(cmd) => {
|
|
eprintln!("Invalid command: {cmd}");
|
|
std::process::exit(1);
|
|
}
|
|
None => {
|
|
eprintln!("Calling l4p without argument is deprecated now. Please use: l4p serve");
|
|
serve();
|
|
}
|
|
}
|
|
}
|
|
|
|
fn serve() {
|
|
let config_path = match find_config() {
|
|
Ok(p) => p,
|
|
Err(paths) => {
|
|
println!("Could not find config file. Tried paths:");
|
|
for p in paths {
|
|
println!("- {}", p);
|
|
}
|
|
std::process::exit(1);
|
|
}
|
|
};
|
|
|
|
let config = match ConfigV1::new(&config_path) {
|
|
Ok(config) => config,
|
|
Err(e) => {
|
|
println!("Could not load config: {:?}", e);
|
|
std::process::exit(1);
|
|
}
|
|
};
|
|
debug!("{:?}", config);
|
|
|
|
let mut server = Server::new_from_v1_config(config.base);
|
|
debug!("{:?}", server);
|
|
|
|
let _ = server.run();
|
|
error!("Server ended with errors");
|
|
}
|
|
|
|
fn find_config() -> Result<String, Vec<String>> {
|
|
let possible_locations = ["/etc/l4p", ""];
|
|
let possible_names = ["l4p.yaml", "config.yaml"];
|
|
|
|
let mut tried_paths = Vec::<String>::new();
|
|
let mut possible_paths = Vec::<PathBuf>::new();
|
|
|
|
if let Ok(env_path) = std::env::var("L4P_CONFIG") {
|
|
possible_paths.push(PathBuf::from(env_path));
|
|
}
|
|
|
|
possible_paths.append(
|
|
&mut possible_locations
|
|
.iter()
|
|
.flat_map(|&path| {
|
|
possible_names
|
|
.iter()
|
|
.map(move |&file| PathBuf::new().join(path).join(file))
|
|
})
|
|
.collect::<Vec<PathBuf>>(),
|
|
);
|
|
|
|
for path in possible_paths {
|
|
let path_str = path.to_string_lossy().to_string();
|
|
if path.exists() {
|
|
return Ok(path_str);
|
|
}
|
|
|
|
tried_paths.push(path_str);
|
|
}
|
|
|
|
Err(tried_paths)
|
|
}
|