use std::{io::Cursor, path::PathBuf}; use path_slash::PathBufExt as _; use tempfile::TempDir; use crate::{misc, GenerationError, Warning, WARNINGS}; /// Converts a given path to windows style if needed. pub fn convert_path_to_os_specific(path: PathBuf) -> PathBuf { // If target os is windows. #[cfg(target_os = "windows")] { return PathBuf::from_slash_lossy(path.into_os_string()); } // If target os is not windows. #[cfg(not(target_os = "windows"))] { return PathBuf::from_backslash_lossy(path.into_os_string()); } } /// Remove chars that are illegal to be used in filenames on both unix & windows. pub fn remove_illegal_chars(mut string: String) -> String { const ILLEGAL_CHARS: [char; 9] = ['/', '\u{005C}', '<', '>', ':', '\u{0022}', '|', '?', '*']; for char in ILLEGAL_CHARS { string = string.replace(char, " "); } return string; } /// Setup html2xhtml in the operating system's temp directory. pub fn setup_html2xhtml() -> Result { #[cfg(target_os = "windows")] { const HTML2XHTML: &[u8; 245025] = include_bytes!("../html2xhtml-windows.zip"); // This will not compile on windows due to this and no I don't give a shit. // Compile it on linux for windows like a sane person. let html2xhtml_temp_dir = create_temp_dir()?; match zip_extract::extract(Cursor::new(HTML2XHTML), html2xhtml_temp_dir.path(), true) { Ok(_) => (), Err(error) => return Err(GenerationError::Html2XhtmlExtractionError { error }), } return Ok(html2xhtml_temp_dir); } #[cfg(target_os = "linux")] { const HTML2XHTML: &[u8; 186938] = include_bytes!("../html2xhtml-linux.zip"); let html2xhtml_temp_dir = create_temp_dir()?; match zip_extract::extract(Cursor::new(HTML2XHTML), html2xhtml_temp_dir.path(), true) { Ok(_) => (), Err(error) => return Err(GenerationError::Html2XhtmlExtractionError { error }), } return Ok(html2xhtml_temp_dir); } #[cfg(target_os = "macos")] { Err(GenerationError::OsUnsupportedError { os: misc::Oses::MacOs, }) } // In the event the OS is unknown. #[allow(unreachable_code)] Err(GenerationError::OsUnsupportedError { os: misc::Oses::OtherUnknownOs, }) } /// Function to create a temporary directory. fn create_temp_dir() -> Result { match TempDir::new() { Ok(temp_dir) => return Ok(temp_dir), Err(error) => return Err(GenerationError::TempDirCreationError { error }), } } /// Delete a temporary directory. pub fn delete_temp_dir(temp_dir: TempDir) { let temp_dir_path = temp_dir.path().to_path_buf(); match temp_dir.close() { Ok(_) => (), Err(warning) => { let warning = Warning::TempDirDeletionError { warning_msg: "Unable to close and delete temp directory".to_string(), temp_directory_path: temp_dir_path, error: warning, }; WARNINGS.lock().unwrap().add_warning(warning); }, } }