tishlang_lsp 1.9.2

//! Built-in and JSX-intrinsic "go to definition" for the Tish LSP.
//!
//! Global / namespace member anchors are loaded from `stdlib/builtins.d.tish` in the `tish`
//! repository via `// @tish-source <symbol> <rel-path> <1-based-line>` pragmas. The type
//! surface in that file is the canonical declaration for ambient builtins.
//!
//! HTML / SVG intrinsic tag names are still listed here (sorted) for fast lookup; each maps
//! to the same vnode factory as in the pragma table for `div`.
//!
//! Definitions resolve to `file://` URIs only when `TISHLANG_SOURCE_ROOT` is set or the client
//! passes `tishlangSourceRoot` in LSP `initializationOptions` (VS Code: `tish.tishlangSourceRoot`).

use std::collections::HashMap;
use std::path::Path;
use std::sync::OnceLock;

use regex::Regex;
use tower_lsp::lsp_types::{Location, Position, Range, Url};

/// Stable path relative to the `tish` repository root (where the workspace `Cargo.toml` lives).
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct BuiltinDef {
    pub rel_path: String,
    /// 0-based LSP line in the target file.
    pub line: u32,
    /// 0-based UTF-16 code unit offset on that line (ASCII-only targets use byte column).
    pub character: u32,
}

fn is_ident_char(c: char) -> bool {
    c.is_alphanumeric() || c == '_'
}

/// HTML / SVG intrinsic tag names (lowercase) that JSX lowers to `h("tag", …)`; must stay sorted for `binary_search`.
const HTML_INTRINSIC_TAGS: &[&str] = &[
    "a", "abbr", "address", "area", "article", "aside", "audio", "b", "base", "bdi", "bdo",
    "blockquote", "body", "br", "button", "canvas", "caption", "cite", "code", "col", "colgroup",
    "data", "datalist", "dd", "del", "details", "dfn", "dialog", "div", "dl", "dt", "em", "embed",
    "fieldset", "figcaption", "figure", "footer", "form", "h1", "h2", "h3", "h4", "h5", "h6",
    "head", "header", "hr", "html", "i", "iframe", "img", "input", "ins", "kbd", "label", "legend",
    "li", "link", "main", "map", "mark", "meta", "meter", "nav", "noscript", "object", "ol",
    "optgroup", "option", "output", "p", "param", "picture", "pre", "progress", "q", "rp", "rt",
    "ruby", "s", "samp", "script", "section", "select", "slot", "small", "source", "span", "strong",
    "style", "sub", "summary", "sup", "svg", "table", "tbody", "td", "template", "textarea",
    "tfoot", "th", "thead", "time", "title", "tr", "track", "u", "ul", "var", "video", "wbr",
];

static SOURCE_MAP: OnceLock<HashMap<String, BuiltinDef>> = OnceLock::new();

/// Parse `// @tish-source <symbol> <rel-path> <1-based-line>` lines (same format as `stdlib/builtins.d.tish`).
pub fn parse_tish_source_pragmas(src: &str) -> HashMap<String, BuiltinDef> {
    let re = Regex::new(r"(?m)^\s*//\s*@tish-source\s+(\S+)\s+(\S+)\s+(\d+)\s*$")
        .expect("builtin pragma regex");
    let mut m = HashMap::new();
    for cap in re.captures_iter(src) {
        let sym = cap[1].to_string();
        let rel = cap[2].to_string();
        let line_1: u32 = cap[3].parse().unwrap_or(1);
        m.insert(
            sym,
            BuiltinDef {
                rel_path: rel,
                line: line_1.saturating_sub(1),
                character: 0,
            },
        );
    }
    m
}

fn source_map() -> &'static HashMap<String, BuiltinDef> {
    SOURCE_MAP.get_or_init(|| {
        let src = include_str!(concat!(
            env!("CARGO_MANIFEST_DIR"),
            "/../../stdlib/builtins.d.tish"
        ));
        parse_tish_source_pragmas(src)
    })
}

/// If `col` lies on the name of a JSX opening (or closing) tag on this line, returns the span of that name in **character indices** (same convention as [`tower_lsp::lsp_types::Position::character`] for ASCII lines).
fn jsx_tag_name_char_span(line: &str, col: usize) -> Option<(usize, usize)> {
    let chars: Vec<char> = line.chars().collect();
    if chars.is_empty() {
        return None;
    }
    let col = col.min(chars.len().saturating_sub(1));
    let mut j = col;
    while j > 0 {
        j -= 1;
        if chars[j] == '>' {
            return None;
        }
        if chars[j] == '<' {
            let mut k = j + 1;
            while k < chars.len() && chars[k].is_whitespace() {
                k += 1;
            }
            if k < chars.len() && chars[k] == '/' {
                k += 1;
                while k < chars.len() && chars[k].is_whitespace() {
                    k += 1;
                }
            }
            let name_start = k;
            while k < chars.len() {
                let c = chars[k];
                if c.is_whitespace() || c == '>' || c == '/' || c == '{' {
                    break;
                }
                if !(c.is_alphanumeric() || c == '_' || c == '-') {
                    break;
                }
                k += 1;
            }
            let name_end = k;
            if name_start < name_end && col >= name_start && col < name_end {
                return Some((name_start, name_end));
            }
            return None;
        }
    }
    None
}

