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

use crate::functions::follow_type::follow_type_id;
use crate::functions::get_type_alt_j::get_type_id;
use crate::records::union_type::UnionType;
use crate::type_aliases::type_id::TypeId;
use crate::type_aliases::type_id_predicate::TypeIdPredicate;
use alloc::vec::Vec;
use std::collections::HashSet;

pub fn filter_map(type_: TypeId, predicate: TypeIdPredicate) -> Vec<TypeId> {
    let type_ = unsafe { follow_type_id(type_) };

    unsafe {
        if !get_type_id::<UnionType>(type_).is_null() {
            let utv = get_type_id::<UnionType>(type_);
            let mut options: HashSet<TypeId> = HashSet::new();

            for &option in (*utv).options.iter() {
                let followed_option = follow_type_id(option);
                if let Some(out) = predicate(followed_option) {
                    options.insert(out);
                }
            }

            options.into_iter().collect()
        } else if let Some(out) = predicate(type_) {
            vec![out]
        } else {
            Vec::new()
        }
    }
}