rustdoc/passes/

propagate_doc_cfg.rs

//! Propagates `#[doc(cfg(…))]` ([RFC 3631]) to child items.
//!
//! [RFC 3631]: https://rust-lang.github.io/rfcs/3631-rustdoc-cfgs-handling.html

use rustc_attr_ir::{Attribute, AttributeKind, DocAttribute, find_attr};
use rustc_data_structures::fx::FxHashMap;
use rustc_span::{ExpnKind, MacroKind};

use crate::clean::inline::{load_attrs, merge_attrs};
use crate::clean::{CfgInfo, Crate, Item, ItemId, ItemKind};
use crate::core::DocContext;
use crate::fold::DocFolder;

pub(super) fn propagate_doc_cfg(cr: Crate, cx: &mut DocContext<'_>) -> Crate {
    if cx.tcx.features().doc_cfg() {
        CfgPropagator { cx, cfg_info: CfgInfo::default(), impl_cfg_info: FxHashMap::default() }
            .fold_crate(cr)
    } else {
        cr
    }
}

struct CfgPropagator<'a, 'tcx> {
    cx: &'a mut DocContext<'tcx>,
    cfg_info: CfgInfo,

    /// To ensure the `doc_cfg` feature works with how `rustdoc` handles impls, we need to store
    /// the `cfg` info of `impl`s placeholder to use them later on the "real" impl item.
    impl_cfg_info: FxHashMap<ItemId, CfgInfo>,
}

/// This function goes through the attributes list (`new_attrs`) and extract the `cfg` tokens from
/// it and put them into `attrs`.
fn add_only_cfg_attributes(attrs: &mut Vec<Attribute>, new_attrs: &[Attribute]) {
    for attr in new_attrs {
        if let Attribute::Parsed(AttributeKind::Doc(d)) = attr
            && !d.cfg.is_empty()
        {
            let mut new_attr = DocAttribute::default();
            new_attr.cfg = d.cfg.clone();
            attrs.push(Attribute::Parsed(AttributeKind::Doc(Box::new(new_attr))));
        } else if let Attribute::Parsed(AttributeKind::CfgTrace(..)) = attr {
            // If it's a `cfg()` attribute, we keep it.
            attrs.push(attr.clone());
        }
    }
}

/// This function goes through the attributes list (`new_attrs`) and extracts the attributes that
/// affect the cfg state propagated to detached items.
fn add_cfg_state_attributes(attrs: &mut Vec<Attribute>, new_attrs: &[Attribute]) {
    for attr in new_attrs {
        if let Attribute::Parsed(AttributeKind::Doc(d)) = attr
            && (!d.cfg.is_empty() || !d.auto_cfg.is_empty() || !d.auto_cfg_change.is_empty())
        {
            let mut new_attr = DocAttribute::default();
            new_attr.cfg = d.cfg.clone();
            new_attr.auto_cfg = d.auto_cfg.clone();
            new_attr.auto_cfg_change = d.auto_cfg_change.clone();
            attrs.push(Attribute::Parsed(AttributeKind::Doc(Box::new(new_attr))));
        } else if let Attribute::Parsed(AttributeKind::CfgTrace(..)) = attr {
            // If it's a `cfg()` attribute, we keep it.
            attrs.push(attr.clone());
        }
    }
}