fn tag_name_at_span(line: &str, start: usize, end: usize) -> String {
    line.chars()
        .enumerate()
        .filter(|(i, _)| *i >= start && *i < end)
        .map(|(_, c)| c)
        .collect()
}

/// `base.member` when the cursor is on `member` (same line, character-index column as `word_at_position`).
fn split_property_access(line: &str, col: usize) -> Option<(String, String)> {
    let chars: Vec<char> = line.chars().collect();
    if chars.is_empty() {
        return None;
    }
    let col = col.min(chars.len().saturating_sub(1));
    if !is_ident_char(chars[col]) {
        return None;
    }
    let mut end = col;
    while end + 1 < chars.len() && is_ident_char(chars[end + 1]) {
        end += 1;
    }
    let mut start = col;
    while start > 0 && is_ident_char(chars[start - 1]) {
        start -= 1;
    }
    if start == 0 {
        return None;
    }
    if chars[start - 1] != '.' {
        return None;
    }
    let member: String = chars[start..=end].iter().collect();
    let mut k = start - 2;
    while k > 0 && is_ident_char(chars[k - 1]) {
        k -= 1;
    }
    let base: String = chars[k..start - 1].iter().collect();
    if base.is_empty() {
        return None;
    }
    Some((base, member))
}

fn lookup_dotted(base: &str, member: &str) -> Option<BuiltinDef> {
    let key = format!("{base}.{member}");
    source_map().get(&key).cloned()
}

fn lookup_global(word: &str) -> Option<BuiltinDef> {
    source_map().get(word).cloned()
}

/// Built-in or JSX-intrinsic definition for the identifier at `position`, if any.
pub fn definition_for_builtin(
    text: &str,
    line: u32,
    character: u32,
    word: &str,
) -> Option<BuiltinDef> {
    let line_str = text.lines().nth(line as usize)?;
    let col = character as usize;

    if let Some((ns, ne)) = jsx_tag_name_char_span(line_str, col) {
        let tag = tag_name_at_span(line_str, ns, ne);
        if tag == word {
            if word == "Fragment" {
                return lookup_global("Fragment");
            }
            if HTML_INTRINSIC_TAGS.binary_search(&word).is_ok() {
                // Intrinsic tags share the same `ui_h` entry point as `div`.
                return lookup_global("div");
            }
        }
    }

    if let Some((base, member)) = split_property_access(line_str, col) {
        if let Some(d) = lookup_dotted(base.as_str(), member.as_str()) {
            return Some(d);
        }
    }

    lookup_global(word)
}

pub fn to_file_location(root: &Path, def: &BuiltinDef) -> Option<Location> {
    let path = root.join(&def.rel_path);
    if !path.is_file() {
        return None;
    }
    let uri = Url::from_file_path(&path).ok()?;
    let p = Position {
        line: def.line,
        character: def.character,
    };
    Some(Location {
        uri,
        range: Range { start: p, end: p },
    })
}

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

    #[test]
    fn pragma_map_loads_console_log() {
        let d = lookup_global("console").expect("console");
        assert!(d.rel_path.contains("eval.rs"));
        let m = lookup_dotted("console", "log").expect("console.log");
        assert!(m.rel_path.contains("natives.rs"));
    }

    #[test]
    fn jsx_div_maps_to_ui_h() {
        let text = "let x = <div />";
        let line = 0u32;
        let defn = definition_for_builtin(text, line, 9, "div").expect("div builtin");
        assert!(defn.rel_path.contains("runtime"));
        assert_eq!(defn.line, 40);
    }

    #[test]
    fn set_timeout_global() {
        let defn =
            definition_for_builtin("setTimeout(0, fn() {})\n", 0, 0, "setTimeout").expect("timer");
        assert_eq!(defn.rel_path, "crates/tish_eval/src/timers.rs");
    }

    #[test]
    fn console_log_qualified() {
        let text = "console.log(1)";
        let defn = definition_for_builtin(text, 0, 10, "log").expect("console.log");
        assert_eq!(defn.rel_path, "crates/tish_eval/src/natives.rs");
    }
}