omena_query/style/

source_refs.rs

use std::collections::{BTreeMap, BTreeSet};
use std::path::{Component, PathBuf};

use super::dynamic_classname::{
    OMENA_QUERY_WORKSPACE_DYNAMIC_CLASSNAME_CONTEXT_DEPTH,
    harvest_omena_query_dynamic_classname_m_tier_diagnostics,
};
use super::*;

pub enum OmenaWorkspaceMonikerInput<'a> {
    CssModuleSelector {
        target_style_uri: Option<&'a str>,
        selector_name: &'a str,
    },
    CssCustomProperty {
        workspace_folder_uri: Option<&'a str>,
        name: &'a str,
    },
    SassSymbol {
        definition_uri: &'a str,
        family: &'a str,
        name: &'a str,
    },
    SassUnresolvedSymbol {
        workspace_folder_uri: Option<&'a str>,
        family: &'a str,
        namespace: Option<&'a str>,
        name: &'a str,
    },
}

pub fn omena_workspace_moniker(input: OmenaWorkspaceMonikerInput<'_>) -> String {
    match input {
        OmenaWorkspaceMonikerInput::CssModuleSelector {
            target_style_uri,
            selector_name,
        } => {
            let target = target_style_uri.unwrap_or("*");
            format!("css-module-selector:{target}#.{selector_name}")
        }
        OmenaWorkspaceMonikerInput::CssCustomProperty {
            workspace_folder_uri,
            name,
        } => {
            let scope = workspace_folder_uri.unwrap_or("global");
            format!("css-custom-property:{scope}#{name}")
        }
        OmenaWorkspaceMonikerInput::SassSymbol {
            definition_uri,
            family,
            name,
        } => format!("sass-symbol:{definition_uri}#{family}:{name}"),
        OmenaWorkspaceMonikerInput::SassUnresolvedSymbol {
            workspace_folder_uri,
            family,
            namespace,
            name,
        } => {
            let scope = workspace_folder_uri.unwrap_or("global");
            let namespace = namespace.unwrap_or("*");
            format!("sass-symbol-unresolved:{scope}#{family}:{namespace}:{name}")
        }
    }
}

pub fn summarize_omena_query_refs_for_class(
    selector_name: &str,
    target_style_uri: Option<&str>,
    include_declaration: bool,
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
    references: &[OmenaQuerySourceSelectorReferenceCandidateV0],
) -> OmenaQueryRefsForClassV0 {
    let mut locations = Vec::new();

    if include_declaration {
        locations.extend(
            definitions
                .iter()
                .filter(|definition| class_names_match(definition.name.as_str(), selector_name))
                .filter(|definition| {
                    target_style_uri.is_none_or(|target_uri| {
                        file_uri_equivalent(target_uri, definition.uri.as_str())
                    })
                })
                .map(|definition| OmenaQueryReferenceLocationV0 {
                    uri: definition.uri.clone(),
                    range: definition.range,
                    name: definition.name.clone(),
                    role: "definition",
                    source: "omenaQueryStyleSelectorDefinitions",
                }),
        );
    }

    for reference in references {
        let reference_candidate = OmenaQuerySourceSelectorCandidateV0 {
            kind: reference.kind,
            name: reference.name.clone(),
            range: reference.range,
            source: reference.source,
            target_style_uri: reference.target_style_uri.clone(),
        };
        if !source_selector_candidate_matches_target_uri(&reference_candidate, target_style_uri) {
            continue;
        }
        let selector_names = resolve_omena_query_source_candidate_selector_names(
            &reference_candidate,
            definitions,
            target_style_uri,
        );
        if selector_names
            .iter()
            .any(|name| class_names_match(name.as_str(), selector_name))
        {
            locations.push(OmenaQueryReferenceLocationV0 {
                uri: reference.uri.clone(),
                range: reference.range,
                name: selector_name.to_string(),
                role: "reference",
                source: "omenaQuerySourceSelectorReferences",
            });
        }
    }

    locations.sort_by_key(|location| {
        (
            reference_location_role_rank(location.role),
            location.uri.clone(),
            location.range.start.line,
            location.range.start.character,
        )
    });
    locations.dedup_by(|left, right| left.uri == right.uri && left.range == right.range);

    OmenaQueryRefsForClassV0 {
        schema_version: "0",
        product: "omena-query.refs-for-class",
        selector_name: selector_name.to_string(),
        target_style_uri: target_style_uri.map(ToString::to_string),
        include_declaration,
        location_count: locations.len(),
        locations,
        ready_surfaces: vec!["refsForClass", "workspaceWideSelectorReferences"],
    }
}

pub fn summarize_omena_query_rename_plan(
    selector_name: &str,
    new_name: &str,
    target_style_uri: Option<&str>,
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
    references: &[OmenaQuerySourceSelectorReferenceEditTargetV0],
) -> OmenaQueryRenamePlanV0 {
    let edits = resolve_omena_query_selector_rename_edits(
        selector_name,
        new_name,
        target_style_uri,
        definitions,
        references,
    );
    OmenaQueryRenamePlanV0 {
        schema_version: "0",
        product: "omena-query.rename-plan",
        selector_name: selector_name.to_string(),
        new_name: new_name.to_string(),
        target_style_uri: target_style_uri.map(ToString::to_string),
        edit_count: edits.len(),
        edits,
        ready_surfaces: vec!["renamePlan", "workspaceWideSelectorRename"],
    }
}

pub fn summarize_omena_query_source_selector_occurrence_index(
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
    references: &[OmenaQuerySourceSelectorReferenceCandidateV0],
) -> OmenaQuerySourceSelectorOccurrenceIndexV0 {
    let mut occurrences = Vec::new();
    for reference in references {
        let reference_candidate = OmenaQuerySourceSelectorCandidateV0 {
            kind: reference.kind,
            name: reference.name.clone(),
            range: reference.range,
            source: reference.source,
            target_style_uri: reference.target_style_uri.clone(),
        };
        for selector_name in resolve_omena_query_source_candidate_selector_names(
            &reference_candidate,
            definitions,
            reference.target_style_uri.as_deref(),
        ) {
            let moniker = source_selector_occurrence_moniker(
                selector_name.as_str(),
                reference.target_style_uri.as_deref(),
            );
            occurrences.push(OmenaQuerySourceSelectorOccurrenceV0 {
                moniker,
                uri: reference.uri.clone(),
                selector_name: selector_name.clone(),
                range: reference.range,
                kind: workspace_occurrence_kind_from_source_reference_kind(reference.kind)
                    .unwrap_or(OmenaWorkspaceOccurrenceKindV0::SourceSelectorReference),
                role: OmenaWorkspaceOccurrenceRoleV0::Reference,
                source: source_reference_occurrence_surface(reference.projection_surface()),
                target_style_uri: reference.target_style_uri.clone(),
                rename_target: reference.kind == "sourceSelectorReference"
                    && reference.name == selector_name,
            });
        }
    }

    occurrences.sort();
    occurrences.dedup();
    let moniker_count = occurrences
        .iter()
        .map(|occurrence| occurrence.moniker.as_str())
        .collect::<BTreeSet<_>>()
        .len();
    let workspace_index = summarize_omena_query_workspace_occurrence_index_from_source_occurrences(
        occurrences.as_slice(),
        vec![
            "workspaceOccurrenceIndex",
            "sourceSelectorOccurrenceIndex",
            "workspaceWideSelectorReferences",
            "workspaceWideSelectorRename",
        ],
    );
    OmenaQuerySourceSelectorOccurrenceIndexV0 {
        schema_version: "0",
        product: "omena-query.source-selector-occurrence-index",
        moniker_count,
        occurrence_count: occurrences.len(),
        workspace_index,
        occurrences,
        ready_surfaces: vec![
            "sourceSelectorOccurrenceIndex",
            "workspaceWideSelectorReferences",
            "workspaceWideSelectorRename",
        ],
    }
}

fn source_reference_occurrence_surface(
    surface: OmenaQuerySourceSelectorReferenceSurfaceV0,
) -> OmenaWorkspaceOccurrenceSurfaceV0 {
    match surface {
        OmenaQuerySourceSelectorReferenceSurfaceV0::OmenaQuerySourceSyntaxIndex
        | OmenaQuerySourceSelectorReferenceSurfaceV0::OmenaTsgoTypeFactProjection => {
            // The published workspace V0 groups source-side projections together.
            OmenaWorkspaceOccurrenceSurfaceV0::OmenaQuerySourceSyntaxIndex
        }
    }
}

