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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

//! C++ `struct ContainsAnyGeneric_DEPRECATED : TypeOnceVisitor`
//! (`Analysis/include/Luau/TypeUtils.h:404-420`, `Analysis/src/TypeUtils.cpp:947-981`).
//!
//! Faithful port: a `TypeOnceVisitor` (skipBoundTypes = true) that traverses a
//! type/type-pack and sets `found` if it encounters any `GenericType` /
//! `GenericTypePack`. Extern types are not descended into. The two static
//! `hasAnyGeneric` entry points construct the visitor, `traverse`, and return
//! `found`.
use crate::records::extern_type::ExternType;
use crate::records::generic_type::GenericType;
use crate::records::generic_type_pack::GenericTypePack;
use crate::records::generic_type_visitor::{GenericTypeVisitor, GenericTypeVisitorTrait};
use crate::records::type_once_visitor::TypeOnceVisitor;
use crate::type_aliases::type_id::TypeId;
use crate::type_aliases::type_pack_id::TypePackId;
use alloc::string::String;
use core::ffi::c_void;
use luaur_common::records::dense_hash_set::DenseHashSet;

#[derive(Debug, Clone)]
pub struct ContainsAnyGenericDeprecated {
    pub base: TypeOnceVisitor,
    pub found: bool,
}

impl ContainsAnyGenericDeprecated {
    pub fn new() -> Self {
        Self {
            // C++: TypeOnceVisitor("ContainsAnyGeneric", /* skipBoundTypes */ true)
            base: TypeOnceVisitor::new(String::from("ContainsAnyGeneric"), true),
            found: false,
        }
    }

    pub fn has_any_generic(ty: TypeId) -> bool {
        let mut visitor = Self::new();
        visitor.traverse_type_id(ty);
        visitor.found
    }

    pub fn has_any_generic_type_pack_id(tp: TypePackId) -> bool {
        let mut visitor = Self::new();
        visitor.traverse_type_pack_id(tp);
        visitor.found
    }
}

impl GenericTypeVisitorTrait for ContainsAnyGenericDeprecated {
    type Seen = DenseHashSet<*mut c_void>;

    fn visitor_base(&mut self) -> &mut GenericTypeVisitor<Self::Seen> {
        &mut self.base.base
    }

    // C++ `bool visit(TypeId ty)`: `found = found || is<GenericType>(ty); return !found;`
    fn visit_type_id(&mut self, _ty: TypeId) -> bool {
        !self.found
    }

    // C++ `bool visit(TypePackId ty)`: `found = found || is<GenericTypePack>(follow(ty)); return !found;`
    fn visit_type_pack_id(&mut self, _tp: TypePackId) -> bool {
        !self.found
    }

    // The `is<GenericType>` / `is<GenericTypePack>` checks of the bare C++
    // `visit` overloads are expressed here via the per-member dispatch hooks the
    // traversal reaches for generic types/packs.
    fn visit_type_id_generic_type(&mut self, _ty: TypeId, _gtv: &GenericType) -> bool {
        self.found = true;
        false
    }

    fn visit_type_pack_id_generic_type_pack(
        &mut self,
        _tp: TypePackId,
        _gtp: &GenericTypePack,
    ) -> bool {
        self.found = true;
        false
    }

    // C++ `bool visit(TypeId ty, const ExternType&) { return false; }`
    fn visit_type_id_extern_type(&mut self, _ty: TypeId, _etv: &ExternType) -> bool {
        false
    }
}