gear-mesh 0.1.0

Next-generation Rust to TypeScript type definition sharing library
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

use gear_mesh_core::{GearMeshType, TypeRef};
use std::collections::HashSet;

/// Extract all custom type names referenced by a type
pub fn extract_type_dependencies(ty: &GearMeshType) -> HashSet<String> {
    let mut deps = HashSet::new();

    match &ty.kind {
        gear_mesh_core::TypeKind::Struct(s) => {
            for field in &s.fields {
                collect_type_refs(&field.ty, &mut deps);
            }
        }
        gear_mesh_core::TypeKind::Enum(e) => {
            for variant in &e.variants {
                match &variant.content {
                    gear_mesh_core::VariantContent::Tuple(types) => {
                        for ty_ref in types {
                            collect_type_refs(ty_ref, &mut deps);
                        }
                    }
                    gear_mesh_core::VariantContent::Struct(fields) => {
                        for field in fields {
                            collect_type_refs(&field.ty, &mut deps);
                        }
                    }
                    gear_mesh_core::VariantContent::Unit => {}
                }
            }
        }
        gear_mesh_core::TypeKind::Newtype(n) => {
            collect_type_refs(&n.inner, &mut deps);
        }
        _ => {}
    }

    deps
}

fn collect_type_refs(ty_ref: &TypeRef, deps: &mut HashSet<String>) {
    // Check if this is a container type (Vec, Option, etc.)
    let is_container = matches!(
        ty_ref.name.as_str(),
        "Vec" | "Option" | "Result" | "HashMap" | "HashSet" | "Box" | "Arc" | "Rc" | "Cow"
    );

    // If it's not a container and not a primitive, add it to dependencies
    if !is_container && !is_primitive(&ty_ref.name) {
        deps.insert(ty_ref.name.clone());
    }

    // Always recursively collect from generics (even for containers)
    for generic in &ty_ref.generics {
        collect_type_refs(generic, deps);
    }
}

fn is_primitive(type_name: &str) -> bool {
    matches!(
        type_name,
        "String"
            | "str"
            | "bool"
            | "char"
            | "i8"
            | "i16"
            | "i32"
            | "i64"
            | "i128"
            | "isize"
            | "u8"
            | "u16"
            | "u32"
            | "u64"
            | "u128"
            | "usize"
            | "f32"
            | "f64"
    )
}