rabex 0.1.1

Tools for parsing and writing unity files, bundles and typetrees
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

use std::borrow::Cow;

use crate::UnityVersion;
use crate::objects::ClassId;
use crate::tpk::TpkTypeTreeBlob;

use super::TypeTreeNode;

/// Interface for retreiving a type tree for a given [`ClassId`].
///
/// It is implemented by [`TpkTypeTreeBlob`](crate::tpk::TpkTypeTreeBlob),
/// but you should consider using the [`TypeTreeCache`](crate::typetree::typetree_cache::TypeTreeCache)
/// or it's thread-safe variant [`sync::TypeTreeCache`](crate::typetree::typetree_cache::sync::TypeTreeCache)
/// for performance reasons.
pub trait TypeTreeProvider {
    fn get_typetree_node(
        &self,
        class_id: ClassId,
        target_version: &UnityVersion,
    ) -> Option<Cow<'_, TypeTreeNode>>;
}

impl TypeTreeProvider for TpkTypeTreeBlob {
    /// Look up and construct a [`TypeTreeNode`] for the class and unity version.
    /// If you call this method often, consider using [`TpkTypeTreeCache`](crate::typetree::typetree_cache::TypeTreeCache).
    fn get_typetree_node(
        &self,
        class_id: ClassId,
        target_version: &UnityVersion,
    ) -> Option<Cow<'_, TypeTreeNode>> {
        TpkTypeTreeBlob::get_typetree_node(self, class_id, target_version).map(Cow::Owned)
    }
}
impl<T: TypeTreeProvider> TypeTreeProvider for &T {
    fn get_typetree_node(
        &self,
        class_id: ClassId,
        target_version: &UnityVersion,
    ) -> Option<Cow<'_, TypeTreeNode>> {
        (*self).get_typetree_node(class_id, target_version)
    }
}

/// [`TypeTreeProvider`] returning no data.
pub struct NullTypeTreeProvider;

impl TypeTreeProvider for NullTypeTreeProvider {
    fn get_typetree_node(&self, _: ClassId, _: &UnityVersion) -> Option<Cow<'_, TypeTreeNode>> {
        None
    }
}