Skip to main content

rumdl_lib/lint_context/
types.rs

1use pulldown_cmark::LinkType;
2use std::borrow::Cow;
3
4/// Pre-computed information about a line
5#[derive(Debug, Clone)]
6pub struct LineInfo {
7    /// Byte offset where this line starts in the document
8    pub byte_offset: usize,
9    /// Length of the line in bytes (without newline)
10    pub byte_len: usize,
11    /// Number of bytes of leading whitespace (for substring extraction)
12    pub indent: usize,
13    /// Visual column width of leading whitespace (with proper tab expansion)
14    /// Per CommonMark, tabs expand to the next column that is a multiple of 4.
15    /// Use this for numeric comparisons like checking for indented code blocks (>= 4).
16    pub visual_indent: usize,
17    /// Whether the line is blank (empty or only whitespace)
18    pub is_blank: bool,
19    /// Whether this line is inside a code block
20    pub in_code_block: bool,
21    /// Whether this line is inside front matter
22    pub in_front_matter: bool,
23    /// Whether this line is inside an HTML block
24    pub in_html_block: bool,
25    /// Whether this line is inside an HTML comment
26    pub in_html_comment: bool,
27    /// List item information if this line starts a list item
28    /// Boxed to reduce LineInfo size: most lines are not list items
29    pub list_item: Option<Box<ListItemInfo>>,
30    /// Heading information if this line is a heading
31    /// Boxed to reduce LineInfo size: most lines are not headings
32    pub heading: Option<Box<HeadingInfo>>,
33    /// Blockquote information if this line is a blockquote
34    /// Boxed to reduce LineInfo size: most lines are not blockquotes
35    pub blockquote: Option<Box<BlockquoteInfo>>,
36    /// Whether this line is inside a mkdocstrings autodoc block
37    pub in_mkdocstrings: bool,
38    /// Whether this line is part of an ESM import/export block (MDX only)
39    pub in_esm_block: bool,
40    /// Whether this line is a continuation of a multi-line code span from a previous line
41    pub in_code_span_continuation: bool,
42    /// Whether this line is a horizontal rule (---, ***, ___, etc.)
43    /// Pre-computed for consistent detection across all rules
44    pub is_horizontal_rule: bool,
45    /// Whether this line is inside a math block ($$ ... $$)
46    pub in_math_block: bool,
47    /// Whether this line is inside a Quarto div block (::: ... :::)
48    pub in_quarto_div: bool,
49    /// Whether this line is a Quarto/Pandoc div marker (opening ::: {.class} or closing :::)
50    /// Analogous to `is_horizontal_rule` — marks structural delimiters that are not paragraph text
51    pub is_div_marker: bool,
52    /// Whether this line contains or is inside a JSX expression (MDX only)
53    pub in_jsx_expression: bool,
54    /// Whether this line is inside an MDX comment {/* ... */} (MDX only)
55    pub in_mdx_comment: bool,
56    /// Whether this line is inside an MkDocs admonition block (!!! or ???)
57    pub in_admonition: bool,
58    /// Whether this line is inside an MkDocs content tab block (===)
59    pub in_content_tab: bool,
60    /// Whether this line is inside an HTML block with markdown attribute (MkDocs grid cards, etc.)
61    pub in_mkdocs_html_markdown: bool,
62    /// Whether this line is a definition list item (: definition)
63    pub in_definition_list: bool,
64    /// Whether this line is inside an Obsidian comment (%%...%% syntax, Obsidian flavor only)
65    pub in_obsidian_comment: bool,
66    /// Whether this line is inside a PyMdown Blocks region (/// ... ///, MkDocs flavor only)
67    pub in_pymdown_block: bool,
68    /// Whether this line is inside a kramdown extension block ({::comment}...{:/comment}, {::nomarkdown}...{:/nomarkdown})
69    pub in_kramdown_extension_block: bool,
70    /// Whether this line is a kramdown block IAL ({:.class #id}) or ALD ({:ref: .class})
71    pub is_kramdown_block_ial: bool,
72    /// Whether this line is inside a JSX component block (MDX only, e.g. `<Tabs>...</Tabs>`)
73    pub in_jsx_block: bool,
74}
75
76impl LineInfo {
77    /// Get the line content as a string slice from the source document
78    pub fn content<'a>(&self, source: &'a str) -> &'a str {
79        &source[self.byte_offset..self.byte_offset + self.byte_len]
80    }
81
82    /// Check if this line is inside MkDocs-specific indented content (admonitions, tabs, or markdown HTML).
83    /// This content uses 4-space indentation which pulldown-cmark would interpret as code blocks,
84    /// but in MkDocs flavor it's actually container content that should be preserved.
85    #[inline]
86    pub fn in_mkdocs_container(&self) -> bool {
87        self.in_admonition || self.in_content_tab || self.in_mkdocs_html_markdown
88    }
89}
90
91/// Information about a list item
92#[derive(Debug, Clone)]
93pub struct ListItemInfo {
94    /// The marker used (*, -, +, or number with . or ))
95    pub marker: String,
96    /// Whether it's ordered (true) or unordered (false)
97    pub is_ordered: bool,
98    /// The number for ordered lists
99    pub number: Option<usize>,
100    /// Column where the marker starts (0-based)
101    pub marker_column: usize,
102    /// Column where content after marker starts
103    pub content_column: usize,
104}
105
106/// Heading style type
107#[derive(Debug, Clone, PartialEq)]
108pub enum HeadingStyle {
109    /// ATX style heading (# Heading)
110    ATX,
111    /// Setext style heading with = underline
112    Setext1,
113    /// Setext style heading with - underline
114    Setext2,
115}
116
117/// Parsed link information
118#[derive(Debug, Clone)]
119pub struct ParsedLink<'a> {
120    /// Line number (1-indexed)
121    pub line: usize,
122    /// Start column (0-indexed) in the line
123    pub start_col: usize,
124    /// End column (0-indexed) in the line
125    pub end_col: usize,
126    /// Byte offset in document
127    pub byte_offset: usize,
128    /// End byte offset in document
129    pub byte_end: usize,
130    /// Link text
131    pub text: Cow<'a, str>,
132    /// Link URL or reference
133    pub url: Cow<'a, str>,
134    /// Whether this is a reference link `[text][ref]` vs inline `[text](url)`
135    pub is_reference: bool,
136    /// Reference ID for reference links
137    pub reference_id: Option<Cow<'a, str>>,
138    /// Link type from pulldown-cmark
139    pub link_type: LinkType,
140}
141
142/// Information about a broken link reported by pulldown-cmark
143#[derive(Debug, Clone)]
144pub struct BrokenLinkInfo {
145    /// The reference text that couldn't be resolved
146    pub reference: String,
147    /// Byte span in the source document
148    pub span: std::ops::Range<usize>,
149}
150
151/// Parsed footnote reference (e.g., `[^1]`, `[^note]`)
152#[derive(Debug, Clone)]
153pub struct FootnoteRef {
154    /// The footnote ID (without the ^ prefix)
155    pub id: String,
156    /// Line number (1-indexed)
157    pub line: usize,
158    /// Start byte offset in document
159    pub byte_offset: usize,
160    /// End byte offset in document
161    pub byte_end: usize,
162}
163
164/// Parsed image information
165#[derive(Debug, Clone)]
166pub struct ParsedImage<'a> {
167    /// Line number (1-indexed)
168    pub line: usize,
169    /// Start column (0-indexed) in the line
170    pub start_col: usize,
171    /// End column (0-indexed) in the line
172    pub end_col: usize,
173    /// Byte offset in document
174    pub byte_offset: usize,
175    /// End byte offset in document
176    pub byte_end: usize,
177    /// Alt text
178    pub alt_text: Cow<'a, str>,
179    /// Image URL or reference
180    pub url: Cow<'a, str>,
181    /// Whether this is a reference image ![alt][ref] vs inline ![alt](url)
182    pub is_reference: bool,
183    /// Reference ID for reference images
184    pub reference_id: Option<Cow<'a, str>>,
185    /// Link type from pulldown-cmark
186    pub link_type: LinkType,
187}
188
189/// Reference definition `[ref]: url "title"`
190#[derive(Debug, Clone)]
191pub struct ReferenceDef {
192    /// Line number (1-indexed)
193    pub line: usize,
194    /// Reference ID (normalized to lowercase)
195    pub id: String,
196    /// URL
197    pub url: String,
198    /// Optional title
199    pub title: Option<String>,
200    /// Byte offset where the reference definition starts
201    pub byte_offset: usize,
202    /// Byte offset where the reference definition ends
203    pub byte_end: usize,
204    /// Byte offset where the title starts (if present, includes quote)
205    pub title_byte_start: Option<usize>,
206    /// Byte offset where the title ends (if present, includes quote)
207    pub title_byte_end: Option<usize>,
208}
209
210/// Parsed code span information
211#[derive(Debug, Clone)]
212pub struct CodeSpan {
213    /// Line number where the code span starts (1-indexed)
214    pub line: usize,
215    /// Line number where the code span ends (1-indexed)
216    pub end_line: usize,
217    /// Start column (0-indexed) in the line
218    pub start_col: usize,
219    /// End column (0-indexed) in the line
220    pub end_col: usize,
221    /// Byte offset in document
222    pub byte_offset: usize,
223    /// End byte offset in document
224    pub byte_end: usize,
225    /// Number of backticks used (1, 2, 3, etc.)
226    pub backtick_count: usize,
227    /// Content inside the code span (without backticks)
228    pub content: String,
229}
230
231/// Parsed math span information (inline $...$ or display $$...$$)
232#[derive(Debug, Clone)]
233pub struct MathSpan {
234    /// Line number where the math span starts (1-indexed)
235    pub line: usize,
236    /// Line number where the math span ends (1-indexed)
237    pub end_line: usize,
238    /// Start column (0-indexed) in the line
239    pub start_col: usize,
240    /// End column (0-indexed) in the line
241    pub end_col: usize,
242    /// Byte offset in document
243    pub byte_offset: usize,
244    /// End byte offset in document
245    pub byte_end: usize,
246    /// Whether this is display math ($$...$$) vs inline ($...$)
247    pub is_display: bool,
248    /// Content inside the math delimiters
249    pub content: String,
250}
251
252/// Information about a heading
253#[derive(Debug, Clone)]
254pub struct HeadingInfo {
255    /// Heading level (1-6 for ATX, 1-2 for Setext)
256    pub level: u8,
257    /// Style of heading
258    pub style: HeadingStyle,
259    /// The heading marker (# characters or underline)
260    pub marker: String,
261    /// Column where the marker starts (0-based)
262    pub marker_column: usize,
263    /// Column where heading text starts
264    pub content_column: usize,
265    /// The heading text (without markers and without custom ID syntax)
266    pub text: String,
267    /// Custom header ID if present (e.g., from {#custom-id} syntax)
268    pub custom_id: Option<String>,
269    /// Original heading text including custom ID syntax
270    pub raw_text: String,
271    /// Whether it has a closing sequence (for ATX)
272    pub has_closing_sequence: bool,
273    /// The closing sequence if present
274    pub closing_sequence: String,
275    /// Whether this is a valid CommonMark heading (ATX headings require space after #)
276    /// False for malformed headings like `#NoSpace` that MD018 should flag
277    pub is_valid: bool,
278}
279
280/// A valid heading from a filtered iteration
281///
282/// Only includes headings that are CommonMark-compliant (have space after #).
283/// Hashtag-like patterns (`#tag`, `#123`) are excluded.
284#[derive(Debug, Clone)]
285pub struct ValidHeading<'a> {
286    /// The 1-indexed line number in the document
287    pub line_num: usize,
288    /// Reference to the heading information
289    pub heading: &'a HeadingInfo,
290    /// Reference to the full line info (for rules that need additional context)
291    pub line_info: &'a LineInfo,
292}
293
294/// Iterator over valid CommonMark headings in a document
295///
296/// Filters out malformed headings like `#NoSpace` that should be flagged by MD018
297/// but should not be processed by other heading rules.
298pub struct ValidHeadingsIter<'a> {
299    lines: &'a [LineInfo],
300    current_index: usize,
301}
302
303impl<'a> ValidHeadingsIter<'a> {
304    pub(super) fn new(lines: &'a [LineInfo]) -> Self {
305        Self {
306            lines,
307            current_index: 0,
308        }
309    }
310}
311
312impl<'a> Iterator for ValidHeadingsIter<'a> {
313    type Item = ValidHeading<'a>;
314
315    fn next(&mut self) -> Option<Self::Item> {
316        while self.current_index < self.lines.len() {
317            let idx = self.current_index;
318            self.current_index += 1;
319
320            let line_info = &self.lines[idx];
321            if let Some(heading) = line_info.heading.as_deref()
322                && heading.is_valid
323            {
324                return Some(ValidHeading {
325                    line_num: idx + 1, // Convert 0-indexed to 1-indexed
326                    heading,
327                    line_info,
328                });
329            }
330        }
331        None
332    }
333}
334
335/// Information about a blockquote line
336#[derive(Debug, Clone)]
337pub struct BlockquoteInfo {
338    /// Nesting level (1 for >, 2 for >>, etc.)
339    pub nesting_level: usize,
340    /// The indentation before the blockquote marker
341    pub indent: String,
342    /// Column where the first > starts (0-based)
343    pub marker_column: usize,
344    /// The blockquote prefix (e.g., "> ", ">> ", etc.)
345    pub prefix: String,
346    /// Content after the blockquote marker(s)
347    pub content: String,
348    /// Whether the line has no space after the marker
349    pub has_no_space_after_marker: bool,
350    /// Whether the line has multiple spaces after the marker
351    pub has_multiple_spaces_after_marker: bool,
352    /// Whether this is an empty blockquote line needing MD028 fix
353    pub needs_md028_fix: bool,
354}
355
356/// Information about a list block
357#[derive(Debug, Clone)]
358pub struct ListBlock {
359    /// Line number where the list starts (1-indexed)
360    pub start_line: usize,
361    /// Line number where the list ends (1-indexed)
362    pub end_line: usize,
363    /// Whether it's ordered or unordered
364    pub is_ordered: bool,
365    /// The consistent marker for unordered lists (if any)
366    pub marker: Option<String>,
367    /// Blockquote prefix for this list (empty if not in blockquote)
368    pub blockquote_prefix: String,
369    /// Lines that are list items within this block
370    pub item_lines: Vec<usize>,
371    /// Nesting level (0 for top-level lists)
372    pub nesting_level: usize,
373    /// Maximum marker width seen in this block (e.g., 3 for "1. ", 4 for "10. ")
374    pub max_marker_width: usize,
375}
376
377/// Character frequency data for fast content analysis
378#[derive(Debug, Clone, Default)]
379pub struct CharFrequency {
380    /// Count of # characters (headings)
381    pub hash_count: usize,
382    /// Count of * characters (emphasis, lists, horizontal rules)
383    pub asterisk_count: usize,
384    /// Count of _ characters (emphasis, horizontal rules)
385    pub underscore_count: usize,
386    /// Count of - characters (lists, horizontal rules, setext headings)
387    pub hyphen_count: usize,
388    /// Count of + characters (lists)
389    pub plus_count: usize,
390    /// Count of > characters (blockquotes)
391    pub gt_count: usize,
392    /// Count of | characters (tables)
393    pub pipe_count: usize,
394    /// Count of [ characters (links, images)
395    pub bracket_count: usize,
396    /// Count of ` characters (code spans, code blocks)
397    pub backtick_count: usize,
398    /// Count of < characters (HTML tags, autolinks)
399    pub lt_count: usize,
400    /// Count of ! characters (images)
401    pub exclamation_count: usize,
402    /// Count of newline characters
403    pub newline_count: usize,
404}
405
406/// Pre-parsed HTML tag information
407#[derive(Debug, Clone)]
408pub struct HtmlTag {
409    /// Line number (1-indexed)
410    pub line: usize,
411    /// Start column (0-indexed) in the line
412    pub start_col: usize,
413    /// End column (0-indexed) in the line
414    pub end_col: usize,
415    /// Byte offset in document
416    pub byte_offset: usize,
417    /// End byte offset in document
418    pub byte_end: usize,
419    /// Tag name (e.g., "div", "img", "br")
420    pub tag_name: String,
421    /// Whether it's a closing tag (`</tag>`)
422    pub is_closing: bool,
423    /// Whether it's self-closing (`<tag />`)
424    pub is_self_closing: bool,
425    /// Raw tag content
426    pub raw_content: String,
427}
428
429/// Pre-parsed emphasis span information
430#[derive(Debug, Clone)]
431pub struct EmphasisSpan {
432    /// Line number (1-indexed)
433    pub line: usize,
434    /// Start column (0-indexed) in the line
435    pub start_col: usize,
436    /// End column (0-indexed) in the line
437    pub end_col: usize,
438    /// Byte offset in document
439    pub byte_offset: usize,
440    /// End byte offset in document
441    pub byte_end: usize,
442    /// Type of emphasis ('*' or '_')
443    pub marker: char,
444    /// Number of markers (1 for italic, 2 for bold, 3+ for bold+italic)
445    pub marker_count: usize,
446    /// Content inside the emphasis
447    pub content: String,
448}
449
450/// Pre-parsed table row information
451#[derive(Debug, Clone)]
452pub struct TableRow {
453    /// Line number (1-indexed)
454    pub line: usize,
455    /// Whether this is a separator row (contains only |, -, :, and spaces)
456    pub is_separator: bool,
457    /// Number of columns (pipe-separated cells)
458    pub column_count: usize,
459    /// Alignment info from separator row
460    pub column_alignments: Vec<String>, // "left", "center", "right", "none"
461}
462
463/// Pre-parsed bare URL information (not in links)
464#[derive(Debug, Clone)]
465pub struct BareUrl {
466    /// Line number (1-indexed)
467    pub line: usize,
468    /// Start column (0-indexed) in the line
469    pub start_col: usize,
470    /// End column (0-indexed) in the line
471    pub end_col: usize,
472    /// Byte offset in document
473    pub byte_offset: usize,
474    /// End byte offset in document
475    pub byte_end: usize,
476    /// The URL string
477    pub url: String,
478    /// Type of URL ("http", "https", "ftp", "email")
479    pub url_type: String,
480}
481
482/// Check if a line is a horizontal rule (---, ***, ___) per CommonMark spec.
483/// CommonMark rules for thematic breaks (horizontal rules):
484/// - May have 0-3 spaces of leading indentation (but NOT tabs)
485/// - Must have 3+ of the same character (-, *, or _)
486/// - May have spaces between characters
487/// - No other characters allowed
488pub fn is_horizontal_rule_line(line: &str) -> bool {
489    // CommonMark: HRs can have 0-3 spaces of leading indentation, not tabs
490    let leading_spaces = line.len() - line.trim_start_matches(' ').len();
491    if leading_spaces > 3 || line.starts_with('\t') {
492        return false;
493    }
494
495    is_horizontal_rule_content(line.trim())
496}
497
498/// Check if trimmed content matches horizontal rule pattern.
499/// Use `is_horizontal_rule_line` for full CommonMark compliance including indentation check.
500pub fn is_horizontal_rule_content(trimmed: &str) -> bool {
501    if trimmed.len() < 3 {
502        return false;
503    }
504
505    let mut chars = trimmed.chars();
506    let first_char = match chars.next() {
507        Some(c @ ('-' | '*' | '_')) => c,
508        _ => return false,
509    };
510
511    // Count occurrences of the rule character, rejecting non-whitespace interlopers
512    let mut count = 1; // Already matched the first character
513    for ch in chars {
514        if ch == first_char {
515            count += 1;
516        } else if ch != ' ' && ch != '\t' {
517            return false;
518        }
519    }
520    count >= 3
521}
522
523/// Backwards-compatible alias for `is_horizontal_rule_content`
524pub fn is_horizontal_rule(trimmed: &str) -> bool {
525    is_horizontal_rule_content(trimmed)
526}