use std::collections::{HashMap, HashSet};
use std::env;
use std::process::Command;
fn get_base_trait_name(after_as: &str) -> Option<String> {
let mut parts = after_as.split('>');
let path = parts.next()?;
let mut rest: Vec<&str> = parts.collect();
let basename: &str = *path.split("::").collect::<Vec<&str>>().last()?;
rest.insert(0, basename);
Some(rest.join(">"))
}
#[allow(clippy::missing_panics_doc, clippy::unwrap_used)]
#[must_use]
pub fn simplify_trait_paths(path: &str) -> String {
let parts: Vec<&str> = path.split(" as ").collect();
if parts.len() == 1 {
path.to_string()
} else {
parts.into_iter()
.enumerate()
.map(|(i, after_as)|
if i % 2 == 0 {after_as.to_string()} else { get_base_trait_name(after_as).unwrap() }
)
.collect::<Vec<String>>()
.join(" as ")
}
}
#[cfg(test)]
mod tests {
use crate::utils::simplify_trait_paths;
#[test]
fn test_1() {
assert_eq!(simplify_trait_paths("<&mut std::collections::hash::table::RawTable<K,V> as std::collections::hash::table::Put<K,V>>::borrow_table_mut"), "<&mut std::collections::hash::table::RawTable<K,V> as Put<K,V>>::borrow_table_mut");
}
#[test]
fn test_2() {
assert_eq!(
simplify_trait_paths("<futures::lock::TryLock<T> as core::ops::deref::Deref>::deref"),
"<futures::lock::TryLock<T> as Deref>::deref"
);
}
#[test]
fn test_3() {
assert_eq!(simplify_trait_paths("<network::proto::state_synchronizer::RequestChunk as ::protobuf::Message>::default_instance"), "<network::proto::state_synchronizer::RequestChunk as Message>::default_instance");
}
#[test]
fn test_4() {
assert_eq!(
simplify_trait_paths("<T as failure::as_fail::AsFail>::as_fail"),
"<T as AsFail>::as_fail"
);
}
}
#[derive(Clone, Default)]
pub struct LabelInfo {
pub short_label: Option<String>,
pub caller_labels: HashSet<String>,
pub debugloc: Option<llvm_ir::DebugLoc>,
}
impl LabelInfo {
pub fn new() -> Self {
Self::default()
}
}
pub struct CallGraph {
pub label_to_label_info: HashMap<String, LabelInfo>,
pub short_label_to_labels: HashMap<String, HashSet<String>>,
}
#[allow(clippy::missing_panics_doc, clippy::expect_used, clippy::unwrap_used)]
pub fn configure_rustup_toolchain() {
let rsup_default = Command::new("rustup")
.arg("default")
.output()
.expect("failed to run rustup to configure toolchain");
let utf8_toolchain = std::str::from_utf8(&rsup_default.stdout)
.unwrap()
.trim_end()
.strip_suffix(" (default)")
.unwrap();
env::set_var("RUSTUP_TOOLCHAIN", utf8_toolchain);
let rustc_version = rustc_version::version_meta().unwrap();
let rustc_llvm_version = rustc_version.llvm_version.unwrap();
assert_eq!(
llvm_ir::llvm_version(),
rustc_llvm_version.major.to_string(),
"Siderophile was configured to use LLVM {}, but the default rustc emits LLVM {}.",
llvm_ir::llvm_version(),
rustc_llvm_version.major,
);
}