impl CfgPropagator<'_, '_> {
    // Some items need to merge their attributes with their parents' otherwise a few of them
    // (mostly `cfg` ones) will be missing.
    fn merge_with_parent_attributes(&mut self, item: &mut Item) {
        let mut attrs = Vec::new();
        // We need to merge an item attributes with its parent's in case it's an impl as an
        // impl might not be defined in the same module as the item it implements.
        //
        // Same if it's an inlined item: we need to get the full original `cfg`.
        //
        // Otherwise, `cfg_info` already tracks everything we need so nothing else to do!
        if matches!(item.kind, ItemKind::ImplItem(_)) || item.inline_stmt_id.is_some() {
            if let Some(mut next_def_id) = item.item_id.as_local_def_id() {
                while let Some(parent_def_id) = self.cx.tcx.opt_local_parent(next_def_id) {
                    let x = load_attrs(self.cx.tcx, parent_def_id.to_def_id());
                    add_only_cfg_attributes(&mut attrs, x);
                    next_def_id = parent_def_id;
                }
            }
        }
        // We also need to merge an item attributes with its parent's in case it's a macro with
        // the `#[macro_export]` attribute, because it might not be defined at crate root.
        else if matches!(item.kind, ItemKind::MacroItem(_, _))
            && item
                .inner
                .attrs
                .other_attrs
                .iter()
                .any(|attr| matches!(attr, Attribute::Parsed(AttributeKind::MacroExport { .. })))
        {
            for parent_def_id in &item.cfg_parent_ids_for_detached_item(self.cx.tcx) {
                let mut parent_attrs = Vec::new();
                add_cfg_state_attributes(
                    &mut parent_attrs,
                    load_attrs(self.cx.tcx, parent_def_id.to_def_id()),
                );
                merge_attrs(self.cx.tcx, &[], Some((&parent_attrs, None)), &mut self.cfg_info);
            }
        }

        let (_, cfg) = merge_attrs(
            self.cx.tcx,
            item.attrs.other_attrs.as_slice(),
            Some((&attrs, None)),
            &mut self.cfg_info,
        );
        item.inner.cfg = cfg;
    }
}

impl DocFolder for CfgPropagator<'_, '_> {
    fn fold_item(&mut self, mut item: Item) -> Option<Item> {
        let old_cfg_info = self.cfg_info.clone();

        // If we have an impl, we check if it has an associated `cfg` "context", and if so we will
        // use that context instead of the actual (wrong) one.
        if let ItemKind::ImplItem(_) = item.kind
            && let Some(cfg_info) = self.impl_cfg_info.remove(&item.item_id)
        {
            self.cfg_info = cfg_info;
        }
        if let ItemKind::PlaceholderImplItem = item.kind {
            if let Some(impl_def_id) = item.item_id.as_def_id() {
                let tcx = self.cx.tcx;
                let expn_data = tcx.expn_that_defined(impl_def_id).expn_data();
                if matches!(expn_data.kind, ExpnKind::Macro(MacroKind::Derive, _))
                    // This impl block comes from a `derive` expansion, so we want to retrieve
                    // the `cfg_attr` if any.
                    && let Some(self_ty_def_id) = tcx
                        .type_of(impl_def_id)
                        .instantiate_identity()
                        .skip_norm_wip()
                        .ty_adt_def()
                        .map(|adt| adt.did())
                    && let self_ty_attrs = load_attrs(tcx, self_ty_def_id)
                    && let Some(cfgs_attr_trace) =
                        find_attr!(self_ty_attrs, CfgAttrTrace(cfgs) => cfgs)
                    && !cfgs_attr_trace.is_empty()
                {
                    // We retrieve the `cfg_attr` of the `derive` this `impl` comes from.
                    let derive_span = expn_data.call_site;
                    let attrs_iter = Attribute::Parsed(AttributeKind::CfgTrace(
                        cfgs_attr_trace
                            .iter()
                            .filter(|(_, span)| span.contains(derive_span))
                            .cloned()
                            .collect(),
                    ));
                    crate::clean::extract_cfg_from_attrs(
                        std::iter::once(&attrs_iter),
                        tcx,
                        &mut self.cfg_info,
                    );
                }
            }
            // If we have a placeholder impl, we store the current `cfg` "context" to be used
            // on the actual impl later on (the impls are generated after we go through the whole
            // AST so they're stored in the `krate` object at the end).
            self.impl_cfg_info.insert(item.item_id, self.cfg_info.clone());
        } else {
            self.merge_with_parent_attributes(&mut item);
        }

        let result = self.fold_item_recur(item);
        self.cfg_info = old_cfg_info;

        Some(result)
    }
}