use serde::Deserialize; use std::{fs, path::PathBuf}; use glob::glob; use std::{fs::File, io::Write, path::Path}; use crate::template::Template; /// Store the configuration of config/config.toml #[derive(Deserialize, Clone, Default, Debug)] pub struct FileConfig { /// http/https pub scheme: Option, /// Port used pub port: Option, /// Mail of owner pub mail: Option, /// Lang used pub lang: Option, /// Adress .onion for Tor pub onion: Option, /// App name pub app_name: Option, /// Name of website owner pub name: Option, /// Fullname of website owner pub fullname: Option, } impl FileConfig { /// Initialize with default values fn new() -> Self { Self { scheme: Some("http".into()), port: Some(8080), app_name: Some("EWP".into()), ..FileConfig::default() } } /// Complete default structure with an existing one fn complete(a: Self) -> Self { // Default config let d = FileConfig::new(); /// Return the default value if nothing is value is none fn test(val: Option, default: Option) -> Option { if val.is_some() { val } else { default } } Self { scheme: test(a.scheme, d.scheme), port: test(a.port, d.port), mail: test(a.mail, d.mail), lang: test(a.lang, d.lang), onion: test(a.onion, d.onion), app_name: test(a.app_name, d.app_name), name: test(a.name, d.name), fullname: test(a.fullname, d.fullname), } } } /// Configuration used internally in the app #[derive(Clone, Debug)] pub struct Config { /// Information given in the config file pub fc: FileConfig, /// Location where the static files are stored pub static_location: String, /// Informations about templates pub tmpl: Template, } /// Load the config file fn get_file_config(file_path: &str) -> FileConfig { match fs::read_to_string(file_path) { Ok(file) => match toml::from_str(&file) { Ok(stored_config) => FileConfig::complete(stored_config), Err(file_error) => { panic!("Error in config file: {file_error}"); } }, Err(_) => { // No config file FileConfig::new() } } } /// Build the configuration pub fn get_config(file_path: &str) -> Config { let internal_config = get_file_config(file_path); let static_dir = "static".to_owned(); let templates_dir = "templates".to_owned(); let files_root = init( "dist".into(), static_dir.to_owned(), templates_dir.to_owned(), ); Config { fc: internal_config.to_owned(), static_location: format!("{}/{}", files_root, static_dir), tmpl: Template { directory: format!("{}/{}", files_root, templates_dir), app_name: internal_config.app_name.unwrap(), }, } } /// Preparation before running the http server fn init(dist_dir: String, static_dir: String, templates_dir: String) -> String { // The static folder is minimized only in release mode if cfg!(debug_assertions) { ".".into() } else { let cfg = minify_html::Cfg { keep_closing_tags: true, minify_css: true, minify_js: true, ..minify_html::Cfg::spec_compliant() }; // Static files for entry in glob(&format!("{static_dir}/**/*.*")).unwrap() { let path = entry.unwrap(); let path_with_dist = path .to_string_lossy() .replace(&static_dir, &format!("{dist_dir}/{static_dir}")); minify_and_copy(&cfg, path, path_with_dist); } // Template files for entry in glob(&format!("{templates_dir}/**/*.*")).unwrap() { let path = entry.unwrap(); let path_with_dist = path .to_string_lossy() .replace(&templates_dir, &format!("{dist_dir}/{templates_dir}")); minify_and_copy(&cfg, path, path_with_dist); } dist_dir } } /// Minify some assets for production fn minify_and_copy(cfg: &minify_html::Cfg, path: PathBuf, path_with_dist: String) { // Create folders let new_path = Path::new(&path_with_dist); fs::create_dir_all(new_path.parent().unwrap()).unwrap(); let session = minify_js::Session::new(); let mut copy = true; if let Some(ext) = path.extension() { let js_ext = "js"; let current_ext = ext.to_string_lossy().to_lowercase(); // List of files who should be minified if ["html", "css", js_ext, "svg", "webmanifest", "xml"].contains(¤t_ext.as_str()) { // We won't copy, we'll minify copy = false; // Minify let data = fs::read(&path).unwrap(); let minified = if current_ext == js_ext { let mut out = Vec::new(); minify_js::minify(&session, minify_js::TopLevelMode::Global, &data, &mut out) .unwrap(); out } else { minify_html::minify(&data, cfg) }; // Write files let file = File::create(&path_with_dist); file.expect("Error when minify the file") .write_all(&minified) .unwrap(); } } if copy { // If no minification is needed fs::copy(path, path_with_dist).unwrap(); } }