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

use once_cell::sync::Lazy;
use std::borrow::Cow;

use bevy_ecs::system::System;
use bevy_render::color::Color;
use bevy_utils::HashMap;

static CRATE_COLORS: Lazy<HashMap<&str, &str>> = Lazy::new(|| {
    [
        // Beige/Red
        ("bevy_transform", "FFE7B9"),
        ("bevy_animation", "FFBDB9"),
        // Greys
        ("bevy_asset", "D1CBC5"),
        ("bevy_scene", "BACFCB"),
        ("bevy_time", "C7DDBD"),
        // Greens
        ("bevy_core", "3E583C"),
        ("bevy_app", "639D18"),
        ("bevy_ecs", "B0D34A"),
        ("bevy_hierarchy", "E4FBA3"),
        // Turquesa
        ("bevy_audio", "98F1D1"),
        // Purples/Pinks
        ("bevy_winit", "664F72"),
        ("bevy_a11y", "9163A6"),
        ("bevy_window", "BB85D4"),
        ("bevy_text", "E9BBFF"),
        ("bevy_gilrs", "973977"),
        ("bevy_input", "D36AAF"),
        ("bevy_ui", "FFB1E5"),
        // Blues
        ("bevy_render", "70B9FC"),
        ("bevy_pbr", "ABD5FC"),
    ]
    .into_iter()
    .collect()
});

pub struct SystemStyle {
    pub bg_color: Color,
    pub text_color: Option<Color>,
    pub border_color: Option<Color>,
    pub border_width: f32,
}

pub fn color_to_hex(color: Color) -> String {
    format!(
        "#{:0>2x}{:0>2x}{:0>2x}",
        (color.r() * 255.0) as u8,
        (color.g() * 255.0) as u8,
        (color.b() * 255.0) as u8,
    )
}

pub fn system_to_style(system: &dyn System<In = (), Out = ()>) -> SystemStyle {
    let name = system.name();
    let pretty_name: Cow<str> = pretty_type_name::pretty_type_name_str(&name).into();
    let is_apply_system_buffers = pretty_name == "apply_system_buffers";
    let name_without_event = name
        .trim_start_matches("bevy_ecs::event::Events<")
        .trim_end_matches(">::update_system");
    let crate_name = name_without_event.split("::").next();

    if is_apply_system_buffers {
        SystemStyle {
            bg_color: Color::hex("E70000").unwrap(),
            text_color: Some(Color::hex("ffffff").unwrap()),
            border_color: Some(Color::hex("5A0000").unwrap()),
            border_width: 2.0,
        }
    } else {
        let bg_color = crate_name
            .and_then(|n| CRATE_COLORS.get(n))
            .map(Color::hex)
            .unwrap_or(Color::hex("eff1f3"))
            .unwrap();

        SystemStyle {
            bg_color,
            text_color: None,
            border_color: None,
            border_width: 1.0,
        }
    }
}