src/librustdoc/passes/stripper.rs

   1 //! A collection of utility functions for the `strip_*` passes.
   2 use rustc_hir::def_id::DefId;
   3 use rustc_middle::ty::{TyCtxt, Visibility};
   4 use rustc_span::symbol::sym;
   5 use std::mem;
   6
   7 use crate::clean::{self, Item, ItemId, ItemIdSet, NestedAttributesExt};
   8 use crate::fold::{strip_item, DocFolder};
   9 use crate::formats::cache::Cache;
  10 use crate::visit_lib::RustdocEffectiveVisibilities;
  11
  12 pub(crate) struct Stripper<'a, 'tcx> {
  13     pub(crate) retained: &'a mut ItemIdSet,
  14     pub(crate) effective_visibilities: &'a RustdocEffectiveVisibilities,
  15     pub(crate) update_retained: bool,
  16     pub(crate) is_json_output: bool,
  17     pub(crate) tcx: TyCtxt<'tcx>,
  18 }
  19
  20 // We need to handle this differently for the JSON output because some non exported items could
  21 // be used in public API. And so, we need these items as well. `is_exported` only checks if they
  22 // are in the public API, which is not enough.
  23 #[inline]
  24 fn is_item_reachable(
  25     tcx: TyCtxt<'_>,
  26     is_json_output: bool,
  27     effective_visibilities: &RustdocEffectiveVisibilities,
  28     item_id: ItemId,
  29 ) -> bool {
  30     if is_json_output {
  31         effective_visibilities.is_reachable(tcx, item_id.expect_def_id())
  32     } else {
  33         effective_visibilities.is_exported(tcx, item_id.expect_def_id())
  34     }
  35 }
  36
  37 impl<'a, 'tcx> DocFolder for Stripper<'a, 'tcx> {
  38     fn fold_item(&mut self, i: Item) -> Option<Item> {
  39         match *i.kind {
  40             clean::StrippedItem(..) => {
  41                 // We need to recurse into stripped modules to strip things
  42                 // like impl methods but when doing so we must not add any
  43                 // items to the `retained` set.
  44                 debug!("Stripper: recursing into stripped {:?} {:?}", i.type_(), i.name);
  45                 let old = mem::replace(&mut self.update_retained, false);
  46                 let ret = self.fold_item_recur(i);
  47                 self.update_retained = old;
  48                 return Some(ret);
  49             }
  50             // These items can all get re-exported
  51             clean::OpaqueTyItem(..)
  52             | clean::TypedefItem(..)
  53             | clean::StaticItem(..)
  54             | clean::StructItem(..)
  55             | clean::EnumItem(..)
  56             | clean::TraitItem(..)
  57             | clean::FunctionItem(..)
  58             | clean::VariantItem(..)
  59             | clean::MethodItem(..)
  60             | clean::ForeignFunctionItem(..)
  61             | clean::ForeignStaticItem(..)
  62             | clean::ConstantItem(..)
  63             | clean::UnionItem(..)
  64             | clean::AssocConstItem(..)
  65             | clean::AssocTypeItem(..)
  66             | clean::TraitAliasItem(..)
  67             | clean::MacroItem(..)
  68             | clean::ForeignTypeItem => {
  69                 let item_id = i.item_id;
  70                 if item_id.is_local()
  71                     && !is_item_reachable(
  72                         self.tcx,
  73                         self.is_json_output,
  74                         self.effective_visibilities,
  75                         item_id,
  76                     )
  77                 {
  78                     debug!("Stripper: stripping {:?} {:?}", i.type_(), i.name);
  79                     return None;
  80                 }
  81             }
  82
  83             clean::StructFieldItem(..) => {
  84                 if i.visibility(self.tcx) != Some(Visibility::Public) {
  85                     return Some(strip_item(i));
  86                 }
  87             }
  88
  89             clean::ModuleItem(..) => {
  90                 if i.item_id.is_local() && i.visibility(self.tcx) != Some(Visibility::Public) {
  91                     debug!("Stripper: stripping module {:?}", i.name);
  92                     let old = mem::replace(&mut self.update_retained, false);
  93                     let ret = strip_item(self.fold_item_recur(i));
  94                     self.update_retained = old;
  95                     return Some(ret);
  96                 }
  97             }
  98
  99             // handled in the `strip-priv-imports` pass
 100             clean::ExternCrateItem { .. } => {}
 101             clean::ImportItem(ref imp) => {
 102                 // Because json doesn't inline imports from private modules, we need to mark
 103                 // the imported item as retained so it's impls won't be stripped.
 104                 //
 105                 // FIXME: Is it necessary to check for json output here: See
 106                 // https://github.com/rust-lang/rust/pull/100325#discussion_r941495215
 107                 if let Some(did) = imp.source.did && self.is_json_output {
 108                     self.retained.insert(did.into());
 109                 }
 110             }
 111
 112             clean::ImplItem(..) => {}
 113
 114             // tymethods etc. have no control over privacy
 115             clean::TyMethodItem(..) | clean::TyAssocConstItem(..) | clean::TyAssocTypeItem(..) => {}
 116
 117             // Proc-macros are always public
 118             clean::ProcMacroItem(..) => {}
 119
 120             // Primitives are never stripped
 121             clean::PrimitiveItem(..) => {}
 122
 123             // Keywords are never stripped
 124             clean::KeywordItem => {}
 125         }
 126
 127         let fastreturn = match *i.kind {
 128             // nothing left to do for traits (don't want to filter their
 129             // methods out, visibility controlled by the trait)
 130             clean::TraitItem(..) => true,
 131
 132             // implementations of traits are always public.
 133             clean::ImplItem(ref imp) if imp.trait_.is_some() => true,
 134             // Variant fields have inherited visibility
 135             clean::VariantItem(clean::Variant::Struct(..) | clean::Variant::Tuple(..)) => true,
 136             _ => false,
 137         };
 138
 139         let i = if fastreturn {
 140             if self.update_retained {
 141                 self.retained.insert(i.item_id);
 142             }
 143             return Some(i);
 144         } else {
 145             self.fold_item_recur(i)
 146         };
 147
 148         if self.update_retained {
 149             self.retained.insert(i.item_id);
 150         }
 151         Some(i)
 152     }
 153 }
 154
 155 /// This stripper discards all impls which reference stripped items
 156 pub(crate) struct ImplStripper<'a, 'tcx> {
 157     pub(crate) tcx: TyCtxt<'tcx>,
 158     pub(crate) retained: &'a ItemIdSet,
 159     pub(crate) cache: &'a Cache,
 160     pub(crate) is_json_output: bool,
 161     pub(crate) document_private: bool,
 162 }
 163
 164 impl<'a> ImplStripper<'a, '_> {
 165     #[inline]
 166     fn should_keep_impl(&self, item: &Item, for_def_id: DefId) -> bool {
 167         if !for_def_id.is_local() || self.retained.contains(&for_def_id.into()) {
 168             true
 169         } else if self.is_json_output {
 170             // If the "for" item is exported and the impl block isn't `#[doc(hidden)]`, then we
 171             // need to keep it.
 172             self.cache.effective_visibilities.is_exported(self.tcx, for_def_id)
 173                 && !item.attrs.lists(sym::doc).has_word(sym::hidden)
 174         } else {
 175             false
 176         }
 177     }
 178 }
 179
 180 impl<'a> DocFolder for ImplStripper<'a, '_> {
 181     fn fold_item(&mut self, i: Item) -> Option<Item> {
 182         if let clean::ImplItem(ref imp) = *i.kind {
 183             // Impl blocks can be skipped if they are: empty; not a trait impl; and have no
 184             // documentation.
 185             //
 186             // There is one special case: if the impl block contains only private items.
 187             if imp.trait_.is_none() {
 188                 // If the only items present are private ones and we're not rendering private items,
 189                 // we don't document it.
 190                 if !imp.items.is_empty()
 191                     && !self.document_private
 192                     && imp.items.iter().all(|i| {
 193                         let item_id = i.item_id;
 194                         item_id.is_local()
 195                             && !is_item_reachable(
 196                                 self.tcx,
 197                                 self.is_json_output,
 198                                 &self.cache.effective_visibilities,
 199                                 item_id,
 200                             )
 201                     })
 202                 {
 203                     return None;
 204                 } else if imp.items.is_empty() && i.doc_value().is_none() {
 205                     return None;
 206                 }
 207             }
 208             // Because we don't inline in `maybe_inline_local` if the output format is JSON,
 209             // we need to make a special check for JSON output: we want to keep it unless it has
 210             // a `#[doc(hidden)]` attribute if the `for_` type is exported.
 211             if let Some(did) = imp.for_.def_id(self.cache) {
 212                 if !imp.for_.is_assoc_ty() && !self.should_keep_impl(&i, did) {
 213                     debug!("ImplStripper: impl item for stripped type; removing");
 214                     return None;
 215                 }
 216             }
 217             if let Some(did) = imp.trait_.as_ref().map(|t| t.def_id()) {
 218                 if !self.should_keep_impl(&i, did) {
 219                     debug!("ImplStripper: impl item for stripped trait; removing");
 220                     return None;
 221                 }
 222             }
 223             if let Some(generics) = imp.trait_.as_ref().and_then(|t| t.generics()) {
 224                 for typaram in generics {
 225                     if let Some(did) = typaram.def_id(self.cache) {
 226                         if !self.should_keep_impl(&i, did) {
 227                             debug!(
 228                                 "ImplStripper: stripped item in trait's generics; removing impl"
 229                             );
 230                             return None;
 231                         }
 232                     }
 233                 }
 234             }
 235         }
 236         Some(self.fold_item_recur(i))
 237     }
 238 }
 239
 240 /// This stripper discards all private import statements (`use`, `extern crate`)
 241 pub(crate) struct ImportStripper<'tcx> {
 242     pub(crate) tcx: TyCtxt<'tcx>,
 243 }
 244
 245 impl<'tcx> DocFolder for ImportStripper<'tcx> {
 246     fn fold_item(&mut self, i: Item) -> Option<Item> {
 247         match *i.kind {
 248             clean::ExternCrateItem { .. } | clean::ImportItem(..)
 249                 if i.visibility(self.tcx) != Some(Visibility::Public) =>
 250             {
 251                 None
 252             }
 253             _ => Some(self.fold_item_recur(i)),
 254         }
 255     }
 256 }