rusteal 0.8.0

Rust bindings for Unreal Engine 5.8.2
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

use bindings::tool_menus::{FToolMenuContext, ToolMenuEntryScript};
use bindings::unreal_ed::{EditorAssetSubsystem, EditorAssetSubsystemExt};
use rusteal_runtime::runtime::{UStructRef, console, task};
use rusteal_runtime::{uclass, uclass_impl};

use crate::editor_subsystem;

#[uclass(parent = ToolMenuEntryScript)]
pub struct {{ plugin }}MenuEntry {}

#[uclass_impl]
impl {{ plugin }}MenuEntry {
    #[ufunction(Override)]
    fn execute(&mut self, _context: UStructRef<FToolMenuContext>) {
        count_assets("/Game");
    }
}

pub fn count_assets(path: &str) {
    let assets = match editor_subsystem::<EditorAssetSubsystem>().and_then(|s| s.checked()) {
        Ok(subsystem) => subsystem.list_assets(path, Some(true), Some(false)),
        Err(e) => {
            log::error!("[{{ plugin }}] no asset subsystem: {e:?}");
            return;
        }
    };

    let path = path.to_string();

    task::spawn(
        move || {
            let mut by_prefix = std::collections::BTreeMap::<String, usize>::new();

            for asset in &assets {
                let name = asset.rsplit('/').next().unwrap_or(asset);
                let prefix = name.split_once('_').map_or("(no prefix)", |(p, _)| p);
                *by_prefix.entry(prefix.to_string()).or_default() += 1;
            }

            (assets.len(), by_prefix)
        },
        move |(total, by_prefix)| {
            log::info!("[{{ plugin }}] {total} assets under {path}");

            for (prefix, count) in by_prefix {
                log::info!("[{{ plugin }}]   {prefix}: {count}");
            }
        },
    );
}

fn startup() {
    let result = console::register_command(
        "{{ crate_name | replace(from="-", to="_") }}.count_assets",
        "Count the assets under a path (default /Game) by name prefix",
        |args, _| count_assets(args.first().copied().unwrap_or("/Game")),
    );

    if let Err(e) = result {
        log::error!("[{{ plugin }}] no console command: {e:?}");
    }
}
rusteal_runtime::on_load!(startup);