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;
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;
}
}