src/librustc_metadata/link_args.rs

   1 use rustc::hir::itemlikevisit::ItemLikeVisitor;
   2 use rustc::hir;
   3 use rustc::ty::TyCtxt;
   4 use rustc_target::spec::abi::Abi;
   5 use syntax::symbol::sym;
   6
   7 pub fn collect<'tcx>(tcx: TyCtxt<'tcx, 'tcx>) -> Vec<String> {
   8     let mut collector = Collector {
   9         args: Vec::new(),
  10     };
  11     tcx.hir().krate().visit_all_item_likes(&mut collector);
  12
  13     for attr in tcx.hir().krate().attrs.iter() {
  14         if attr.path == sym::link_args {
  15             if let Some(linkarg) = attr.value_str() {
  16                 collector.add_link_args(&linkarg.as_str());
  17             }
  18         }
  19     }
  20
  21     return collector.args
  22 }
  23
  24 struct Collector {
  25     args: Vec<String>,
  26 }
  27
  28 impl<'tcx> ItemLikeVisitor<'tcx> for Collector {
  29     fn visit_item(&mut self, it: &'tcx hir::Item) {
  30         let fm = match it.node {
  31             hir::ItemKind::ForeignMod(ref fm) => fm,
  32             _ => return,
  33         };
  34         if fm.abi == Abi::Rust ||
  35             fm.abi == Abi::RustIntrinsic ||
  36             fm.abi == Abi::PlatformIntrinsic {
  37             return
  38         }
  39
  40         // First, add all of the custom #[link_args] attributes
  41         for m in it.attrs.iter().filter(|a| a.check_name(sym::link_args)) {
  42             if let Some(linkarg) = m.value_str() {
  43                 self.add_link_args(&linkarg.as_str());
  44             }
  45         }
  46     }
  47
  48     fn visit_trait_item(&mut self, _it: &'tcx hir::TraitItem) {}
  49     fn visit_impl_item(&mut self, _it: &'tcx hir::ImplItem) {}
  50 }
  51
  52 impl Collector {
  53     fn add_link_args(&mut self, args: &str) {
  54         self.args.extend(args.split(' ').filter(|s| !s.is_empty()).map(|s| s.to_string()))
  55     }
  56 }