101 lines
3.2 KiB
Rust
101 lines
3.2 KiB
Rust
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<TempDir, GenerationError> {
|
|
#[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<TempDir, GenerationError> {
|
|
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);
|
|
},
|
|
}
|
|
}
|