hara-native 0.1.21

HAL-free native host runtime and package launcher for Hara
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

use super::kind::UnitKind;
use crate::kernel::Form;
use std::collections::BTreeSet;

pub fn raw_provided_vars(form: &Form, module: &str) -> BTreeSet<String> {
    let mut output = BTreeSet::new();
    collect_provided_vars(form, module, &mut output);
    output
}

pub(crate) fn unit_kind(form: &Form) -> UnitKind {
    match list_head(without_metadata(form)) {
        Some("defmacro") => UnitKind::Macro,
        Some(
            "defprotocol"
                | "extend-type"
                | "defmulti"
                | "defmethod"
                | "defstruct"
                | "defmutable",
        ) => UnitKind::Registration,
        Some("def" | "defn" | "declare") => UnitKind::Definition,
        _ => UnitKind::Initializer,
    }
}

pub(super) fn provided_vars(form: &Form, module: &str) -> BTreeSet<String> {
    raw_provided_vars(form, module)
}

fn collect_provided_vars(form: &Form, module: &str, output: &mut BTreeSet<String>) {
    let Form::List(values) = without_metadata(form) else {
        return;
    };
    let Some(Form::Symbol(head)) = values.first() else {
        return;
    };
    match head.as_str() {
        "do" => {
            for value in values.iter().skip(1) {
                collect_provided_vars(value, module, output);
            }
        }
        "declare" => {
            for value in values.iter().skip(1) {
                if let Some(name) = binding_name(value) {
                    output.insert(qualify(module, name));
                }
            }
        }
        "def" | "defn" | "defmacro" | "defmulti" => {
            if let Some(name) = values.get(1).and_then(binding_name) {
                output.insert(qualify(module, name));
            }
        }
        "defstruct" | "defmutable" => {
            if let Some(name) = values.get(1).and_then(binding_name) {
                output.insert(qualify(module, name));
                output.insert(qualify(module, &format!("->{name}")));
                output.insert(qualify(module, &format!("map->{name}")));
            }
        }
        "defprotocol" => {
            if let Some(name) = values.get(1).and_then(binding_name) {
                output.insert(qualify(module, name));
            }
            for method in values.iter().skip(2) {
                let Form::List(parts) = without_metadata(method) else {
                    continue;
                };
                if let Some(Form::Symbol(name)) = parts.first() {
                    output.insert(qualify(module, name));
                }
            }
        }
        _ => {}
    }
}

pub(super) fn binding_name(form: &Form) -> Option<&str> {
    match without_metadata(form) {
        Form::Symbol(name) if !name.contains('/') => Some(name),
        _ => None,
    }
}

pub(super) fn list_head(form: &Form) -> Option<&str> {
    match form {
        Form::List(values) => match values.first() {
            Some(Form::Symbol(head)) => Some(head),
            _ => None,
        },
        _ => None,
    }
}

pub(super) fn qualify(module: &str, name: &str) -> String {
    if name.contains('/') {
        name.into()
    } else {
        format!("{module}/{name}")
    }
}

pub(super) fn without_metadata(form: &Form) -> &Form {
    match form {
        Form::Metadata(_, value) => without_metadata(value),
        value => value,
    }
}