shlf 0.10.9

Empower Development Journey with AI-Driven Tools
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use std::error::Error;
use std::fs;
use std::path::Path;

const SOURCE_ASSETS_DIR: &str = "assets/prompts";
const CONFIG_SUBPATH: &str = "shelf/assets/prompts";

fn main() -> Result<(), Box<dyn Error>> {
    let base_dirs = directories::BaseDirs::new()
        .ok_or("Could not find home directory to construct config path")?;
    let target_assets_dir = base_dirs.config_dir().join(CONFIG_SUBPATH);

    ensure_dir_exists(&target_assets_dir)?;
    copy_assets_from_source_to_target(SOURCE_ASSETS_DIR, &target_assets_dir)?;

    Ok(())
}

/// Ensures `dir` exists, creating it and any missing parents if necessary.
fn ensure_dir_exists(dir: &Path) -> Result<(), Box<dyn Error>> {
    if !dir.exists() {
        fs::create_dir_all(dir)
            .map_err(|e| format!("Failed to create assets directory {dir:?}: {e}"))?;
    }
    Ok(())
}

/// Copies regular files from `source_dir` (relative path) into `target_dir`.
/// Only files (not directories, symlinks, etc.) are copied.
fn copy_assets_from_source_to_target(
    source_dir: &str,
    target_dir: &Path,
) -> Result<(), Box<dyn Error>> {
    let entries = fs::read_dir(source_dir)
        .map_err(|e| format!("Failed to read source assets directory '{source_dir}': {e}",))?;

    for entry_res in entries {
        let entry = entry_res.map_err(|e| format!("Failed to read directory entry: {e}"))?;
        let file_type = entry
            .file_type()
            .map_err(|e| format!("Failed to get file type for {:?}: {}", entry.path(), e))?;

        if file_type.is_file() {
            let file_name = entry.file_name();
            let dest_path = target_dir.join(&file_name);
            fs::copy(entry.path(), &dest_path).map_err(|e| {
                format!(
                    "Failed to copy {:?} to {:?}: {}",
                    entry.path(),
                    dest_path,
                    e
                )
            })?;
        }
    }

    Ok(())
}