fallow_extract/

css.rs

//! CSS/SCSS file parsing and CSS Module class name extraction.
//!
//! Handles `@import`, `@use`, `@forward`, `@plugin`, `@apply`, `@tailwind` directives,
//! and extracts class names as named exports from `.module.css`, `.module.scss`,
//! `.module.sass`, and `.module.less` files.
//!
//! Extraction is a deliberate hybrid, not a half-finished migration. lightningcss
//! owns the membership decision for standard CSS (which `.token` occurrences are
//! genuine class selectors, via `lightningcss_class_set`); the regex scanners own
//! span location and the entire SCSS path. lightningcss parses standard CSS only,
//! not SCSS syntax (`@use`, `@forward`, `//` line comments, `$variables`), so SCSS
//! files are gated away from the parser and the regex chain stays as permanent
//! infrastructure rather than a transitional step toward an all-parser tokenizer.

use std::path::Path;
use std::sync::LazyLock;

use oxc_span::Span;
use rustc_hash::FxHashSet;

use crate::css_mask::{
    mask_sass_module_members, mask_view_transition_arguments, mask_with_whitespace,
};
use crate::css_module_classes::lightningcss_class_set;
use crate::{ExportInfo, ExportName, ImportInfo, ImportedName, ModuleInfo, VisibilityTag};
use fallow_types::discover::FileId;

/// Regex to extract CSS @import sources.
/// Matches: @import "path"; @import 'path'; @import url("path"); @import url('path'); @import url(path);
static CSS_IMPORT_RE: LazyLock<regex::Regex> = LazyLock::new(|| {
    crate::static_regex(
        r#"@import\s+(?:url\(\s*(?:["']([^"']+)["']|([^)]+))\s*\)|["']([^"']+)["'])"#,
    )
});

