lux_cli/
search.rs

1use std::collections::HashMap;
2
3use clap::Args;
4use eyre::Result;
5use itertools::Itertools;
6use text_trees::{FormatCharacters, StringTreeNode, TreeFormatting};
7
8use lux_lib::{
9    config::Config,
10    package::{PackageName, PackageReq, PackageVersion},
11    progress::MultiProgress,
12    remote_package_db::RemotePackageDB,
13};
14
15use crate::args::OutputFormat;
16
17#[derive(Args)]
18pub struct Search {
19    lua_package_req: PackageReq,
20    // TODO(vhyrro): Add options.
21    #[arg(long, default_value = "text", value_enum, ignore_case = true)]
22    output_format: OutputFormat,
23}
24
25pub async fn search(data: Search, config: Config) -> Result<()> {
26    let progress = MultiProgress::new(&config);
27    let bar = progress.map(MultiProgress::new_bar);
28    let formatting = TreeFormatting::dir_tree(FormatCharacters::box_chars());
29
30    let package_db = RemotePackageDB::from_config(&config, &bar).await?;
31
32    bar.map(|b| b.set_message(format!("🔎 Searching for `{}`...", data.lua_package_req)));
33
34    let lua_package_req = data.lua_package_req;
35
36    let result = package_db.search(&lua_package_req);
37
38    bar.map(|b| b.finish_and_clear());
39
40    match data.output_format {
41        OutputFormat::Json => {
42            let rock_to_version_map: HashMap<&PackageName, Vec<&PackageVersion>> =
43                HashMap::from_iter(result);
44            println!("{}", serde_json::to_string(&rock_to_version_map)?);
45        }
46        OutputFormat::Text => {
47            for (key, versions) in result.into_iter().sorted() {
48                let mut tree = StringTreeNode::new(key.to_string().to_owned());
49
50                for version in versions {
51                    tree.push(version.to_string());
52                }
53
54                println!("{}", tree.to_string_with_format(&formatting)?);
55            }
56        }
57    }
58
59    Ok(())
60}
lux_cli/search.rs

lux_cli/
search.rs
