bynk-lsp 0.140.0

//! v0.31 (ADR 0064): locals navigation — resolve the local binding under the
//! cursor and all its sites (its definition plus every use that resolves to
//! it), for `references`, go-to-`definition`, and `documentHighlight`.
//!
//! Slice 1 records bindings with scope ranges (not use sites); the use sites
//! are recovered here by lexing the file and keeping the identifier tokens of
//! the binding's name within its scope that resolve back to it (so a shadowing
//! inner binding's uses — and every binding's *def* token — are excluded).
//! Pure over the analysed snapshot, like `index_queries`.

use bynk_check::locals::{LocalBinding, LocalKind, binding_at_def, locals_at};
use bynk_syntax::lexer::{self, TokenKind};
use bynk_syntax::span::Span;

/// The identifier-token name covering `offset`, if any. Interpolation holes are
/// expanded (issue #473), so a cursor inside `"… \(name) …"` resolves to the
/// hole's `name` identifier rather than the opaque `InterpStr` token.
fn ident_at(text: &str, offset: usize) -> Option<(&str, Span)> {
    let toks = lexer::tokenize_expanding_holes(text).ok()?;
    toks.into_iter()
        .find(|t| t.kind == TokenKind::Ident && t.span.start <= offset && offset <= t.span.end)
        .map(|t| (&text[t.span.start..t.span.end], t.span))
}

/// The binding the cursor refers to — whether it sits on the definition name
/// or on a use — within `locals` (a file's bindings).
fn target_at<'a>(
    locals: &'a [LocalBinding],
    text: &str,
    offset: usize,
) -> Option<&'a LocalBinding> {
    let (name, _) = ident_at(text, offset)?;
    binding_at_def(locals, offset)
        .filter(|b| b.name == name)
        .or_else(|| {
            locals_at(locals, offset)
                .into_iter()
                .find(|b| b.name == name)
        })
}

/// All sites of the local under the cursor — its definition first, then every
/// use that resolves to it (shadowing-safe). `None` when the cursor is not on
/// a local.
pub fn local_sites_at(locals: &[LocalBinding], text: &str, offset: usize) -> Option<Vec<Span>> {
    let target = target_at(locals, text, offset)?;
    // Hole-aware (issue #473): use sites inside `\(…)` holes count too.
    let toks = lexer::tokenize_expanding_holes(text).ok()?;
    let mut sites = vec![target.def_span];
    for t in &toks {
        if t.kind != TokenKind::Ident || text[t.span.start..t.span.end] != target.name {
            continue;
        }
        if t.span == target.def_span {
            continue; // the definition, already added
        }
        // A binding's own def token is not a use of anything.
        if locals.iter().any(|b| b.def_span == t.span) {
            continue;
        }
        if t.span.start < target.scope.start || t.span.end > target.scope.end {
            continue; // outside the binding's scope
        }
        // Does this use resolve to `target` (not a shadowing inner binding)?
        let resolves = locals_at(locals, t.span.start)
            .into_iter()
            .find(|b| b.name == target.name)
            .map(|b| b.def_span);
        if resolves == Some(target.def_span) {
            sites.push(t.span);
        }
    }
    Some(sites)
}

/// The definition site of the local under the cursor, if any.
pub fn local_definition_at(locals: &[LocalBinding], text: &str, offset: usize) -> Option<Span> {
    target_at(locals, text, offset).map(|b| b.def_span)
}

/// v0.122 (editor-currency slice 1): a hover summary for the local binding /
/// parameter under the cursor — `let x: <ty>` / `param n: <ty>`, rendered from
/// the checker's captured `LocalBinding` (its `ty` is already the surface
/// `type_ref` form, matching inlay hints and signature help). `None` when the
/// cursor is not on a local. Reuses the same `target_at` resolution as
/// go-to-definition / references, so hover cannot disagree with them.
pub fn describe_local_at(locals: &[LocalBinding], text: &str, offset: usize) -> Option<String> {
    let b = target_at(locals, text, offset)?;
    let keyword = match b.kind {
        LocalKind::Let => "let",
        LocalKind::Param => "param",
    };
    Some(format!("```bynk\n{keyword} {}: {}\n```", b.name, b.ty))
}

