bevy_dioxus/src/lib.rs

#![feature(trait_alias)]
mod apply_mutations;
pub mod colors;
mod deferred_system;
mod ecs_hooks;
pub mod elements;
#[macro_use]
mod events;
#[cfg(feature = "hot_reload")]
mod hot_reload;
mod parse_attributes;
mod tick;
mod use_state_sendable;

use self::{
    apply_mutations::BevyTemplate,
    deferred_system::DeferredSystemRunQueue,
    ecs_hooks::EcsSubscriptions,
    events::{generate_mouse_enter_leave_events, EventReaders, MouseEnter, MouseExit},
    tick::tick_dioxus_ui,
};
use bevy::{
    app::{App, Last, Plugin, PreUpdate},
    ecs::{bundle::Bundle, component::Component, entity::{Entity, EntityHashMap}, schedule::IntoSystemConfigs},
    prelude::Deref,
    ui::{node_bundles::NodeBundle, ui_focus_system},
    utils::HashMap,
};
use dioxus::core::{Element, ElementId, Scope, VirtualDom};

pub mod prelude {
    pub use super::deferred_system::use_system_scheduler;
    pub use super::ecs_hooks::{
        use_event_reader, use_query, use_query_filtered, use_resource, use_world,
    };
    pub use super::elements::dioxus_elements;
    pub use super::use_state_sendable::*;
    pub use super::{DioxusUiBundle, DioxusUiPlugin, DioxusUiRoot};
    pub use bevy_mod_picking::pointer::PointerButton;
    pub use dioxus;
    pub use dioxus::prelude::{Event as DioxusEvent, *};
}

pub struct DioxusUiPlugin;

impl Plugin for DioxusUiPlugin {
    fn build(&self, app: &mut App) {
        #[cfg(feature = "hot_reload")]
        dioxus_hot_reload::hot_reload_init!(dioxus_hot_reload::Config::<
            hot_reload::HotReloadContext,
        >::default());

        app.init_non_send_resource::<UiContext>()
            .init_resource::<DeferredSystemRunQueue>()
            .init_resource::<EventReaders>()
            .add_event::<MouseEnter>()
            .add_event::<MouseExit>()
            .add_systems(
                PreUpdate,
                generate_mouse_enter_leave_events.after(ui_focus_system),
            )
            .add_systems(Last, tick_dioxus_ui);
    }
}

#[derive(Bundle)]
pub struct DioxusUiBundle {
    pub dioxus_ui_root: DioxusUiRoot,
    pub node_bundle: NodeBundle,
}

#[derive(Component, Deref, Hash, PartialEq, Eq, Clone, Copy)]
pub struct DioxusUiRoot(pub fn(Scope) -> Element);

#[derive(Default)]
struct UiContext {
    roots: HashMap<(Entity, DioxusUiRoot), UiRoot>,
    subscriptions: EcsSubscriptions,
}

struct UiRoot {
    virtual_dom: VirtualDom,
    element_id_to_bevy_ui_entity: HashMap<ElementId, Entity>,
    bevy_ui_entity_to_element_id: EntityHashMap<ElementId>,
    templates: HashMap<String, BevyTemplate>,
    needs_rebuild: bool,
}

impl UiRoot {
    fn new(root_component: DioxusUiRoot) -> Self {
        Self {
            virtual_dom: VirtualDom::new(root_component.0),
            element_id_to_bevy_ui_entity: HashMap::new(),
            bevy_ui_entity_to_element_id: EntityHashMap::default(),
            templates: HashMap::new(),
            needs_rebuild: true,
        }
    }
}