tyrel/rust-ssg
1
0
Fork 0
Code Issues 3 Pull Requests Packages Projects 4 Releases Wiki Activity

specify what dirs to create

Browse Source
This commit is contained in:
Tyrel Souza 2023-10-14 14:41:51 -04:00
parent b1ef76e2de
commit bfeae64c84
No known key found for this signature in database
GPG Key ID: F3614B02ACBE438E
2 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/file_utils.rs
View File

@ -24,13 +24,17 @@ pub fn write_file(html: String) {
fs::write("./public/links/index.html", html).expect("Could not write to index.html"); fs::write("./public/links/index.html", html).expect("Could not write to index.html");
} }
pub fn create_dirs() { pub fn create_dirs(dirs: &[&str]) {
if !Path::new("./public").exists() { if !Path::new("./public").exists() {
fs::create_dir("./public").expect("Could not create public directory"); fs::create_dir("./public").expect("Could not create public directory");
} }
if !Path::new("./public/links").exists() { for dir in dirs {
fs::create_dir("./public/links").expect("Could not create public directory"); let path = format!("./public/{}", dir);
if !Path::new(&path).exists() {
fs::create_dir(&path).expect("Could not create public directory");
}
} }
} }
pub fn delete_public_dir() { pub fn delete_public_dir() {

6
src/main.rs
View File

@ -4,8 +4,12 @@ use crate::links::{load_links,render_links};
fn main() { fn main() {
// absolutely destroy any /public data, so we have a clean slate to write to
file_utils::delete_public_dir(); file_utils::delete_public_dir();
file_utils::create_dirs(); file_utils::create_dirs(&[
"links",
"blog"
]);
// Generate and Write Links // Generate and Write Links
let links_input = load_links("data/links/links.yaml"); let links_input = load_links("data/links/links.yaml");