44 lines
1.3 KiB
Rust
44 lines
1.3 KiB
Rust
pub mod db;
|
|
|
|
use std::path::{Path, PathBuf};
|
|
use color_eyre::eyre::eyre;
|
|
use color_eyre::owo_colors::OwoColorize;
|
|
use tokio::fs;
|
|
use crate::config::types::ApplicationConfig;
|
|
use crate::constants::{APP_CONFIG_DIR, APP_DATA_DIR};
|
|
use crate::crawler::DLSITE_IMG_FOLDER;
|
|
|
|
|
|
pub async fn initialize_folders() -> color_eyre::Result<()> {
|
|
let app_conf = ApplicationConfig::get_config()?;
|
|
if !APP_CONFIG_DIR.exists() {
|
|
fs::create_dir_all(APP_CONFIG_DIR.as_path()).await?;
|
|
}
|
|
if !APP_DATA_DIR.exists() {
|
|
fs::create_dir_all(APP_DATA_DIR.as_path()).await?;
|
|
}
|
|
if !DLSITE_IMG_FOLDER.exists() {
|
|
fs::create_dir_all(DLSITE_IMG_FOLDER.as_path()).await?;
|
|
}
|
|
let db_path = Path::new(&app_conf.basic_config.db_path);
|
|
if !db_path.exists() {
|
|
fs::create_dir_all(db_path).await?;
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn get_all_folders(paths: Vec<&Path>) -> color_eyre::Result<Vec<PathBuf>> {
|
|
let mut folders: Vec<PathBuf> = Vec::new();
|
|
for path in paths {
|
|
let path = path.to_path_buf();
|
|
if !path.exists() {
|
|
return Err(eyre!("{:?} {}", path.blue(), "does not exist".red()));
|
|
}
|
|
|
|
let mut dirs = fs::read_dir(path).await?;
|
|
while let Some(dir) = dirs.next_entry().await? {
|
|
folders.push(dir.path());
|
|
}
|
|
}
|
|
Ok(folders)
|
|
} |