lune/packages/lib-roblox/src/shared/instance.rs

use std::borrow::{Borrow, Cow};

use rbx_dom_weak::types::VariantType as DomType;
use rbx_reflection::{ClassTag, DataType};

/**
    Checks if the given property is an enum.

    Returns `None` if the class or property does not exist.
*/
pub fn property_is_enum(
    instance_class: impl AsRef<str>,
    property_name: impl AsRef<str>,
) -> Option<bool> {
    let db = rbx_reflection_database::get();
    let class = db.classes.get(instance_class.as_ref())?;
    let prop = class.properties.get(property_name.as_ref())?;

    Some(matches!(prop.data_type, DataType::Enum(_)))
}

/**
    Finds the type of a property of the given class.

    Returns `None` if the class or property does not exist or if the property is an enum.
*/
pub fn find_property_type(
    instance_class: impl AsRef<str>,
    property_name: impl AsRef<str>,
) -> Option<DomType> {
    let db = rbx_reflection_database::get();
    let class = db.classes.get(instance_class.as_ref())?;
    let prop = class.properties.get(property_name.as_ref())?;

    if let DataType::Value(typ) = prop.data_type {
        Some(typ)
    } else {
        None
    }
}

/**
    Finds the enum name of a property of the given class.

    Returns `None` if the class or property does not exist or if the property is *not* an enum.
*/
pub fn find_property_enum(
    instance_class: impl AsRef<str>,
    property_name: impl AsRef<str>,
) -> Option<Cow<'static, str>> {
    let db = rbx_reflection_database::get();
    let class = db.classes.get(instance_class.as_ref())?;
    let prop = class.properties.get(property_name.as_ref())?;

    if let DataType::Enum(name) = &prop.data_type {
        Some(Cow::Borrowed(name))
    } else {
        None
    }
}

/**
    Checks if an instance class exists in the reflection database.
*/
pub fn class_exists(class_name: impl AsRef<str>) -> bool {
    let db = rbx_reflection_database::get();
    db.classes.contains_key(class_name.as_ref())
}

/**
    Checks if an instance class matches a given class or superclass, similar to
    [Instance::IsA](https://create.roblox.com/docs/reference/engine/classes/Instance#IsA)
    from the Roblox standard library.

    Note that this function may return `None` if it encounters a class or superclass
    that does not exist in the currently known class reflection database.
*/
pub fn class_is_a(instance_class: impl AsRef<str>, class_name: impl AsRef<str>) -> Option<bool> {
    let mut instance_class = instance_class.as_ref();
    let class_name = class_name.as_ref();

    if class_name == "Instance" || instance_class == class_name {
        Some(true)
    } else {
        let db = rbx_reflection_database::get();

        while instance_class != class_name {
            let class_descriptor = db.classes.get(instance_class)?;
            if let Some(sup) = &class_descriptor.superclass {
                instance_class = sup.borrow();
            } else {
                return Some(false);
            }
        }

        Some(true)
    }
}

/**
    Checks if an instance class is a service.

    This is separate from [`class_is_a`] since services do not share a
    common base class, and are instead determined through reflection tags.

    Note that this function may return `None` if it encounters a class or superclass
    that does not exist in the currently known class reflection database.
*/
pub fn class_is_a_service(instance_class: impl AsRef<str>) -> Option<bool> {
    let mut instance_class = instance_class.as_ref();

    let db = rbx_reflection_database::get();

    loop {
        let class_descriptor = db.classes.get(instance_class)?;
        if class_descriptor.tags.contains(&ClassTag::Service) {
            return Some(true);
        } else if let Some(sup) = &class_descriptor.superclass {
            instance_class = sup.borrow();
        } else {
            break;
        }
    }

    Some(false)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn is_a_class_valid() {
        assert_eq!(class_is_a("Part", "Part"), Some(true));
        assert_eq!(class_is_a("Part", "BasePart"), Some(true));
        assert_eq!(class_is_a("Part", "PVInstance"), Some(true));
        assert_eq!(class_is_a("Part", "Instance"), Some(true));

        assert_eq!(class_is_a("Workspace", "Workspace"), Some(true));
        assert_eq!(class_is_a("Workspace", "Model"), Some(true));
        assert_eq!(class_is_a("Workspace", "Instance"), Some(true));
    }

    #[test]
    fn is_a_class_invalid() {
        assert_eq!(class_is_a("Part", "part"), Some(false));
        assert_eq!(class_is_a("Part", "Base-Part"), Some(false));
        assert_eq!(class_is_a("Part", "Model"), Some(false));
        assert_eq!(class_is_a("Part", "Paart"), Some(false));

        assert_eq!(class_is_a("Workspace", "Service"), Some(false));
        assert_eq!(class_is_a("Workspace", "."), Some(false));
        assert_eq!(class_is_a("Workspace", ""), Some(false));
    }

    #[test]
    fn is_a_service_valid() {
        assert_eq!(class_is_a_service("Workspace"), Some(true));
        assert_eq!(class_is_a_service("PhysicsService"), Some(true));
        assert_eq!(class_is_a_service("ReplicatedFirst"), Some(true));
        assert_eq!(class_is_a_service("CSGDictionaryService"), Some(true));
    }

    #[test]
    fn is_a_service_invalid() {
        assert_eq!(class_is_a_service("Camera"), Some(false));
        assert_eq!(class_is_a_service("Terrain"), Some(false));
        assert_eq!(class_is_a_service("Work-space"), None);
        assert_eq!(class_is_a_service("CSG Dictionary Service"), None);
    }
}