gmarche-rs/src/config.rs

use serde::{Deserialize, Serialize};
use std::{
	net::{IpAddr, Ipv4Addr, SocketAddr},
	path::Path,
};

const CONFIG_FILE: &str = "config.json";

#[derive(Deserialize, Serialize)]
pub struct Config {
	pub admin_passwords: Vec<String>,
	pub cookies_https_only: bool,
	pub cookies_domain: Option<String>,
	pub listen: SocketAddr,
	pub root_url: String,
}

impl Default for Config {
	fn default() -> Self {
		Self {
			admin_passwords: vec![],
			cookies_https_only: false,
			cookies_domain: None,
			listen: SocketAddr::new(IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)), 10353),
			root_url: String::from("/"),
		}
	}
}

pub fn read_config(dir: &Path) -> Config {
	let path = dir.join(CONFIG_FILE);

	if !path.is_file() {
		let config = Config::default();
		std::fs::write(path, serde_json::to_string_pretty(&config).unwrap())
			.expect("Cannot write config file");
		config
	} else {
		serde_json::from_str(
			std::str::from_utf8(&std::fs::read(path).expect("Cannot read config file"))
				.expect("Bad encoding in config file"),
		)
		.expect("Bad JSON in config file")
	}
}