knave 0.10.3

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

use rustc_hash::FxHashSet;

use crate::core::{BindingId, ReferenceKind};
use crate::types::{Borrow, Type, TypedExpr, TypedExprKind, TypedProgram, TypedStructExpr};

pub struct InPlaceChanges;

impl InPlaceChanges {
    pub fn of(program: &TypedProgram) -> FxHashSet<BindingId> {
        let mut changed = FxHashSet::default();
        for definition in program
            .modules
            .iter()
            .flat_map(|module| &module.definitions)
        {
            Self::written_mut_types(&definition.scheme.body, &mut changed);
            Self::expression(&definition.value, &mut changed);
        }
        changed.retain(|binding| {
            program.structs.get(*binding).is_some() || program.enums.get(*binding).is_some()
        });
        changed
    }

    fn expression(expr: &TypedExpr, changed: &mut FxHashSet<BindingId>) {
        Self::written_mut_types(&expr.ty, changed);
        for binder in expr.binders() {
            Self::written_mut_types(&binder.ty, changed);
        }
        if let TypedExprKind::Call(_, arguments) | TypedExprKind::Recur(arguments) = &expr.kind {
            for argument in arguments {
                if matches!(argument.borrow, Borrow::Mutable | Borrow::Reborrow) {
                    Self::place(&argument.value, changed);
                }
            }
        }
        if let TypedExprKind::Borrow(ReferenceKind::Mutable, value) = &expr.kind {
            Self::place(value, changed);
        }
        for child in expr.children() {
            Self::expression(child, changed);
        }
    }

    fn place(place: &TypedExpr, changed: &mut FxHashSet<BindingId>) {
        match &place.kind {
            TypedExprKind::Struct(TypedStructExpr::Field(owner, _)) => {
                Self::changed_type(&owner.ty, changed);
                Self::place(owner, changed);
            }
            _ => Self::changed_type(&place.ty, changed),
        }
    }

    fn written_mut_types(ty: &Type, changed: &mut FxHashSet<BindingId>) {
        if let Type::Mut(_, inner) = ty {
            Self::changed_type(inner, changed);
        }
        for part in ty.parts() {
            Self::written_mut_types(part, changed);
        }
    }

    fn changed_type(ty: &Type, changed: &mut FxHashSet<BindingId>) {
        let mut resolved = ty;
        while let Type::Ref(_, inner) | Type::Mut(_, inner) = resolved {
            resolved = inner;
        }
        if let Type::Named(binding, _) = resolved {
            changed.insert(*binding);
        }
    }
}