pub fn summarize_omena_query_refs_for_class_from_occurrence_index(
    selector_name: &str,
    target_style_uri: Option<&str>,
    include_declaration: bool,
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
    occurrence_index: &OmenaQuerySourceSelectorOccurrenceIndexV0,
) -> OmenaQueryRefsForClassV0 {
    let mut locations = Vec::new();

    if include_declaration {
        locations.extend(
            definitions
                .iter()
                .filter(|definition| class_names_match(definition.name.as_str(), selector_name))
                .filter(|definition| {
                    target_style_uri.is_none_or(|target_uri| {
                        file_uri_equivalent(target_uri, definition.uri.as_str())
                    })
                })
                .map(|definition| OmenaQueryReferenceLocationV0 {
                    uri: definition.uri.clone(),
                    range: definition.range,
                    name: definition.name.clone(),
                    role: "definition",
                    source: "omenaQueryStyleSelectorDefinitions",
                }),
        );
    }

    locations.extend(
        source_selector_occurrences_for_query(occurrence_index, selector_name, target_style_uri)
            .into_iter()
            .map(|occurrence| OmenaQueryReferenceLocationV0 {
                uri: occurrence.uri.clone(),
                range: occurrence.range,
                name: occurrence.selector_name.clone(),
                role: occurrence.role.as_str(),
                source: "omenaQuerySourceSelectorOccurrenceIndex",
            }),
    );

    locations.sort_by_key(|location| {
        (
            reference_location_role_rank(location.role),
            location.uri.clone(),
            location.range.start.line,
            location.range.start.character,
        )
    });
    locations.dedup_by(|left, right| left.uri == right.uri && left.range == right.range);

    OmenaQueryRefsForClassV0 {
        schema_version: "0",
        product: "omena-query.refs-for-class",
        selector_name: selector_name.to_string(),
        target_style_uri: target_style_uri.map(ToString::to_string),
        include_declaration,
        location_count: locations.len(),
        locations,
        ready_surfaces: vec![
            "refsForClass",
            "workspaceWideSelectorReferences",
            "sourceSelectorOccurrenceIndex",
        ],
    }
}

pub fn summarize_omena_query_rename_plan_from_occurrence_index(
    selector_name: &str,
    new_name: &str,
    target_style_uri: Option<&str>,
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
    occurrence_index: &OmenaQuerySourceSelectorOccurrenceIndexV0,
) -> OmenaQueryRenamePlanV0 {
    let references =
        source_selector_occurrences_for_query(occurrence_index, selector_name, target_style_uri)
            .into_iter()
            .filter(|occurrence| occurrence.rename_target)
            .map(|occurrence| OmenaQuerySourceSelectorReferenceEditTargetV0 {
                uri: occurrence.uri.clone(),
                name: occurrence.selector_name.clone(),
                range: occurrence.range,
                target_style_uri: occurrence.target_style_uri.clone(),
            })
            .collect::<Vec<_>>();
    let mut plan = summarize_omena_query_rename_plan(
        selector_name,
        new_name,
        target_style_uri,
        definitions,
        references.as_slice(),
    );
    plan.ready_surfaces.push("sourceSelectorOccurrenceIndex");
    plan
}

pub fn occurrences_for_monikers<'a>(
    index: &'a OmenaWorkspaceOccurrenceIndexV0,
    monikers: &BTreeSet<String>,
) -> Vec<&'a OmenaWorkspaceOccurrenceV0> {
    monikers
        .iter()
        .filter_map(|moniker| index.by_moniker.get(moniker.as_str()))
        .flat_map(|occurrences| occurrences.iter())
        .collect()
}

pub fn summarize_omena_query_refs_for_workspace_class(
    selector_name: &str,
    target_style_uri: Option<&str>,
    include_declaration: bool,
    style_sources: &[OmenaQueryStyleSourceInputV0],
    source_documents: &[OmenaQuerySourceDocumentInputV0],
    package_manifests: &[OmenaQueryStylePackageManifestV0],
) -> OmenaQueryRefsForClassV0 {
    summarize_omena_query_refs_for_workspace_class_with_resolution_inputs(
        selector_name,
        target_style_uri,
        include_declaration,
        style_sources,
        source_documents,
        package_manifests,
        &OmenaQueryStyleResolutionInputsV0::default(),
    )
}

#[allow(clippy::too_many_arguments)]
pub fn summarize_omena_query_refs_for_workspace_class_with_resolution_inputs(
    selector_name: &str,
    target_style_uri: Option<&str>,
    include_declaration: bool,
    style_sources: &[OmenaQueryStyleSourceInputV0],
    source_documents: &[OmenaQuerySourceDocumentInputV0],
    package_manifests: &[OmenaQueryStylePackageManifestV0],
    resolution_inputs: &OmenaQueryStyleResolutionInputsV0,
) -> OmenaQueryRefsForClassV0 {
    let definitions = summarize_omena_query_style_selector_definitions(style_sources);
    let references = collect_omena_query_source_selector_reference_candidates(
        style_sources,
        source_documents,
        package_manifests,
        resolution_inputs,
    );
    summarize_omena_query_refs_for_class(
        selector_name,
        target_style_uri,
        include_declaration,
        definitions.as_slice(),
        references.as_slice(),
    )
}

pub fn summarize_omena_query_rename_plan_for_workspace_class(
    selector_name: &str,
    new_name: &str,
    target_style_uri: Option<&str>,
    style_sources: &[OmenaQueryStyleSourceInputV0],
    source_documents: &[OmenaQuerySourceDocumentInputV0],
    package_manifests: &[OmenaQueryStylePackageManifestV0],
) -> OmenaQueryRenamePlanV0 {
    summarize_omena_query_rename_plan_for_workspace_class_with_resolution_inputs(
        selector_name,
        new_name,
        target_style_uri,
        style_sources,
        source_documents,
        package_manifests,
        &OmenaQueryStyleResolutionInputsV0::default(),
    )
}

#[allow(clippy::too_many_arguments)]
pub fn summarize_omena_query_rename_plan_for_workspace_class_with_resolution_inputs(
    selector_name: &str,
    new_name: &str,
    target_style_uri: Option<&str>,
    style_sources: &[OmenaQueryStyleSourceInputV0],
    source_documents: &[OmenaQuerySourceDocumentInputV0],
    package_manifests: &[OmenaQueryStylePackageManifestV0],
    resolution_inputs: &OmenaQueryStyleResolutionInputsV0,
) -> OmenaQueryRenamePlanV0 {
    let definitions = summarize_omena_query_style_selector_definitions(style_sources);
    let references = collect_omena_query_source_selector_reference_edit_targets(
        style_sources,
        source_documents,
        package_manifests,
        resolution_inputs,
    );
    summarize_omena_query_rename_plan(
        selector_name,
        new_name,
        target_style_uri,
        definitions.as_slice(),
        references.as_slice(),
    )
}

pub fn summarize_omena_query_missing_selector_diagnostic(
    target_style_uri: &str,
    target_style_source: &str,
    selector_name: &str,
    source_reference_range: ParserRangeV0,
) -> OmenaQuerySourceDiagnosticV0 {
    let insertion_range = end_of_source_range(target_style_source);
    let has_existing_style_content = !target_style_source.trim().is_empty();
    OmenaQuerySourceDiagnosticV0 {
        code: "missingSelector",
        severity: "warning",
        provenance: omena_query_evidence_graph_provenance![
            "omena-query.source-syntax-index",
            "omena-query.style-selector-definitions",
        ],
        range: source_reference_range,
        message: format!(
            "CSS Module selector '.{selector_name}' not found in indexed style tokens."
        ),
        precision: Some(source_diagnostic_precision(
            "classValueResolution",
            "sourceSyntaxIndex",
            "perSourceReference",
        )),
        suggestion: None,
        create_selector: Some(OmenaQueryCreateSelectorActionV0 {
            uri: target_style_uri.to_string(),
            range: insertion_range,
            new_text: if has_existing_style_content {
                format!("\n\n.{selector_name} {{\n}}\n")
            } else {
                format!(".{selector_name} {{\n}}\n")
            },
            selector_name: selector_name.to_string(),
        }),
    }
}