/// Every local-binding occurrence in the file — `(span, is_definition)` — for
/// semantic-token colouring. A token is a definition if it sits on a binding's
/// def span, else a use if it resolves to a local in scope at that point.
pub fn local_token_sites(locals: &[LocalBinding], text: &str) -> Vec<(Span, bool)> {
    // Hole-aware (issue #473): locals used inside `\(…)` holes colour too.
    let Ok(toks) = lexer::tokenize_expanding_holes(text) else {
        return Vec::new();
    };
    let mut out = Vec::new();
    for t in &toks {
        if t.kind != TokenKind::Ident {
            continue;
        }
        let name = &text[t.span.start..t.span.end];
        if locals.iter().any(|b| b.def_span == t.span) {
            out.push((t.span, true)); // a binding's def
        } else if locals_at(locals, t.span.start)
            .into_iter()
            .any(|b| b.name == name)
        {
            out.push((t.span, false)); // a use that resolves to a local
        }
    }
    out
}

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

    // `fn f(n: Int) -> Int { let x = n  <uses> }` laid out so offsets are easy.
    fn bindings() -> Vec<LocalBinding> {
        // text: see `TEXT`; n: param scope over body, x: let scope after its stmt.
        vec![
            LocalBinding {
                name: "n".into(),
                def_span: Span { start: 5, end: 6 },
                kind: LocalKind::Param,
                ty: "Int".into(),
                scope: Span { start: 20, end: 60 },
            },
            LocalBinding {
                name: "x".into(),
                def_span: Span { start: 26, end: 27 },
                kind: LocalKind::Let,
                ty: "Int".into(),
                scope: Span { start: 34, end: 60 },
            },
        ]
    }

    const TEXT: &str = "fn f(n: Int) -> Int { let x = n\n  x + x\n}";
    //                   0         1         2         3
    //                   0123456789012345678901234567890123456789

    #[test]
    fn sites_for_a_use_collect_def_plus_uses() {
        let locals = bindings();
        // Cursor on the first `x` use (offset 36, in `  x + x`).
        let x_use = TEXT.match_indices('x').nth(1).unwrap().0; // first use of x
        let sites = local_sites_at(&locals, TEXT, x_use).expect("on a local");
        assert!(
            sites.contains(&Span { start: 26, end: 27 }),
            "includes def: {sites:?}"
        );
        assert!(sites.len() >= 2, "def + at least one use: {sites:?}");
    }

    #[test]
    fn definition_resolves_from_a_use() {
        let locals = bindings();
        let n_use = TEXT.rfind('n').unwrap(); // the `n` in `let x = n`
        assert_eq!(
            local_definition_at(&locals, TEXT, n_use),
            Some(Span { start: 5, end: 6 })
        );
    }

    #[test]
    fn not_on_a_local_yields_none() {
        let locals = bindings();
        assert!(local_sites_at(&locals, TEXT, 0).is_none()); // on `fn`
    }

    #[test]
    fn token_sites_mark_definitions_and_uses() {
        let sites = local_token_sites(&bindings(), TEXT);
        assert!(
            sites.iter().any(|(_, decl)| *decl),
            "has a definition token"
        );
        assert!(sites.iter().any(|(_, decl)| !*decl), "has a use token");
        // The `x` def is a declaration token.
        assert!(
            sites.contains(&(Span { start: 26, end: 27 }, true)),
            "x def is a declaration: {sites:?}"
        );
    }

    // End-to-end against real checker output — the lexer's token spans must
    // line up with the checker's recorded def spans.
    #[test]
    fn resolves_a_real_local_from_diagnose_project() {
        let root = std::path::Path::new(env!("CARGO_MANIFEST_DIR"))
            .join("../bynkc/tests/fixtures/inlay/clean/src");
        let r = bynk_ide::diagnose_project(&root, &std::collections::HashMap::new());
        let file = r
            .files
            .iter()
            .find(|f| f.source_path.to_string_lossy().ends_with("util.bynk"))
            .expect("util.bynk analysed");
        let text = &file.text;
        let locals = r
            .locals
            .iter()
            .find(|(p, _)| p.to_string_lossy().ends_with("util.bynk"))
            .map(|(_, l)| l.clone())
            .expect("util.bynk locals");

        // `let total = …` then `total` — cursor on the use resolves to def + use.
        let use_off = text.rfind("total").expect("total use");
        let sites = local_sites_at(&locals, text, use_off).expect("on a local");
        assert!(sites.len() >= 2, "def + use: {sites:?}");
        // The definition is first and is the `let total` name.
        let def = text.find("total").expect("total def");
        assert_eq!(sites[0].start, def, "def first");
    }

    // v0.122 (slice 1): hover renders a local's `let`/`param` prefix and its
    // inferred type, from the same real checker output.
    #[test]
    fn describe_local_renders_kind_and_type() {
        let root = std::path::Path::new(env!("CARGO_MANIFEST_DIR"))
            .join("../bynkc/tests/fixtures/inlay/clean/src");
        let r = bynk_ide::diagnose_project(&root, &std::collections::HashMap::new());
        let file = r
            .files
            .iter()
            .find(|f| f.source_path.to_string_lossy().ends_with("util.bynk"))
            .expect("util.bynk analysed");
        let text = &file.text;
        let locals = r
            .locals
            .iter()
            .find(|(p, _)| p.to_string_lossy().ends_with("util.bynk"))
            .map(|(_, l)| l.clone())
            .expect("util.bynk locals");

        // A `let` binding — `let total = xs.fold(…)` → `let total: Int`.
        let total = text.find("total").expect("total def");
        assert_eq!(
            describe_local_at(&locals, text, total).as_deref(),
            Some("```bynk\nlet total: Int\n```")
        );
        // A parameter — `fn sum(xs: List[Int])` → `param xs: List[Int]`. Use the
        // def site so we land on the parameter, not a shadowing local.
        let xs_param = text.find("xs: List[Int]").expect("xs param");
        assert_eq!(
            describe_local_at(&locals, text, xs_param).as_deref(),
            Some("```bynk\nparam xs: List[Int]\n```")
        );
        // Not on a local (the `fn` keyword) → nothing.
        assert!(describe_local_at(&locals, text, text.find("fn").unwrap()).is_none());
    }

    // Issue #473: a parameter used *inside* an interpolation hole
    // (`"… \(name) …"`) must resolve the same as anywhere else. Drives the real
    // hover / go-to-definition / references resolution against live checker
    // output — the position→symbol step that was previously blind to holes
    // because the file lexes the string to one opaque `InterpStr` token.

    // `greet`'s param `name` is used only inside a `\(shout(name))` hole.
    const HOLE_SRC: &str = "\
commons demo.text

