syster-base 0.1.12-alpha

Core library for SysML v2 and KerML parsing, AST, and semantic analysis
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

//! Folding range extraction for SysML files

use crate::semantic::types::FoldingRangeInfo;
use crate::syntax::sysml::ast::{DefinitionMember, Element, SysMLFile, UsageMember};

/// Extract all foldable ranges from a SysML file
pub fn extract_folding_ranges(file: &SysMLFile) -> Vec<FoldingRangeInfo> {
    let mut ranges = Vec::new();

    for element in &file.elements {
        collect_ranges(element, &mut ranges);
    }

    // Keep only multiline ranges and sort by start line
    ranges.retain(|r| r.span.end.line > r.span.start.line);
    ranges.sort_by_key(|r| r.span.start.line);
    ranges
}

/// Recursively collect folding ranges from an element and its children
fn collect_ranges(element: &Element, ranges: &mut Vec<FoldingRangeInfo>) {
    match element {
        Element::Package(p) => {
            if let Some(span) = &p.span {
                ranges.push(FoldingRangeInfo::code(*span));
            }
            for child in &p.elements {
                collect_ranges(child, ranges);
            }
        }
        Element::Definition(d) => {
            if let Some(span) = &d.span {
                ranges.push(FoldingRangeInfo::code(*span));
            }
            for member in &d.body {
                match member {
                    DefinitionMember::Usage(u) => {
                        collect_ranges(&Element::Usage((**u).clone()), ranges)
                    }
                    DefinitionMember::Comment(c) => {
                        collect_ranges(&Element::Comment((**c).clone()), ranges)
                    }
                    DefinitionMember::Import(_) => {
                        // Imports are single-line, no folding needed
                    }
                }
            }
        }
        Element::Usage(u) => {
            if let Some(span) = &u.span {
                ranges.push(FoldingRangeInfo::code(*span));
            }
            for member in &u.body {
                match member {
                    UsageMember::Usage(u) => collect_ranges(&Element::Usage((**u).clone()), ranges),
                    UsageMember::Comment(c) => collect_ranges(&Element::Comment(c.clone()), ranges),
                }
            }
        }
        Element::Comment(c) => {
            if let Some(span) = &c.span {
                ranges.push(FoldingRangeInfo::comment(*span));
            }
        }
        Element::Import(_) | Element::Alias(_) | Element::Dependency(_) | Element::Filter(_) => {}
    }
}