1use std::sync::{Arc, LazyLock};
2use std::{io, mem};
3
4use rustc_data_structures::fx::{FxHashMap, FxHashSet, FxIndexMap};
5use rustc_data_structures::unord::UnordSet;
6use rustc_driver::USING_INTERNAL_FEATURES;
7use rustc_errors::TerminalUrl;
8use rustc_errors::codes::*;
9use rustc_errors::emitter::{
10 DynEmitter, HumanEmitter, HumanReadableErrorType, OutputTheme, stderr_destination,
11};
12use rustc_errors::json::JsonEmitter;
13use rustc_feature::UnstableFeatures;
14use rustc_hir::def::Res;
15use rustc_hir::def_id::{DefId, DefIdMap, DefIdSet, LocalDefId};
16use rustc_hir::intravisit::{self, Visitor};
17use rustc_hir::{HirId, Path};
18use rustc_lint::{MissingDoc, late_lint_mod};
19use rustc_middle::hir::nested_filter;
20use rustc_middle::ty::{self, ParamEnv, Ty, TyCtxt};
21use rustc_session::config::{
22 self, CrateType, ErrorOutputType, Input, OutFileName, OutputType, OutputTypes, ResolveDocLinks,
23};
24pub(crate) use rustc_session::config::{Options, UnstableOptions};
25use rustc_session::{Session, lint};
26use rustc_span::source_map;
27use rustc_span::symbol::sym;
28use tracing::{debug, info};
29
30use crate::clean::inline::build_trait;
31use crate::clean::{self, ItemId};
32use crate::config::{Options as RustdocOptions, OutputFormat, RenderOptions};
33use crate::formats::cache::Cache;
34use crate::passes;
35use crate::passes::Condition::*;
36use crate::passes::collect_intra_doc_links::LinkCollector;
37
38pub(crate) struct DocContext<'tcx> {
39 pub(crate) tcx: TyCtxt<'tcx>,
40 pub(crate) param_env: ParamEnv<'tcx>,
44 pub(crate) external_traits: FxIndexMap<DefId, clean::Trait>,
46 pub(crate) active_extern_traits: DefIdSet,
49 pub(crate) args: DefIdMap<clean::GenericArg>,
56 pub(crate) current_type_aliases: DefIdMap<usize>,
57 pub(crate) impl_trait_bounds: FxHashMap<ImplTraitParam, Vec<clean::GenericBound>>,
59 pub(crate) generated_synthetics: FxHashSet<(Ty<'tcx>, DefId)>,
62 pub(crate) auto_traits: Vec<DefId>,
63 pub(crate) render_options: RenderOptions,
65 pub(crate) cache: Cache,
67 pub(crate) inlined: FxHashSet<ItemId>,
69 pub(crate) output_format: OutputFormat,
71 pub(crate) show_coverage: bool,
73}
74
75impl<'tcx> DocContext<'tcx> {
76 pub(crate) fn sess(&self) -> &'tcx Session {
77 self.tcx.sess
78 }
79
80 pub(crate) fn with_param_env<T, F: FnOnce(&mut Self) -> T>(
81 &mut self,
82 def_id: DefId,
83 f: F,
84 ) -> T {
85 let old_param_env = mem::replace(&mut self.param_env, self.tcx.param_env(def_id));
86 let ret = f(self);
87 self.param_env = old_param_env;
88 ret
89 }
90
91 pub(crate) fn typing_env(&self) -> ty::TypingEnv<'tcx> {
92 ty::TypingEnv {
93 typing_mode: ty::TypingMode::non_body_analysis(),
94 param_env: self.param_env,
95 }
96 }
97
98 pub(crate) fn enter_alias<F, R>(
101 &mut self,
102 args: DefIdMap<clean::GenericArg>,
103 def_id: DefId,
104 f: F,
105 ) -> R
106 where
107 F: FnOnce(&mut Self) -> R,
108 {
109 let old_args = mem::replace(&mut self.args, args);
110 *self.current_type_aliases.entry(def_id).or_insert(0) += 1;
111 let r = f(self);
112 self.args = old_args;
113 if let Some(count) = self.current_type_aliases.get_mut(&def_id) {
114 *count -= 1;
115 if *count == 0 {
116 self.current_type_aliases.remove(&def_id);
117 }
118 }
119 r
120 }
121
122 pub(crate) fn as_local_hir_id(tcx: TyCtxt<'_>, item_id: ItemId) -> Option<HirId> {
125 match item_id {
126 ItemId::DefId(real_id) => {
127 real_id.as_local().map(|def_id| tcx.local_def_id_to_hir_id(def_id))
128 }
129 _ => None,
131 }
132 }
133
134 pub(crate) fn is_json_output(&self) -> bool {
138 self.output_format.is_json() && !self.show_coverage
139 }
140}
141
142pub(crate) fn new_dcx(
147 error_format: ErrorOutputType,
148 source_map: Option<Arc<source_map::SourceMap>>,
149 diagnostic_width: Option<usize>,
150 unstable_opts: &UnstableOptions,
151) -> rustc_errors::DiagCtxt {
152 let translator = rustc_driver::default_translator();
153 let emitter: Box<DynEmitter> = match error_format {
154 ErrorOutputType::HumanReadable { kind, color_config } => {
155 let short = kind.short();
156 Box::new(
157 HumanEmitter::new(stderr_destination(color_config), translator)
158 .sm(source_map.map(|sm| sm as _))
159 .short_message(short)
160 .diagnostic_width(diagnostic_width)
161 .track_diagnostics(unstable_opts.track_diagnostics)
162 .theme(if let HumanReadableErrorType::Unicode = kind {
163 OutputTheme::Unicode
164 } else {
165 OutputTheme::Ascii
166 })
167 .ui_testing(unstable_opts.ui_testing),
168 )
169 }
170 ErrorOutputType::Json { pretty, json_rendered, color_config } => {
171 let source_map = source_map.unwrap_or_else(|| {
172 Arc::new(source_map::SourceMap::new(source_map::FilePathMapping::empty()))
173 });
174 Box::new(
175 JsonEmitter::new(
176 Box::new(io::BufWriter::new(io::stderr())),
177 Some(source_map),
178 translator,
179 pretty,
180 json_rendered,
181 color_config,
182 )
183 .ui_testing(unstable_opts.ui_testing)
184 .diagnostic_width(diagnostic_width)
185 .track_diagnostics(unstable_opts.track_diagnostics)
186 .terminal_url(TerminalUrl::No),
187 )
188 }
189 };
190
191 rustc_errors::DiagCtxt::new(emitter).with_flags(unstable_opts.dcx_flags(true))
192}
193
194pub(crate) fn create_config(
196 input: Input,
197 RustdocOptions {
198 crate_name,
199 proc_macro_crate,
200 error_format,
201 diagnostic_width,
202 libs,
203 externs,
204 mut cfgs,
205 check_cfgs,
206 codegen_options,
207 unstable_opts,
208 target,
209 edition,
210 sysroot,
211 lint_opts,
212 describe_lints,
213 lint_cap,
214 scrape_examples_options,
215 expanded_args,
216 remap_path_prefix,
217 target_modifiers,
218 ..
219 }: RustdocOptions,
220 render_options: &RenderOptions,
221) -> rustc_interface::Config {
222 cfgs.push("doc".to_string());
224
225 let mut lints_to_show = vec![
228 rustc_lint::builtin::MISSING_DOCS.name.to_string(),
230 rustc_lint::builtin::INVALID_DOC_ATTRIBUTES.name.to_string(),
231 rustc_lint::builtin::RENAMED_AND_REMOVED_LINTS.name.to_string(),
233 rustc_lint::builtin::UNKNOWN_LINTS.name.to_string(),
234 rustc_lint::builtin::UNEXPECTED_CFGS.name.to_string(),
235 rustc_lint::builtin::UNFULFILLED_LINT_EXPECTATIONS.name.to_string(),
237 ];
238 lints_to_show.extend(crate::lint::RUSTDOC_LINTS.iter().map(|lint| lint.name.to_string()));
239
240 let (lint_opts, lint_caps) = crate::lint::init_lints(lints_to_show, lint_opts, |lint| {
241 Some((lint.name_lower(), lint::Allow))
242 });
243
244 let crate_types =
245 if proc_macro_crate { vec![CrateType::ProcMacro] } else { vec![CrateType::Rlib] };
246 let resolve_doc_links = if render_options.document_private {
247 ResolveDocLinks::All
248 } else {
249 ResolveDocLinks::Exported
250 };
251 let test = scrape_examples_options.map(|opts| opts.scrape_tests).unwrap_or(false);
252 let sessopts = config::Options {
254 sysroot,
255 search_paths: libs,
256 crate_types,
257 lint_opts,
258 lint_cap,
259 cg: codegen_options,
260 externs,
261 target_triple: target,
262 unstable_features: UnstableFeatures::from_environment(crate_name.as_deref()),
263 actually_rustdoc: true,
264 resolve_doc_links,
265 unstable_opts,
266 error_format,
267 diagnostic_width,
268 edition,
269 describe_lints,
270 crate_name,
271 test,
272 remap_path_prefix,
273 output_types: if let Some(file) = render_options.dep_info() {
274 OutputTypes::new(&[(
275 OutputType::DepInfo,
276 file.map(|f| OutFileName::Real(f.to_path_buf())),
277 )])
278 } else {
279 OutputTypes::new(&[])
280 },
281 target_modifiers,
282 ..Options::default()
283 };
284
285 rustc_interface::Config {
286 opts: sessopts,
287 crate_cfg: cfgs,
288 crate_check_cfg: check_cfgs,
289 input,
290 output_file: None,
291 output_dir: None,
292 file_loader: None,
293 locale_resources: rustc_driver::DEFAULT_LOCALE_RESOURCES.to_vec(),
294 lint_caps,
295 psess_created: None,
296 hash_untracked_state: None,
297 register_lints: Some(Box::new(crate::lint::register_lints)),
298 override_queries: Some(|_sess, providers| {
299 providers.lint_mod = |tcx, module_def_id| late_lint_mod(tcx, module_def_id, MissingDoc);
302 providers.used_trait_imports = |_, _| {
304 static EMPTY_SET: LazyLock<UnordSet<LocalDefId>> = LazyLock::new(UnordSet::default);
305 &EMPTY_SET
306 };
307 providers.typeck = move |tcx, def_id| {
309 let typeck_root_def_id = tcx.typeck_root_def_id(def_id.to_def_id()).expect_local();
313 if typeck_root_def_id != def_id {
314 return tcx.typeck(typeck_root_def_id);
315 }
316
317 let body = tcx.hir_body_owned_by(def_id);
318 debug!("visiting body for {def_id:?}");
319 EmitIgnoredResolutionErrors::new(tcx).visit_body(body);
320 (rustc_interface::DEFAULT_QUERY_PROVIDERS.typeck)(tcx, def_id)
321 };
322 }),
323 extra_symbols: Vec::new(),
324 make_codegen_backend: None,
325 registry: rustc_driver::diagnostics_registry(),
326 ice_file: None,
327 using_internal_features: &USING_INTERNAL_FEATURES,
328 expanded_args,
329 }
330}
331
332pub(crate) fn run_global_ctxt(
333 tcx: TyCtxt<'_>,
334 show_coverage: bool,
335 render_options: RenderOptions,
336 output_format: OutputFormat,
337) -> (clean::Crate, RenderOptions, Cache) {
338 let _ = tcx.sess.time("wf_checking", || tcx.ensure_ok().check_type_wf(()));
348
349 tcx.dcx().abort_if_errors();
350
351 tcx.sess.time("missing_docs", || rustc_lint::check_crate(tcx));
352 tcx.sess.time("check_mod_attrs", || {
353 tcx.hir_for_each_module(|module| tcx.ensure_ok().check_mod_attrs(module))
354 });
355 rustc_passes::stability::check_unused_or_stable_features(tcx);
356
357 let auto_traits =
358 tcx.visible_traits().filter(|&trait_def_id| tcx.trait_is_auto(trait_def_id)).collect();
359
360 let mut ctxt = DocContext {
361 tcx,
362 param_env: ParamEnv::empty(),
363 external_traits: Default::default(),
364 active_extern_traits: Default::default(),
365 args: Default::default(),
366 current_type_aliases: Default::default(),
367 impl_trait_bounds: Default::default(),
368 generated_synthetics: Default::default(),
369 auto_traits,
370 cache: Cache::new(render_options.document_private, render_options.document_hidden),
371 inlined: FxHashSet::default(),
372 output_format,
373 render_options,
374 show_coverage,
375 };
376
377 for cnum in tcx.crates(()) {
378 crate::visit_lib::lib_embargo_visit_item(&mut ctxt, cnum.as_def_id());
379 }
380
381 if let Some(sized_trait_did) = ctxt.tcx.lang_items().sized_trait() {
385 let sized_trait = build_trait(&mut ctxt, sized_trait_did);
386 ctxt.external_traits.insert(sized_trait_did, sized_trait);
387 }
388
389 let mut krate = tcx.sess.time("clean_crate", || clean::krate(&mut ctxt));
390
391 if krate.module.doc_value().is_empty() {
392 let help = format!(
393 "The following guide may be of use:\n\
394 {}/rustdoc/how-to-write-documentation.html",
395 crate::DOC_RUST_LANG_ORG_VERSION
396 );
397 tcx.node_lint(
398 crate::lint::MISSING_CRATE_LEVEL_DOCS,
399 DocContext::as_local_hir_id(tcx, krate.module.item_id).unwrap(),
400 |lint| {
401 lint.primary_message("no documentation found for this crate's top-level module");
402 lint.help(help);
403 },
404 );
405 }
406
407 for attr in krate.module.attrs.lists(sym::doc) {
410 if attr.is_word() && attr.has_name(sym::document_private_items) {
411 ctxt.render_options.document_private = true;
412 }
413 }
414
415 info!("Executing passes");
416
417 let mut visited = FxHashMap::default();
418 let mut ambiguous = FxIndexMap::default();
419
420 for p in passes::defaults(show_coverage) {
421 let run = match p.condition {
422 Always => true,
423 WhenDocumentPrivate => ctxt.render_options.document_private,
424 WhenNotDocumentPrivate => !ctxt.render_options.document_private,
425 WhenNotDocumentHidden => !ctxt.render_options.document_hidden,
426 };
427 if run {
428 debug!("running pass {}", p.pass.name);
429 if let Some(run_fn) = p.pass.run {
430 krate = tcx.sess.time(p.pass.name, || run_fn(krate, &mut ctxt));
431 } else {
432 let (k, LinkCollector { visited_links, ambiguous_links, .. }) =
433 passes::collect_intra_doc_links::collect_intra_doc_links(krate, &mut ctxt);
434 krate = k;
435 visited = visited_links;
436 ambiguous = ambiguous_links;
437 }
438 }
439 }
440
441 tcx.sess.time("check_lint_expectations", || tcx.check_expectations(Some(sym::rustdoc)));
442
443 krate = tcx.sess.time("create_format_cache", || Cache::populate(&mut ctxt, krate));
444
445 let mut collector =
446 LinkCollector { cx: &mut ctxt, visited_links: visited, ambiguous_links: ambiguous };
447 collector.resolve_ambiguities();
448
449 tcx.dcx().abort_if_errors();
450
451 (krate, ctxt.render_options, ctxt.cache)
452}
453
454struct EmitIgnoredResolutionErrors<'tcx> {
459 tcx: TyCtxt<'tcx>,
460}
461
462impl<'tcx> EmitIgnoredResolutionErrors<'tcx> {
463 fn new(tcx: TyCtxt<'tcx>) -> Self {
464 Self { tcx }
465 }
466}
467
468impl<'tcx> Visitor<'tcx> for EmitIgnoredResolutionErrors<'tcx> {
469 type NestedFilter = nested_filter::OnlyBodies;
470
471 fn maybe_tcx(&mut self) -> Self::MaybeTyCtxt {
472 self.tcx
475 }
476
477 fn visit_path(&mut self, path: &Path<'tcx>, _id: HirId) {
478 debug!("visiting path {path:?}");
479 if path.res == Res::Err {
480 let label = format!(
484 "could not resolve path `{}`",
485 path.segments
486 .iter()
487 .map(|segment| segment.ident.as_str())
488 .intersperse("::")
489 .collect::<String>()
490 );
491 rustc_errors::struct_span_code_err!(
492 self.tcx.dcx(),
493 path.span,
494 E0433,
495 "failed to resolve: {label}",
496 )
497 .with_span_label(path.span, label)
498 .with_note("this error was originally ignored because you are running `rustdoc`")
499 .with_note("try running again with `rustc` or `cargo check` and you may get a more detailed error")
500 .emit();
501 }
502 intravisit::walk_path(self, path);
506 }
507}
508
509#[derive(Clone, Copy, PartialEq, Eq, Hash)]
512pub(crate) enum ImplTraitParam {
513 DefId(DefId),
514 ParamIndex(u32),
515}
516
517impl From<DefId> for ImplTraitParam {
518 fn from(did: DefId) -> Self {
519 ImplTraitParam::DefId(did)
520 }
521}
522
523impl From<u32> for ImplTraitParam {
524 fn from(idx: u32) -> Self {
525 ImplTraitParam::ParamIndex(idx)
526 }
527}