refact(rustup): fetch installed toolchains and components

Refactor the functions used for fetching installed
components and toolchains.

Signed-off-by: innocentzero <isfarulhaque@proton.me>
This commit is contained in:
innocentzero
2024-04-02 04:51:15 +05:30
parent d4551a6a2b
commit ea2c51620a
+55 -13
View File
@@ -26,13 +26,26 @@ impl Backend for Rustup {
impl_backend_constants!();
fn get_all_installed_packages(&self) -> anyhow::Result<HashSet<Package>> {
let mut cmd = Command::new(self.get_binary());
let packages: HashSet<Package> = run_rustup_command(&mut cmd, SWITCHES_INFO)
let mut toolchains_vec = self
.run_toolchain_command(&[&"toolchain", &"list"])
.context("Getting installed toolchains")?;
let mut toolchains: HashSet<Package> = toolchains_vec
.iter()
.map(|name| ["toolchain", name].join("/").into())
.collect();
let packages: HashSet<Package> = self
.run_component_command(
&[&"component", &"list", &"--installed", &"--toolchain"],
&mut toolchains_vec,
)
.context("Getting installed components")?
.iter()
.map(|name| ["component/", name].join("").into())
.map(|name| ["component", name].join("/").into())
.collect();
Ok(packages)
toolchains.extend(packages.into_iter());
Ok(toolchains)
}
fn get_explicitly_installed_packages(&self) -> anyhow::Result<HashSet<Package>> {
@@ -43,27 +56,56 @@ impl Backend for Rustup {
fn make_dependency(&self, _: &[Package]) -> anyhow::Result<std::process::ExitStatus> {
panic!("Not supported by {}", BINARY)
}
}
fn run_rustup_command(cmd: &mut Command, args: &[&str]) -> Result<Vec<String>, anyhow::Error> {
cmd.args(args);
let output = String::from_utf8(cmd.output()?.stdout)?;
impl Rustup {
pub(crate) fn new() -> Self {
Self {
packages: HashSet::new(),
}
}
fn run_component_command(
&self,
args: &[&str],
toolchains: &mut Vec<String>,
) -> Result<Vec<String>, anyhow::Error> {
let mut val = Vec::new();
for toolchain in toolchains {
let mut cmd = Command::new(self.get_binary());
cmd.args(args);
cmd.arg(&toolchain);
let output = String::from_utf8(cmd.output()?.stdout)?;
for i in output.lines() {
let mut it = i.splitn(3, "-");
let component = it.next().expect("Component name is empty!");
match component {
"cargo" | "rustfmt" | "clippy" | "miri" | "rls" | "rustc" => {
val.push(component.to_string());
val.push([toolchain, component].join("/"));
}
_ => {
val.push(
component.to_string()
+ "-"
+ it.next().expect("No such component is managed by rustup"),
);
let component = [
component,
it.next().expect("No such component is managed by rustup"),
]
.join("-");
val.push([toolchain, component.as_str()].join("/"));
}
}
}
}
Ok(val)
}
fn run_toolchain_command(&self, args: &[&str]) -> Result<Vec<String>, anyhow::Error> {
let mut cmd = Command::new(self.get_binary());
cmd.args(args);
let output = String::from_utf8(cmd.output()?.stdout)?;
let mut val = Vec::new();
for i in output.lines() {
let mut it = i.splitn(2, "-");
val.push(it.next().expect("Toolchain name is empty.").to_string());
}
Ok(val)
}
}