94 lines
3.1 KiB
Rust
94 lines
3.1 KiB
Rust
/*
|
|
This file is part of Rasp Manager.
|
|
Rasp Manager is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
Rasp Manager is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
You should have received a copy of the GNU General Public License
|
|
along with Rasp Manager. If not, see <https://www.gnu.org/licenses/>
|
|
*/
|
|
|
|
use std::collections::HashMap;
|
|
|
|
use serde::{Deserialize};
|
|
use toml::from_str;
|
|
use clap::{App, Arg};
|
|
use tide::convert::Serialize;
|
|
|
|
|
|
#[derive(Clone, Deserialize, Serialize)]
|
|
pub struct Command {
|
|
pub label: String,
|
|
pub command: String
|
|
}
|
|
|
|
#[derive(Clone, Deserialize, Serialize)]
|
|
pub struct Config {
|
|
pub static_dir: Option<String>,
|
|
pub addr: String,
|
|
pub port: i32,
|
|
pub commands: HashMap<String, Command>
|
|
}
|
|
|
|
impl Config {
|
|
pub fn generate() -> Config {
|
|
let matches = App::new("Rasp Manager")
|
|
.about("A simple server manager for local newtrok")
|
|
.version(env!("CARGO_PKG_VERSION"))
|
|
.author(env!("CARGO_PKG_AUTHORS"))
|
|
.arg(Arg::with_name("config")
|
|
.short("c")
|
|
.long("config")
|
|
.value_name("CONFIG")
|
|
.help("Config file"))
|
|
.arg(Arg::with_name("addr")
|
|
.short("a")
|
|
.long("addr")
|
|
.value_name("ADDR")
|
|
.help("Address to listen port on"))
|
|
.arg(Arg::with_name("port")
|
|
.short("p")
|
|
.long("port")
|
|
.value_name("PORT")
|
|
.help("Port to listen"))
|
|
.arg(Arg::with_name("static_dir")
|
|
.short("s")
|
|
.long("static_dir")
|
|
.value_name("DIR")
|
|
.help("Directory to host as static"))
|
|
.get_matches();
|
|
|
|
|
|
let config = if let Some(config) = matches.value_of("config") {
|
|
let content = std::fs::read_to_string(config).expect(
|
|
&format!("The config file doesn't exist at: {}", config)
|
|
);
|
|
let tomlcfg: Config = from_str(&content).expect(
|
|
"The config file appears malformed!"
|
|
);
|
|
tomlcfg
|
|
} else {
|
|
Config {
|
|
static_dir: None,
|
|
addr: "0.0.0.0".to_string(),
|
|
port: 80,
|
|
commands: HashMap::new()
|
|
}
|
|
};
|
|
|
|
Config {
|
|
static_dir: match matches.value_of("static_dir") {
|
|
Some(val) => Some(val.to_owned()),
|
|
None => config.static_dir.to_owned()
|
|
},
|
|
addr: matches.value_of("addr").unwrap_or(&config.addr).to_owned(),
|
|
port: matches.value_of("port").unwrap_or_default().parse().unwrap_or(config.port),
|
|
commands: config.commands
|
|
}
|
|
}
|
|
}
|