use config::{Config, FileFormat}; use gtk::cairo::IoError; use snafu::prelude::*; use log::{error, info}; use serde::{Deserialize, Serialize}; use std::fs::File; use std::io::Write; #[derive(Debug, Serialize, Deserialize, Clone)] pub struct AppConfig { pub camera_ip: String, pub camera_port: u32, pub tracker_ip: String, pub tracker_port: u32, pub tracker_refresh_rate_millis: u32, } impl Default for AppConfig { fn default() -> Self { AppConfig { camera_ip: "10.0.0.33".to_string(), camera_port: 8765, tracker_ip: "10.0.0.210".to_string(), tracker_port: 6543, tracker_refresh_rate_millis: 100, } } } pub fn load_config() -> AppConfig { Config::builder() .add_source(config::File::new("./settings.toml", FileFormat::Toml)) .build() .and_then(|val| val.try_deserialize()) .unwrap_or_default() } #[derive(Debug, Snafu)] pub enum SaveConfigError { #[snafu(display("Could not serialize app state: {source}"))] SerdeError {source: toml::ser::Error }, #[snafu(display("Could not write app state to file: {path}"))] IoError {source: std::io::Error, path: String }, } pub fn save_config(config: &AppConfig) -> Result<(), SaveConfigError> { let toml_str = toml::to_string(&config).context(SerdeSnafu)?; let mut file = File::create("./settings.toml").context(IoSnafu {path: "./settings.toml" })?; file.write_all(toml_str.as_bytes()).context(IoSnafu {path: "./settings.toml" })?; info!("Config file saved successfully"); Ok(()) }