aboutsummaryrefslogtreecommitdiff
path: root/mingling_cli/src/pkg_mgr/cmd_internal_loadpkgs.rs
blob: b4928ce462389d47aca6da16275edd68b5ba24a8 (plain) (blame)
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
use std::{
    fs, io,
    path::{Path, PathBuf},
};

use mingling::{
    Routable,
    macros::{buffer, command, pack, r_println, renderer, routeify},
};

use crate::{
    Next,
    pkg_mgr::{ErrorNoDataDirectory, ResPackagesDir},
};

// Version directory paths of every enabled package.
pack!(ResultLoadPkgsPaths = Vec<PathBuf>);

// Completion script paths of every enabled package.
pack!(ResultLoadPkgsComps = Vec<PathBuf>);

#[command(node = "__loadpkgs_path", routeify)]
pub fn load_packages_paths(packages_dir: &ResPackagesDir) -> Next {
    let packages_dir = &packages_dir.path;
    if packages_dir.as_os_str().is_empty() {
        return ErrorNoDataDirectory::default().to_chain();
    }
    let paths = enabled_version_dirs(packages_dir).map_err(|e| {
        io::Error::new(
            e.kind(),
            format!("failed to read {}: {e}", packages_dir.display()),
        )
    })?;
    ResultLoadPkgsPaths::new(paths).to_chain()
}

#[command(node = "__loadpkgs_comp_scripts", routeify)]
pub fn load_packages_comp_scripts(packages_dir: &ResPackagesDir) -> Next {
    let packages_dir = &packages_dir.path;
    if packages_dir.as_os_str().is_empty() {
        return ErrorNoDataDirectory::default().to_chain();
    }
    let scripts = comp_scripts(packages_dir).map_err(|e| {
        io::Error::new(
            e.kind(),
            format!("failed to read {}: {e}", packages_dir.display()),
        )
    })?;
    ResultLoadPkgsComps::new(scripts).to_chain()
}

#[renderer(buffer)]
pub fn render_result_load_pkgs_paths(r: ResultLoadPkgsPaths) {
    for path in r.inner {
        r_println!("{}", path.display());
    }
}

#[renderer(buffer)]
pub fn render_result_load_pkgs_comps(r: ResultLoadPkgsComps) {
    for path in r.inner {
        r_println!("{}", path.display());
    }
}

/// Version directories of every enabled package.
///
/// An enable file is a `name` file (no `@`) whose content is the enabled version.
fn enabled_version_dirs(packages_dir: &Path) -> Result<Vec<PathBuf>, io::Error> {
    let mut dirs = Vec::new();
    for entry in fs::read_dir(packages_dir)? {
        let entry = entry?;
        if !entry.file_type().is_ok_and(|t| t.is_file()) {
            continue;
        }
        let name = entry.file_name().to_string_lossy().into_owned();
        if name.contains('@') {
            continue;
        }
        let version = fs::read_to_string(entry.path())?;
        dirs.push(packages_dir.join(format!("{name}@{}", version.trim())));
    }
    dirs.sort();
    Ok(dirs)
}

/// Completion scripts inside every enabled version directory.
fn comp_scripts(packages_dir: &Path) -> Result<Vec<PathBuf>, io::Error> {
    let mut scripts = Vec::new();
    for dir in enabled_version_dirs(packages_dir)? {
        if !dir.is_dir() {
            continue;
        }
        for entry in fs::read_dir(&dir)? {
            let entry = entry?;
            let name = entry.file_name().to_string_lossy().into_owned();
            if name.contains("_comp") {
                scripts.push(entry.path());
            }
        }
    }
    scripts.sort();
    Ok(scripts)
}