From 60b475ea88af637f197c1c710c964846346fa85b Mon Sep 17 00:00:00 2001
From: Brian Anderson <banderson@mozilla.com>
Date: Wed, 18 Jan 2012 14:20:26 -0800
Subject: [PATCH] rustdoc: Parse crate name from #[link] attr

---
 src/rustdoc/attr_parser.rs | 42 +++++++++++++++++++++++++++++++++++---
 1 file changed, 39 insertions(+), 3 deletions(-)

diff --git a/src/rustdoc/attr_parser.rs b/src/rustdoc/attr_parser.rs
index 2e82d987b26..1b58d8382f5 100644
--- a/src/rustdoc/attr_parser.rs
+++ b/src/rustdoc/attr_parser.rs
@@ -9,8 +9,12 @@ import rustc::syntax::ast;
 import rustc::front::attr;
 import core::tuple;
 
-export fn_attrs, arg_attrs;
-export parse_fn;
+export crate_attrs, fn_attrs, arg_attrs;
+export parse_crate, parse_fn;
+
+type crate_attrs = {
+    name: option<str>
+};
 
 type fn_attrs = {
     brief: option<str>,
@@ -44,6 +48,38 @@ fn doc_meta(
     }
 }
 
+fn parse_crate(attrs: [ast::attribute]) -> crate_attrs {
+    let link_metas = attr::find_linkage_metas(attrs);
+
+    {
+        name: attr::meta_item_value_from_list(link_metas, "name")
+    }
+}
+
+#[test]
+fn should_extract_crate_name_from_link_attribute() {
+    let source = "#[link(name = \"snuggles\")]";
+    let attrs = test::parse_attributes(source);
+    let attrs = parse_crate(attrs);
+    assert attrs.name == some("snuggles");
+}
+
+#[test]
+fn should_not_extract_crate_name_if_no_link_attribute() {
+    let source = "";
+    let attrs = test::parse_attributes(source);
+    let attrs = parse_crate(attrs);
+    assert attrs.name == none;
+}
+
+#[test]
+fn should_not_extract_crate_name_if_no_name_value_in_link_attribute() {
+    let source = "#[link(whatever)]";
+    let attrs = test::parse_attributes(source);
+    let attrs = parse_crate(attrs);
+    assert attrs.name == none;
+}
+
 fn parse_fn(
     attrs: [ast::attribute]
 ) -> fn_attrs {
@@ -114,7 +150,7 @@ fn parse_fn_(
 }
 
 #[cfg(test)]
-mod tests {
+mod test {
 
     fn parse_attributes(source: str) -> [ast::attribute] {
         import rustc::syntax::parse::parser;