lean-ctx-syntax 0.1.0

Tree-sitter based code structure for LeanCTX | OSS: signatures, deep AST queries, import resolution and the post-edit syntax gate.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

// SPDX-License-Identifier: Apache-2.0

use super::extract::extract_signatures_ts;
use crate::signatures::Signature;

pub(crate) fn extract_sfc_signatures(content: &str) -> Option<Vec<Signature>> {
    let (script_content, line_offset) = extract_script_block_with_offset(content)?;
    let is_ts = content.contains("lang=\"ts\"") || content.contains("lang=\"typescript\"");
    let ext = if is_ts { "ts" } else { "js" };
    let mut sigs = extract_signatures_ts(&script_content, ext)?;
    // Spans are relative to the extracted <script> block; shift them back to
    // file-absolute lines so navigation modes point at the real source line.
    for sig in &mut sigs {
        sig.start_line = sig.start_line.map(|line| line + line_offset);
        sig.end_line = sig.end_line.map(|line| line + line_offset);
    }
    Some(sigs)
}

pub fn extract_script_block(content: &str) -> Option<String> {
    extract_script_block_with_offset(content).map(|(script, _)| script)
}

fn extract_script_block_with_offset(content: &str) -> Option<(String, usize)> {
    let lower = content.to_lowercase();
    let start_tag_pos = lower.find("<script")?;
    let tag_end = content[start_tag_pos..].find('>')? + start_tag_pos + 1;
    let end_tag = "</script>";
    let end_pos = lower[tag_end..].find(end_tag)? + tag_end;
    let script = &content[tag_end..end_pos];
    if script.trim().is_empty() {
        return None;
    }
    let line_offset = content[..tag_end].bytes().filter(|b| *b == b'\n').count();
    Some((script.to_string(), line_offset))
}