windjammer 0.48.0

A simple language inspired by Go, Ruby, and Elixir that transpiles to Rust - 80% of Rust's power with 20% of the complexity
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150

//! Stdlib method heuristics for automatic `&` insertion when no analyzer signature exists.

use crate::parser::{Expression, OwnershipHint, Type};

use super::{MethodCallAnalyzer, MethodCallContext};

impl MethodCallAnalyzer {
    /// Check if this method call needs & based on stdlib patterns
    pub(super) fn needs_stdlib_ref(
        method: &str,
        arg: &Expression,
        ctx: &MethodCallContext<'_, '_>,
        arg_count: usize,
        receiver_type_name: Option<&str>,
        local_var_types: Option<&std::collections::HashMap<String, Type>>,
    ) -> bool {
        let usize_variables = ctx.usize_variables;
        let current_function_params = ctx.current_function_params;
        let borrowed_iterator_vars = ctx.borrowed_iterator_vars;
        let inferred_borrowed_params = ctx.inferred_borrowed_params;

        let arg_is_already_borrowed = if let Expression::Identifier { name, .. } = arg {
            let is_ref_param = current_function_params.iter().any(|p| {
                &p.name == name && matches!(p.ownership, OwnershipHint::Ref | OwnershipHint::Mut)
            });
            let is_borrowed_iter_var = borrowed_iterator_vars.contains(name);
            let param_is_ref = current_function_params.iter().any(|p| {
                p.name == *name
                    && crate::codegen::rust::types::param_generates_as_rust_ref(
                        &p.type_,
                        &p.name,
                        inferred_borrowed_params,
                    )
            });
            let is_str_ref_optimized = ctx.str_ref_optimized_params.contains(name.as_str());

            is_ref_param || is_borrowed_iter_var || param_is_ref || is_str_ref_optimized
        } else {
            false
        };

        if arg_is_already_borrowed {
            return false;
        }

        if arg_count > 1 && super::super::stdlib_method_traits::is_map_key_method(method) {
            return false;
        }

        if super::super::stdlib_method_traits::is_map_key_method(method) {
            let is_known_map = receiver_type_name.is_some_and(|n| {
                let base = n.split('<').next().unwrap_or(n);
                matches!(base, "HashMap" | "BTreeMap" | "IndexMap")
            });
            let is_known_vec =
                receiver_type_name.is_some_and(|n| n.split('<').next().unwrap_or(n) == "Vec");

            if is_known_vec {
                return false;
            }
            if is_known_map {
                if let Expression::Identifier { name, .. } = arg {
                    let is_already_ref = current_function_params.iter().any(|p| {
                        p.name == *name
                            && crate::codegen::rust::types::param_generates_as_rust_ref(
                                &p.type_,
                                &p.name,
                                inferred_borrowed_params,
                            )
                    });
                    if is_already_ref {
                        return false;
                    }
                }
                return true;
            }

            if super::super::stdlib_method_traits::is_map_key_method(method)
                && Self::is_copy_type_with_locals(
                    arg,
                    usize_variables,
                    current_function_params,
                    local_var_types,
                )
            {
                let arg_name = if let Expression::Identifier { name, .. } = arg {
                    Some(name.as_str())
                } else {
                    None
                };

                let looks_like_hashmap_key = arg_name.is_some_and(|name| {
                    name == "id"
                        || name == "key"
                        || name == "entity"
                        || name.ends_with("_id")
                        || name.ends_with("_key")
                });

                if looks_like_hashmap_key {
                    if let Some(name) = arg_name {
                        let is_already_map_key_ref = current_function_params.iter().any(|p| {
                            p.name == name
                                && crate::codegen::rust::types::param_generates_as_rust_ref(
                                    &p.type_,
                                    &p.name,
                                    inferred_borrowed_params,
                                )
                        });
                        if is_already_map_key_ref {
                            return false;
                        }
                    }
                    return true;
                }

                return false;
            }

            if super::super::stdlib_method_traits::is_map_key_method(method) {
                if let Expression::Cast { type_, .. } = arg {
                    if Self::is_copy_type_annotation_internal(type_) {
                        return false;
                    }
                }
            }

            return true;
        }

        if Self::is_copy_type_with_locals(
            arg,
            usize_variables,
            current_function_params,
            local_var_types,
        ) {
            return false;
        }

        if matches!(method, "contains" | "binary_search") {
            return true;
        }

        if matches!(method, "contains" | "starts_with" | "ends_with") {
            return true;
        }

        false
    }
}