/// Two-tier reference universe, tier two: the name failed the bound CSS
/// Module's export set but resolves in the GLOBAL class universe (class
/// selectors of indexed non-module stylesheets). At runtime a bind-style
/// lookup falls through to the literal class name, which the global
/// stylesheet styles — so this is not a missing selector; it is a scoping
/// fact worth disclosing (the emitted class is literal and unscoped).
pub fn summarize_omena_query_global_class_fallthrough_diagnostic(
    selector_name: &str,
    global_definition_uri: &str,
    target_style_uri: &str,
    target_style_source: &str,
    source_reference_range: ParserRangeV0,
) -> OmenaQuerySourceDiagnosticV0 {
    let global_file_label = global_definition_uri
        .rsplit('/')
        .next()
        .filter(|label| !label.is_empty())
        .unwrap_or(global_definition_uri);
    // LSP-provided URIs percent-encode non-ASCII filenames; decode after the
    // split so the user-facing message shows the readable name.
    let global_file =
        percent_decode_uri_path(global_file_label).unwrap_or_else(|| global_file_label.to_string());
    // The one action that changes the scoping fact: adding the selector to
    // the bound module makes the reference module-scoped again. Reuses the
    // missing-selector machinery so the edit shape cannot drift.
    let create_selector = summarize_omena_query_missing_selector_diagnostic(
        target_style_uri,
        target_style_source,
        selector_name,
        source_reference_range,
    )
    .create_selector;
    OmenaQuerySourceDiagnosticV0 {
        code: "globalClassFallthrough",
        severity: "hint",
        provenance: omena_query_evidence_graph_provenance![
            "omena-query.source-syntax-index",
            "omena-query.style-selector-definitions",
        ],
        range: source_reference_range,
        message: format!(
            "'.{selector_name}' is not exported by the bound CSS Module; it resolves to the global stylesheet '{global_file}' and is emitted as a literal, unscoped class name."
        ),
        precision: Some(source_diagnostic_precision(
            "classValueResolution",
            "globalClassUniverse",
            "perSourceReference",
        )),
        suggestion: None,
        create_selector,
    }
}

pub fn summarize_omena_query_source_diagnostics_for_file(
    source_uri: &str,
    candidates: &[OmenaQuerySourceMissingSelectorDiagnosticCandidateV0],
) -> OmenaQuerySourceDiagnosticsForFileV0 {
    let mut diagnostics = candidates
        .iter()
        .map(|candidate| {
            summarize_omena_query_missing_selector_diagnostic(
                candidate.target_style_uri.as_str(),
                candidate.target_style_source.as_str(),
                candidate.selector_name.as_str(),
                candidate.source_reference_range,
            )
        })
        .collect::<Vec<_>>();
    apply_omena_query_checker_product_gate_to_source_diagnostics(&mut diagnostics);
    OmenaQuerySourceDiagnosticsForFileV0 {
        schema_version: "0",
        product: "omena-query.diagnostics-for-file",
        file_uri: source_uri.to_string(),
        file_kind: "source",
        diagnostic_count: diagnostics.len(),
        diagnostics,
        ready_surfaces: vec![
            "sourceMissingSelectorDiagnostics",
            "crossLanguageDiagnostics",
            "checkerProductDiagnosticGate",
        ],
    }
}

pub fn summarize_omena_query_source_diagnostics_for_workspace_file(
    source_path: &str,
    source_source: &str,
    style_sources: &[OmenaQueryStyleSourceInputV0],
    package_manifests: &[OmenaQueryStylePackageManifestV0],
) -> OmenaQuerySourceDiagnosticsForFileV0 {
    summarize_omena_query_source_diagnostics_for_workspace_file_with_resolution_inputs(
        source_path,
        source_source,
        style_sources,
        package_manifests,
        &OmenaQueryStyleResolutionInputsV0::default(),
    )
}

pub fn summarize_omena_query_source_diagnostics_for_workspace_file_with_resolution_inputs(
    source_path: &str,
    source_source: &str,
    style_sources: &[OmenaQueryStyleSourceInputV0],
    package_manifests: &[OmenaQueryStylePackageManifestV0],
    resolution_inputs: &OmenaQueryStyleResolutionInputsV0,
) -> OmenaQuerySourceDiagnosticsForFileV0 {
    summarize_omena_query_source_diagnostics_for_workspace_file_with_resolution_inputs_and_context_depth(
        source_path,
        source_source,
        style_sources,
        package_manifests,
        resolution_inputs,
        OMENA_QUERY_WORKSPACE_DYNAMIC_CLASSNAME_CONTEXT_DEPTH,
    )
}

fn summarize_omena_query_source_diagnostics_for_workspace_file_with_resolution_inputs_and_context_depth(
    source_path: &str,
    source_source: &str,
    style_sources: &[OmenaQueryStyleSourceInputV0],
    package_manifests: &[OmenaQueryStylePackageManifestV0],
    resolution_inputs: &OmenaQueryStyleResolutionInputsV0,
    max_context_depth: usize,
) -> OmenaQuerySourceDiagnosticsForFileV0 {
    let available_style_paths = style_sources
        .iter()
        .map(|source| source.style_path.as_str())
        .collect::<BTreeSet<_>>();
    let definitions = summarize_omena_query_style_selector_definitions(style_sources);
    let imports = summarize_omena_query_source_import_declarations_for_source_language(
        source_path,
        source_source,
        None,
    );
    let mut imported_style_bindings = Vec::new();
    let mut classnames_bind_bindings = Vec::new();
    let mut diagnostics = Vec::new();

    for import in imports.imports {
        if import.specifier == "classnames/bind" {
            classnames_bind_bindings.push(import.binding);
            continue;
        }

        if !is_query_source_style_module_specifier(import.specifier.as_str()) {
            continue;
        }

        match resolve_style_module_source_with_path_mappings(
            source_path,
            import.specifier.as_str(),
            &available_style_paths,
            package_manifests,
            resolution_inputs.bundler_path_mappings.as_slice(),
            resolution_inputs.tsconfig_path_mappings.as_slice(),
            resolution_inputs.disk_style_path_identities.as_slice(),
        ) {
            Some(style_path) => {
                imported_style_bindings.push(OmenaQuerySourceImportedStyleBindingV0 {
                    binding: import.binding,
                    style_uri: style_path,
                })
            }
            None => diagnostics.push(OmenaQuerySourceDiagnosticV0 {
                code: "missing-module",
                severity: "warning",
                provenance: omena_query_evidence_graph_provenance![
                    "omena-query.source-import-declarations",
                    "omena-resolver.style-module-resolution",
                ],
                range: parser_range_for_byte_span(source_source, import.specifier_byte_span),
                message: if resolution_inputs.disk_style_path_identities.is_empty() {
                    format!(
                        "Cannot resolve CSS Module '{}' from the provided workspace inputs.",
                        import.specifier
                    )
                } else {
                    format!(
                        "Cannot resolve CSS Module '{}'. The file does not exist.",
                        import.specifier
                    )
                },
                precision: Some(source_diagnostic_precision(
                    "styleModuleResolution",
                    "sourceImportResolution",
                    "perImportSpecifier",
                )),
                suggestion: None,
                create_selector: None,
            }),
        }
    }

    let index = summarize_omena_query_source_syntax_index_for_source_language(
        source_path,
        source_source,
        None,
        imported_style_bindings,
        classnames_bind_bindings,
    );
    summarize_omena_query_source_diagnostics_from_syntax_index(
        source_path,
        source_source,
        &index,
        OmenaQuerySourceDiagnosticsWorkspaceFacts {
            definitions: definitions.as_slice(),
            style_sources,
        },
        diagnostics,
        OmenaQuerySourceDiagnosticsAssemblyOptions {
            max_context_depth,
            ready_surfaces: vec![
                "sourceMissingModuleDiagnostics",
                "sourceMissingSelectorDiagnostics",
                "sourceResolvedClassDiagnostics",
                "crossLanguageDiagnostics",
                "checkerProductDiagnosticGate",
            ],
            include_dynamic_classname_m_tier: true,
        },
    )
}

/// Workspace source diagnostics with an explicit call-string bound `k` for the
/// harvested dynamic-className M-tier flow. The default LSP entry pins
/// `k = OMENA_QUERY_WORKSPACE_DYNAMIC_CLASSNAME_CONTEXT_DEPTH`; this variant
/// exposes `k` so the context-sensitivity of the harvested k-CFA flow is
/// observable (a context-insensitive `k = 0` run joins call sites that share a
/// callee binding and emits a different M-tier diagnostic set).
pub fn summarize_omena_query_source_diagnostics_for_workspace_file_with_context_depth(
    source_path: &str,
    source_source: &str,
    style_sources: &[OmenaQueryStyleSourceInputV0],
    package_manifests: &[OmenaQueryStylePackageManifestV0],
    max_context_depth: usize,
) -> OmenaQuerySourceDiagnosticsForFileV0 {
    summarize_omena_query_source_diagnostics_for_workspace_file_with_resolution_inputs_and_context_depth(
        source_path,
        source_source,
        style_sources,
        package_manifests,
        &OmenaQueryStyleResolutionInputsV0::default(),
        max_context_depth,
    )
}

