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

//! Node: `cxx:Function:Luau.Analysis:Analysis/src/ToString.cpp:1397:assign_cycle_names`
//! Source: `Analysis/src/ToString.cpp:1397-1440` (hand-ported)

use crate::functions::follow_type::follow;
use crate::functions::get_type_alt_j::get;
use crate::records::table_type::TableType;
use crate::type_aliases::type_id::TypeId;
use crate::type_aliases::type_pack_id::TypePackId;
use alloc::collections::BTreeSet;
use alloc::format;
use alloc::string::String;
use luaur_common::records::dense_hash_map::DenseHashMap;

/// C++ `static void assignCycleNames(...)`.
pub fn assign_cycle_names(
    cycles: &BTreeSet<TypeId>,
    cycle_tps: &BTreeSet<TypePackId>,
    cycle_names: &mut DenseHashMap<TypeId, String>,
    cycle_tp_names: &mut DenseHashMap<TypePackId, String>,
    exhaustive: bool,
) {
    unsafe {
        let mut next_index = 1;

        for &cycle_ty in cycles.iter() {
            // TODO: use the stringified type list if there are no cycles
            let ttv = get::<TableType>(follow(cycle_ty));
            if !exhaustive
                && !ttv.is_null()
                && ((*ttv).synthetic_name.is_some() || (*ttv).name.is_some())
            {
                // If we have a cycle type in type parameters, assign a cycle name for this named table
                if (*ttv)
                    .instantiated_type_params
                    .iter()
                    .any(|&el| cycles.contains(&follow(el)))
                {
                    *cycle_names.get_or_insert(cycle_ty) = match &(*ttv).name {
                        Some(name) => name.clone(),
                        None => (*ttv).synthetic_name.clone().unwrap(),
                    };
                }

                continue;
            }

            let name = format!("t{}", next_index);
            next_index += 1;

            *cycle_names.get_or_insert(cycle_ty) = name;
        }

        for &tp in cycle_tps.iter() {
            let name = format!("tp{}", next_index);
            next_index += 1;
            *cycle_tp_names.get_or_insert(tp) = name;
        }
    }
}