fn shout(s: String) -> String {
  s
}

fn greet(name: String) -> String {
  \"Hi, \\(shout(name))!\"
}
";

    /// `(text, locals)` for `demo/text.bynk` after a real project analysis.
    fn analyse_hole_fixture(test_name: &str) -> (String, Vec<LocalBinding>) {
        let root = std::env::temp_dir().join(format!(
            "bynk-locals-hole-{test_name}-{}",
            std::process::id()
        ));
        let _ = std::fs::remove_dir_all(&root);
        let file = root.join("demo/text.bynk");
        std::fs::create_dir_all(file.parent().unwrap()).expect("create dirs");
        std::fs::write(&file, HOLE_SRC).expect("write fixture");
        let root = root.canonicalize().unwrap_or(root);

        let r = bynk_ide::diagnose_project(&root, &std::collections::HashMap::new());
        let text = r
            .files
            .iter()
            .find(|f| f.source_path.to_string_lossy().ends_with("text.bynk"))
            .expect("text.bynk analysed")
            .text
            .clone();
        let locals = r
            .locals
            .iter()
            .find(|(p, _)| p.to_string_lossy().ends_with("text.bynk"))
            .map(|(_, l)| l.clone())
            .expect("text.bynk locals");
        (text, locals)
    }

    /// The byte offset of the `n`th occurrence of `needle` in `text`.
    fn nth_offset(text: &str, needle: &str, n: usize) -> usize {
        text.match_indices(needle).nth(n).expect("occurrence").0
    }

    #[test]
    fn hover_describes_a_param_inside_a_hole() {
        let (text, locals) = analyse_hole_fixture("hover");
        // 2nd `name`: the use inside `\(shout(name))` (1st is the declaration).
        let in_hole = nth_offset(&text, "name", 1) + 1; // mid-identifier
        assert_eq!(
            describe_local_at(&locals, &text, in_hole).as_deref(),
            Some("```bynk\nparam name: String\n```"),
            "hover inside the hole renders the param summary"
        );
    }

    #[test]
    fn definition_of_a_param_resolves_from_inside_a_hole() {
        let (text, locals) = analyse_hole_fixture("def");
        let in_hole = nth_offset(&text, "name", 1) + 1;
        let def = local_definition_at(&locals, &text, in_hole).expect("resolves to a def");
        let decl = nth_offset(&text, "name", 0); // the parameter declaration
        assert_eq!(def.start, decl, "def points at the `name` parameter");
    }

    #[test]
    fn references_include_a_param_use_inside_a_hole() {
        let (text, locals) = analyse_hole_fixture("refs");
        let decl = nth_offset(&text, "name", 0);
        let sites = local_sites_at(&locals, &text, decl).expect("on the param");
        let in_hole = nth_offset(&text, "name", 1);
        assert!(
            sites.iter().any(|s| s.start <= in_hole && in_hole < s.end),
            "references include the in-hole use at {in_hole}; got {sites:?}"
        );
    }
}