pub fn summarize_omena_query_source_diagnostics_for_workspace_file_with_source_syntax_index(
    source_path: &str,
    source_source: &str,
    source_syntax_index: &OmenaQuerySourceSyntaxIndexV0,
    style_sources: &[OmenaQueryStyleSourceInputV0],
) -> OmenaQuerySourceDiagnosticsForFileV0 {
    summarize_omena_query_source_diagnostics_for_workspace_file_with_source_syntax_index_and_context_depth(
        source_path,
        source_source,
        source_syntax_index,
        style_sources,
        OMENA_QUERY_WORKSPACE_DYNAMIC_CLASSNAME_CONTEXT_DEPTH,
    )
}

pub fn summarize_omena_query_source_diagnostics_for_workspace_file_with_source_syntax_index_and_definitions(
    source_path: &str,
    source_source: &str,
    source_syntax_index: &OmenaQuerySourceSyntaxIndexV0,
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
    style_sources: &[OmenaQueryStyleSourceInputV0],
) -> OmenaQuerySourceDiagnosticsForFileV0 {
    summarize_omena_query_source_diagnostics_from_syntax_index(
        source_path,
        source_source,
        source_syntax_index,
        OmenaQuerySourceDiagnosticsWorkspaceFacts {
            definitions,
            style_sources,
        },
        Vec::new(),
        OmenaQuerySourceDiagnosticsAssemblyOptions {
            max_context_depth: OMENA_QUERY_WORKSPACE_DYNAMIC_CLASSNAME_CONTEXT_DEPTH,
            ready_surfaces: vec![
                "sourceIndexedSyntaxDiagnostics",
                "sourceMissingSelectorDiagnostics",
                "sourceResolvedClassDiagnostics",
                "crossLanguageDiagnostics",
                "checkerProductDiagnosticGate",
            ],
            include_dynamic_classname_m_tier: true,
        },
    )
}

pub fn summarize_omena_query_source_baseline_diagnostics_for_workspace_file_with_source_syntax_index_and_definitions(
    source_path: &str,
    source_source: &str,
    source_syntax_index: &OmenaQuerySourceSyntaxIndexV0,
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
    style_sources: &[OmenaQueryStyleSourceInputV0],
) -> OmenaQuerySourceDiagnosticsForFileV0 {
    summarize_omena_query_source_diagnostics_from_syntax_index(
        source_path,
        source_source,
        source_syntax_index,
        OmenaQuerySourceDiagnosticsWorkspaceFacts {
            definitions,
            style_sources,
        },
        Vec::new(),
        OmenaQuerySourceDiagnosticsAssemblyOptions {
            max_context_depth: OMENA_QUERY_WORKSPACE_DYNAMIC_CLASSNAME_CONTEXT_DEPTH,
            ready_surfaces: vec![
                "sourceIndexedSyntaxDiagnostics",
                "sourceMissingSelectorDiagnostics",
                "sourceBaselineDiagnostics",
                "crossLanguageDiagnostics",
                "checkerProductDiagnosticGate",
            ],
            include_dynamic_classname_m_tier: false,
        },
    )
}

pub fn summarize_omena_query_source_diagnostics_for_workspace_file_with_source_syntax_index_and_context_depth(
    source_path: &str,
    source_source: &str,
    source_syntax_index: &OmenaQuerySourceSyntaxIndexV0,
    style_sources: &[OmenaQueryStyleSourceInputV0],
    max_context_depth: usize,
) -> OmenaQuerySourceDiagnosticsForFileV0 {
    let definitions = summarize_omena_query_style_selector_definitions(style_sources);
    summarize_omena_query_source_diagnostics_from_syntax_index(
        source_path,
        source_source,
        source_syntax_index,
        OmenaQuerySourceDiagnosticsWorkspaceFacts {
            definitions: definitions.as_slice(),
            style_sources,
        },
        Vec::new(),
        OmenaQuerySourceDiagnosticsAssemblyOptions {
            max_context_depth,
            ready_surfaces: vec![
                "sourceIndexedSyntaxDiagnostics",
                "sourceMissingSelectorDiagnostics",
                "sourceResolvedClassDiagnostics",
                "crossLanguageDiagnostics",
                "checkerProductDiagnosticGate",
            ],
            include_dynamic_classname_m_tier: true,
        },
    )
}

struct OmenaQuerySourceDiagnosticsWorkspaceFacts<'a> {
    definitions: &'a [OmenaQueryStyleSelectorDefinitionV0],
    style_sources: &'a [OmenaQueryStyleSourceInputV0],
}

struct OmenaQuerySourceDiagnosticsAssemblyOptions {
    max_context_depth: usize,
    ready_surfaces: Vec<&'static str>,
    include_dynamic_classname_m_tier: bool,
}

fn summarize_omena_query_source_diagnostics_from_syntax_index(
    source_path: &str,
    source_source: &str,
    index: &OmenaQuerySourceSyntaxIndexV0,
    workspace_facts: OmenaQuerySourceDiagnosticsWorkspaceFacts<'_>,
    mut diagnostics: Vec<OmenaQuerySourceDiagnosticV0>,
    options: OmenaQuerySourceDiagnosticsAssemblyOptions,
) -> OmenaQuerySourceDiagnosticsForFileV0 {
    let style_sources_by_path = workspace_facts
        .style_sources
        .iter()
        .map(|source| (source.style_path.as_str(), source.style_source.as_str()))
        .collect::<BTreeMap<_, _>>();
    diagnostics.extend(summarize_omena_query_domain_class_reference_diagnostics(
        source_source,
        index,
    ));
    diagnostics.extend(
        summarize_omena_query_type_fact_provider_unavailable_diagnostics(source_source, index),
    );

    if !index.imported_style_bindings.is_empty() {
        if options.include_dynamic_classname_m_tier {
            // Harvest dynamic-className call sites (template-interpolation projections)
            // from the same syntax index and route them through the real k-limited
            // (k-CFA) M-tier flow gate, so the LSP-consumed default path emits the
            // context-sensitive no-unknown-dynamic-class / no-impossible-selector
            // diagnostics without an external producer.
            //
            // `no-unknown-dynamic-class` is module-scoped: a target carrying a
            // resolved `target_style_uri` is matched against ONLY that module's
            // selectors (`selector_universe_by_uri`), so a `btn-` prefix is not
            // cross-attributed to a `btn-*` selector in a different imported module.
            // Targets with no resolved binding fall back to the union universe.
            let selector_universe = workspace_facts
                .definitions
                .iter()
                .map(|definition| definition.name.clone())
                .collect::<BTreeSet<_>>()
                .into_iter()
                .collect::<Vec<_>>();
            let mut selector_universe_by_uri: BTreeMap<String, Vec<String>> = BTreeMap::new();
            for definition in workspace_facts.definitions {
                selector_universe_by_uri
                    .entry(definition.uri.clone())
                    .or_default()
                    .push(definition.name.clone());
            }
            for names in selector_universe_by_uri.values_mut() {
                names.sort();
                names.dedup();
            }
            diagnostics.extend(harvest_omena_query_dynamic_classname_m_tier_diagnostics(
                source_path,
                source_source,
                index,
                selector_universe.as_slice(),
                &selector_universe_by_uri,
                options.max_context_depth,
            ));
        }

        for reference in &index.selector_references {
            let Some(target_style_uri) = reference.target_style_uri.as_deref() else {
                continue;
            };
            let target_style_is_known =
                workspace_facts.definitions.iter().any(|definition| {
                    file_uri_equivalent(definition.uri.as_str(), target_style_uri)
                }) || style_sources_by_path
                    .keys()
                    .any(|style_uri| file_uri_equivalent(style_uri, target_style_uri));
            if !target_style_is_known {
                continue;
            }
            let Some(selector_name) = reference.selector_name.clone().or_else(|| {
                source_reference_text_selector_name(source_source, reference.byte_span)
            }) else {
                continue;
            };
            let candidate = OmenaQuerySourceSelectorCandidateV0 {
                kind: match reference.match_kind {
                    OmenaQuerySourceSelectorReferenceMatchKindV0::Exact => {
                        "sourceSelectorReference"
                    }
                    OmenaQuerySourceSelectorReferenceMatchKindV0::Prefix => {
                        "sourceSelectorPrefixReference"
                    }
                },
                name: selector_name.clone(),
                range: parser_range_for_byte_span(source_source, reference.byte_span),
                source: "omenaQuerySourceSyntaxIndex",
                target_style_uri: Some(target_style_uri.to_string()),
            };
            if !resolve_omena_query_style_selector_definitions_for_source_candidate(
                &candidate,
                workspace_facts.definitions,
            )
            .is_empty()
            {
                continue;
            }
            let target_style_source = style_sources_by_path
                .get(target_style_uri)
                .copied()
                .or_else(|| {
                    style_sources_by_path
                        .iter()
                        .find(|(style_uri, _)| file_uri_equivalent(style_uri, target_style_uri))
                        .map(|(_, source)| *source)
                });
            let value_domain_size = index
                .selector_references
                .iter()
                .filter(|candidate| {
                    candidate.byte_span == reference.byte_span
                        && candidate.target_style_uri == reference.target_style_uri
                        && candidate.match_kind == reference.match_kind
                })
                .filter_map(|candidate| candidate.selector_name.as_deref())
                .collect::<BTreeSet<_>>()
                .len();
            diagnostics.push(
                summarize_omena_query_unresolved_source_reference_diagnostic(
                    source_source,
                    reference,
                    selector_name.as_str(),
                    target_style_source,
                    workspace_facts.definitions,
                    value_domain_size,
                ),
            );
        }
    }

    diagnostics.sort_by_key(|diagnostic| {
        (
            diagnostic.range.start.line,
            diagnostic.range.start.character,
            diagnostic.code,
            diagnostic.message.clone(),
        )
    });
    diagnostics.dedup_by(|left, right| {
        left.code == right.code && left.range == right.range && left.message == right.message
    });
    apply_omena_query_checker_product_gate_to_source_diagnostics(&mut diagnostics);

    OmenaQuerySourceDiagnosticsForFileV0 {
        schema_version: "0",
        product: "omena-query.diagnostics-for-file",
        file_uri: source_path.to_string(),
        file_kind: "source",
        diagnostic_count: diagnostics.len(),
        diagnostics,
        ready_surfaces: options.ready_surfaces,
    }
}

