omnicoach 0.3.0

Lightweight, high-signal static analysis and command linter for Python and Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

//! Python comments and whitespace.

use crate::code_lint::ast::tree::{ParsedFile, span_from_ruff_range};
use crate::diagnostic::SourceSpan;
use ruff_python_ast::token::TokenKind;
use ruff_text_size::Ranged as _;

/// Collects the spans of all comment tokens in `file` in source order, or none if `file` failed
/// to parse.
pub fn collect_comment_spans(file: &ParsedFile) -> Vec<SourceSpan> {
    let Some(parsed) = file.py_module() else {
        return Vec::new();
    };
    parsed
        .tokens()
        .iter()
        .filter(|token| token.kind() == TokenKind::Comment)
        .map(|token| span_from_ruff_range(token.range()))
        .collect()
}