2014-05-26 14:48:54 -07:00
|
|
|
//! Used by `rustc` when compiling a plugin crate.
|
|
|
|
|
2014-05-24 16:16:10 -07:00
|
|
|
use syntax::ast;
|
2015-09-14 21:58:20 +12:00
|
|
|
use syntax::attr;
|
2016-06-21 18:08:13 -04:00
|
|
|
use syntax_pos::Span;
|
2016-11-02 18:22:59 -04:00
|
|
|
use rustc::hir::itemlikevisit::ItemLikeVisitor;
|
2016-03-29 08:50:44 +03:00
|
|
|
use rustc::hir;
|
2019-01-13 01:06:50 +01:00
|
|
|
use rustc::hir::def_id::{CrateNum, DefId, LOCAL_CRATE};
|
|
|
|
use rustc::ty::TyCtxt;
|
|
|
|
use rustc::ty::query::Providers;
|
2013-12-25 11:10:33 -07:00
|
|
|
|
2014-05-24 16:16:10 -07:00
|
|
|
struct RegistrarFinder {
|
2014-02-28 13:09:09 -08:00
|
|
|
registrars: Vec<(ast::NodeId, Span)> ,
|
2013-12-25 11:10:33 -07:00
|
|
|
}
|
|
|
|
|
2016-11-02 18:22:59 -04:00
|
|
|
impl<'v> ItemLikeVisitor<'v> for RegistrarFinder {
|
2015-07-31 00:04:06 -07:00
|
|
|
fn visit_item(&mut self, item: &hir::Item) {
|
2018-07-11 23:36:06 +08:00
|
|
|
if let hir::ItemKind::Fn(..) = item.node {
|
2015-02-01 21:53:25 -05:00
|
|
|
if attr::contains_name(&item.attrs,
|
2014-11-29 16:41:21 -05:00
|
|
|
"plugin_registrar") {
|
|
|
|
self.registrars.push((item.id, item.span));
|
2013-12-25 11:10:33 -07:00
|
|
|
}
|
|
|
|
}
|
2016-12-04 04:21:06 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
fn visit_trait_item(&mut self, _trait_item: &hir::TraitItem) {
|
2013-12-25 11:10:33 -07:00
|
|
|
}
|
2016-11-04 18:20:15 -04:00
|
|
|
|
|
|
|
fn visit_impl_item(&mut self, _impl_item: &hir::ImplItem) {
|
|
|
|
}
|
2013-12-25 11:10:33 -07:00
|
|
|
}
|
|
|
|
|
2014-05-24 16:16:10 -07:00
|
|
|
/// Find the function marked with `#[plugin_registrar]`, if any.
|
2019-01-13 01:06:50 +01:00
|
|
|
pub fn find_plugin_registrar<'tcx>(tcx: TyCtxt<'_, 'tcx, 'tcx>) -> Option<DefId> {
|
|
|
|
tcx.plugin_registrar_fn(LOCAL_CRATE)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn plugin_registrar_fn<'tcx>(
|
|
|
|
tcx: TyCtxt<'_, 'tcx, 'tcx>,
|
|
|
|
cnum: CrateNum,
|
|
|
|
) -> Option<DefId> {
|
|
|
|
assert_eq!(cnum, LOCAL_CRATE);
|
2016-01-29 15:04:07 -05:00
|
|
|
|
2014-05-24 16:16:10 -07:00
|
|
|
let mut finder = RegistrarFinder { registrars: Vec::new() };
|
2019-01-13 01:06:50 +01:00
|
|
|
tcx.hir().krate().visit_all_item_likes(&mut finder);
|
2013-12-25 11:10:33 -07:00
|
|
|
|
2014-05-24 16:16:10 -07:00
|
|
|
match finder.registrars.len() {
|
2013-12-25 11:10:33 -07:00
|
|
|
0 => None,
|
|
|
|
1 => {
|
2014-05-24 16:16:10 -07:00
|
|
|
let (node_id, _) = finder.registrars.pop().unwrap();
|
2019-01-13 01:06:50 +01:00
|
|
|
Some(tcx.hir().local_def_id(node_id))
|
2013-12-25 11:10:33 -07:00
|
|
|
},
|
|
|
|
_ => {
|
2019-01-13 01:06:50 +01:00
|
|
|
let diagnostic = tcx.sess.diagnostic();
|
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();
|
2015-12-14 11:17:55 +13:00
|
|
|
diagnostic.abort_if_errors();
|
2013-12-25 11:10:33 -07:00
|
|
|
unreachable!();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2019-01-13 01:06:50 +01:00
|
|
|
|
|
|
|
|
|
|
|
pub fn provide(providers: &mut Providers<'_>) {
|
|
|
|
*providers = Providers {
|
|
|
|
plugin_registrar_fn,
|
|
|
|
..*providers
|
|
|
|
};
|
|
|
|
}
|