knave 0.13.1

A Lisp programming language 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

use crate::core::BindingId;
use crate::source::Span;
use crate::types::{Type, TypedArgument, TypedExpr, TypedExprKind, TypedLambda, TypedLeaf};

pub struct WholeValueChanges;

impl WholeValueChanges {
    pub fn site(lambda: &TypedLambda, param: BindingId) -> Option<Span> {
        lambda
            .body
            .expressions()
            .find_map(|expr| Self::site_in(expr, param))
    }

    fn site_in(expr: &TypedExpr, param: BindingId) -> Option<Span> {
        if let TypedExprKind::Call(callee, arguments) = &expr.kind
            && Self::replaces(callee, arguments, param)
        {
            return Some(callee.span);
        }
        expr.children()
            .into_iter()
            .find_map(|child| Self::site_in(child, param))
    }

    fn replaces(callee: &TypedExpr, arguments: &[TypedArgument], param: BindingId) -> bool {
        let Type::Function(_, params, _) = &callee.ty else {
            return false;
        };
        let [Type::Mut(_, referent), replacement] = params.as_slice() else {
            return false;
        };
        referent.as_ref() == replacement
            && matches!(
                arguments.first().map(|argument| &argument.value.kind),
                Some(TypedExprKind::Leaf(TypedLeaf::Local(binding))) if *binding == param
            )
    }
}