From 58a34a4dde1aac009d5413b4eb414194764c5859 Mon Sep 17 00:00:00 2001 From: Guillaume Gomez Date: Thu, 28 Jan 2021 17:05:22 +0100 Subject: [PATCH] Remove is_spotlight field from `Trait` --- src/librustdoc/clean/inline.rs | 6 +++--- src/librustdoc/clean/mod.rs | 3 --- src/librustdoc/clean/types.rs | 3 +-- src/librustdoc/clean/utils.rs | 18 +++++++++++++++++- src/librustdoc/core.rs | 2 +- src/librustdoc/fold.rs | 4 ++-- src/librustdoc/formats/cache.rs | 10 ++++++++-- src/librustdoc/html/render/mod.rs | 9 +++++---- src/librustdoc/json/conversions.rs | 2 +- src/librustdoc/json/mod.rs | 2 +- 10 files changed, 39 insertions(+), 20 deletions(-) diff --git a/src/librustdoc/clean/inline.rs b/src/librustdoc/clean/inline.rs index ea75d1614bd..519ec7216e3 100644 --- a/src/librustdoc/clean/inline.rs +++ b/src/librustdoc/clean/inline.rs @@ -195,14 +195,12 @@ crate fn build_external_trait(cx: &mut DocContext<'_>, did: DefId) -> clean::Tra let generics = (cx.tcx.generics_of(did), predicates).clean(cx); let generics = filter_non_trait_generics(did, generics); let (generics, supertrait_bounds) = separate_supertrait_bounds(generics); - let is_spotlight = load_attrs(cx, did).clean(cx).has_doc_flag(sym::spotlight); let is_auto = cx.tcx.trait_is_auto(did); clean::Trait { unsafety: cx.tcx.trait_def(did).unsafety, generics, items: trait_items, bounds: supertrait_bounds, - is_spotlight, is_auto, } } @@ -626,6 +624,8 @@ crate fn record_extern_trait(cx: &mut DocContext<'_>, did: DefId) { debug!("record_extern_trait: {:?}", did); let trait_ = build_external_trait(cx, did); - cx.external_traits.borrow_mut().insert(did, trait_); + cx.external_traits + .borrow_mut() + .insert(did, (trait_, clean::utils::has_doc_flag(cx.tcx.get_attrs(did), sym::spotlight))); cx.active_extern_traits.remove(&did); } diff --git a/src/librustdoc/clean/mod.rs b/src/librustdoc/clean/mod.rs index b6e70462105..7aa34333e17 100644 --- a/src/librustdoc/clean/mod.rs +++ b/src/librustdoc/clean/mod.rs @@ -2003,14 +2003,11 @@ impl Clean> for (&hir::Item<'_>, Option) { .iter() .map(|ti| cx.tcx.hir().trait_item(ti.id).clean(cx)) .collect(); - let attrs = item.attrs.clean(cx); - let is_spotlight = attrs.has_doc_flag(sym::spotlight); TraitItem(Trait { unsafety, items, generics: generics.clean(cx), bounds: bounds.clean(cx), - is_spotlight, is_auto: is_auto.clean(cx), }) } diff --git a/src/librustdoc/clean/types.rs b/src/librustdoc/clean/types.rs index 9a2319f6e37..0289ee9afb7 100644 --- a/src/librustdoc/clean/types.rs +++ b/src/librustdoc/clean/types.rs @@ -57,7 +57,7 @@ crate struct Crate { crate primitives: Vec<(DefId, PrimitiveType)>, // These are later on moved into `CACHEKEY`, leaving the map empty. // Only here so that they can be filtered through the rustdoc passes. - crate external_traits: Rc>>, + crate external_traits: Rc>>, crate masked_crates: FxHashSet, crate collapsed: bool, } @@ -1185,7 +1185,6 @@ crate struct Trait { crate items: Vec, crate generics: Generics, crate bounds: Vec, - crate is_spotlight: bool, crate is_auto: bool, } diff --git a/src/librustdoc/clean/utils.rs b/src/librustdoc/clean/utils.rs index d2eee49f0c9..3473cc6931e 100644 --- a/src/librustdoc/clean/utils.rs +++ b/src/librustdoc/clean/utils.rs @@ -7,13 +7,14 @@ use crate::clean::{ }; use crate::core::DocContext; +use rustc_attr::list_contains_name; use rustc_data_structures::fx::FxHashSet; use rustc_hir as hir; use rustc_hir::def::{DefKind, Res}; use rustc_hir::def_id::{DefId, LOCAL_CRATE}; use rustc_middle::mir::interpret::ConstValue; use rustc_middle::ty::subst::{GenericArgKind, SubstsRef}; -use rustc_middle::ty::{self, DefIdTree, Ty, TyCtxt}; +use rustc_middle::ty::{self, Attributes, DefIdTree, Ty, TyCtxt}; use rustc_span::symbol::{kw, sym, Symbol}; use std::mem; @@ -520,3 +521,18 @@ crate fn find_nearest_parent_module(tcx: TyCtxt<'_>, def_id: DefId) -> Option, flag: Symbol) -> bool { + attrs.iter().any(|attr| { + attr.has_name(sym::doc) + && attr.meta_item_list().map_or(false, |l| list_contains_name(&l, flag)) + }) +} diff --git a/src/librustdoc/core.rs b/src/librustdoc/core.rs index 8fceb00eeae..5f24ec7b3f7 100644 --- a/src/librustdoc/core.rs +++ b/src/librustdoc/core.rs @@ -55,7 +55,7 @@ crate struct DocContext<'tcx> { /// Later on moved into `cache` crate renderinfo: RenderInfo, /// Later on moved through `clean::Crate` into `cache` - crate external_traits: Rc>>, + crate external_traits: Rc>>, /// Used while populating `external_traits` to ensure we don't process the same trait twice at /// the same time. crate active_extern_traits: FxHashSet, diff --git a/src/librustdoc/fold.rs b/src/librustdoc/fold.rs index b2773a29e29..752233ade37 100644 --- a/src/librustdoc/fold.rs +++ b/src/librustdoc/fold.rs @@ -91,9 +91,9 @@ crate trait DocFolder: Sized { { let external_traits = { std::mem::take(&mut *c.external_traits.borrow_mut()) }; - for (k, mut v) in external_traits { + for (k, (mut v, is_spotlight)) in external_traits { v.items = v.items.into_iter().filter_map(|i| self.fold_item(i)).collect(); - c.external_traits.borrow_mut().insert(k, v); + c.external_traits.borrow_mut().insert(k, (v, is_spotlight)); } } c diff --git a/src/librustdoc/formats/cache.rs b/src/librustdoc/formats/cache.rs index ce1204c7be1..a64eeb5bc58 100644 --- a/src/librustdoc/formats/cache.rs +++ b/src/librustdoc/formats/cache.rs @@ -7,6 +7,7 @@ use rustc_hir::def_id::{CrateNum, DefId, CRATE_DEF_INDEX}; use rustc_middle::middle::privacy::AccessLevels; use rustc_middle::ty::TyCtxt; use rustc_span::source_map::FileName; +use rustc_span::symbol::sym; use rustc_span::Symbol; use crate::clean::{self, GetDefId}; @@ -64,7 +65,9 @@ crate struct Cache { /// Implementations of a crate should inherit the documentation of the /// parent trait if no extra documentation is specified, and default methods /// should show up in documentation about trait implementations. - crate traits: FxHashMap, + /// + /// The `bool` parameter is about if the trait is `spotlight`. + crate traits: FxHashMap, /// When rendering traits, it's often useful to be able to list all /// implementors of the trait, and this mapping is exactly, that: a mapping @@ -244,10 +247,13 @@ impl<'a, 'tcx> DocFolder for CacheBuilder<'a, 'tcx> { } } + let tcx = self.tcx; // Propagate a trait method's documentation to all implementors of the // trait. if let clean::TraitItem(ref t) = *item.kind { - self.cache.traits.entry(item.def_id).or_insert_with(|| t.clone()); + self.cache.traits.entry(item.def_id).or_insert_with(|| { + (t.clone(), clean::utils::has_doc_flag(tcx.get_attrs(item.def_id), sym::spotlight)) + }); } // Collect all the implementors of traits. diff --git a/src/librustdoc/html/render/mod.rs b/src/librustdoc/html/render/mod.rs index 7f122bb8cb5..ab3c9bab64f 100644 --- a/src/librustdoc/html/render/mod.rs +++ b/src/librustdoc/html/render/mod.rs @@ -3687,8 +3687,9 @@ fn spotlight_decl(decl: &clean::FnDecl, cache: &Cache) -> String { if let Some(impls) = cache.impls.get(&did) { for i in impls { let impl_ = i.inner_impl(); - if impl_.trait_.def_id_full(cache).map_or(false, |d| cache.traits[&d].is_spotlight) - { + if impl_.trait_.def_id().map_or(false, |d| { + cache.traits.get(&d).map(|(_, is_spotlight)| *is_spotlight).unwrap_or(false) + }) { if out.is_empty() { write!( &mut out, @@ -3979,7 +3980,7 @@ fn render_impl( false, outer_version, outer_const_version, - trait_, + trait_.map(|(t, _)| t), show_def_docs, ); } @@ -4024,7 +4025,7 @@ fn render_impl( // We don't emit documentation for default items if they appear in the // Implementations on Foreign Types or Implementors sections. if show_default_items { - if let Some(t) = trait_ { + if let Some((t, _)) = trait_ { render_default_items( w, cx, diff --git a/src/librustdoc/json/conversions.rs b/src/librustdoc/json/conversions.rs index b248fcdefbe..da09d49a006 100644 --- a/src/librustdoc/json/conversions.rs +++ b/src/librustdoc/json/conversions.rs @@ -408,7 +408,7 @@ impl From for FnDecl { impl From for Trait { fn from(trait_: clean::Trait) -> Self { - let clean::Trait { unsafety, items, generics, bounds, is_spotlight: _, is_auto } = trait_; + let clean::Trait { unsafety, items, generics, bounds, is_auto } = trait_; Trait { is_auto, is_unsafe: unsafety == rustc_hir::Unsafety::Unsafe, diff --git a/src/librustdoc/json/mod.rs b/src/librustdoc/json/mod.rs index ce88e09b174..aa093ba0a77 100644 --- a/src/librustdoc/json/mod.rs +++ b/src/librustdoc/json/mod.rs @@ -84,7 +84,7 @@ impl JsonRenderer<'tcx> { Rc::clone(&self.cache) .traits .iter() - .filter_map(|(&id, trait_item)| { + .filter_map(|(&id, (trait_item, _))| { // only need to synthesize items for external traits if !id.is_local() { trait_item.items.clone().into_iter().for_each(|i| self.item(i).unwrap());