80 lines
2.4 KiB
Rust
80 lines
2.4 KiB
Rust
|
use std::path::PathBuf;
|
||
|
|
use std::fs::{self, File, metadata};
|
||
|
|
|
||
|
|
pub struct MDBook {
|
||
|
|
dest: PathBuf,
|
||
|
|
src: PathBuf,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl MDBook {
|
||
|
|
pub fn new() -> Self {
|
||
|
|
MDBook {
|
||
|
|
dest: PathBuf::from("book"),
|
||
|
|
src: PathBuf::from("src"),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn init(&self, dir: &PathBuf) -> Result<(),&str> {
|
||
|
|
|
||
|
|
// Hacky way to check if the directory exists... Until PathExt moves to stable
|
||
|
|
match metadata(dir) {
|
||
|
|
Err(_) => return Err("Destination path does not exist"),
|
||
|
|
_ => {}
|
||
|
|
}
|
||
|
|
|
||
|
|
let dest = if self.dest.is_relative() {
|
||
|
|
dir.join(&self.dest)
|
||
|
|
} else {
|
||
|
|
self.dest.clone()
|
||
|
|
};
|
||
|
|
|
||
|
|
let src = if self.src.is_relative() {
|
||
|
|
dir.join(&self.src)
|
||
|
|
} else {
|
||
|
|
self.src.clone()
|
||
|
|
};
|
||
|
|
|
||
|
|
// Hacky way to check if the directory exists... Until PathExt moves to stable
|
||
|
|
match metadata(&dest) {
|
||
|
|
Err(_) => {
|
||
|
|
// There is a very high chance that the error is due to the fact that
|
||
|
|
// the directory / file does not exist
|
||
|
|
fs::create_dir(&dest).unwrap();
|
||
|
|
},
|
||
|
|
Ok(_) => { /* If there is no error, the directory / file does exist */ }
|
||
|
|
}
|
||
|
|
|
||
|
|
// Hacky way to check if the directory exists... Until PathExt moves to stable
|
||
|
|
match metadata(&src) {
|
||
|
|
Err(_) => {
|
||
|
|
// There is a very high chance that the error is due to the fact that
|
||
|
|
// the directory / file does not exist
|
||
|
|
fs::create_dir(&src).unwrap();
|
||
|
|
},
|
||
|
|
Ok(_) => { /* If there is no error, the directory / file does exist */ }
|
||
|
|
}
|
||
|
|
|
||
|
|
// Hacky way to check if the directory exists... Until PathExt moves to stable
|
||
|
|
match metadata(dir.join("src/SUMMARY.md")) {
|
||
|
|
Err(_) => {
|
||
|
|
// There is a very high chance that the error is due to the fact that
|
||
|
|
// the directory / file does not exist
|
||
|
|
File::create("src/SUMMARY.md").unwrap();
|
||
|
|
},
|
||
|
|
Ok(_) => { /* If there is no error, the directory / file does exist */ }
|
||
|
|
}
|
||
|
|
|
||
|
|
return Ok(());
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn set_dest(mut self, dest: PathBuf) -> Self {
|
||
|
|
self.dest = dest;
|
||
|
|
self
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn set_src(mut self, src: PathBuf) -> Self {
|
||
|
|
self.src = src;
|
||
|
|
self
|
||
|
|
}
|
||
|
|
}
|