64 lines
2.0 KiB
Rust
Raw Normal View History

2013-12-25 11:10:33 -07:00
// Copyright 2012-2013 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.
2014-05-26 14:48:54 -07:00
//! Used by `rustc` when compiling a plugin crate.
use syntax::ast;
use syntax::attr;
use syntax::codemap::Span;
use syntax::errors;
use rustc::dep_graph::DepNode;
use rustc::front::map::Map;
use rustc_front::intravisit::Visitor;
2015-07-31 00:04:06 -07:00
use rustc_front::hir;
2013-12-25 11:10:33 -07:00
struct RegistrarFinder {
registrars: Vec<(ast::NodeId, Span)> ,
2013-12-25 11:10:33 -07:00
}
impl<'v> Visitor<'v> for RegistrarFinder {
2015-07-31 00:04:06 -07:00
fn visit_item(&mut self, item: &hir::Item) {
if let hir::ItemFn(..) = item.node {
if attr::contains_name(&item.attrs,
"plugin_registrar") {
self.registrars.push((item.id, item.span));
2013-12-25 11:10:33 -07:00
}
}
}
}
/// Find the function marked with `#[plugin_registrar]`, if any.
pub fn find_plugin_registrar(diagnostic: &errors::Handler,
hir_map: &Map)
2015-07-31 00:04:06 -07:00
-> Option<ast::NodeId> {
let _task = hir_map.dep_graph.in_task(DepNode::PluginRegistrar);
let krate = hir_map.krate();
let mut finder = RegistrarFinder { registrars: Vec::new() };
krate.visit_all_items(&mut finder);
2013-12-25 11:10:33 -07:00
match finder.registrars.len() {
2013-12-25 11:10:33 -07:00
0 => None,
1 => {
let (node_id, _) = finder.registrars.pop().unwrap();
Some(node_id)
2013-12-25 11:10:33 -07:00
},
_ => {
2015-12-21 10:00:43 +13:00
let mut e = diagnostic.struct_err("multiple plugin registration functions found");
2015-01-31 12:20:46 -05:00
for &(_, span) in &finder.registrars {
2015-12-21 10:00:43 +13:00
e.span_note(span, "one is here");
2013-12-25 11:10:33 -07:00
}
2015-12-21 10:00:43 +13:00
e.emit();
diagnostic.abort_if_errors();
2013-12-25 11:10:33 -07:00
unreachable!();
}
}
}