/// Undecidability disclosures, split by WHOSE property the cause is.
///
/// `unresolvable` is a property of the CODE (the value's type is an open
/// string, so no finite class-name domain exists — no type checker can
/// enumerate it): disclosed per site at HINT severity, with the one action
/// that lifts it (narrow to a string-literal union). Every other reason is
/// a property of the SESSION (the provider is missing or broken): stamping
/// every dynamic site with the tool's own outage is noise, so those
/// collapse to ONE disclosure per file at the first affected site.
fn summarize_omena_query_type_fact_provider_unavailable_diagnostics(
    source: &str,
    index: &OmenaQuerySourceSyntaxIndexV0,
) -> Vec<OmenaQuerySourceDiagnosticV0> {
    let provenance = || {
        omena_query_evidence_graph_provenance![
            "omena-query.source-syntax-index",
            "omena-tsgo-client.provider-capabilities",
            OMENA_QUERY_TSGO_PROVIDER_UNAVAILABLE_PROVENANCE,
        ]
    };
    let precision = || {
        Some(source_diagnostic_precision(
            OMENA_QUERY_TYPE_ORACLE_UNKNOWN_VALUE_DOMAIN,
            "typeOracleProviderUnavailable",
            "perTypeFactTarget",
        ))
    };
    let mut diagnostics = Vec::new();
    let mut session_facts = Vec::new();
    for fact in index
        .type_fact_provider_unavailable
        .iter()
        .filter(|fact| fact.provider_id == "tsgo")
    {
        if fact.reason == "unresolvable" {
            diagnostics.push(OmenaQuerySourceDiagnosticV0 {
                code: "unknownClassValueDomain",
                severity: "hint",
                provenance: provenance(),
                range: parser_range_for_byte_span(source, fact.byte_span),
                message: "This class value has an open string type, so its class names cannot be checked here.".to_string(),
                precision: precision(),
                suggestion: Some(
                    "Narrow the value's type to a string-literal union (for example 'primary' | 'danger') to enable class checking at this site.".to_string(),
                ),
                create_selector: None,
            });
        } else {
            session_facts.push(fact);
        }
    }
    if let Some(first) = session_facts.first() {
        let cause = match first.reason {
            "projectMiss" => "tsgo could not find a project for this source",
            "noTransport" => "no tsgo provider transport is available",
            "processUnavailable" => "the tsgo provider process could not start",
            _ => "the tsgo provider request failed",
        };
        let site_count = session_facts.len();
        diagnostics.push(OmenaQuerySourceDiagnosticV0 {
            code: "unknownClassValueDomain",
            severity: "warning",
            provenance: provenance(),
            range: parser_range_for_byte_span(source, first.byte_span),
            message: format!(
                "CSS Module class value domain is unknown because {cause}. Dynamic class values in this file ({site_count} site{}) are not checked until the provider is available.",
                if site_count == 1 { "" } else { "s" }
            ),
            precision: precision(),
            suggestion: None,
            create_selector: None,
        });
    }
    diagnostics
}

fn summarize_omena_query_domain_class_reference_diagnostics(
    source: &str,
    index: &OmenaQuerySourceSyntaxIndexV0,
) -> Vec<OmenaQuerySourceDiagnosticV0> {
    let mut diagnostics = Vec::new();
    for reference in &index.domain_class_references {
        let Some(option_name) = reference.option_name.as_ref() else {
            continue;
        };
        let Some(universe) = index.class_value_universes.iter().find(|universe| {
            universe.plugin_id == reference.plugin_id
                && universe.domain == reference.domain
                && universe.owner_name == reference.owner_name
        }) else {
            continue;
        };
        let Some(axis) = universe
            .axes
            .iter()
            .find(|axis| axis.axis_name == reference.axis_name)
        else {
            continue;
        };
        if axis.values.iter().any(|value| value == option_name) {
            continue;
        }
        diagnostics.push(OmenaQuerySourceDiagnosticV0 {
            code: "missingClassValueOption",
            severity: "warning",
            provenance: omena_query_evidence_graph_provenance![
                "omena-bridge.class-value-universe-provider",
                "omena-query.source-domain-class-references",
            ],
            range: parser_range_for_byte_span(source, reference.byte_span),
            message: format!(
                "Class value option '{}' is not defined for {}.{}.",
                option_name, reference.owner_name, reference.axis_name
            ),
            precision: Some(source_diagnostic_precision(
                "classValueUniverse",
                "sourceDomainReference",
                "perDomainAxis",
            )),
            suggestion: None,
            create_selector: None,
        });
    }
    diagnostics
}

pub(super) fn summarize_omena_query_style_selector_definitions(
    style_sources: &[OmenaQueryStyleSourceInputV0],
) -> Vec<OmenaQueryStyleSelectorDefinitionV0> {
    let mut definitions = Vec::new();
    for source in style_sources {
        let Some(candidates) = summarize_omena_query_style_hover_candidates(
            source.style_path.as_str(),
            source.style_source.as_str(),
        ) else {
            continue;
        };
        definitions.extend(candidates.candidates.into_iter().filter_map(|candidate| {
            (candidate.kind == "selector").then(|| OmenaQueryStyleSelectorDefinitionV0 {
                uri: source.style_path.clone(),
                name: candidate.name,
                range: candidate.range,
            })
        }));
    }
    definitions.sort_by_key(|definition| {
        (
            definition.uri.clone(),
            definition.range.start.line,
            definition.range.start.character,
            definition.name.clone(),
        )
    });
    definitions.dedup();
    definitions
}

fn collect_omena_query_source_selector_reference_candidates(
    style_sources: &[OmenaQueryStyleSourceInputV0],
    source_documents: &[OmenaQuerySourceDocumentInputV0],
    package_manifests: &[OmenaQueryStylePackageManifestV0],
    resolution_inputs: &OmenaQueryStyleResolutionInputsV0,
) -> Vec<OmenaQuerySourceSelectorReferenceCandidateV0> {
    collect_omena_query_source_selector_references_with_resolution_inputs(
        style_sources,
        source_documents,
        package_manifests,
        resolution_inputs,
    )
    .into_iter()
    .map(|reference| reference.candidate)
    .collect()
}

fn collect_omena_query_source_selector_reference_edit_targets(
    style_sources: &[OmenaQueryStyleSourceInputV0],
    source_documents: &[OmenaQuerySourceDocumentInputV0],
    package_manifests: &[OmenaQueryStylePackageManifestV0],
    resolution_inputs: &OmenaQueryStyleResolutionInputsV0,
) -> Vec<OmenaQuerySourceSelectorReferenceEditTargetV0> {
    collect_omena_query_source_selector_references_with_resolution_inputs(
        style_sources,
        source_documents,
        package_manifests,
        resolution_inputs,
    )
    .into_iter()
    .filter_map(|reference| {
        reference
            .is_exact
            .then_some(OmenaQuerySourceSelectorReferenceEditTargetV0 {
                uri: reference.candidate.uri,
                name: reference.candidate.name,
                range: reference.candidate.range,
                target_style_uri: reference.candidate.target_style_uri,
            })
    })
    .collect()
}

