rabex 0.2.0

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
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
81
82
83
84
85
86
87
88
89
90

use std::collections::HashMap;
use std::io::Cursor;

use anyhow::Result;
use rabex::files::serializedfile::build_common_offset_map;
use rabex::files::serializedfile::builder::SerializedFileBuilder;
use rabex::objects::{ClassId, ClassIdType, PPtr};
use rabex::typetree::typetree_cache::TypeTreeCache;
use serde_derive::Serialize;

fn main() -> Result<()> {
    let bundle_name = "new_bundle";
    let unity_version = "2022.2.2f1".parse().unwrap();
    let tpk = TypeTreeCache::embedded();
    let common_offset_map = build_common_offset_map(&tpk.inner, &unity_version);

    let mut sharedassets =
        SerializedFileBuilder::new(&unity_version, &tpk, &common_offset_map, true);

    let container = ["A", "B"]
        .iter()
        .map(|scene| {
            let path = format!("unity-scene-repacker/{scene}.unity");
            (path, AssetInfo::default())
        })
        .collect();

    sharedassets.add_object(&PreloadData {
        // TODO?
        /*m_Assets: vec![PPtr {
            m_FileID: 1.into(),
            m_PathID: 10001,
        }],*/
        ..Default::default()
    })?;

    sharedassets.add_object(&AssetBundle {
        m_Name: bundle_name.to_owned(),
        m_Container: container,
        m_MainAsset: AssetInfo::default(),
        m_RuntimeCompatibility: 1,
        m_IsStreamedSceneAssetBundle: true,
        m_PathFlags: 7,
        ..Default::default()
    })?;

    let mut out = Cursor::new(Vec::new());
    sharedassets.write(&mut out)?;

    Ok(())
}

#[derive(Debug, Serialize, Default)]
#[allow(non_snake_case)]
pub struct PreloadData {
    pub m_Name: String,
    pub m_Assets: Vec<PPtr>,
    pub m_Dependencies: Vec<String>,
    pub m_ExplicitDataLayout: bool,
}
impl ClassIdType for PreloadData {
    const CLASS_ID: ClassId = ClassId::PreloadData;
}

#[derive(Debug, Serialize, Default)]
#[allow(non_snake_case)]
pub struct AssetBundle {
    pub m_Name: String,
    pub m_PreloadTable: Vec<PPtr>,
    pub m_Container: HashMap<String, AssetInfo>,
    pub m_MainAsset: AssetInfo,
    pub m_RuntimeCompatibility: u32,
    pub m_AssetBundleName: String,
    pub m_Dependencies: Vec<String>,
    pub m_IsStreamedSceneAssetBundle: bool,
    pub m_ExplicitDataLayout: i32,
    pub m_PathFlags: i32,
    pub m_SceneHashes: HashMap<String, String>,
}
impl ClassIdType for AssetBundle {
    const CLASS_ID: ClassId = ClassId::AssetBundle;
}

#[derive(Debug, Serialize, Default)]
#[allow(non_snake_case)]
pub struct AssetInfo {
    pub preloadIndex: i32,
    pub preloadSize: i32,
    pub asset: PPtr,
}