summaryrefslogtreecommitdiff
path: root/cli/tools/doc.rs
blob: e27e495704c383492b4b5b2dba2d6e7e2e755c03 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.

use crate::args::DocFlags;
use crate::args::DocSourceFileFlag;
use crate::args::Flags;
use crate::colors;
use crate::display::write_json_to_stdout;
use crate::display::write_to_stdout_ignore_sigpipe;
use crate::file_fetcher::File;
use crate::proc_state::ProcState;
use crate::tsc::get_types_declaration_file_text;
use deno_ast::MediaType;
use deno_core::anyhow::bail;
use deno_core::error::AnyError;
use deno_core::resolve_url_or_path;
use deno_doc as doc;
use deno_graph::ModuleSpecifier;
use std::path::PathBuf;

pub async fn print_docs(
  flags: Flags,
  doc_flags: DocFlags,
) -> Result<(), AnyError> {
  let ps = ProcState::build(flags).await?;

  let mut doc_nodes = match doc_flags.source_file {
    DocSourceFileFlag::Builtin => {
      let source_file_specifier =
        ModuleSpecifier::parse("deno://lib.deno.d.ts").unwrap();
      let content = get_types_declaration_file_text(ps.options.unstable());
      let mut loader = deno_graph::source::MemoryLoader::new(
        vec![(
          source_file_specifier.to_string(),
          deno_graph::source::Source::Module {
            specifier: source_file_specifier.to_string(),
            content,
            maybe_headers: None,
          },
        )],
        Vec::new(),
      );
      let analyzer = deno_graph::CapturingModuleAnalyzer::default();
      let graph = deno_graph::create_graph(
        vec![source_file_specifier.clone()],
        &mut loader,
        deno_graph::GraphOptions {
          is_dynamic: false,
          imports: None,
          resolver: None,
          module_analyzer: Some(&analyzer),
          reporter: None,
        },
      )
      .await;
      let doc_parser = doc::DocParser::new(
        graph,
        doc_flags.private,
        analyzer.as_capturing_parser(),
      );
      doc_parser.parse_module(&source_file_specifier)?.definitions
    }
    DocSourceFileFlag::Path(source_file) => {
      let module_specifier = resolve_url_or_path(&source_file)?;

      // If the root module has external types, the module graph won't redirect it,
      // so instead create a dummy file which exports everything from the actual file being documented.
      let root_specifier = resolve_url_or_path("./$deno$doc.ts").unwrap();
      let root = File {
        local: PathBuf::from("./$deno$doc.ts"),
        maybe_types: None,
        media_type: MediaType::TypeScript,
        source: format!("export * from \"{module_specifier}\";").into(),
        specifier: root_specifier.clone(),
        maybe_headers: None,
      };

      // Save our fake file into file fetcher cache.
      ps.file_fetcher.insert_cached(root);

      let graph = ps.create_graph(vec![root_specifier.clone()]).await?;
      let doc_parser = doc::DocParser::new(
        graph,
        doc_flags.private,
        ps.parsed_source_cache.as_capturing_parser(),
      );
      doc_parser.parse_with_reexports(&root_specifier)?
    }
  };

  if doc_flags.json {
    write_json_to_stdout(&doc_nodes)
  } else {
    doc_nodes.retain(|doc_node| doc_node.kind != doc::DocNodeKind::Import);
    let details = if let Some(filter) = doc_flags.filter {
      let nodes =
        doc::find_nodes_by_name_recursively(doc_nodes, filter.clone());
      if nodes.is_empty() {
        bail!("Node {} was not found!", filter);
      }
      format!(
        "{}",
        doc::DocPrinter::new(&nodes, colors::use_color(), doc_flags.private)
      )
    } else {
      format!(
        "{}",
        doc::DocPrinter::new(
          &doc_nodes,
          colors::use_color(),
          doc_flags.private
        )
      )
    };

    write_to_stdout_ignore_sigpipe(details.as_bytes()).map_err(AnyError::from)
  }
}