/// Regex to extract SCSS @use and @forward sources.
/// Matches: @use "path"; @use 'path'; @forward "path"; @forward 'path';
static SCSS_USE_RE: LazyLock<regex::Regex> =
    LazyLock::new(|| crate::static_regex(r#"@(?:use|forward)\s+["']([^"']+)["']"#));

/// Regex to extract Tailwind CSS @plugin sources.
/// Matches: @plugin "package"; @plugin 'package'; @plugin "./local-plugin.js";
static CSS_PLUGIN_RE: LazyLock<regex::Regex> =
    LazyLock::new(|| crate::static_regex(r#"@plugin\s+["']([^"']+)["']"#));

/// Regex to extract @apply class references.
/// Matches: @apply class1 class2 class3;
static CSS_APPLY_RE: LazyLock<regex::Regex> =
    LazyLock::new(|| crate::static_regex(r"@apply\s+[^;}\n]+"));

/// Regex to extract @tailwind directives.
/// Matches: @tailwind base; @tailwind components; @tailwind utilities;
static CSS_TAILWIND_RE: LazyLock<regex::Regex> =
    LazyLock::new(|| crate::static_regex(r"@tailwind\s+\w+"));

/// Regex to match CSS block comments (`/* ... */`) for stripping before extraction.
static CSS_COMMENT_RE: LazyLock<regex::Regex> =
    LazyLock::new(|| crate::static_regex(r"(?s)/\*.*?\*/"));

/// Regex to match SCSS single-line comments (`// ...`) for stripping before extraction.
static SCSS_LINE_COMMENT_RE: LazyLock<regex::Regex> =
    LazyLock::new(|| crate::static_regex(r"//[^\n]*"));

/// Regex to extract CSS class names from selectors.
/// Matches `.className` in selectors. Applied after stripping comments, strings, and URLs.
static CSS_CLASS_RE: LazyLock<regex::Regex> =
    LazyLock::new(|| crate::static_regex(r"\.([a-zA-Z_][a-zA-Z0-9_-]*)"));

/// Regex to strip quoted strings and `url(...)` content from CSS before class extraction.
/// Prevents false positives from `content: ".foo"` and `url(./path/file.ext)`.
static CSS_NON_SELECTOR_RE: LazyLock<regex::Regex> =
    LazyLock::new(|| crate::static_regex(r#"(?s)"[^"]*"|'[^']*'|url\([^)]*\)"#));

/// Regex to strip the prelude of `@layer` and `@import` at-rules before
/// CSS-Modules class extraction. Matches the `@keyword` plus everything up to
/// (but not including) the next `;` or `{`, so block bodies are preserved.
///
/// Narrow allowlist by design (issue #540): only at-rules whose preludes
/// legitimately carry dot-separated identifiers without selector semantics are
/// stripped. `@layer foo.bar` (CSS Cascading & Inheritance L5) lists layer
/// names; `@import url("x.css") layer(theme.button)` carries a parenthesised
/// layer reference. `@scope (.foo) to (.bar)` keeps its existing behavior
/// because the prelude IS a selector list and `.foo` / `.bar` are real class
/// references that the user may want to surface as exports.
static CSS_AT_RULE_PRELUDE_RE: LazyLock<regex::Regex> =
    LazyLock::new(|| crate::static_regex(r"@(?:layer|import)\b[^;{]*"));

pub(crate) fn is_css_file(path: &Path) -> bool {
    path.extension()
        .and_then(|e| e.to_str())
        .is_some_and(|ext| matches!(ext, "css" | "scss" | "sass" | "less"))
}

/// A CSS import source with both the literal source and fallow's resolver-normalized form.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct CssImportSource {
    /// The import source exactly as it appeared in `@import` / `@use` / `@forward` / `@plugin`.
    pub raw: String,
    /// The source normalized for fallow's resolver (`variables` -> `./variables` in SCSS).
    pub normalized: String,
    /// Whether this source came from Tailwind CSS `@plugin`.
    pub is_plugin: bool,
    /// Span of the source specifier in the original CSS/SCSS input.
    pub span: Span,
}

fn is_css_module_file(path: &Path) -> bool {
    is_css_file(path)
        && path
            .file_stem()
            .and_then(|s| s.to_str())
            .is_some_and(|stem| stem.ends_with(".module"))
}

/// Returns true if a CSS import source is a remote URL or data URI that should be skipped.
fn is_css_url_import(source: &str) -> bool {
    source.starts_with("http://") || source.starts_with("https://") || source.starts_with("data:")
}

/// Normalize a CSS/SCSS import path to use `./` prefix for relative paths.
/// Bare file names such as `reset.css` stay relative for CSS ergonomics, while
/// package subpaths such as `tailwindcss/theme.css` stay bare so bundler-style
/// package CSS imports resolve through `node_modules`.
///
/// When `is_scss` is true, extensionless specifiers that are not SCSS built-in
/// modules (`sass:*`) are treated as relative imports (SCSS partial convention).
/// This handles `@use 'variables'` resolving to `./_variables.scss`.
///
/// Scoped npm packages (`@scope/pkg`) are always kept bare, even when they have
/// CSS extensions (e.g., `@fontsource/monaspace-neon/400.css`). Bundlers like
/// Vite resolve these from node_modules, not as relative paths.
fn normalize_css_import_path(path: String, is_scss: bool) -> String {
    if path.starts_with('.') || path.starts_with('/') || path.contains("://") {
        return path;
    }
    if path.starts_with('@') && path.contains('/') {
        return path;
    }
    let path_ref = std::path::Path::new(&path);
    if !is_scss
        && path.contains('/')
        && path_ref
            .extension()
            .and_then(|e| e.to_str())
            .is_some_and(is_style_extension)
    {
        return path;
    }
    let ext = std::path::Path::new(&path)
        .extension()
        .and_then(|e| e.to_str());
    match ext {
        Some(e) if is_style_extension(e) => format!("./{path}"),
        _ => {
            if is_scss && !path.contains(':') {
                format!("./{path}")
            } else {
                path
            }
        }
    }
}

fn is_style_extension(ext: &str) -> bool {
    ext.eq_ignore_ascii_case("css")
        || ext.eq_ignore_ascii_case("scss")
        || ext.eq_ignore_ascii_case("sass")
        || ext.eq_ignore_ascii_case("less")
}

#[cfg(test)]
thread_local! {
    /// Comment mask passes on this thread, so tests can pin one pass per file.
    static COMMENT_MASK_PASSES: std::cell::Cell<u32> = const { std::cell::Cell::new(0) };
    /// Source bytes that the line lookups read on this thread, so tests can pin
    /// a linear cost for the located token scans.
    static LINE_LOOKUP_BYTES: std::cell::Cell<usize> = const { std::cell::Cell::new(0) };
}

#[cfg(test)]
fn note_line_lookup_bytes(bytes: usize) {
    LINE_LOOKUP_BYTES.with(|read| read.set(read.get() + bytes));
}

fn mask_css_comments(source: &str, is_scss: bool) -> String {
    #[cfg(test)]
    COMMENT_MASK_PASSES.with(|passes| passes.set(passes.get() + 1));
    crate::css_masked_bytes::add(source.len() as u64);
    let mut masked = mask_with_whitespace(source, &CSS_COMMENT_RE);
    if is_scss {
        masked = mask_with_whitespace(&masked, &SCSS_LINE_COMMENT_RE);
    }
    masked
}

/// Normalize a Tailwind CSS `@plugin` target.
///
/// Unlike SCSS `@use`, extensionless targets such as `daisyui` are package
/// specifiers, not local partials. Keep bare specifiers bare and only preserve
/// explicit relative/root-relative paths.
fn normalize_css_plugin_path(path: String) -> String {
    path
}

/// Extract `@import` / `@use` / `@forward` / `@plugin` source paths from a CSS/SCSS string.
///
/// Returns both the raw source and the normalized source. URL imports
/// (`http://`, `https://`, `data:`) are skipped. Use [`extract_css_imports`]
/// when only the normalized form is needed.
///
/// Regex-based by design: this path also handles the SCSS `@use` / `@forward`
/// forms, which lightningcss does not parse, so unlike class extraction there is
/// no parser-backed set to defer the membership decision to.
#[must_use]
pub fn extract_css_import_sources(source: &str, is_scss: bool) -> Vec<CssImportSource> {
    import_sources_from_masked(&mask_css_comments(source, is_scss), is_scss)
}

/// [`extract_css_import_sources`] on a source whose comments are already
/// masked by [`mask_css_comments`].
fn import_sources_from_masked(stripped: &str, is_scss: bool) -> Vec<CssImportSource> {
    let mut out = Vec::new();

    for cap in CSS_IMPORT_RE.captures_iter(stripped) {
        let raw = cap.get(1).or_else(|| cap.get(2)).or_else(|| cap.get(3));
        if let Some(m) = raw {
            let (src, span) = trimmed_match_with_span(m);
            if !src.is_empty() && !is_css_url_import(&src) {
                out.push(CssImportSource {
                    normalized: normalize_css_import_path(src.clone(), is_scss),
                    raw: src,
                    is_plugin: false,
                    span,
                });
            }
        }
    }

    if is_scss {
        for cap in SCSS_USE_RE.captures_iter(stripped) {
            if let Some(m) = cap.get(1) {
                let (raw, span) = trimmed_match_with_span(m);
                out.push(CssImportSource {
                    normalized: normalize_css_import_path(raw.clone(), true),
                    raw,
                    is_plugin: false,
                    span,
                });
            }
        }
    }

    for cap in CSS_PLUGIN_RE.captures_iter(stripped) {
        if let Some(m) = cap.get(1) {
            let (raw, span) = trimmed_match_with_span(m);
            if !raw.is_empty() && !is_css_url_import(&raw) {
                out.push(CssImportSource {
                    normalized: normalize_css_plugin_path(raw.clone()),
                    raw,
                    is_plugin: true,
                    span,
                });
            }
        }
    }

    out
}

fn trimmed_match_with_span(m: regex::Match<'_>) -> (String, Span) {
    let raw = m.as_str();
    let trimmed_start = raw.len() - raw.trim_start().len();
    let trimmed_end = raw.trim_end().len();
    let start = m.start() + trimmed_start;
    let end = m.start() + trimmed_end;
    (raw.trim().to_string(), Span::new(start as u32, end as u32))
}

/// Extract normalized `@import` / `@use` / `@forward` / `@plugin` source paths from a CSS/SCSS string.
///
/// Returns specifiers normalized via `normalize_css_import_path`. URL imports
/// (`http://`, `https://`, `data:`) are skipped. Used by callers that only need
/// entry/dependency source paths; callers that need import kind information
/// should use [`extract_css_import_sources`].
#[must_use]
pub fn extract_css_imports(source: &str, is_scss: bool) -> Vec<String> {
    extract_css_import_sources(source, is_scss)
        .into_iter()
        .map(|source| source.normalized)
        .collect()
}

/// Opening of a Tailwind v4 `@theme` block: `@theme`, optional modifier keywords
/// (`inline` / `static` / `reference` / `default`), then the `{`. Matches up to
/// and including the brace so the caller can brace-match the body from `end()`.
static CSS_THEME_OPEN_RE: LazyLock<regex::Regex> = LazyLock::new(|| {
    crate::static_regex(r"@theme(?:\s+(?:inline|static|reference|default))*\s*\{")
});

/// A `var(--custom-property)` reference, capturing the dashed-ident name without
/// the leading `--`. Used only to credit a theme token read by another theme
/// token inside a `@theme` interior (lightningcss skips the unknown at-rule).
static CSS_VAR_REF_RE: LazyLock<regex::Regex> =
    LazyLock::new(|| crate::static_regex(r"var\(\s*--([A-Za-z0-9_-]+)"));

/// A Tailwind v4 `@theme` token definition: the custom-property name WITHOUT the
/// leading `--` (e.g. `color-brand`) and its 1-based line in the source.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ThemeTokenDef {
    /// The custom-property name with the `--` prefix stripped (`color-brand`).
    pub name: String,
    /// The normalized top-level declaration value, with internal whitespace
    /// collapsed. Empty only when the value could not be recovered.
    pub value: String,
    /// 1-based line of the declaration in the original source.
    pub line: u32,
}

/// Result of scanning a CSS source for Tailwind v4 `@theme` blocks.
#[derive(Debug, Clone, Default, PartialEq, Eq)]
pub struct ThemeScan {
    /// Custom-property tokens DEFINED at the top level of a `@theme` block, with
    /// the `*`-reset form (`--color-*: initial`) and bare-namespace declarations
    /// excluded. Deduped by name (first definition wins for the line).
    pub tokens: Vec<ThemeTokenDef>,
    /// Custom-property names (without `--`) READ via `var()` anywhere inside a
    /// `@theme` block interior, each paired with the 1-based source line of the
    /// `var(` token. lightningcss does not descend into the unknown `@theme`
    /// at-rule, so these reads are invisible to `CssAnalytics`; a token backing
    /// another token (`--color-button: var(--color-brand)`) keeps the backing
    /// token live.
    pub theme_var_reads: Vec<(String, u32)>,
}

/// Scan a CSS source for Tailwind v4 `@theme` blocks, returning the defined
/// design tokens plus the custom properties read via `var()` inside those blocks.
///
/// Tailwind v4 is CSS-first, so `@theme { --color-brand: #f00; }` is the unit of
/// a user-authored design token. lightningcss treats `@theme` as an unknown
/// at-rule and skips it, so this is a separate brace-matching pass (comments and
/// strings masked first so braces / semicolons inside them never break the block
/// boundary). Only top-level `--ident: value` declarations are tokens; declarations
/// inside a nested block (e.g. `@keyframes` for `--animate-*`) are not.
#[must_use]
pub fn scan_theme_blocks(source: &str) -> ThemeScan {
    // Fast path: skip the masking allocation for the common no-`@theme` file.
    if !source.contains("@theme") {
        return ThemeScan::default();
    }
    theme_scan_from_masked(source, &mask_theme_source(source))
}

/// [`scan_theme_blocks`] on a source masked by [`mask_theme_source`].
fn theme_scan_from_masked(source: &str, masked: &str) -> ThemeScan {
    let mut out = ThemeScan::default();
    let mut seen: FxHashSet<String> = FxHashSet::default();
    for open in CSS_THEME_OPEN_RE.find_iter(masked) {
        let body_start = open.end();
        let body_end = find_theme_body_end(masked, body_start);
        collect_theme_declarations(&mut ThemeDeclarationScan {
            source,
            masked,
            start: body_start,
            end: body_end,
            out: &mut out.tokens,
            seen: &mut seen,
        });
        collect_theme_var_reads(
            source,
            masked,
            body_start,
            body_end,
            &mut out.theme_var_reads,
        );
    }
    out
}

/// Located regular-CSS `var(--token)` reads OUTSIDE any `@theme` block interior:
/// `(name, line)` per read, with the `--` stripped from the name. `@theme`-
/// interior reads are deliberately excluded here (they are located separately by
/// [`scan_theme_blocks`] as the distinct `theme-var` surface), so the two read
/// kinds never double-count. Comments / strings / `url()` are masked first, so a
/// `var()` inside those regions is never matched.
#[must_use]
pub fn extract_css_var_reads_located(source: &str) -> Vec<(String, u32)> {
    if !source.contains("var(") {
        return Vec::new();
    }
    css_var_reads_from_masked(source, &mask_theme_source(source))
}

/// [`extract_css_var_reads_located`] on a source masked by
/// [`mask_theme_source`].
fn css_var_reads_from_masked(source: &str, masked: &str) -> Vec<(String, u32)> {
    // Byte ranges of every `@theme { ... }` interior, so reads inside them are
    // skipped (they are the `theme-var` surface, located elsewhere).
    let mut theme_bodies: Vec<(usize, usize)> = Vec::new();
    if masked.contains("@theme") {
        for open in CSS_THEME_OPEN_RE.find_iter(masked) {
            let body_start = open.end();
            let body_end = find_theme_body_end(masked, body_start);
            theme_bodies.push((body_start, body_end));
        }
    }
    let in_theme = |offset: usize| theme_bodies.iter().any(|&(s, e)| offset >= s && offset < e);
    let mut out = Vec::new();
    // Incremental line counter: `captures_iter` yields matches in source order, so
    // advance from the previous read's offset instead of rescanning the whole
    // prefix per read (issue #1843 follow-up). Masking preserves byte offsets 1:1,
    // but newlines are counted over `source` (comment masking blanks newlines in
    // `masked`), matching the original `line_at_offset(source, ..)`.
    let mut last_pos = 0usize;
    let mut last_line = 1u32;
    for cap in CSS_VAR_REF_RE.captures_iter(masked) {
        let (Some(whole), Some(name)) = (cap.get(0), cap.get(1)) else {
            continue;
        };
        if in_theme(whole.start()) {
            continue;
        }
        let offset = whole.start();
        last_line = last_line.saturating_add(newlines_between(source, last_pos, offset));
        last_pos = offset;
        out.push((name.as_str().to_owned(), last_line));
    }
    out
}

/// Mask comments, strings, and `url(...)` while preserving byte offsets so
/// braces inside those regions never affect `@theme` block matching.
fn mask_theme_source(source: &str) -> String {
    mask_with_whitespace(&mask_css_comments(source, false), &CSS_NON_SELECTOR_RE)
}

/// Brace-match from just after a `@theme {` opener to its partner.
fn find_theme_body_end(masked: &str, body_start: usize) -> usize {
    let bytes = masked.as_bytes();
    let mut depth = 1usize;
    let mut i = body_start;
    while i < bytes.len() {
        match bytes[i] {
            b'{' => depth += 1,
            b'}' => {
                depth -= 1;
                if depth == 0 {
                    break;
                }
            }
            _ => {}
        }
        i += 1;
    }
    i.min(bytes.len())
}

fn collect_theme_var_reads(
    source: &str,
    masked: &str,
    body_start: usize,
    body_end: usize,
    out: &mut Vec<(String, u32)>,
) {
    let Some(body) = masked.get(body_start..body_end) else {
        return;
    };
    // Incremental line counter: matches arrive in source order, so advance from
    // the previous read's offset instead of rescanning the whole prefix per read
    // (issue #1843 follow-up). Starting from offset 0 keeps the first read's line
    // identical to `line_at_offset(source, offset)`.
    let mut last_pos = 0usize;
    let mut last_line = 1u32;
    for cap in CSS_VAR_REF_RE.captures_iter(body) {
        let (Some(whole), Some(name)) = (cap.get(0), cap.get(1)) else {
            continue;
        };
        // Absolute byte offset of the `var(` token start in the original source
        // (masking preserves byte offsets 1:1).
        let offset = body_start + whole.start();
        last_line = last_line.saturating_add(newlines_between(source, last_pos, offset));
        last_pos = offset;
        out.push((name.as_str().to_owned(), last_line));
    }
}

/// 1-based line number of `offset` in `source`, counting `\n` up to (but not
/// including) the byte at `offset`. Out-of-range offsets clamp to line 1. The
/// tests use it as the reference for the incremental line counters.
#[cfg(test)]
fn line_at_offset(source: &str, offset: usize) -> u32 {
    note_line_lookup_bytes(offset);
    let count = source
        .get(..offset)
        .map_or(0, |s| s.bytes().filter(|&b| b == b'\n').count());
    u32::try_from(1 + count).unwrap_or(u32::MAX)
}

/// Count the `\n` bytes in `source[from..to]`, returning 0 for a reversed or
/// out-of-range span. Feeds an incremental 1-based line counter across regex
/// matches that arrive in source order, replacing the O(matches * n) per-match
/// `source[..offset]` prefix rescan (issue #1843 follow-up: worst on a single
/// long line with no newlines).
fn newlines_between(source: &str, from: usize, to: usize) -> u32 {
    #[cfg(test)]
    note_line_lookup_bytes(to.saturating_sub(from));
    let count = source
        .get(from..to)
        .map_or(0, |s| s.bytes().filter(|&b| b == b'\n').count());
    u32::try_from(count).unwrap_or(u32::MAX)
}

/// Walk a masked `@theme` body collecting top-level `--ident: value` declarations
/// as tokens. Tracks brace depth so declarations inside a nested block (e.g. an
/// `@keyframes` for `--animate-*`) are skipped, and statement position so only a
/// `--ident` at a declaration start counts. The `*`-reset form (`--color-*`) is
/// excluded because the `*` breaks the ident scan before the `:`.
fn collect_theme_declarations(scan: &mut ThemeDeclarationScan<'_, '_>) {
    let bytes = scan.masked.as_bytes();
    let mut depth = 0usize;
    let mut expect_decl = true;
    let mut i = scan.start;
    while i < scan.end {
        let b = bytes[i];
        match b {
            b'{' => {
                depth += 1;
                expect_decl = false;
                i += 1;
            }
            b'}' => {
                depth = depth.saturating_sub(1);
                if depth == 0 {
                    expect_decl = true;
                }
                i += 1;
            }
            b';' => {
                if depth == 0 {
                    expect_decl = true;
                }
                i += 1;
            }
            _ if b.is_ascii_whitespace() => i += 1,
            _ => {
                if depth == 0 && expect_decl {
                    expect_decl = false;
                    i = scan_theme_declaration(scan, b, i);
                } else {
                    i += 1;
                }
            }
        }
    }
}

struct ThemeDeclarationScan<'a, 'b> {
    source: &'a str,
    masked: &'a str,
    start: usize,
    end: usize,
    out: &'b mut Vec<ThemeTokenDef>,
    seen: &'b mut FxHashSet<String>,
}

/// At a declaration start, harvest a `--ident:` custom-property name and return
/// the cursor advanced past the scanned ident. Returns `i + 1` for any non-`--`
/// declaration start.
fn scan_theme_declaration(scan: &mut ThemeDeclarationScan<'_, '_>, b: u8, i: usize) -> usize {
    let bytes = scan.masked.as_bytes();
    if !(b == b'-' && bytes.get(i + 1) == Some(&b'-')) {
        return i + 1;
    }
    let id_start = i;
    let mut j = i;
    while j < scan.end {
        let c = bytes[j];
        if c == b'-' || c == b'_' || c.is_ascii_alphanumeric() {
            j += 1;
        } else {
            break;
        }
    }
    let mut k = j;
    while k < scan.end && bytes[k].is_ascii_whitespace() {
        k += 1;
    }
    // Only a `--ident:` (no `*` before the colon) is a token.
    if k < scan.end && bytes[k] == b':' {
        let name = &scan.masked[id_start + 2..j];
        if !name.is_empty() && scan.seen.insert(name.to_owned()) {
            let value = theme_declaration_value(scan.source, scan.masked, k + 1, scan.end);
            let line = 1 + scan
                .source
                .get(..id_start)
                .map_or(0, |s| s.bytes().filter(|&x| x == b'\n').count());
            scan.out.push(ThemeTokenDef {
                name: name.to_owned(),
                value,
                line: u32::try_from(line).unwrap_or(u32::MAX),
            });
        }
    }
    j
}

fn theme_declaration_value(source: &str, masked: &str, start: usize, end: usize) -> String {
    let bytes = masked.as_bytes();
    let mut depth = 0usize;
    let mut i = start;
    while i < end {
        match bytes[i] {
            b'{' => depth += 1,
            b'}' => {
                if depth == 0 {
                    break;
                }
                depth -= 1;
            }
            b';' if depth == 0 => break,
            _ => {}
        }
        i += 1;
    }
    source
        .get(start..i)
        .unwrap_or_default()
        .split_whitespace()
        .collect::<Vec<_>>()
        .join(" ")
}

/// Extract the utility tokens referenced in `@apply` directive bodies across a
/// CSS source (comment / string masked). `@apply rounded-card font-bold;` yields
/// `["rounded-card", "font-bold"]`. The leading-`!` and trailing-`!` important
/// modifiers and a bare `!important` token are stripped, so a theme token whose
/// utility is applied only via `@apply` is credited as used.
#[must_use]
pub fn extract_apply_tokens(source: &str) -> Vec<String> {
    extract_apply_tokens_located(source)
        .into_iter()
        .map(|(token, _line)| token)
        .collect()
}

/// Like [`extract_apply_tokens`], but pairs each class-shaped token with the
/// 1-based source line of its `@apply` directive. Used by the token-consumer
/// reverse index to locate `@apply`-surface consumers; masking preserves byte
/// offsets so the directive line is recoverable from the match start.
#[must_use]
pub fn extract_apply_tokens_located(source: &str) -> Vec<(String, u32)> {
    // Fast path: skip the masking allocation for the common no-`@apply` file.
    if !source.contains("@apply") {
        return Vec::new();
    }
    apply_tokens_from_masked(source, &mask_theme_source(source))
}

/// [`extract_apply_tokens_located`] on a source masked by
/// [`mask_theme_source`].
fn apply_tokens_from_masked(source: &str, masked: &str) -> Vec<(String, u32)> {
    let mut out = Vec::new();
    // Matches arrive in source order, so the line advances from the previous
    // match instead of a rescan of the whole prefix for each match.
    let mut last_pos = 0usize;
    let mut line = 1u32;
    for m in CSS_APPLY_RE.find_iter(masked) {
        line = line.saturating_add(newlines_between(source, last_pos, m.start()));
        last_pos = m.start();
        let body = m.as_str().trim_start_matches("@apply");
        for token in body.split_whitespace() {
            let token = token.trim_matches('!');
            if token.is_empty() || token == "important" {
                continue;
            }
            out.push((token.to_owned(), line));
        }
    }
    out
}

/// The Tailwind token surfaces of one stylesheet, from one masked copy of the
/// source.
#[derive(Debug, Clone, Default, PartialEq, Eq)]
pub struct StylesheetTokens {
    /// The result of [`scan_theme_blocks`].
    pub theme: ThemeScan,
    /// The result of [`extract_apply_tokens_located`].
    pub apply_tokens_located: Vec<(String, u32)>,
    /// The result of [`extract_css_var_reads_located`].
    pub css_var_reads_located: Vec<(String, u32)>,
}

/// Scan one stylesheet for `@theme` tokens, `@apply` tokens and `var()` reads.
///
/// The result is equal to the three separate scans, but the comment and string
/// mask runs at most once for the source.
#[must_use]
pub fn scan_stylesheet_tokens(source: &str) -> StylesheetTokens {
    let has_theme = source.contains("@theme");
    let has_apply = source.contains("@apply");
    let has_var = source.contains("var(");
    if !(has_theme || has_apply || has_var) {
        return StylesheetTokens::default();
    }
    let masked = mask_theme_source(source);
    StylesheetTokens {
        theme: if has_theme {
            theme_scan_from_masked(source, &masked)
        } else {
            ThemeScan::default()
        },
        apply_tokens_located: if has_apply {
            apply_tokens_from_masked(source, &masked)
        } else {
            Vec::new()
        },
        css_var_reads_located: if has_var {
            css_var_reads_from_masked(source, &masked)
        } else {
            Vec::new()
        },
    }
}

/// Extract class names from a CSS module file as named exports.
///
/// For standard CSS, lightningcss parses the source into an AST and supplies the
/// authoritative set of class-selector names; the byte-offset scanner then
/// locates each name's [`Span`] in the ORIGINAL `source` (pointing at the bare
/// class name, no leading dot) so downstream `compute_line_offsets` resolves the
/// real declaration line and column instead of falling back to line:1 col:0
/// (issue #549). For SCSS (Sass syntax lightningcss does not parse) and for any
/// CSS that fails to parse outright, the regex-only scanner is used unchanged.
pub fn extract_css_module_exports(source: &str, is_scss: bool) -> Vec<ExportInfo> {
    css_module_exports(source, &mask_css_comments(source, is_scss), is_scss)
}

/// [`extract_css_module_exports`] with the comment mask of `source` from
/// [`mask_css_comments`], so a caller that has the mask does not compute it
/// again.
fn css_module_exports(source: &str, comment_masked: &str, is_scss: bool) -> Vec<ExportInfo> {
    if !is_scss && let Some(class_set) = lightningcss_class_set(source) {
        return scan_css_module_exports(comment_masked, Some(&class_set));
    }
    scan_css_module_exports(comment_masked, None)
}

/// Scan the comment-masked source for `.class` tokens and emit one
/// [`ExportInfo`] per distinct class (first occurrence wins), with a [`Span`]
/// pointing at the post-dot identifier. The mask keeps byte offsets, so the
/// span is also valid in the original source.
///
/// When `class_filter` is `Some`, only tokens present in the AST-derived set are
/// emitted, so the parser owns the membership decision and the scanner owns only
/// span location. When `class_filter` is `None` (SCSS / parse-failure fallback),
/// the at-rule prelude is masked to keep `@layer foo.bar` / `@import ...
/// layer(...)` segments from being mistaken for classes.
fn scan_css_module_exports(
    comment_masked: &str,
    class_filter: Option<&FxHashSet<String>>,
) -> Vec<ExportInfo> {
    let masked = mask_css_module_class_candidates(comment_masked, class_filter.is_some());
    let mut seen = FxHashSet::default();
    let mut exports = Vec::new();
    for cap in CSS_CLASS_RE.captures_iter(&masked) {
        if let Some(m) = cap.get(1) {
            push_css_class_export(m, class_filter, &mut seen, &mut exports);
        }
    }
    exports
}

fn mask_css_module_class_candidates(comment_masked: &str, has_class_filter: bool) -> String {
    let mut masked = mask_with_whitespace(comment_masked, &CSS_NON_SELECTOR_RE);
    masked = mask_view_transition_arguments(&masked);
    if !has_class_filter {
        masked = mask_with_whitespace(&masked, &CSS_AT_RULE_PRELUDE_RE);
        masked = mask_sass_module_members(&masked);
    }
    mask_css_module_global_scopes(masked)
}

const GLOBAL_KEYWORD: &str = ":global";
const LOCAL_KEYWORD: &str = ":local";

/// A CSS Modules scope keyword (`:global` or `:local`) at a byte offset.
#[derive(Clone, Copy)]
struct ScopeKeyword {
    is_global: bool,
    len: usize,
    /// The keyword opens a `(...)` argument, as in `:global(.foo)`.
    has_argument: bool,
}

/// Return the CSS Modules scope keyword that starts at `bytes[i]`, if any.
/// A longer identifier such as `:global-ish` is not a scope keyword.
fn scope_keyword_at(bytes: &[u8], i: usize) -> Option<ScopeKeyword> {
    let rest = &bytes[i..];
    let (is_global, len) = if rest.starts_with(GLOBAL_KEYWORD.as_bytes()) {
        (true, GLOBAL_KEYWORD.len())
    } else if rest.starts_with(LOCAL_KEYWORD.as_bytes()) {
        (false, LOCAL_KEYWORD.len())
    } else {
        return None;
    };
    let next = rest.get(len).copied();
    if next.is_some_and(|b| b.is_ascii_alphanumeric() || b == b'_' || b == b'-') {
        return None;
    }
    Some(ScopeKeyword {
        is_global,
        len,
        has_argument: next == Some(b'('),
    })
}

/// Replace each bare `:global` / `:local` keyword (without a `(...)` argument)
/// with spaces of equal length, so lightningcss can parse the rule.
pub(crate) fn blank_bare_css_module_scope_keywords(source: &str) -> std::borrow::Cow<'_, str> {
    if !source.contains(GLOBAL_KEYWORD) && !source.contains(LOCAL_KEYWORD) {
        return std::borrow::Cow::Borrowed(source);
    }
    let bytes = source.as_bytes();
    let mut out = source.to_string();
    let mut i = 0;
    while i < bytes.len() {
        match scope_keyword_at(bytes, i) {
            Some(keyword) => {
                if !keyword.has_argument {
                    out.replace_range(i..i + keyword.len, &" ".repeat(keyword.len));
                }
                i += keyword.len;
            }
            None => i += 1,
        }
    }
    std::borrow::Cow::Owned(out)
}

/// The scope of one open parenthesis in a selector.
#[derive(Clone, Copy, PartialEq, Eq)]
enum ParenScope {
    Global,
    Local,
    Other,
}

/// Selector scope state for [`mask_css_module_global_scopes`].
struct GlobalScopeScan {
    /// The scope that each open `{` block gives to its nested rules.
    block_global: Vec<bool>,
    /// The scope of the current selector, changed by a bare `:global`/`:local`.
    selector_global: bool,
    parens: Vec<ParenScope>,
}

impl GlobalScopeScan {
    fn inherited(&self) -> bool {
        self.block_global.last().copied().unwrap_or(false)
    }

    /// End the current selector or declaration. The next one starts in the
    /// scope of the enclosing block.
    fn reset_selector(&mut self) {
        self.selector_global = self.inherited();
        self.parens.clear();
    }

    /// A `:global(...)` or `:local(...)` argument sets the scope for its
    /// content. Otherwise the selector scope applies.
    fn is_global(&self) -> bool {
        self.parens
            .iter()
            .rev()
            .find(|scope| **scope != ParenScope::Other)
            .map_or(self.selector_global, |scope| *scope == ParenScope::Global)
    }
}

/// Blank the `.` of each class token in a CSS Modules global scope, so the
/// class scanner does not report it. CSS Modules do not put such a class in
/// the class map of the module (issue #3311).
///
/// The input must have comments and strings masked already. These forms make
/// a global scope:
/// - `:global(.a .b)`: the classes in the argument are global.
/// - a bare `:global`: the rest of the selector is global, until a bare
///   `:local`, a `:local(...)` argument, or the next selector in the list.
/// - a rule whose selector ends in a global scope: its nested rules start in
///   the global scope, as Sass and `postcss-nested` compile them.
///
/// A source without a block `{` (a `#{` interpolation does not count) uses the
/// indented Sass syntax, so a line end also ends the selector there.
fn mask_css_module_global_scopes(masked: String) -> String {
    if !masked.contains(GLOBAL_KEYWORD) {
        return masked;
    }
    let indented = !masked
        .match_indices('{')
        .any(|(i, _)| i == 0 || masked.as_bytes()[i - 1] != b'#');
    let bytes = masked.as_bytes();
    let mut blank = Vec::new();
    let mut scan = GlobalScopeScan {
        block_global: Vec::new(),
        selector_global: false,
        parens: Vec::new(),
    };
    let mut i = 0;
    while i < bytes.len() {
        if let Some(keyword) = scope_keyword_at(bytes, i) {
            if keyword.has_argument {
                scan.parens.push(if keyword.is_global {
                    ParenScope::Global
                } else {
                    ParenScope::Local
                });
                i += keyword.len + 1;
                continue;
            }
            if scan.parens.is_empty() {
                scan.selector_global = keyword.is_global;
            }
            i += keyword.len;
            continue;
        }
        match bytes[i] {
            b'#' if bytes.get(i + 1) == Some(&b'{') => {
                i = scss_interpolation_end(bytes, i + 2);
                continue;
            }
            b'(' => scan.parens.push(ParenScope::Other),
            b')' => {
                scan.parens.pop();
            }
            b',' if scan.parens.is_empty() => scan.selector_global = scan.inherited(),
            b'{' => {
                scan.block_global.push(scan.selector_global);
                scan.reset_selector();
            }
            b'}' => {
                scan.block_global.pop();
                scan.reset_selector();
            }
            b';' => scan.reset_selector(),
            b'\n' if indented => scan.reset_selector(),
            b'.' if scan.is_global()
                && bytes
                    .get(i + 1)
                    .is_some_and(|b| b.is_ascii_alphabetic() || *b == b'_') =>
            {
                blank.push(i);
            }
            _ => {}
        }
        i += 1;
    }
    let mut out = masked;
    for offset in blank {
        out.replace_range(offset..=offset, " ");
    }
    out
}

/// Return the offset after the `}` that closes a SCSS `#{...}` interpolation
/// whose content starts at `start`.
fn scss_interpolation_end(bytes: &[u8], start: usize) -> usize {
    let mut depth = 1usize;
    let mut i = start;
    while i < bytes.len() {
        match bytes[i] {
            b'{' => depth += 1,
            b'}' => {
                depth -= 1;
                if depth == 0 {
                    return i + 1;
                }
            }
            _ => {}
        }
        i += 1;
    }
    bytes.len()
}

fn push_css_class_export(
    class_match: regex::Match<'_>,
    class_filter: Option<&FxHashSet<String>>,
    seen: &mut FxHashSet<String>,
    exports: &mut Vec<ExportInfo>,
) {
    let class_name = class_match.as_str().to_string();
    if class_filter.is_some_and(|filter| !filter.contains(&class_name)) {
        return;
    }
    if seen.insert(class_name.clone()) {
        exports.push(css_class_export(class_name, class_match));
    }
}

fn css_class_export(class_name: String, class_match: regex::Match<'_>) -> ExportInfo {
    #[expect(
        clippy::cast_possible_truncation,
        reason = "CSS files exceeding u32::MAX bytes are not a realistic input"
    )]
    let span = Span::new(class_match.start() as u32, class_match.end() as u32);
    ExportInfo {
        name: ExportName::Named(class_name),
        local_name: None,
        is_type_only: false,
        visibility: VisibilityTag::None,
        expected_unused_reason: None,
        span,
        members: Vec::new(),
        is_side_effect_used: false,
        super_class: None,
        deprecated: false,
        deprecated_reason: None,
    }
}

/// Build the import edges for a CSS/SCSS source: every `@import`/`@use`/etc.
/// directive plus a synthetic `tailwindcss` side-effect import when `@apply` or
/// `@tailwind` is present.
fn build_css_imports(stripped: &str, is_scss: bool) -> Vec<ImportInfo> {
    let mut imports = Vec::new();

    for css_source in import_sources_from_masked(stripped, is_scss) {
        imports.push(ImportInfo {
            source: css_source.normalized,
            imported_name: if css_source.is_plugin {
                ImportedName::Default
            } else {
                ImportedName::SideEffect
            },
            local_name: String::new(),
            is_type_only: false,
            is_type_only_star: false,
            from_style: false,
            span: css_source.span,
            source_span: css_source.span,
        });
    }

    let has_apply = CSS_APPLY_RE.is_match(stripped);
    let has_tailwind = CSS_TAILWIND_RE.is_match(stripped);
    if has_apply || has_tailwind {
        imports.push(ImportInfo {
            source: "tailwindcss".to_string(),
            imported_name: ImportedName::SideEffect,
            local_name: String::new(),
            is_type_only: false,
            is_type_only_star: false,
            from_style: false,
            span: Span::default(),
            source_span: Span::default(),
        });
    }

    imports
}

/// Parse a CSS/SCSS file, extracting @import, @use, @forward, @plugin, @apply, and @tailwind directives.
pub(crate) fn parse_css_to_module(
    file_id: FileId,
    path: &Path,
    source: &str,
    content_hash: u64,
) -> ModuleInfo {
    let parsed_suppressions = crate::suppress::parse_suppressions_from_source(source);
    let is_scss = path
        .extension()
        .and_then(|e| e.to_str())
        .is_some_and(|ext| matches!(ext, "scss" | "sass" | "less"));

    let stripped = crate::css_mask::mask_sass_shorthand(mask_css_comments(source, is_scss), path);
    let imports = build_css_imports(&stripped, is_scss);

    let exports = if is_css_module_file(path) {
        css_module_exports(source, &stripped, is_scss)
    } else {
        Vec::new()
    };

    css_module_info(
        file_id,
        content_hash,
        source,
        parsed_suppressions,
        imports,
        exports,
    )
}

/// Assemble the `ModuleInfo` for a CSS/SCSS file: the import/export edges plus
/// the line offsets and suppressions; all AST-derived fields stay empty since
/// CSS carries no JS-level structure. Pure plumbing struct literal.
fn css_module_info(
    file_id: FileId,
    content_hash: u64,
    source: &str,
    parsed_suppressions: crate::suppress::ParsedSuppressions,
    imports: Vec<ImportInfo>,
    exports: Vec<ExportInfo>,
) -> ModuleInfo {
    crate::module_info::non_js_module_info(crate::module_info::NonJsModuleInfoInput {
        file_id,
        content_hash,
        source,
        parsed_suppressions,
        imports,
        exports,
    })
}

#[cfg(all(test, not(miri)))]
mod tests {
    use super::*;

    /// Helper to collect export names as strings from `extract_css_module_exports`.
    fn export_names(source: &str) -> Vec<String> {
        extract_css_module_exports(source, false)
            .into_iter()
            .filter_map(|e| match e.name {
                ExportName::Named(n) => Some(n),
                ExportName::Default => None,
            })
            .collect()
    }

    #[test]
    fn is_css_file_css() {
        assert!(is_css_file(Path::new("styles.css")));
    }

    #[test]
    fn is_css_file_scss() {
        assert!(is_css_file(Path::new("styles.scss")));
    }

    #[test]
    fn is_css_file_sass() {
        assert!(is_css_file(Path::new("styles.sass")));
    }

    #[test]
    fn is_css_file_less() {
        assert!(is_css_file(Path::new("styles.less")));
    }

    #[test]
    fn is_css_file_rejects_js() {
        assert!(!is_css_file(Path::new("app.js")));
    }

    #[test]
    fn is_css_file_rejects_ts() {
        assert!(!is_css_file(Path::new("app.ts")));
    }

    #[test]
    fn is_css_file_rejects_no_extension() {
        assert!(!is_css_file(Path::new("Makefile")));
    }

    #[test]
    fn is_css_module_file_module_css() {
        assert!(is_css_module_file(Path::new("Component.module.css")));
    }

    #[test]
    fn is_css_module_file_module_scss() {
        assert!(is_css_module_file(Path::new("Component.module.scss")));
    }

    #[test]
    fn is_css_module_file_rejects_plain_css() {
        assert!(!is_css_module_file(Path::new("styles.css")));
    }

    #[test]
    fn is_css_module_file_rejects_plain_scss() {
        assert!(!is_css_module_file(Path::new("styles.scss")));
    }

    #[test]
    fn is_css_module_file_rejects_module_js() {
        assert!(!is_css_module_file(Path::new("utils.module.js")));
    }

    #[test]
    fn extracts_single_class() {
        let names = export_names(".foo { color: red; }");
        assert_eq!(names, vec!["foo"]);
    }

    #[test]
    fn extracts_multiple_classes() {
        let names = export_names(".foo { } .bar { }");
        assert_eq!(names, vec!["foo", "bar"]);
    }

    #[test]
    fn extracts_nested_classes() {
        let names = export_names(".foo .bar { color: red; }");
        assert!(names.contains(&"foo".to_string()));
        assert!(names.contains(&"bar".to_string()));
    }

    #[test]
    fn extracts_hyphenated_class() {
        let names = export_names(".my-class { }");
        assert_eq!(names, vec!["my-class"]);
    }

    #[test]
    fn extracts_camel_case_class() {
        let names = export_names(".myClass { }");
        assert_eq!(names, vec!["myClass"]);
    }

    fn scss_export_names(source: &str) -> Vec<String> {
        extract_css_module_exports(source, true)
            .into_iter()
            .filter_map(|e| match e.name {
                ExportName::Named(n) => Some(n),
                ExportName::Default => None,
            })
            .collect()
    }

    #[test]
    fn skips_class_inside_global_pseudo() {
        // CSS Modules do not put a `:global(.foo)` class in the class map.
        let names = export_names(":global(.globalClass) { color: red; }");
        assert!(names.is_empty(), "got {names:?}");
    }

    #[test]
    fn keeps_local_class_next_to_global_pseudo() {
        // Issue #3311.
        let names = export_names(".wrapper :global(.selector) { color: red; }");
        assert_eq!(names, vec!["wrapper"]);
    }

    #[test]
    fn skips_every_class_inside_global_pseudo_with_descendants() {
        let names = export_names(".panel :global(.outer .inner.mod) { color: red; }");
        assert_eq!(names, vec!["panel"]);
    }

    #[test]
    fn skips_classes_after_bare_global() {
        let names = export_names(".lead :global .first .second { color: red; }");
        assert_eq!(names, vec!["lead"]);
    }

    #[test]
    fn bare_global_stops_at_local() {
        let names = export_names(":global .g1 :local(.l1) .g2 :local .l2 .l3, .l4 { color: red; }");
        assert_eq!(names, vec!["l1", "l2", "l3", "l4"]);
    }

    #[test]
    fn bare_global_stops_at_comma() {
        let names = export_names(":global .g1, .l1 :is(.l2, .l3) { color: red; }");
        assert_eq!(names, vec!["l1", "l2", "l3"]);
    }

    #[test]
    fn global_class_inside_media_query_is_skipped() {
        let names = export_names("@media (min-width: 1px) { .media :global(.g) { color: red; } }");
        assert_eq!(names, vec!["media"]);
    }

    #[test]
    fn class_both_local_and_global_stays_exported_at_local_span() {
        let source = ":global(.both) { color: red; }\n.both { color: blue; }";
        let exports = extract_css_module_exports(source, false);
        assert_eq!(exports.len(), 1);
        let span = exports[0].span;
        assert_eq!(&source[span.start as usize..span.end as usize], "both");
        assert!(
            span.start > 20,
            "span must point at the local rule: {span:?}"
        );
    }

    #[test]
    fn composes_keeps_local_classes() {
        let names = export_names(
            ".base { color: red; }\n.button { composes: base; composes: g from global; }",
        );
        assert_eq!(names, vec!["base", "button"]);
    }

    #[test]
    fn global_in_strings_and_comments_has_no_effect() {
        let names =
            export_names("/* :global */ .a { content: \":global\"; }\n.b :global(.c) { }\n.d { }");
        assert_eq!(names, vec!["a", "b", "d"]);
    }

    #[test]
    fn scss_skips_global_pseudo_and_global_block() {
        let names = scss_export_names(
            ".card {\n  :global(.g1) { color: red; }\n  .title { color: blue; }\n}\n\
             :global {\n  .g2 { color: red; }\n  .g3 { :local(.l1) { color: red; } }\n}\n\
             .after { color: blue; }",
        );
        assert_eq!(names, vec!["card", "title", "l1", "after"]);
    }

    #[test]
    fn scss_bare_global_in_parent_reaches_nested_rules() {
        let names = scss_export_names(
            ".a :global .b {\n  .c { color: red; }\n  &:hover { color: blue; }\n}\n.d { }",
        );
        assert_eq!(names, vec!["a", "d"]);
    }

    #[test]
    fn scss_interpolation_does_not_end_global_scope() {
        let names = scss_export_names("$n: x;\n.a :global .b-#{$n} .c { }\n.d { }");
        assert_eq!(names, vec!["a", "d"]);
    }

    #[test]
    fn sass_indented_bare_global_ends_at_line_end() {
        let names = scss_export_names(".a :global .b\n  color: red\n.c\n  color: blue\n");
        assert_eq!(names, vec!["a", "c"]);
    }

    #[test]
    fn sass_indented_with_interpolation_still_ends_global_at_line_end() {
        let names = scss_export_names(
            ".a :global .b\n  color: red\n.c-#{$x}\n  color: blue\n.d\n  color: blue\n",
        );
        // The scanner keeps the static prefix `c-` of an interpolated class.
        assert_eq!(names, vec!["a", "c-", "d"]);
    }

    #[test]
    fn global_like_pseudo_names_are_not_global() {
        let names = export_names(".a:global-ish .b { }");
        assert!(names.contains(&"b".to_string()), "got {names:?}");
    }

    #[test]
    fn extracts_class_inside_local_pseudo() {
        let names = export_names(":local(.localClass) { color: red; }");
        assert_eq!(names, vec!["localClass"]);
    }

    #[test]
    fn extracts_classes_inside_negation() {
        let names = export_names(".btn:not(.disabled) { }");
        assert!(names.contains(&"btn".to_string()), "got {names:?}");
        assert!(names.contains(&"disabled".to_string()), "got {names:?}");
    }

    #[test]
    fn extracts_classes_inside_is_and_where() {
        let names = export_names(":is(.a, .b) :where(.c) { }");
        for expected in ["a", "b", "c"] {
            assert!(
                names.contains(&expected.to_string()),
                "missing {expected} in {names:?}"
            );
        }
    }

    #[test]
    fn extracts_underscore_class() {
        let names = export_names("._hidden { } .__wrapper { }");
        assert!(names.contains(&"_hidden".to_string()));
        assert!(names.contains(&"__wrapper".to_string()));
    }

    #[test]
    fn pseudo_selector_hover() {
        let names = export_names(".foo:hover { color: blue; }");
        assert_eq!(names, vec!["foo"]);
    }

    #[test]
    fn pseudo_selector_focus() {
        let names = export_names(".input:focus { outline: none; }");
        assert_eq!(names, vec!["input"]);
    }

    #[test]
    fn pseudo_element_before() {
        let names = export_names(".icon::before { content: ''; }");
        assert_eq!(names, vec!["icon"]);
    }

    #[test]
    fn combined_pseudo_selectors() {
        let names = export_names(".btn:hover, .btn:active, .btn:focus { }");
        assert_eq!(names, vec!["btn"]);
    }

    #[test]
    fn classes_inside_media_query() {
        let names = export_names(
            "@media (max-width: 768px) { .mobile-nav { display: block; } .desktop-nav { display: none; } }",
        );
        assert!(names.contains(&"mobile-nav".to_string()));
        assert!(names.contains(&"desktop-nav".to_string()));
    }

    #[test]
    fn classes_inside_multi_line_media_query() {
        let names =
            export_names("@media\n  screen and (min-width: 600px)\n{\n  .real { color: red; }\n}");
        assert_eq!(names, vec!["real"]);
    }

    #[test]
    fn at_layer_statement_does_not_export() {
        let names = export_names("@layer foo.bar;");
        assert!(names.is_empty(), "got {names:?}");
        let names = export_names("@layer foo.bar, foo.baz;");
        assert!(names.is_empty(), "got {names:?}");
    }

    #[test]
    fn at_layer_block_keeps_body_classes() {
        let names = export_names("@layer foo.bar { .root { color: red; } }");
        assert_eq!(names, vec!["root"]);
    }

    #[test]
    fn at_layer_multiline_prelude_keeps_body_classes() {
        let names = export_names("@layer\n  foo.bar\n{ .root { color: red; } }");
        assert_eq!(names, vec!["root"]);
    }

    #[test]
    fn at_layer_with_nested_media_keeps_body() {
        let names =
            export_names("@layer foo.bar { @media (max-width: 768px) { .real { color: red; } } }");
        assert_eq!(names, vec!["real"]);
    }

    #[test]
    fn at_import_with_layer_attribute_does_not_export() {
        let names = export_names(r#"@import url("x.css") layer(theme.button);"#);
        assert!(names.is_empty(), "got {names:?}");
    }

    #[test]
    fn class_then_at_layer_does_not_leak_prelude() {
        let names =
            export_names(".outer { color: blue; } @layer foo.bar { .inner { color: red; } }");
        assert_eq!(names, vec!["outer", "inner"]);
    }

    #[test]
    fn at_scope_keeps_selector_list_classes() {
        let names = export_names("@scope (.parent) to (.child) { .title { color: red; } }");
        assert!(names.contains(&"parent".to_string()), "got {names:?}");
        assert!(names.contains(&"child".to_string()), "got {names:?}");
        assert!(names.contains(&"title".to_string()), "got {names:?}");
    }

    #[test]
    fn at_keyframes_numeric_step_is_not_class() {
        let names = export_names(
            "@keyframes slide { 0% { transform: scale(.5); } 100% { transform: scale(1); } }",
        );
        assert!(names.is_empty(), "got {names:?}");
    }

    #[test]
    fn at_webkit_keyframes_keeps_body_classes() {
        let names = export_names("@-webkit-keyframes slide { 0% { } 100% { } } .real { }");
        assert_eq!(names, vec!["real"]);
    }

    #[test]
    fn deduplicates_repeated_class() {
        let names = export_names(".btn { color: red; } .btn { font-size: 14px; }");
        assert_eq!(names.iter().filter(|n| *n == "btn").count(), 1);
    }

    #[test]
    fn empty_source() {
        let names = export_names("");
        assert!(names.is_empty());
    }

    #[test]
    fn no_classes() {
        let names = export_names("body { margin: 0; } * { box-sizing: border-box; }");
        assert!(names.is_empty());
    }

    #[test]
    fn ignores_classes_in_block_comments() {
        let names = export_names("/* .fake { } */ .real { }");
        assert!(!names.contains(&"fake".to_string()));
        assert!(names.contains(&"real".to_string()));
    }

    #[test]
    fn ignores_classes_in_scss_line_comments() {
        let exports = extract_css_module_exports("// .fake\n.real { }", true);
        let names: Vec<_> = exports
            .iter()
            .filter_map(|e| match &e.name {
                ExportName::Named(n) => Some(n.as_str()),
                ExportName::Default => None,
            })
            .collect();
        assert_eq!(names, vec!["real"]);
    }

    #[test]
    fn ignores_classes_in_strings() {
        let names = export_names(r#".real { content: ".fake"; }"#);
        assert!(names.contains(&"real".to_string()));
        assert!(!names.contains(&"fake".to_string()));
    }

    #[test]
    fn ignores_classes_in_url() {
        let names = export_names(".real { background: url(./images/hero.png); }");
        assert!(names.contains(&"real".to_string()));
        assert!(!names.contains(&"png".to_string()));
    }

    #[test]
    fn mask_css_block_comment() {
        let source = "/* removed */ .kept { }";
        let result = mask_css_comments(source, false);
        assert!(!result.contains("removed"));
        assert_eq!(result.len(), source.len());
        assert_eq!(result.find(".kept"), source.find(".kept"));
    }

    #[test]
    fn mask_scss_line_comment() {
        let source = "// removed\n.kept { }";
        let result = mask_css_comments(source, true);
        assert!(!result.contains("removed"));
        assert_eq!(result.len(), source.len());
        assert_eq!(result.find(".kept"), source.find(".kept"));
    }

    #[test]
    fn mask_scss_preserves_css_outside_comments() {
        let source = "// line comment\n/* block comment */\n.visible { color: red; }";
        let result = mask_css_comments(source, true);
        assert!(result.contains(".visible"));
    }

    #[test]
    fn url_import_http() {
        assert!(is_css_url_import("http://example.com/style.css"));
    }

    #[test]
    fn url_import_https() {
        assert!(is_css_url_import("https://fonts.googleapis.com/css"));
    }

    #[test]
    fn url_import_data() {
        assert!(is_css_url_import("data:text/css;base64,abc"));
    }

    #[test]
    fn url_import_local_not_skipped() {
        assert!(!is_css_url_import("./local.css"));
    }

    #[test]
    fn url_import_bare_specifier_not_skipped() {
        assert!(!is_css_url_import("tailwindcss"));
    }

    #[test]
    fn normalize_relative_dot_path_unchanged() {
        assert_eq!(
            normalize_css_import_path("./reset.css".to_string(), false),
            "./reset.css"
        );
    }

    #[test]
    fn normalize_parent_relative_path_unchanged() {
        assert_eq!(
            normalize_css_import_path("../shared.scss".to_string(), false),
            "../shared.scss"
        );
    }

    #[test]
    fn normalize_absolute_path_unchanged() {
        assert_eq!(
            normalize_css_import_path("/styles/main.css".to_string(), false),
            "/styles/main.css"
        );
    }

    #[test]
    fn normalize_url_unchanged() {
        assert_eq!(
            normalize_css_import_path("https://example.com/style.css".to_string(), false),
            "https://example.com/style.css"
        );
    }

    #[test]
    fn normalize_bare_css_gets_dot_slash() {
        assert_eq!(
            normalize_css_import_path("app.css".to_string(), false),
            "./app.css"
        );
    }

    #[test]
    fn normalize_css_package_subpath_stays_bare() {
        assert_eq!(
            normalize_css_import_path("tailwindcss/theme.css".to_string(), false),
            "tailwindcss/theme.css"
        );
    }

    #[test]
    fn normalize_css_package_subpath_with_dotted_name_stays_bare() {
        assert_eq!(
            normalize_css_import_path("highlight.js/styles/github.css".to_string(), false),
            "highlight.js/styles/github.css"
        );
    }

    #[test]
    fn normalize_bare_scss_gets_dot_slash() {
        assert_eq!(
            normalize_css_import_path("vars.scss".to_string(), false),
            "./vars.scss"
        );
    }

    #[test]
    fn normalize_bare_sass_gets_dot_slash() {
        assert_eq!(
            normalize_css_import_path("main.sass".to_string(), false),
            "./main.sass"
        );
    }

    #[test]
    fn normalize_bare_less_gets_dot_slash() {
        assert_eq!(
            normalize_css_import_path("theme.less".to_string(), false),
            "./theme.less"
        );
    }

    #[test]
    fn normalize_bare_js_extension_stays_bare() {
        assert_eq!(
            normalize_css_import_path("module.js".to_string(), false),
            "module.js"
        );
    }

    #[test]
    fn normalize_scss_bare_partial_gets_dot_slash() {
        assert_eq!(
            normalize_css_import_path("variables".to_string(), true),
            "./variables"
        );
    }

    #[test]
    fn normalize_scss_bare_partial_with_subdir_gets_dot_slash() {
        assert_eq!(
            normalize_css_import_path("base/reset".to_string(), true),
            "./base/reset"
        );
    }

    #[test]
    fn normalize_scss_builtin_stays_bare() {
        assert_eq!(
            normalize_css_import_path("sass:math".to_string(), true),
            "sass:math"
        );
    }

    #[test]
    fn normalize_scss_relative_path_unchanged() {
        assert_eq!(
            normalize_css_import_path("../styles/variables".to_string(), true),
            "../styles/variables"
        );
    }

    #[test]
    fn normalize_css_bare_extensionless_stays_bare() {
        assert_eq!(
            normalize_css_import_path("tailwindcss".to_string(), false),
            "tailwindcss"
        );
    }

    #[test]
    fn normalize_scoped_package_with_css_extension_stays_bare() {
        assert_eq!(
            normalize_css_import_path("@fontsource/monaspace-neon/400.css".to_string(), false),
            "@fontsource/monaspace-neon/400.css"
        );
    }

    #[test]
    fn normalize_scoped_package_with_scss_extension_stays_bare() {
        assert_eq!(
            normalize_css_import_path("@company/design-system/tokens.scss".to_string(), true),
            "@company/design-system/tokens.scss"
        );
    }

    #[test]
    fn normalize_scoped_package_without_extension_stays_bare() {
        assert_eq!(
            normalize_css_import_path("@fallow/design-system/styles".to_string(), false),
            "@fallow/design-system/styles"
        );
    }

    #[test]
    fn normalize_scoped_package_extensionless_scss_stays_bare() {
        assert_eq!(
            normalize_css_import_path("@company/tokens".to_string(), true),
            "@company/tokens"
        );
    }

    #[test]
    fn normalize_path_alias_with_css_extension_stays_bare() {
        assert_eq!(
            normalize_css_import_path("@/components/Button.css".to_string(), false),
            "@/components/Button.css"
        );
    }

    #[test]
    fn normalize_path_alias_extensionless_stays_bare() {
        assert_eq!(
            normalize_css_import_path("@/styles/variables".to_string(), false),
            "@/styles/variables"
        );
    }

    #[test]
    fn mask_css_no_comments() {
        let source = ".foo { color: red; }";
        assert_eq!(mask_css_comments(source, false), source);
    }

    #[test]
    fn mask_css_multiple_block_comments() {
        let source = "/* comment-one */ .foo { } /* comment-two */ .bar { }";
        let result = mask_css_comments(source, false);
        assert!(!result.contains("comment-one"));
        assert!(!result.contains("comment-two"));
        assert!(result.contains(".foo"));
        assert!(result.contains(".bar"));
    }

    #[test]
    fn mask_scss_does_not_affect_non_scss() {
        let source = "// this stays\n.foo { }";
        let result = mask_css_comments(source, false);
        assert!(result.contains("// this stays"));
    }

    #[test]
    fn css_module_parses_suppressions() {
        let info = parse_css_to_module(
            fallow_types::discover::FileId(0),
            Path::new("Component.module.css"),
            "/* fallow-ignore-file */\n.btn { color: red; }",
            0,
        );
        assert!(!info.suppressions.is_empty());
        assert_eq!(info.suppressions[0].line, 0);
    }

    #[test]
    fn extracts_class_starting_with_underscore() {
        let names = export_names("._private { } .__dunder { }");
        assert!(names.contains(&"_private".to_string()));
        assert!(names.contains(&"__dunder".to_string()));
    }

    #[test]
    fn ignores_id_selectors() {
        let names = export_names("#myId { color: red; }");
        assert!(!names.contains(&"myId".to_string()));
    }

    #[test]
    fn ignores_element_selectors() {
        let names = export_names("div { color: red; } span { }");
        assert!(names.is_empty());
    }

    #[test]
    fn extract_css_imports_at_import_quoted() {
        let imports = extract_css_imports(r#"@import "./reset.css";"#, false);
        assert_eq!(imports, vec!["./reset.css"]);
    }

    #[test]
    fn extract_css_imports_package_subpath_stays_bare() {
        let imports =
            extract_css_imports(r#"@import "tailwindcss/theme.css" layer(theme);"#, false);
        assert_eq!(imports, vec!["tailwindcss/theme.css"]);
    }

    #[test]
    fn extract_css_imports_at_import_url() {
        let imports = extract_css_imports(r#"@import url("./reset.css");"#, false);
        assert_eq!(imports, vec!["./reset.css"]);
    }

    #[test]
    fn extract_css_imports_skips_remote_urls() {
        let imports =
            extract_css_imports(r#"@import "https://fonts.example.com/font.css";"#, false);
        assert!(imports.is_empty());
    }

    #[test]
    fn extract_css_imports_scss_use_normalizes_partial() {
        let imports = extract_css_imports(r#"@use "variables";"#, true);
        assert_eq!(imports, vec!["./variables"]);
    }

    #[test]
    fn extract_css_imports_scss_forward_normalizes_partial() {
        let imports = extract_css_imports(r#"@forward "tokens";"#, true);
        assert_eq!(imports, vec!["./tokens"]);
    }

    #[test]
    fn extract_css_imports_skips_comments() {
        let imports = extract_css_imports(
            r#"/* @import "./hidden.scss"; */
@use "real";"#,
            true,
        );
        assert_eq!(imports, vec!["./real"]);
    }

    #[test]
    fn extract_css_imports_at_plugin_keeps_package_bare() {
        let imports = extract_css_imports(r#"@plugin "daisyui";"#, true);
        assert_eq!(imports, vec!["daisyui"]);
    }

    #[test]
    fn extract_css_imports_at_plugin_tracks_relative_file() {
        let imports = extract_css_imports(r#"@plugin "./tailwind-plugin.js";"#, false);
        assert_eq!(imports, vec!["./tailwind-plugin.js"]);
    }

    #[test]
    fn extract_css_imports_scss_at_import_kept_relative() {
        let imports = extract_css_imports(r"@import 'Foo';", true);
        assert_eq!(imports, vec!["./Foo"]);
    }

    #[test]
    fn extract_css_imports_additional_data_string_body() {
        let body = r#"@use "./src/styles/global.scss";"#;
        let imports = extract_css_imports(body, true);
        assert_eq!(imports, vec!["./src/styles/global.scss"]);
    }

    /// Resolve a span's start to (line, col) using the same primitives the
    /// downstream pipeline uses in `crates/core/src/analyze/unused_exports.rs`.
    fn span_line_col(source: &str, start: u32) -> (u32, u32) {
        let offsets = fallow_types::extract::compute_line_offsets(source);
        fallow_types::extract::byte_offset_to_line_col(&offsets, start)
    }

    #[test]
    fn span_points_at_real_class_declaration_line() {
        let source = "\n\n\n\n.foo { color: red; }\n";
        let exports = extract_css_module_exports(source, false);
        assert_eq!(exports.len(), 1);
        let span = exports[0].span;
        let (line, col) = span_line_col(source, span.start);
        assert_eq!(line, 5, "`.foo` on line 5 must produce line 5, not line 1");
        assert_eq!(
            col, 1,
            "column points at `f` in `.foo` (post-dot identifier)"
        );
        assert_eq!(
            &source[span.start as usize..span.end as usize],
            "foo",
            "span range must slice to the class identifier in the original source"
        );
    }

    #[test]
    fn span_survives_multibyte_comment_prefix() {
        let source = "/* \u{2713} */\n.foo { }";
        let exports = extract_css_module_exports(source, false);
        assert_eq!(exports.len(), 1);
        let span = exports[0].span;
        assert!(
            source.is_char_boundary(span.start as usize),
            "span.start must lie on a UTF-8 char boundary"
        );
        assert_eq!(&source[span.start as usize..span.end as usize], "foo");
    }

    #[test]
    fn span_skips_at_layer_prelude_dot_segments() {
        let source = "@layer foo.bar { }\n.root { }\n";
        let exports = extract_css_module_exports(source, false);
        let names: Vec<_> = exports
            .iter()
            .filter_map(|e| match &e.name {
                ExportName::Named(n) => Some(n.as_str()),
                ExportName::Default => None,
            })
            .collect();
        assert_eq!(names, vec!["root"], "@layer sub-segments must not export");
        let span = exports[0].span;
        let (line, _col) = span_line_col(source, span.start);
        assert_eq!(line, 2, "`.root` lives on line 2 of the original source");
        assert_eq!(&source[span.start as usize..span.end as usize], "root");
    }

    #[test]
    fn span_skips_classes_in_strings() {
        let source = ".real { content: \".fake\"; }\n.also-real { }\n";
        let exports = extract_css_module_exports(source, false);
        let names: Vec<_> = exports
            .iter()
            .filter_map(|e| match &e.name {
                ExportName::Named(n) => Some(n.as_str()),
                ExportName::Default => None,
            })
            .collect();
        assert_eq!(names, vec!["real", "also-real"]);
        for export in &exports {
            let span = export.span;
            let slice = &source[span.start as usize..span.end as usize];
            match &export.name {
                ExportName::Named(n) => assert_eq!(slice, n.as_str()),
                ExportName::Default => unreachable!("CSS modules emit only named exports"),
            }
        }
    }

    #[test]
    fn span_deduplicates_to_first_occurrence() {
        let source = ".btn { color: red; }\n.btn { color: blue; }\n";
        let exports = extract_css_module_exports(source, false);
        assert_eq!(exports.len(), 1);
        let (line, _col) = span_line_col(source, exports[0].span.start);
        assert_eq!(
            line, 1,
            "first occurrence wins for deduplicated class names"
        );
    }

    #[test]
    fn span_inside_media_query() {
        let source =
            "@media (max-width: 768px) {\n  .mobile { display: block; }\n  .desktop { }\n}\n";
        let exports = extract_css_module_exports(source, false);
        let by_name: rustc_hash::FxHashMap<&str, oxc_span::Span> = exports
            .iter()
            .filter_map(|e| match &e.name {
                ExportName::Named(n) => Some((n.as_str(), e.span)),
                ExportName::Default => None,
            })
            .collect();
        let mobile_line = span_line_col(source, by_name["mobile"].start).0;
        let desktop_line = span_line_col(source, by_name["desktop"].start).0;
        assert_eq!(mobile_line, 2);
        assert_eq!(desktop_line, 3);
    }

    #[test]
    fn at_layer_only_module_emits_no_exports() {
        let exports = extract_css_module_exports("@layer foo.bar, foo.baz;\n", false);
        assert!(exports.is_empty());
    }

    #[test]
    fn parse_css_to_module_resolves_real_line_offsets() {
        let source = "\n\n\n\n.foo { color: red; }\n";
        let info = parse_css_to_module(
            fallow_types::discover::FileId(0),
            Path::new("Component.module.css"),
            source,
            0,
        );
        assert_eq!(info.exports.len(), 1);
        let (line, _col) = fallow_types::extract::byte_offset_to_line_col(
            &info.line_offsets,
            info.exports[0].span.start,
        );
        assert_eq!(line, 5, "downstream line must equal the source line");
    }

    fn theme_token_names(source: &str) -> Vec<String> {
        scan_theme_blocks(source)
            .tokens
            .into_iter()
            .map(|t| t.name)
            .collect()
    }

    #[test]
    fn theme_single_block_collects_tokens() {
        let names = theme_token_names("@theme { --color-brand: #f00; --radius-card: 8px; }");
        assert_eq!(names, vec!["color-brand", "radius-card"]);
    }

    #[test]
    fn theme_token_values_are_normalized() {
        let scan = scan_theme_blocks("@theme {\n  --color-brand: rgb( 255 0 0 );\n}");
        assert_eq!(scan.tokens[0].name, "color-brand");
        assert_eq!(scan.tokens[0].value, "rgb( 255 0 0 )");
    }

    #[test]
    fn theme_dashed_multi_segment_names() {
        let names = theme_token_names(
            "@theme {\n  --font-weight-heavy: 900;\n  --inset-shadow-glow: 0 0 4px red;\n}",
        );
        assert_eq!(names, vec!["font-weight-heavy", "inset-shadow-glow"]);
    }

    #[test]
    fn theme_inline_and_static_modifiers() {
        assert_eq!(
            theme_token_names("@theme inline { --color-a: red; }"),
            vec!["color-a"]
        );
        assert_eq!(
            theme_token_names("@theme static { --color-b: red; }"),
            vec!["color-b"]
        );
    }

    #[test]
    fn theme_multiple_blocks_union() {
        let names = theme_token_names(
            "@theme { --color-a: red; }\n.x { color: blue; }\n@theme { --spacing-gutter: 1rem; }",
        );
        assert_eq!(names, vec!["color-a", "spacing-gutter"]);
    }

    #[test]
    fn theme_reset_form_excluded() {
        // `--color-*: initial` is a namespace reset directive, not a token.
        let names = theme_token_names("@theme { --color-*: initial; --color-brand: red; }");
        assert_eq!(names, vec!["color-brand"]);
    }

    #[test]
    fn theme_no_block_yields_nothing() {
        assert!(theme_token_names(".x { --color-brand: red; }").is_empty());
    }

    #[test]
    fn theme_line_numbers() {
        let scan = scan_theme_blocks("@theme {\n  --color-a: red;\n  --radius-b: 4px;\n}");
        assert_eq!(scan.tokens[0].line, 2);
        assert_eq!(scan.tokens[1].line, 3);
    }

    #[test]
    fn theme_token_backs_token_via_var() {
        let scan = scan_theme_blocks(
            "@theme {\n  --color-brand: #f00;\n  --color-button: var(--color-brand);\n}",
        );
        assert!(
            scan.theme_var_reads
                .iter()
                .any(|(name, _)| name == "color-brand")
        );
    }

    #[test]
    fn theme_var_read_carries_line() {
        // The `var(--color-brand)` read sits on line 3 of the source; the located
        // theme-var read must carry that 1-based line for the reverse index.
        let scan = scan_theme_blocks(
            "@theme {\n  --color-brand: #f00;\n  --color-button: var(--color-brand);\n}",
        );
        assert_eq!(
            scan.theme_var_reads,
            vec![("color-brand".to_string(), 3u32)]
        );
    }

    #[test]
    fn css_var_reads_locate_outside_theme_and_exclude_interior() {
        // A regular-CSS `var(--color-brand)` read is located (css-var surface);
        // a read inside the `@theme` interior is the distinct theme-var surface
        // and MUST be excluded here so the two kinds never double-count.
        let source = "@theme {\n  --color-brand: #f00;\n  --color-button: var(--color-brand);\n}\n\n.btn {\n  color: var(--color-brand);\n}\n";
        assert_eq!(
            extract_css_var_reads_located(source),
            vec![("color-brand".to_string(), 7u32)],
            "only the .btn read (line 7) is a css-var; the @theme-interior read is excluded"
        );

        // A source whose only `var()` read is inside `@theme` yields no css-var.
        assert!(
            extract_css_var_reads_located("@theme {\n  --a: #fff;\n  --b: var(--a);\n}",)
                .is_empty(),
            "a @theme-interior-only var() read is not a css-var consumer"
        );
    }

    #[test]
    fn css_var_reads_line_match_naive_reference_on_dense_line() {
        // Many `var()` reads packed onto a single long line (the pathological
        // zero-newline prefix) plus one trailing read on the next line: the
        // incremental line counter must agree byte-for-byte with the naive
        // per-match prefix rescan. No `@theme`, comments, strings, or `url()`,
        // so masking is identity and every read is a css-var read.
        use std::fmt::Write as _;
        let mut src = String::from(".x {");
        for i in 0..500 {
            let _ = write!(src, " color: var(--t{i});");
        }
        src.push_str(" }\n.y { color: var(--tail); }\n");

        let got = extract_css_var_reads_located(&src);

        // Reference: recompute each read's line via a full prefix rescan.
        let want: Vec<(String, u32)> = CSS_VAR_REF_RE
            .captures_iter(&src)
            .filter_map(|cap| cap.get(0).zip(cap.get(1)))
            .map(|(whole, name)| {
                (
                    name.as_str().to_owned(),
                    line_at_offset(&src, whole.start()),
                )
            })
            .collect();

        assert_eq!(got, want);
        assert!(got.len() > 500, "expected the dense line plus the trailer");
        // The trailer sits on line 2; the packed reads all sit on line 1.
        assert_eq!(got.last().map(|(_, l)| *l), Some(2));
        assert!(got[..got.len() - 1].iter().all(|(_, l)| *l == 1));
    }

    #[test]
    fn apply_token_lines_match_naive_reference_on_dense_line() {
        use std::fmt::Write as _;
        let mut src = String::from("/* a\n b */\n.x {");
        for i in 0..300 {
            let _ = write!(src, " @apply p-{i};");
        }
        src.push_str(" }\n\n.y { @apply tail-a tail-b; }\r\n.z { @apply last; }\n");

        let got = extract_apply_tokens_located(&src);

        let masked = mask_theme_source(&src);
        let want: Vec<(String, u32)> = CSS_APPLY_RE
            .find_iter(&masked)
            .flat_map(|m| {
                let line = line_at_offset(&src, m.start());
                m.as_str()
                    .trim_start_matches("@apply")
                    .split_whitespace()
                    .map(move |token| (token.to_owned(), line))
            })
            .collect();

        assert_eq!(got, want);
        assert_eq!(got.len(), 303);
        assert!(got[..300].iter().all(|(_, line)| *line == 3));
        assert_eq!(
            &got[300..],
            &[
                ("tail-a".to_owned(), 5),
                ("tail-b".to_owned(), 5),
                ("last".to_owned(), 6),
            ]
        );
    }

    #[test]
    fn theme_string_braces_do_not_truncate_block() {
        let scan = scan_theme_blocks(
            "@theme {\n  --font-label: \"}\";\n  --color-brand: #f00;\n  --color-button: var(--color-brand);\n}",
        );
        assert_eq!(
            scan.tokens
                .iter()
                .map(|token| token.name.as_str())
                .collect::<Vec<_>>(),
            vec!["font-label", "color-brand", "color-button"]
        );
        assert!(
            scan.theme_var_reads
                .iter()
                .any(|(name, _)| name == "color-brand")
        );
    }

    #[test]
    fn theme_nested_keyframes_body_not_collected() {
        // `@keyframes` inside `@theme` (for `--animate-*`) must not surface its
        // step selectors or interior as theme tokens.
        let names = theme_token_names(
            "@theme {\n  --animate-spin: spin 1s linear infinite;\n  @keyframes spin { from { --x: 0; } to { --y: 1; } }\n}",
        );
        assert_eq!(names, vec!["animate-spin"]);
    }

    #[test]
    fn theme_comment_block_ignored() {
        let names = theme_token_names("/* @theme { --color-fake: red; } */ .x { color: blue; }");
        assert!(names.is_empty(), "got {names:?}");
    }

    #[test]
    fn theme_deduplicates_repeated_token() {
        let names = theme_token_names("@theme { --color-a: red; --color-a: blue; }");
        assert_eq!(names, vec!["color-a"]);
    }

    #[test]
    fn apply_tokens_basic() {
        let tokens = extract_apply_tokens(".panel { @apply rounded-card font-bold; }");
        assert_eq!(tokens, vec!["rounded-card", "font-bold"]);
    }

    #[test]
    fn apply_tokens_strips_important() {
        let tokens = extract_apply_tokens(".x { @apply text-brand! font-bold !important; }");
        assert_eq!(tokens, vec!["text-brand", "font-bold"]);
    }

    fn line_lookup_bytes(run: impl FnOnce()) -> usize {
        LINE_LOOKUP_BYTES.with(|read| read.set(0));
        run();
        LINE_LOOKUP_BYTES.with(std::cell::Cell::get)
    }

    /// One dense line with `count` `@apply` directives and `var()` reads,
    /// followed by a few short lines.
    fn dense_token_stylesheet(count: usize) -> String {
        use std::fmt::Write as _;
        let mut src = String::from("@theme { --color-brand: red; }\n.x {");
        for i in 0..count {
            let _ = write!(src, " @apply p-{i}; color: var(--color-{i});");
        }
        src.push_str(" }\n.y { @apply tail; color: var(--color-tail); }\n");
        src
    }

    #[test]
    fn located_token_scans_read_each_byte_once_for_lines() {
        for count in [400, 800] {
            let src = dense_token_stylesheet(count);
            let apply = line_lookup_bytes(|| {
                assert_eq!(extract_apply_tokens_located(&src).len(), count + 1);
            });
            let var_reads = line_lookup_bytes(|| {
                assert_eq!(extract_css_var_reads_located(&src).len(), count + 1);
            });
            let combined = line_lookup_bytes(|| {
                let _ = scan_stylesheet_tokens(&src);
            });
            // A prefix rescan for each match reads about count * len / 2
            // bytes, which is far above len.
            assert!(apply <= src.len(), "@apply read {apply} of {}", src.len());
            assert!(
                var_reads <= src.len(),
                "var() read {var_reads} of {}",
                src.len()
            );
            assert!(
                combined <= 2 * src.len(),
                "scan read {combined} of {}",
                src.len()
            );
        }
    }

    fn comment_mask_passes(run: impl FnOnce()) -> u32 {
        COMMENT_MASK_PASSES.with(|passes| passes.set(0));
        run();
        COMMENT_MASK_PASSES.with(std::cell::Cell::get)
    }

    #[test]
    fn a_css_module_parse_masks_comments_once() {
        let source = "/* a */ @import './b.css';\n.card { @apply rounded; } // c\n";
        for name in ["styles.module.css", "styles.module.scss", "styles.css"] {
            let passes = comment_mask_passes(|| {
                parse_css_to_module(FileId(0), Path::new(name), source, 0);
            });
            assert_eq!(passes, 1, "comment mask passes for {name}");
        }
    }

    #[test]
    fn a_stylesheet_token_scan_masks_comments_once() {
        let source = "/* x */\n@theme {\n  --color-brand: red;\n  --color-alt: var(--color-brand);\n}\n.a { @apply bg-brand p-2; color: var(--color-alt); }\n";
        let mut scan = StylesheetTokens::default();
        let passes = comment_mask_passes(|| scan = scan_stylesheet_tokens(source));
        assert_eq!(passes, 1);
        assert_eq!(scan.theme, scan_theme_blocks(source));
        assert_eq!(
            scan.apply_tokens_located,
            extract_apply_tokens_located(source)
        );
        assert_eq!(
            scan.css_var_reads_located,
            extract_css_var_reads_located(source)
        );
    }

    #[test]
    fn a_stylesheet_token_scan_skips_the_mask_without_directives() {
        let source = "/* plain */ .a { color: red; }";
        let mut scan = StylesheetTokens::default();
        let passes = comment_mask_passes(|| scan = scan_stylesheet_tokens(source));
        assert_eq!(passes, 0);
        assert_eq!(scan, StylesheetTokens::default());
    }

    #[test]
    fn apply_tokens_ignored_in_comments() {
        let tokens = extract_apply_tokens("/* @apply hidden-token; */ .x { color: red; }");
        assert!(tokens.is_empty(), "got {tokens:?}");
    }
}