css-module-lexer 0.1.2

Lexes CSS modules returning their dependencies metadata
Documentation
mod css_syntax;
mod dependencies;
mod dependency_types;
mod lexer;

pub use dependencies::DashedIdentCollector;
pub use dependencies::LexDependencies;
pub use dependencies::ModeData;
pub use dependency_types::Dependency;
pub use dependency_types::DependencyContext;
pub use dependency_types::DependencyIndex;
pub use dependency_types::DependencyListRange;
pub use dependency_types::ImportAttributes;
pub use dependency_types::Mode;
pub use dependency_types::Range;
pub use dependency_types::UrlRangeKind;
pub use dependency_types::ValueAtRuleImportItem;
pub use dependency_types::Warning;
pub use dependency_types::WarningKind;
pub use lexer::{Lexer, LexerVisitor, Pos, Token, TokenFlags, TokenKind, TokenWithTrivia, Trivia};

pub trait HandleWarning<'s> {
    fn handle_warning(&mut self, warning: Warning<'s>);
}

impl<'s, F: FnMut(Warning<'s>)> HandleWarning<'s> for F {
    fn handle_warning(&mut self, warning: Warning<'s>) {
        self(warning);
    }
}

pub fn lex_dependencies<'s>(
    input: &'s str,
    mode: Mode,
    mut handle_dependency: impl FnMut(&Dependency<'s>),
    handle_warning: impl HandleWarning<'s>,
) -> DependencyContext<'s> {
    let mut lexer = Lexer::new(input, DashedIdentCollector::default());
    let mut visitor = LexDependencies::new(handle_warning, mode);
    visitor.lex_streaming(&mut lexer);
    let dependency_context = visitor.into_dependency_context();
    for dependency in dependency_context.iter() {
        handle_dependency(dependency);
    }
    dependency_context
}

pub fn collect_dependencies(input: &str, mode: Mode) -> (DependencyContext<'_>, Vec<Warning<'_>>) {
    let mut warnings = Vec::with_capacity(estimate_warning_capacity(input.len(), mode));
    let mut lexer = Lexer::new(input, DashedIdentCollector::default());
    let mut visitor = LexDependencies::new(|v| warnings.push(v), mode);
    visitor.lex_streaming(&mut lexer);
    (visitor.into_dependency_context(), warnings)
}

fn estimate_warning_capacity(input_len: usize, mode: Mode) -> usize {
    let warning_rate = match mode {
        Mode::Pure => (1, 256),
        _ => (1, 512),
    };

    estimate_capacity(input_len, warning_rate.0, warning_rate.1, 2, 512)
}

#[inline]
fn estimate_capacity(
    input_len: usize,
    numerator: usize,
    denominator: usize,
    minimum: usize,
    maximum: usize,
) -> usize {
    input_len
        .saturating_mul(numerator)
        .checked_div(denominator)
        .unwrap_or(0)
        .clamp(minimum, maximum)
}