proto_cli 0.57.3

A multi-language version manager, a unified toolchain.
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

use crate::session::ProtoSession;
use clap::Args;
use iocraft::prelude::element;
use proto_core::flow::link::Linker;
use proto_core::flow::resolve::Resolver;
use starbase::AppResult;
use starbase_console::ui::*;
use starbase_utils::fs;
use tracing::debug;

#[derive(Args, Clone, Debug)]
pub struct RegenArgs {
    #[arg(long, help = "Also recreate binary symlinks")]
    bin: bool,
}

#[tracing::instrument(skip_all)]
pub async fn regen(session: ProtoSession, args: RegenArgs) -> AppResult {
    let store = &session.env.store;
    let progress = session.render_progress_loader().await;

    progress.set_message(if args.bin {
        "Regenerating shims and bins..."
    } else {
        "Regenerating shims..."
    });

    // Delete all shims
    progress.set_message("Removing old shims");

    fs::remove_dir_all(&store.shims_dir)?;

    // Delete all bins (except for proto)
    if args.bin {
        progress.set_message("Removing old bins");

        for file in fs::read_dir_all(&store.bin_dir)? {
            let path = file.path();
            let name = fs::file_name(&path);

            if path.is_dir()
                || name == "proto"
                || name == "proto.exe"
                || name == "proto-shim"
                || name == "proto-shim.exe"
            {
                continue;
            }

            store.unlink_bin(&path)?;
        }
    }

    // Regenerate everything!
    let config = session.env.load_config()?;

    for tool in session.load_tools().await? {
        progress.set_message(format!("Regenerating {}", tool.get_name()));

        if let Some(version) = config.versions.get(&tool.context) {
            let mut spec = version.to_owned();
            let name = tool.get_name().to_owned();

            // Shims - Create once if tool has a configured version
            debug!("Regenerating {name} shim");

            Resolver::resolve(&tool, &mut spec, true).await?;

            let linker = Linker::new(&tool, &spec);

            linker.link_shims(true).await?;

            // Bins - Create for each installed version
            if args.bin {
                debug!("Relinking {name} bin");

                linker.link_bins(true).await?;
            }
        }
    }

    progress.stop().await?;

    session.console.render(element! {
        Notice(variant: Variant::Success) {
            StyledText(
                content: if args.bin {
                    "Regenerated shims and bins!"
                } else {
                    "Regenerated shims!"
                },
            )
        }
    })?;

    Ok(None)
}