bevy_cef_core 0.2.0

Core library for bevy_cef
use crate::browser_process::CefExtensions;
use crate::browser_process::CommandLineConfig;
use crate::browser_process::MessageLoopTimer;
use crate::browser_process::browser_process_handler::BrowserProcessHandlerBuilder;
use crate::util::{SCHEME_CEF, cef_scheme_flags};
use cef::rc::{Rc, RcImpl};
use cef::{
    BrowserProcessHandler, CefString, CommandLine, ImplApp, ImplCommandLine, ImplSchemeRegistrar,
    SchemeRegistrar, WrapApp,
};
use cef_dll_sys::{_cef_app_t, cef_base_ref_counted_t};
use std::sync::mpsc::Sender;

/// ## Reference
///
/// - [`CefApp Class Reference`](https://cef-builds.spotifycdn.com/docs/106.1/classCefApp.html)
pub struct BrowserProcessAppBuilder {
    object: *mut RcImpl<_cef_app_t, Self>,
    message_loop_working_requester: Sender<MessageLoopTimer>,
    config: CommandLineConfig,
    extensions: CefExtensions,
}

impl BrowserProcessAppBuilder {
    pub fn build(
        message_loop_working_requester: Sender<MessageLoopTimer>,
        config: CommandLineConfig,
        extensions: CefExtensions,
    ) -> cef::App {
        cef::App::new(Self {
            object: core::ptr::null_mut(),
            message_loop_working_requester,
            config,
            extensions,
        })
    }
}

impl Clone for BrowserProcessAppBuilder {
    fn clone(&self) -> Self {
        let object = unsafe {
            let rc_impl = &mut *self.object;
            rc_impl.interface.add_ref();
            self.object
        };
        Self {
            object,
            message_loop_working_requester: self.message_loop_working_requester.clone(),
            config: self.config.clone(),
            extensions: self.extensions.clone(),
        }
    }
}

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

impl ImplApp for BrowserProcessAppBuilder {
    fn on_before_command_line_processing(
        &self,
        _: Option<&CefString>,
        command_line: Option<&mut CommandLine>,
    ) {
        let Some(command_line) = command_line else {
            return;
        };
        for switch in &self.config.switches {
            command_line.append_switch(Some(&(*switch).into()));
        }

        for (name, value) in &self.config.switch_values {
            command_line.append_switch_with_value(Some(&(*name).into()), Some(&(*value).into()));
        }
    }

    fn on_register_custom_schemes(&self, registrar: Option<&mut SchemeRegistrar>) {
        if let Some(registrar) = registrar {
            registrar.add_custom_scheme(Some(&SCHEME_CEF.into()), cef_scheme_flags() as _);
        }
    }

    fn browser_process_handler(&self) -> Option<BrowserProcessHandler> {
        Some(BrowserProcessHandlerBuilder::build(
            self.message_loop_working_requester.clone(),
            self.extensions.clone(),
        ))
    }

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

impl WrapApp for BrowserProcessAppBuilder {
    fn wrap_rc(&mut self, object: *mut RcImpl<_cef_app_t, Self>) {
        self.object = object;
    }
}