pub(super) fn collect_omena_query_source_selector_references_with_resolution_inputs(
    style_sources: &[OmenaQueryStyleSourceInputV0],
    source_documents: &[OmenaQuerySourceDocumentInputV0],
    package_manifests: &[OmenaQueryStylePackageManifestV0],
    resolution_inputs: &OmenaQueryStyleResolutionInputsV0,
) -> Vec<OmenaQueryWorkspaceSourceReferenceCandidateV0> {
    let available_style_paths = style_sources
        .iter()
        .map(|source| source.style_path.as_str())
        .collect::<BTreeSet<_>>();
    let mut references = Vec::new();

    for document in source_documents {
        let Some(mut index) = source_selector_reference_index_for_document(
            document,
            &available_style_paths,
            package_manifests,
            resolution_inputs,
        ) else {
            continue;
        };
        canonicalize_omena_query_source_selector_references(&mut index.selector_references);

        for reference in index.selector_references {
            let Some(name) = reference.selector_name.clone().or_else(|| {
                source_reference_text_selector_name(&document.source_source, reference.byte_span)
            }) else {
                continue;
            };
            let is_exact = matches!(
                reference.match_kind,
                OmenaQuerySourceSelectorReferenceMatchKindV0::Exact
            );
            references.push(OmenaQueryWorkspaceSourceReferenceCandidateV0 {
                is_exact,
                candidate: OmenaQuerySourceSelectorReferenceCandidateV0 {
                    uri: document.source_path.clone(),
                    kind: if is_exact {
                        "sourceSelectorReference"
                    } else {
                        "sourceSelectorPrefixReference"
                    },
                    name,
                    range: parser_range_for_byte_span(&document.source_source, reference.byte_span),
                    source: reference.surface.as_str(),
                    target_style_uri: reference.target_style_uri,
                },
            });
        }
    }

    references.sort_by_key(|reference| {
        (
            reference.candidate.uri.clone(),
            reference.candidate.range.start.line,
            reference.candidate.range.start.character,
            reference.candidate.name.clone(),
        )
    });
    references.dedup_by(|left, right| {
        left.candidate.uri == right.candidate.uri
            && left.candidate.range == right.candidate.range
            && left.candidate.name == right.candidate.name
            && left.candidate.target_style_uri == right.candidate.target_style_uri
    });
    references
}

