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
use errors::*; use codeviz_common::ElementFormat; #[derive(Debug, Clone)] pub enum Name { Imported(ImportedName), BuiltIn(BuiltInName), Local(LocalName), } impl Name { pub fn imported(module: &str, name: &str) -> ImportedName { ImportedName { module: module.to_owned(), name: name.to_owned(), alias: None, } } pub fn imported_alias(module: &str, name: &str, alias: &str) -> ImportedName { ImportedName { module: module.to_owned(), name: name.to_owned(), alias: Some(alias.to_owned()), } } pub fn built_in(name: &str) -> BuiltInName { BuiltInName { name: name.to_owned() } } pub fn local(name: &str) -> LocalName { LocalName { name: name.to_owned() } } pub fn format<E>(&self, out: &mut E) -> Result<()> where E: ElementFormat { match *self { Name::Imported(ref imported) => { if let Some(ref alias) = imported.alias { write!(out, "{}.{}", alias, imported.name.clone()) } else { out.write_str(&imported.name) } } Name::BuiltIn(ref built_in) => out.write_str(&built_in.name), Name::Local(ref local) => out.write_str(&local.name), }?; Ok(()) } } #[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)] pub struct ImportedName { pub module: String, pub name: String, pub alias: Option<String>, } #[derive(Debug, Clone)] pub struct BuiltInName { pub name: String, } #[derive(Debug, Clone)] pub struct LocalName { pub name: String, } impl<'a, T> From<&'a T> for Name where T: Into<Name> + Clone { fn from(value: &'a T) -> Name { value.clone().into() } } impl From<ImportedName> for Name { fn from(value: ImportedName) -> Name { Name::Imported(value) } } impl From<BuiltInName> for Name { fn from(value: BuiltInName) -> Name { Name::BuiltIn(value) } } impl From<LocalName> for Name { fn from(value: LocalName) -> Name { Name::Local(value) } }