luaur-analysis 0.1.3

Luau type checker and type inference (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

use crate::functions::follow_type::follow_type_id;
use crate::functions::follow_type_pack::follow_type_pack_id;
use crate::records::type_function_instance_type::TypeFunctionInstanceType;
use crate::records::type_function_instance_type_pack::TypeFunctionInstanceTypePack;
use crate::type_aliases::type_id::TypeId;
use crate::type_aliases::type_pack_id::TypePackId;

/// C++ `template<typename T> bool areEquivalent(const T& a, const T& b)`.
pub trait TypeFunctionInstanceEquivalent {
    fn same_function(&self, other: &Self) -> bool;
    fn type_arguments(&self) -> &[TypeId];
    fn pack_arguments(&self) -> &[TypePackId];
}

impl TypeFunctionInstanceEquivalent for TypeFunctionInstanceType {
    fn same_function(&self, other: &Self) -> bool {
        self.function == other.function
    }

    fn type_arguments(&self) -> &[TypeId] {
        &self.type_arguments
    }

    fn pack_arguments(&self) -> &[TypePackId] {
        &self.pack_arguments
    }
}

impl TypeFunctionInstanceEquivalent for TypeFunctionInstanceTypePack {
    fn same_function(&self, other: &Self) -> bool {
        self.function == other.function
    }

    fn type_arguments(&self) -> &[TypeId] {
        &self.typeArguments
    }

    fn pack_arguments(&self) -> &[TypePackId] {
        &self.packArguments
    }
}

pub fn are_equivalent<T: TypeFunctionInstanceEquivalent>(a: &T, b: &T) -> bool {
    if !a.same_function(b) {
        return false;
    }

    if a.type_arguments().len() != b.type_arguments().len()
        || a.pack_arguments().len() != b.pack_arguments().len()
    {
        return false;
    }

    for (a_arg, b_arg) in a.type_arguments().iter().zip(b.type_arguments().iter()) {
        if unsafe { follow_type_id(*a_arg) } != unsafe { follow_type_id(*b_arg) } {
            return false;
        }
    }

    for (a_arg, b_arg) in a.pack_arguments().iter().zip(b.pack_arguments().iter()) {
        if unsafe { follow_type_pack_id(*a_arg) } != unsafe { follow_type_pack_id(*b_arg) } {
            return false;
        }
    }

    true
}