fn source_selector_reference_index_for_document(
    document: &OmenaQuerySourceDocumentInputV0,
    available_style_paths: &BTreeSet<&str>,
    package_manifests: &[OmenaQueryStylePackageManifestV0],
    resolution_inputs: &OmenaQueryStyleResolutionInputsV0,
) -> Option<OmenaQuerySourceSyntaxIndexV0> {
    if let Some(index) = document.source_syntax_index.clone()
        && (!index.imported_style_bindings.is_empty()
            || index
                .selector_references
                .iter()
                .any(|reference| reference.target_style_uri.is_some()))
    {
        return Some(index);
    }

    let imports = summarize_omena_query_source_import_declarations_for_source_language(
        document.source_path.as_str(),
        &document.source_source,
        None,
    );
    let mut imported_style_bindings = Vec::new();
    let mut classnames_bind_bindings = Vec::new();

    for import in imports.imports {
        if import.specifier == "classnames/bind" {
            classnames_bind_bindings.push(import.binding);
            continue;
        }
        let Some(style_uri) = resolve_style_module_source_with_path_mappings(
            &document.source_path,
            &import.specifier,
            available_style_paths,
            package_manifests,
            resolution_inputs.bundler_path_mappings.as_slice(),
            resolution_inputs.tsconfig_path_mappings.as_slice(),
            resolution_inputs.disk_style_path_identities.as_slice(),
        ) else {
            continue;
        };
        imported_style_bindings.push(OmenaQuerySourceImportedStyleBindingV0 {
            binding: import.binding,
            style_uri,
        });
    }

    if imported_style_bindings.is_empty() {
        return None;
    }

    Some(
        summarize_omena_query_source_syntax_index_for_source_language(
            document.source_path.as_str(),
            &document.source_source,
            None,
            imported_style_bindings,
            classnames_bind_bindings,
        ),
    )
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub(super) struct OmenaQueryWorkspaceSourceReferenceCandidateV0 {
    pub(super) is_exact: bool,
    pub(super) candidate: OmenaQuerySourceSelectorReferenceCandidateV0,
}

fn summarize_omena_query_unresolved_source_reference_diagnostic(
    source: &str,
    reference: &OmenaQuerySourceSelectorReferenceFactV0,
    selector_name: &str,
    target_style_source: Option<&str>,
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
    value_domain_size: usize,
) -> OmenaQuerySourceDiagnosticV0 {
    let range = parser_range_for_byte_span(source, reference.byte_span);
    let reference_text = source
        .get(reference.byte_span.start..reference.byte_span.end)
        .unwrap_or_default()
        .trim_matches(['"', '\'', '`']);
    let code = match reference.match_kind {
        OmenaQuerySourceSelectorReferenceMatchKindV0::Exact if reference_text == selector_name => {
            "missingStaticClass"
        }
        OmenaQuerySourceSelectorReferenceMatchKindV0::Exact => "missingResolvedClassValues",
        OmenaQuerySourceSelectorReferenceMatchKindV0::Prefix if reference_text == selector_name => {
            "missingTemplatePrefix"
        }
        OmenaQuerySourceSelectorReferenceMatchKindV0::Prefix => "missingResolvedClassDomain",
    };
    let create_selector = reference
        .target_style_uri
        .as_deref()
        .zip(target_style_source)
        .filter(|_| {
            matches!(
                reference.match_kind,
                OmenaQuerySourceSelectorReferenceMatchKindV0::Exact
            )
        })
        .and_then(|(target_style_uri, target_style_source)| {
            summarize_omena_query_missing_selector_diagnostic(
                target_style_uri,
                target_style_source,
                selector_name,
                range,
            )
            .create_selector
        });
    let suggestion = if code == "missingStaticClass" {
        reference
            .target_style_uri
            .as_deref()
            .and_then(|target_style_uri| {
                closest_selector_name(
                    selector_name,
                    definitions
                        .iter()
                        .filter(|definition| {
                            file_uri_equivalent(definition.uri.as_str(), target_style_uri)
                        })
                        .map(|definition| definition.name.as_str()),
                    3,
                )
            })
    } else {
        None
    };

    OmenaQuerySourceDiagnosticV0 {
        code,
        severity: "warning",
        provenance: omena_query_evidence_graph_provenance![
            "omena-query.source-syntax-index",
            "omena-query.style-selector-definitions",
        ],
        range,
        message: query_source_diagnostic_message(
            code,
            selector_name,
            suggestion.as_deref(),
            value_domain_size,
        ),
        precision: Some(source_diagnostic_precision(
            "classValueResolution",
            "sourceSelectorReference",
            match code {
                "missingResolvedClassValues" | "missingResolvedClassDomain" => {
                    "resolvedClassValueDomain"
                }
                _ => "perSourceReference",
            },
        )),
        suggestion,
        create_selector,
    }
}

fn query_source_diagnostic_message(
    code: &str,
    selector_name: &str,
    suggestion: Option<&str>,
    value_domain_size: usize,
) -> String {
    match code {
        "missingStaticClass" => {
            let hint = suggestion
                .map(|suggestion| format!(" Did you mean '{suggestion}'?"))
                .unwrap_or_default();
            format!("Class '.{selector_name}' not found in target CSS Module.{hint}")
        }
        "missingTemplatePrefix" => {
            format!("No class starting with '{selector_name}' found in target CSS Module.")
        }
        "missingResolvedClassValues" => {
            format!(
                "Missing class for possible value: '{selector_name}'. Analysis reason: analysis preserved multiple finite candidate values. Analysis shape: bounded finite ({value_domain_size})."
            )
        }
        "missingResolvedClassDomain" => {
            format!("No class matched resolved prefix '{selector_name}'.")
        }
        _ => "Source diagnostic reported by omena-query.".to_string(),
    }
}

fn closest_selector_name<'a>(
    target: &str,
    candidates: impl IntoIterator<Item = &'a str>,
    max_distance: usize,
) -> Option<String> {
    let mut best = None::<(&'a str, usize)>;
    for candidate in candidates {
        let current_bound = best
            .map(|(_, distance)| distance.saturating_sub(1))
            .unwrap_or(max_distance);
        let distance = bounded_levenshtein_distance(target, candidate, current_bound);
        if distance <= max_distance
            && best.is_none_or(|(_, best_distance)| distance < best_distance)
        {
            best = Some((candidate, distance));
        }
    }
    best.map(|(candidate, _)| candidate.to_string())
}

fn bounded_levenshtein_distance(left: &str, right: &str, max_distance: usize) -> usize {
    if left == right {
        return 0;
    }
    if left.is_empty() {
        return right.chars().count();
    }
    if right.is_empty() {
        return left.chars().count();
    }

    let left_chars = left.chars().collect::<Vec<_>>();
    let right_chars = right.chars().collect::<Vec<_>>();
    if left_chars.len().abs_diff(right_chars.len()) > max_distance {
        return max_distance + 1;
    }

    let mut previous = (0..=right_chars.len()).collect::<Vec<_>>();
    let mut current = vec![0; right_chars.len() + 1];
    for (left_index, left_char) in left_chars.iter().enumerate() {
        current[0] = left_index + 1;
        let mut row_min = current[0];
        for (right_index, right_char) in right_chars.iter().enumerate() {
            let cost = usize::from(left_char != right_char);
            let value = (current[right_index] + 1)
                .min(previous[right_index + 1] + 1)
                .min(previous[right_index] + cost);
            current[right_index + 1] = value;
            row_min = row_min.min(value);
        }
        if row_min > max_distance {
            return max_distance + 1;
        }
        previous.copy_from_slice(&current);
    }
    previous[right_chars.len()]
}

fn is_query_source_style_module_specifier(specifier: &str) -> bool {
    specifier.contains(".module.")
        || specifier.ends_with(".css")
        || specifier.ends_with(".scss")
        || specifier.ends_with(".sass")
        || specifier.ends_with(".less")
}

pub fn resolve_omena_query_source_provider_candidates(
    source_candidates: Vec<OmenaQuerySourceSelectorCandidateV0>,
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
) -> OmenaQuerySourceProviderCandidateResolutionV0 {
    if definitions.is_empty() {
        return OmenaQuerySourceProviderCandidateResolutionV0 {
            schema_version: "0",
            product: "omena-query.source-provider-candidate-resolution",
            matched: Vec::new(),
            unresolved: Vec::new(),
        };
    }

    let (mut matched, mut unresolved): (Vec<_>, Vec<_>) =
        source_candidates.into_iter().partition(|candidate| {
            definitions.iter().any(|definition| {
                source_selector_candidate_matches_definition(candidate, definition)
            })
        });
    matched.sort();
    unresolved.sort();
    OmenaQuerySourceProviderCandidateResolutionV0 {
        schema_version: "0",
        product: "omena-query.source-provider-candidate-resolution",
        matched,
        unresolved,
    }
}

pub fn resolve_omena_query_style_selector_definitions_for_source_candidate(
    candidate: &OmenaQuerySourceSelectorCandidateV0,
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
) -> Vec<OmenaQueryStyleSelectorDefinitionV0> {
    let mut matched = definitions
        .iter()
        .filter(|definition| source_selector_candidate_matches_definition(candidate, definition))
        .cloned()
        .collect::<Vec<_>>();
    matched.sort_by_key(|definition| {
        (
            definition.uri.clone(),
            definition.range.start.line,
            definition.range.start.character,
            definition.name.clone(),
        )
    });
    matched.dedup();
    matched
}

pub fn resolve_omena_query_source_candidate_selector_names(
    candidate: &OmenaQuerySourceSelectorCandidateV0,
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
    target_style_uri: Option<&str>,
) -> Vec<String> {
    if candidate.kind != "sourceSelectorPrefixReference" {
        return vec![candidate.name.clone()];
    }

    let mut names = definitions
        .iter()
        .filter(|definition| source_selector_candidate_matches_definition(candidate, definition))
        .filter(|definition| {
            candidate
                .target_style_uri
                .as_deref()
                .or(target_style_uri)
                .is_none_or(|target_uri| file_uri_equivalent(target_uri, definition.uri.as_str()))
        })
        .map(|definition| definition.name.clone())
        .collect::<Vec<_>>();
    names.sort();
    names.dedup();
    names
}

pub fn resolve_omena_query_selector_rename_edits(
    selector_name: &str,
    new_name: &str,
    target_style_uri: Option<&str>,
    definitions: &[OmenaQueryStyleSelectorDefinitionV0],
    references: &[OmenaQuerySourceSelectorReferenceEditTargetV0],
) -> Vec<OmenaQueryWorkspaceTextEditV0> {
    let replacement = new_name.trim_start_matches('.');
    if replacement.is_empty() {
        return Vec::new();
    }

    let mut edits = definitions
        .iter()
        .filter(|definition| class_names_match(definition.name.as_str(), selector_name))
        .filter(|definition| {
            target_style_uri
                .is_none_or(|target_uri| file_uri_equivalent(target_uri, definition.uri.as_str()))
        })
        .map(|definition| OmenaQueryWorkspaceTextEditV0 {
            uri: definition.uri.clone(),
            range: definition.range,
            new_text: replacement.to_string(),
        })
        .chain(
            references
                .iter()
                .filter(|reference| class_names_match(reference.name.as_str(), selector_name))
                .filter(|reference| {
                    source_reference_matches_target_style(reference, target_style_uri)
                })
                .map(|reference| OmenaQueryWorkspaceTextEditV0 {
                    uri: reference.uri.clone(),
                    range: reference.range,
                    new_text: replacement.to_string(),
                }),
        )
        .collect::<Vec<_>>();
    edits.sort_by_key(|edit| {
        (
            edit.uri.clone(),
            edit.range.start.line,
            edit.range.start.character,
            edit.range.end.line,
            edit.range.end.character,
        )
    });
    edits
}

fn source_selector_candidate_matches_definition(
    candidate: &OmenaQuerySourceSelectorCandidateV0,
    definition: &OmenaQueryStyleSelectorDefinitionV0,
) -> bool {
    let selector_matches = if candidate.kind == "sourceSelectorPrefixReference" {
        ClassNameV0::new(definition.name.as_str())
            .decoded()
            .starts_with(ClassNameV0::new(candidate.name.as_str()).decoded())
    } else {
        canonical_class_key(definition.name.as_str())
            == canonical_class_key(candidate.name.as_str())
    };
    selector_matches
        && candidate
            .target_style_uri
            .as_deref()
            .is_none_or(|target_uri| file_uri_equivalent(target_uri, definition.uri.as_str()))
}

fn source_reference_matches_target_style(
    reference: &OmenaQuerySourceSelectorReferenceEditTargetV0,
    target_style_uri: Option<&str>,
) -> bool {
    target_style_uri.is_none_or(|target_uri| {
        reference
            .target_style_uri
            .as_deref()
            .is_none_or(|candidate_target_uri| {
                file_uri_equivalent(candidate_target_uri, target_uri)
            })
    })
}

fn source_selector_occurrences_for_query(
    occurrence_index: &OmenaQuerySourceSelectorOccurrenceIndexV0,
    selector_name: &str,
    target_style_uri: Option<&str>,
) -> Vec<OmenaQuerySourceSelectorOccurrenceV0> {
    let matching_monikers = occurrence_index
        .occurrences
        .iter()
        .filter(|occurrence| class_names_match(occurrence.selector_name.as_str(), selector_name))
        .filter(|occurrence| {
            target_style_uri.is_none_or(|target_uri| {
                occurrence
                    .target_style_uri
                    .as_deref()
                    .is_some_and(|candidate_uri| file_uri_equivalent(candidate_uri, target_uri))
            })
        })
        .map(|occurrence| occurrence.moniker.clone())
        .collect::<BTreeSet<_>>();
    occurrences_for_monikers(&occurrence_index.workspace_index, &matching_monikers)
        .into_iter()
        .filter_map(source_selector_occurrence_from_workspace_occurrence)
        .collect()
}

pub fn summarize_omena_query_workspace_occurrence_index_from_source_occurrences(
    occurrences: &[OmenaQuerySourceSelectorOccurrenceV0],
    ready_surfaces: Vec<&'static str>,
) -> OmenaWorkspaceOccurrenceIndexV0 {
    let occurrences = occurrences
        .iter()
        .map(workspace_occurrence_from_source_occurrence)
        .collect::<Vec<_>>();
    summarize_omena_query_workspace_occurrence_index_from_occurrences(
        occurrences.as_slice(),
        ready_surfaces,
    )
}

pub fn summarize_omena_query_workspace_occurrence_index_from_occurrences(
    occurrences: &[OmenaWorkspaceOccurrenceV0],
    ready_surfaces: Vec<&'static str>,
) -> OmenaWorkspaceOccurrenceIndexV0 {
    let mut by_moniker: BTreeMap<String, Vec<OmenaWorkspaceOccurrenceV0>> = BTreeMap::new();
    let mut by_file: BTreeMap<String, BTreeSet<String>> = BTreeMap::new();
    for occurrence in occurrences {
        let workspace_occurrence = occurrence.clone();
        by_file
            .entry(workspace_occurrence.uri.clone())
            .or_default()
            .insert(workspace_occurrence.moniker.clone());
        by_moniker
            .entry(workspace_occurrence.moniker.clone())
            .or_default()
            .push(workspace_occurrence);
    }
    for occurrences in by_moniker.values_mut() {
        occurrences.sort();
        occurrences.dedup();
    }
    let by_file = by_file
        .into_iter()
        .map(|(uri, monikers)| (uri, monikers.into_iter().collect()))
        .collect::<BTreeMap<_, _>>();
    let moniker_count = by_moniker.len();
    let occurrence_count = by_moniker.values().map(Vec::len).sum();
    OmenaWorkspaceOccurrenceIndexV0 {
        schema_version: "0",
        product: "omena-query.workspace-occurrence-index",
        moniker_count,
        occurrence_count,
        by_moniker,
        by_file,
        ready_surfaces,
    }
}

fn workspace_occurrence_from_source_occurrence(
    occurrence: &OmenaQuerySourceSelectorOccurrenceV0,
) -> OmenaWorkspaceOccurrenceV0 {
    OmenaWorkspaceOccurrenceV0 {
        moniker: occurrence.moniker.clone(),
        uri: occurrence.uri.clone(),
        name: occurrence.selector_name.clone(),
        range: occurrence.range,
        kind: occurrence.kind,
        role: occurrence.role,
        surface: occurrence.source,
        family: Some(OmenaWorkspaceOccurrenceFamilyV0::CssModuleSelector),
        namespace: None,
        target_style_uri: occurrence.target_style_uri.clone(),
        rename_target: occurrence.rename_target,
    }
}

fn source_selector_occurrence_from_workspace_occurrence(
    occurrence: &OmenaWorkspaceOccurrenceV0,
) -> Option<OmenaQuerySourceSelectorOccurrenceV0> {
    (occurrence.family == Some(OmenaWorkspaceOccurrenceFamilyV0::CssModuleSelector)).then(|| {
        OmenaQuerySourceSelectorOccurrenceV0 {
            moniker: occurrence.moniker.clone(),
            uri: occurrence.uri.clone(),
            selector_name: occurrence.name.clone(),
            range: occurrence.range,
            kind: occurrence.kind,
            role: occurrence.role,
            source: occurrence.surface,
            target_style_uri: occurrence.target_style_uri.clone(),
            rename_target: occurrence.rename_target,
        }
    })
}

fn workspace_occurrence_kind_from_source_reference_kind(
    kind: &str,
) -> Option<OmenaWorkspaceOccurrenceKindV0> {
    match kind {
        "sourceSelectorReference" => Some(OmenaWorkspaceOccurrenceKindV0::SourceSelectorReference),
        "sourceSelectorPrefixReference" => {
            Some(OmenaWorkspaceOccurrenceKindV0::SourceSelectorPrefixReference)
        }
        _ => None,
    }
}

fn source_selector_candidate_matches_target_uri(
    candidate: &OmenaQuerySourceSelectorCandidateV0,
    target_style_uri: Option<&str>,
) -> bool {
    target_style_uri.is_none_or(|target_uri| {
        candidate
            .target_style_uri
            .as_deref()
            .is_none_or(|candidate_target_uri| {
                file_uri_equivalent(candidate_target_uri, target_uri)
            })
    })
}

fn source_selector_occurrence_moniker(
    selector_name: &str,
    target_style_uri: Option<&str>,
) -> String {
    omena_workspace_moniker(OmenaWorkspaceMonikerInput::CssModuleSelector {
        target_style_uri,
        selector_name,
    })
}

fn class_names_match(left: &str, right: &str) -> bool {
    ClassNameV0::new(left).same_as(&ClassNameV0::new(right))
}

fn reference_location_role_rank(role: &str) -> u8 {
    match role {
        "definition" => 0,
        "reference" => 1,
        _ => 2,
    }
}

fn file_uri_equivalent(left: &str, right: &str) -> bool {
    if left == right {
        return true;
    }
    match (
        file_uri_to_normalized_path(left),
        file_uri_to_normalized_path(right),
    ) {
        (Some(left_path), Some(right_path)) => left_path == right_path,
        _ => false,
    }
}

fn file_uri_to_normalized_path(uri: &str) -> Option<PathBuf> {
    let raw_path = uri.strip_prefix("file://")?;
    Some(normalize_path(PathBuf::from(percent_decode_uri_path(
        raw_path,
    )?)))
}

fn percent_decode_uri_path(raw_path: &str) -> Option<String> {
    let bytes = raw_path.as_bytes();
    let mut decoded = Vec::with_capacity(bytes.len());
    let mut index = 0usize;
    while index < bytes.len() {
        if bytes[index] == b'%' {
            let high = bytes.get(index + 1).and_then(|byte| hex_value(*byte))?;
            let low = bytes.get(index + 2).and_then(|byte| hex_value(*byte))?;
            decoded.push((high << 4) | low);
            index += 3;
        } else {
            decoded.push(bytes[index]);
            index += 1;
        }
    }
    String::from_utf8(decoded).ok()
}

fn hex_value(byte: u8) -> Option<u8> {
    match byte {
        b'0'..=b'9' => Some(byte - b'0'),
        b'a'..=b'f' => Some(byte - b'a' + 10),
        b'A'..=b'F' => Some(byte - b'A' + 10),
        _ => None,
    }
}

fn normalize_path(path: PathBuf) -> PathBuf {
    let mut normalized = PathBuf::new();
    for component in path.components() {
        match component {
            Component::CurDir => {}
            Component::ParentDir => {
                normalized.pop();
            }
            Component::Normal(_) | Component::RootDir | Component::Prefix(_) => {
                normalized.push(component.as_os_str());
            }
        }
    }
    normalized
}

#[cfg(test)]
mod global_class_fallthrough_label_tests {
    use super::summarize_omena_query_global_class_fallthrough_diagnostic;
    use crate::ParserRangeV0;

    #[test]
    fn message_shows_decoded_non_ascii_global_filename() {
        let diagnostic = summarize_omena_query_global_class_fallthrough_diagnostic(
            "chip",
            "file:///ws/%EC%83%98%ED%94%8C%EB%B0%B0%EB%84%88.css",
            "file:///ws/App.module.css",
            ".root {}\n",
            ParserRangeV0::default(),
        );
        assert!(
            diagnostic.message.contains("샘플배너.css"),
            "{}",
            diagnostic.message
        );
        assert!(
            !diagnostic.message.contains("%EC"),
            "{}",
            diagnostic.message
        );
    }
}

#[cfg(test)]
mod class_reference_identity_tests {
    use super::*;

    #[test]
    fn references_join_cross_spelling_names_in_direct_and_indexed_paths() {
        let style_uri = "file:///workspace/App.module.css";
        let source_uri = "file:///workspace/App.tsx";
        let definitions = vec![OmenaQueryStyleSelectorDefinitionV0 {
            uri: style_uri.to_string(),
            name: r"\62 tn".to_string(),
            range: ParserRangeV0::default(),
        }];
        let references = vec![OmenaQuerySourceSelectorReferenceCandidateV0 {
            uri: source_uri.to_string(),
            kind: "sourceSelectorReference",
            name: "btn".to_string(),
            range: ParserRangeV0::default(),
            source: "omenaQuerySourceSyntaxIndex",
            target_style_uri: Some(style_uri.to_string()),
        }];

        let direct = summarize_omena_query_refs_for_class(
            r"\62 tn",
            Some(style_uri),
            true,
            definitions.as_slice(),
            references.as_slice(),
        );
        let occurrence_index = summarize_omena_query_source_selector_occurrence_index(
            definitions.as_slice(),
            references.as_slice(),
        );
        let indexed = summarize_omena_query_refs_for_class_from_occurrence_index(
            r"\62 tn",
            Some(style_uri),
            true,
            definitions.as_slice(),
            &occurrence_index,
        );

        for summary in [direct, indexed] {
            assert_eq!(summary.location_count, 2);
            assert!(
                summary
                    .locations
                    .iter()
                    .any(|location| { location.role == "reference" && location.uri == source_uri })
            );
        }
    }
}