knave 0.11.6

A small Lisp for humans and agents that infers types and ownership and compiles to readable Rust
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

use rustc_hash::{FxHashMap, FxHashSet};

use super::mangle::NameMangler;
use crate::core::BindingId;

pub struct LocalNames {
    used: FxHashSet<String>,
    assigned: FxHashMap<BindingId, String>,
    scopes: Vec<Vec<String>>,
}

impl LocalNames {
    const FIRST_SUFFIX: usize = 2;

    pub fn new(reserved: impl IntoIterator<Item = String>) -> Self {
        Self {
            used: reserved.into_iter().collect(),
            assigned: FxHashMap::default(),
            scopes: Vec::new(),
        }
    }

    pub fn enter(&mut self) {
        self.scopes.push(Vec::new());
    }

    pub fn leave(&mut self) {
        for name in self.scopes.pop().unwrap_or_default() {
            self.used.remove(&name);
        }
    }

    pub fn bind(&mut self, binding: BindingId, name: &str) -> String {
        let base = NameMangler::local(name);
        let mut candidate = base.clone();
        let mut suffix = Self::FIRST_SUFFIX;
        while self.used.contains(&candidate) {
            candidate = format!("{base}_{suffix}");
            suffix += 1;
        }
        self.used.insert(candidate.clone());
        if let Some(scope) = self.scopes.last_mut() {
            scope.push(candidate.clone());
        }
        self.assigned.insert(binding, candidate.clone());
        candidate
    }

    pub fn bind_exact(&mut self, binding: BindingId, name: &str) {
        self.used.insert(name.to_string());
        if let Some(scope) = self.scopes.last_mut() {
            scope.push(name.to_string());
        }
        self.assigned.insert(binding, name.to_string());
    }

    pub fn reuse(&mut self, binding: BindingId, existing: BindingId) {
        if let Some(name) = self.assigned.get(&existing).cloned() {
            self.assigned.insert(binding, name);
        }
    }

    pub fn bind_receiver(&mut self, binding: BindingId) -> String {
        let name = "self".to_string();
        self.used.insert(name.clone());
        if let Some(scope) = self.scopes.last_mut() {
            scope.push(name.clone());
        }
        self.assigned.insert(binding, name.clone());
        name
    }

    pub fn temporary(&mut self, base: &str) -> String {
        let mut candidate = base.to_string();
        let mut suffix = Self::FIRST_SUFFIX;
        while self.used.contains(&candidate) {
            candidate = format!("{base}_{suffix}");
            suffix += 1;
        }
        self.used.insert(candidate.clone());
        if let Some(scope) = self.scopes.last_mut() {
            scope.push(candidate.clone());
        }
        candidate
    }

    pub fn get(&self, binding: BindingId) -> Option<&str> {
        self.assigned.get(&binding).map(String::as_str)
    }

    pub fn pairs(&self) -> impl Iterator<Item = (BindingId, &str)> {
        self.assigned
            .iter()
            .map(|(binding, name)| (*binding, name.as_str()))
    }
}