src/librustdoc/passes/strip_hidden.rs

   1 //! Strip all doc(hidden) items from the output.
   2 use rustc_span::symbol::sym;
   3 use std::mem;
   4
   5 use crate::clean;
   6 use crate::clean::{Item, ItemIdSet, NestedAttributesExt};
   7 use crate::core::DocContext;
   8 use crate::fold::{strip_item, DocFolder};
   9 use crate::passes::{ImplStripper, Pass};
  10
  11 crate const STRIP_HIDDEN: Pass = Pass {
  12     name: "strip-hidden",
  13     run: strip_hidden,
  14     description: "strips all `#[doc(hidden)]` items from the output",
  15 };
  16
  17 /// Strip items marked `#[doc(hidden)]`
  18 crate fn strip_hidden(krate: clean::Crate, _: &mut DocContext<'_>) -> clean::Crate {
  19     let mut retained = ItemIdSet::default();
  20
  21     // strip all #[doc(hidden)] items
  22     let krate = {
  23         let mut stripper = Stripper { retained: &mut retained, update_retained: true };
  24         stripper.fold_crate(krate)
  25     };
  26
  27     // strip all impls referencing stripped items
  28     let mut stripper = ImplStripper { retained: &retained };
  29     stripper.fold_crate(krate)
  30 }
  31
  32 struct Stripper<'a> {
  33     retained: &'a mut ItemIdSet,
  34     update_retained: bool,
  35 }
  36
  37 impl<'a> DocFolder for Stripper<'a> {
  38     fn fold_item(&mut self, i: Item) -> Option<Item> {
  39         if i.attrs.lists(sym::doc).has_word(sym::hidden) {
  40             debug!("strip_hidden: stripping {:?} {:?}", i.type_(), i.name);
  41             // use a dedicated hidden item for given item type if any
  42             match *i.kind {
  43                 clean::StructFieldItem(..) | clean::ModuleItem(..) => {
  44                     // We need to recurse into stripped modules to
  45                     // strip things like impl methods but when doing so
  46                     // we must not add any items to the `retained` set.
  47                     let old = mem::replace(&mut self.update_retained, false);
  48                     let ret = strip_item(self.fold_item_recur(i));
  49                     self.update_retained = old;
  50                     return Some(ret);
  51                 }
  52                 _ => return None,
  53             }
  54         } else {
  55             if self.update_retained {
  56                 self.retained.insert(i.def_id);
  57             }
  58         }
  59         Some(self.fold_item_recur(i))
  60     }
  61 }