| // Copyright 2017 The Rust Project Developers. See the COPYRIGHT |
| // file at the top-level directory of this distribution and at |
| // http://rust-lang.org/COPYRIGHT. |
| // |
| // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or |
| // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license |
| // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your |
| // option. This file may not be copied, modified, or distributed |
| // except according to those terms. |
| |
| use rustc::hir::itemlikevisit::ItemLikeVisitor; |
| use rustc::hir; |
| use rustc::ty::TyCtxt; |
| use rustc_target::spec::abi::Abi; |
| |
| pub fn collect<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>) -> Vec<String> { |
| let mut collector = Collector { |
| args: Vec::new(), |
| }; |
| tcx.hir.krate().visit_all_item_likes(&mut collector); |
| |
| for attr in tcx.hir.krate().attrs.iter() { |
| if attr.path == "link_args" { |
| if let Some(linkarg) = attr.value_str() { |
| collector.add_link_args(&linkarg.as_str()); |
| } |
| } |
| } |
| |
| return collector.args |
| } |
| |
| struct Collector { |
| args: Vec<String>, |
| } |
| |
| impl<'tcx> ItemLikeVisitor<'tcx> for Collector { |
| fn visit_item(&mut self, it: &'tcx hir::Item) { |
| let fm = match it.node { |
| hir::ItemKind::ForeignMod(ref fm) => fm, |
| _ => return, |
| }; |
| if fm.abi == Abi::Rust || |
| fm.abi == Abi::RustIntrinsic || |
| fm.abi == Abi::PlatformIntrinsic { |
| return |
| } |
| |
| // First, add all of the custom #[link_args] attributes |
| for m in it.attrs.iter().filter(|a| a.check_name("link_args")) { |
| if let Some(linkarg) = m.value_str() { |
| self.add_link_args(&linkarg.as_str()); |
| } |
| } |
| } |
| |
| fn visit_trait_item(&mut self, _it: &'tcx hir::TraitItem) {} |
| fn visit_impl_item(&mut self, _it: &'tcx hir::ImplItem) {} |
| } |
| |
| impl Collector { |
| fn add_link_args(&mut self, args: &str) { |
| self.args.extend(args.split(' ').filter(|s| !s.is_empty()).map(|s| s.to_string())) |
| } |
| } |