bevy_cef_core 0.1.0

Core library for bevy_cef
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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

mod brp_handler;
mod js_emit_event_handler;

use crate::browser_process::ContextMenuHandlerBuilder;
use crate::prelude::IntoString;
use cef::rc::{Rc, RcImpl};
use cef::{
    Browser, Client, ContextMenuHandler, DisplayHandler, Frame, ImplClient, ImplProcessMessage,
    ListValue, ProcessId, ProcessMessage, RenderHandler, WrapClient, sys,
};
use std::os::raw::c_int;

pub use brp_handler::BrpHandler;
pub use js_emit_event_handler::{IpcEventRaw, JsEmitEventHandler};

pub trait ProcessMessageHandler {
    fn process_name(&self) -> &'static str;

    fn handle_message(&self, browser: &mut Browser, frame: &mut Frame, args: Option<ListValue>);
}

/// ## Reference
///
/// - [`CefBrowser Class Reference`](https://cef-builds.spotifycdn.com/docs/106.1/classCefBrowser.html)
pub struct ClientHandlerBuilder {
    object: *mut RcImpl<sys::cef_client_t, Self>,
    render_handler: RenderHandler,
    context_menu_handler: ContextMenuHandler,
    message_handlers: Vec<std::rc::Rc<dyn ProcessMessageHandler>>,
    display_handler: Option<DisplayHandler>,
}

impl ClientHandlerBuilder {
    pub fn new(render_handler: RenderHandler) -> Self {
        Self {
            object: std::ptr::null_mut(),
            render_handler,
            context_menu_handler: ContextMenuHandlerBuilder::build(),
            message_handlers: Vec::new(),
            display_handler: None,
        }
    }

    pub fn with_display_handler(mut self, display_handler: DisplayHandler) -> Self {
        self.display_handler = Some(display_handler);
        self
    }

    pub fn with_message_handler(mut self, handler: impl ProcessMessageHandler + 'static) -> Self {
        self.message_handlers.push(std::rc::Rc::new(handler));
        self
    }

    pub fn build(self) -> Client {
        Client::new(self)
    }
}

impl Rc for ClientHandlerBuilder {
    fn as_base(&self) -> &sys::cef_base_ref_counted_t {
        unsafe {
            let base = &*self.object;
            std::mem::transmute(&base.cef_object)
        }
    }
}

impl WrapClient for ClientHandlerBuilder {
    fn wrap_rc(&mut self, object: *mut RcImpl<sys::cef_client_t, Self>) {
        self.object = object;
    }
}

impl Clone for ClientHandlerBuilder {
    fn clone(&self) -> Self {
        let object = unsafe {
            let rc_impl = &mut *self.object;
            rc_impl.interface.add_ref();
            rc_impl
        };

        Self {
            object,
            render_handler: self.render_handler.clone(),
            context_menu_handler: self.context_menu_handler.clone(),
            message_handlers: self.message_handlers.clone(),
            display_handler: self.display_handler.clone(),
        }
    }
}

impl ImplClient for ClientHandlerBuilder {
    fn render_handler(&self) -> Option<RenderHandler> {
        Some(self.render_handler.clone())
    }

    fn display_handler(&self) -> Option<DisplayHandler> {
        self.display_handler.clone()
    }

    fn on_process_message_received(
        &self,
        browser: Option<&mut Browser>,
        frame: Option<&mut Frame>,
        _: ProcessId,
        message: Option<&mut ProcessMessage>,
    ) -> c_int {
        if let Some(message) = message
            && let Some(browser) = browser
            && let Some(frame) = frame
            && let Some(name) = Some(message.name().into_string())
            && let Some(handler) = self
                .message_handlers
                .iter()
                .find(|h| h.process_name() == name.as_str())
        {
            {
                let args = message.argument_list();
                handler.handle_message(browser, frame, args);
            }
        };
        1
    }

    #[inline]
    fn get_raw(&self) -> *mut sys::_cef_client_t {
        self.object.cast()
    }
}