chromiumoxide/handler/
target.rs

1use std::collections::VecDeque;
2use std::pin::Pin;
3use std::sync::Arc;
4use std::time::Instant;
5
6use chromiumoxide_cdp::cdp::browser_protocol::target::DetachFromTargetParams;
7use futures::channel::oneshot::Sender;
8use futures::stream::Stream;
9use futures::task::{Context, Poll};
10
11use chromiumoxide_cdp::cdp::browser_protocol::page::{FrameId, GetFrameTreeParams};
12use chromiumoxide_cdp::cdp::browser_protocol::{
13    browser::BrowserContextId,
14    log as cdplog, performance,
15    target::{AttachToTargetParams, SessionId, SetAutoAttachParams, TargetId, TargetInfo},
16};
17use chromiumoxide_cdp::cdp::events::CdpEvent;
18use chromiumoxide_cdp::cdp::CdpEventMessage;
19use chromiumoxide_types::{Command, Method, Request, Response};
20
21use super::blockers::intercept_manager::NetworkInterceptManager;
22use crate::auth::Credentials;
23use crate::cdp::browser_protocol::target::CloseTargetParams;
24use crate::cmd::CommandChain;
25use crate::cmd::CommandMessage;
26use crate::error::{CdpError, Result};
27use crate::handler::browser::BrowserContext;
28use crate::handler::domworld::DOMWorldKind;
29use crate::handler::emulation::EmulationManager;
30use crate::handler::frame::{
31    FrameEvent, FrameManager, NavigationError, NavigationId, NavigationOk,
32};
33use crate::handler::frame::{FrameRequestedNavigation, UTILITY_WORLD_NAME};
34use crate::handler::network::{NetworkEvent, NetworkManager};
35use crate::handler::page::PageHandle;
36use crate::handler::viewport::Viewport;
37use crate::handler::{PageInner, REQUEST_TIMEOUT};
38use crate::listeners::{EventListenerRequest, EventListeners};
39use crate::{page::Page, ArcHttpRequest};
40use chromiumoxide_cdp::cdp::js_protocol::runtime::{
41    ExecutionContextId, RunIfWaitingForDebuggerParams,
42};
43use std::time::Duration;
44
45macro_rules! advance_state {
46    ($s:ident, $cx:ident, $now:ident, $cmds: ident, $next_state:expr ) => {{
47        if let Poll::Ready(poll) = $cmds.poll($now) {
48            return match poll {
49                None => {
50                    $s.init_state = $next_state;
51                    $s.poll($cx, $now)
52                }
53                Some(Ok((method, params))) => Some(TargetEvent::Request(Request {
54                    method,
55                    session_id: $s.session_id.clone().map(Into::into),
56                    params,
57                })),
58                Some(Err(_)) => Some($s.on_initialization_failed()),
59            };
60        } else {
61            return None;
62        }
63    }};
64}
65
66lazy_static::lazy_static! {
67    /// Initial start command params.
68    static ref INIT_COMMANDS_PARAMS: Vec<(chromiumoxide_types::MethodId, serde_json::Value)> = {
69        let attach = SetAutoAttachParams::builder()
70            .flatten(true)
71            .auto_attach(true)
72            .wait_for_debugger_on_start(true)
73            .build()
74            .unwrap();
75        let enable_performance = performance::EnableParams::default();
76        let enable_log = cdplog::EnableParams::default();
77
78        vec![
79                (
80                    attach.identifier(),
81                    serde_json::to_value(attach).unwrap_or_default(),
82                ),
83                (
84                    enable_performance.identifier(),
85                    serde_json::to_value(enable_performance).unwrap_or_default(),
86                ),
87                (
88                    enable_log.identifier(),
89                    serde_json::to_value(enable_log).unwrap_or_default(),
90                )
91            ]
92    };
93
94    /// Attach to target commands
95    static ref ATTACH_TARGET: (chromiumoxide_types::MethodId, serde_json::Value) = {
96        let runtime_cmd = RunIfWaitingForDebuggerParams::default();
97
98        (runtime_cmd.identifier(), serde_json::to_value(runtime_cmd).unwrap_or_default())
99    };
100}
101
102#[derive(Debug)]
103pub struct Target {
104    /// Info about this target as returned from the chromium instance
105    info: TargetInfo,
106    /// The type of this target
107    r#type: TargetType,
108    /// Configs for this target
109    config: TargetConfig,
110    /// The context this target is running in
111    browser_context: BrowserContext,
112    /// The frame manager that maintains the state of all frames and handles
113    /// navigations of frames
114    frame_manager: FrameManager,
115    /// Handles all the https
116    network_manager: NetworkManager,
117    emulation_manager: EmulationManager,
118    /// The identifier of the session this target is attached to
119    session_id: Option<SessionId>,
120    /// The handle of the browser page of this target
121    page: Option<PageHandle>,
122    /// Drives this target towards initialization
123    pub(crate) init_state: TargetInit,
124    /// Currently queued events to report to the `Handler`
125    queued_events: VecDeque<TargetEvent>,
126    /// All registered event subscriptions
127    event_listeners: EventListeners,
128    /// Senders that need to be notified once the main frame has loaded
129    wait_for_frame_navigation: Vec<Sender<ArcHttpRequest>>,
130    /// The sender who requested the page.
131    initiator: Option<Sender<Result<Page>>>,
132}
133
134impl Target {
135    /// Create a new target instance with `TargetInfo` after a
136    /// `CreateTargetParams` request.
137    pub fn new(info: TargetInfo, config: TargetConfig, browser_context: BrowserContext) -> Self {
138        let ty = TargetType::new(&info.r#type);
139        let request_timeout = config.request_timeout;
140        let mut network_manager = NetworkManager::new(config.ignore_https_errors, request_timeout);
141
142        if !config.cache_enabled {
143            network_manager.set_cache_enabled(false);
144        }
145
146        if !config.service_worker_enabled {
147            network_manager.set_service_worker_enabled(true);
148        }
149
150        network_manager.set_request_interception(config.request_intercept);
151
152        if let Some(ref headers) = config.extra_headers {
153            network_manager.set_extra_headers(headers.clone());
154        }
155
156        network_manager.ignore_visuals = config.ignore_visuals;
157        network_manager.block_javascript = config.ignore_javascript;
158        network_manager.block_analytics = config.ignore_analytics;
159        network_manager.block_stylesheets = config.ignore_stylesheets;
160        network_manager.only_html = config.only_html;
161        network_manager.intercept_manager = config.intercept_manager;
162
163        Self {
164            info,
165            r#type: ty,
166            config,
167            frame_manager: FrameManager::new(request_timeout),
168            network_manager,
169            emulation_manager: EmulationManager::new(request_timeout),
170            session_id: None,
171            page: None,
172            init_state: TargetInit::AttachToTarget,
173            wait_for_frame_navigation: Default::default(),
174            queued_events: Default::default(),
175            event_listeners: Default::default(),
176            initiator: None,
177            browser_context,
178        }
179    }
180
181    pub fn set_session_id(&mut self, id: SessionId) {
182        self.session_id = Some(id)
183    }
184
185    pub fn session_id(&self) -> Option<&SessionId> {
186        self.session_id.as_ref()
187    }
188
189    pub fn browser_context(&self) -> &BrowserContext {
190        &self.browser_context
191    }
192
193    pub fn session_id_mut(&mut self) -> &mut Option<SessionId> {
194        &mut self.session_id
195    }
196
197    /// The identifier for this target
198    pub fn target_id(&self) -> &TargetId {
199        &self.info.target_id
200    }
201
202    /// The type of this target
203    pub fn r#type(&self) -> &TargetType {
204        &self.r#type
205    }
206
207    /// Whether this target is already initialized
208    pub fn is_initialized(&self) -> bool {
209        matches!(self.init_state, TargetInit::Initialized)
210    }
211
212    /// Navigate a frame
213    pub fn goto(&mut self, req: FrameRequestedNavigation) {
214        self.frame_manager.goto(req)
215    }
216
217    fn create_page(&mut self) {
218        if self.page.is_none() {
219            if let Some(session) = self.session_id.clone() {
220                let handle =
221                    PageHandle::new(self.target_id().clone(), session, self.opener_id().cloned());
222                self.page = Some(handle);
223            }
224        }
225    }
226
227    /// Tries to create the `PageInner` if this target is already initialized
228    pub(crate) fn get_or_create_page(&mut self) -> Option<&Arc<PageInner>> {
229        self.create_page();
230        self.page.as_ref().map(|p| p.inner())
231    }
232
233    pub fn is_page(&self) -> bool {
234        self.r#type().is_page()
235    }
236
237    pub fn browser_context_id(&self) -> Option<&BrowserContextId> {
238        self.info.browser_context_id.as_ref()
239    }
240
241    pub fn info(&self) -> &TargetInfo {
242        &self.info
243    }
244
245    /// Get the target that opened this target. Top-level targets return `None`.
246    pub fn opener_id(&self) -> Option<&TargetId> {
247        self.info.opener_id.as_ref()
248    }
249
250    pub fn frame_manager(&self) -> &FrameManager {
251        &self.frame_manager
252    }
253
254    pub fn frame_manager_mut(&mut self) -> &mut FrameManager {
255        &mut self.frame_manager
256    }
257
258    pub fn event_listeners_mut(&mut self) -> &mut EventListeners {
259        &mut self.event_listeners
260    }
261
262    /// Received a response to a command issued by this target
263    pub fn on_response(&mut self, resp: Response, method: &str) {
264        if let Some(cmds) = self.init_state.commands_mut() {
265            cmds.received_response(method);
266        }
267
268        if let GetFrameTreeParams::IDENTIFIER = method {
269            if let Some(resp) = resp
270                .result
271                .and_then(|val| GetFrameTreeParams::response_from_value(val).ok())
272            {
273                self.frame_manager.on_frame_tree(resp.frame_tree);
274            }
275        }
276        // requests originated from the network manager all return an empty response, hence they
277        // can be ignored here
278    }
279
280    pub fn on_event(&mut self, event: CdpEventMessage) {
281        let CdpEventMessage { params, method, .. } = event;
282
283        match &params {
284            // `FrameManager` events
285            CdpEvent::PageFrameAttached(ev) => self
286                .frame_manager
287                .on_frame_attached(ev.frame_id.clone(), Some(ev.parent_frame_id.clone())),
288            CdpEvent::PageFrameDetached(ev) => self.frame_manager.on_frame_detached(ev),
289            CdpEvent::PageFrameNavigated(ev) => self.frame_manager.on_frame_navigated(&ev.frame),
290            CdpEvent::PageNavigatedWithinDocument(ev) => {
291                self.frame_manager.on_frame_navigated_within_document(ev)
292            }
293            CdpEvent::RuntimeExecutionContextCreated(ev) => {
294                self.frame_manager.on_frame_execution_context_created(ev)
295            }
296            CdpEvent::RuntimeExecutionContextDestroyed(ev) => {
297                self.frame_manager.on_frame_execution_context_destroyed(ev)
298            }
299            CdpEvent::RuntimeExecutionContextsCleared(_) => {
300                self.frame_manager.on_execution_contexts_cleared()
301            }
302            CdpEvent::RuntimeBindingCalled(ev) => {
303                // TODO check if binding registered and payload is json
304                self.frame_manager.on_runtime_binding_called(ev)
305            }
306            CdpEvent::PageLifecycleEvent(ev) => self.frame_manager.on_page_lifecycle_event(ev),
307            CdpEvent::PageFrameStartedLoading(ev) => {
308                self.frame_manager.on_frame_started_loading(ev);
309            }
310
311            // `Target` events
312            CdpEvent::TargetAttachedToTarget(ev) => {
313                if ev.waiting_for_debugger {
314                    let runtime_cmd = ATTACH_TARGET.clone();
315
316                    self.queued_events.push_back(TargetEvent::Request(Request {
317                        method: runtime_cmd.0,
318                        session_id: Some(ev.session_id.clone().into()),
319                        params: runtime_cmd.1,
320                    }));
321                }
322
323                if "service_worker" == &ev.target_info.r#type {
324                    let detach_command = DetachFromTargetParams::builder()
325                        .session_id(ev.session_id.clone())
326                        .build();
327
328                    self.queued_events.push_back(TargetEvent::Request(Request {
329                        method: detach_command.identifier(),
330                        session_id: self.session_id.clone().map(Into::into),
331                        params: serde_json::to_value(detach_command).unwrap_or_default(),
332                    }));
333                }
334            }
335
336            // `NetworkManager` events
337            CdpEvent::FetchRequestPaused(ev) => self.network_manager.on_fetch_request_paused(ev),
338            CdpEvent::FetchAuthRequired(ev) => self.network_manager.on_fetch_auth_required(ev),
339            CdpEvent::NetworkRequestWillBeSent(ev) => {
340                self.network_manager.on_request_will_be_sent(ev)
341            }
342            CdpEvent::NetworkRequestServedFromCache(ev) => {
343                self.network_manager.on_request_served_from_cache(ev)
344            }
345            CdpEvent::NetworkResponseReceived(ev) => self.network_manager.on_response_received(ev),
346            CdpEvent::NetworkLoadingFinished(ev) => {
347                self.network_manager.on_network_loading_finished(ev)
348            }
349            CdpEvent::NetworkLoadingFailed(ev) => {
350                self.network_manager.on_network_loading_failed(ev)
351            }
352            _ => (),
353        }
354        chromiumoxide_cdp::consume_event!(match params {
355           |ev| self.event_listeners.start_send(ev),
356           |json| { let _ = self.event_listeners.try_send_custom(&method, json);}
357        });
358    }
359
360    /// Called when a init command timed out
361    fn on_initialization_failed(&mut self) -> TargetEvent {
362        if let Some(initiator) = self.initiator.take() {
363            let _ = initiator.send(Err(CdpError::Timeout));
364        }
365        self.init_state = TargetInit::Closing;
366        let close_target = CloseTargetParams::new(self.info.target_id.clone());
367        TargetEvent::Request(Request {
368            method: close_target.identifier(),
369            session_id: self.session_id.clone().map(Into::into),
370            params: serde_json::to_value(close_target).unwrap_or_default(),
371        })
372    }
373
374    /// Advance that target's state
375    pub(crate) fn poll(&mut self, cx: &mut Context<'_>, now: Instant) -> Option<TargetEvent> {
376        if !self.is_page() {
377            // can only poll pages
378            return None;
379        }
380
381        match &mut self.init_state {
382            TargetInit::AttachToTarget => {
383                self.init_state = TargetInit::InitializingFrame(FrameManager::init_commands(
384                    self.config.request_timeout,
385                ));
386
387                if let Ok(params) = AttachToTargetParams::builder()
388                    .target_id(self.target_id().clone())
389                    .flatten(true)
390                    .build()
391                {
392                    return Some(TargetEvent::Request(Request::new(
393                        params.identifier(),
394                        serde_json::to_value(params).unwrap_or_default(),
395                    )));
396                } else {
397                    return None;
398                }
399            }
400            TargetInit::InitializingFrame(cmds) => {
401                self.session_id.as_ref()?;
402                if let Poll::Ready(poll) = cmds.poll(now) {
403                    return match poll {
404                        None => {
405                            if let Some(isolated_world_cmds) =
406                                self.frame_manager.ensure_isolated_world(UTILITY_WORLD_NAME)
407                            {
408                                *cmds = isolated_world_cmds;
409                            } else {
410                                self.init_state = TargetInit::InitializingNetwork(
411                                    self.network_manager.init_commands(),
412                                );
413                            }
414                            self.poll(cx, now)
415                        }
416                        Some(Ok((method, params))) => Some(TargetEvent::Request(Request {
417                            method,
418                            session_id: self.session_id.clone().map(Into::into),
419                            params,
420                        })),
421                        Some(Err(_)) => Some(self.on_initialization_failed()),
422                    };
423                } else {
424                    return None;
425                }
426            }
427            TargetInit::InitializingNetwork(cmds) => {
428                advance_state!(
429                    self,
430                    cx,
431                    now,
432                    cmds,
433                    TargetInit::InitializingPage(Self::page_init_commands(
434                        self.config.request_timeout
435                    ))
436                );
437            }
438            TargetInit::InitializingPage(cmds) => {
439                advance_state!(
440                    self,
441                    cx,
442                    now,
443                    cmds,
444                    match self.config.viewport.as_ref() {
445                        Some(viewport) => TargetInit::InitializingEmulation(
446                            self.emulation_manager.init_commands(viewport)
447                        ),
448                        None => TargetInit::Initialized,
449                    }
450                );
451            }
452            TargetInit::InitializingEmulation(cmds) => {
453                advance_state!(self, cx, now, cmds, TargetInit::Initialized);
454            }
455            TargetInit::Initialized => {
456                if let Some(initiator) = self.initiator.take() {
457                    // make sure that the main frame of the page has finished loading
458                    if self
459                        .frame_manager
460                        .main_frame()
461                        .map(|frame| frame.is_loaded())
462                        .unwrap_or_default()
463                    {
464                        if let Some(page) = self.get_or_create_page() {
465                            let _ = initiator.send(Ok(page.clone().into()));
466                        } else {
467                            self.initiator = Some(initiator);
468                        }
469                    } else {
470                        self.initiator = Some(initiator);
471                    }
472                }
473            }
474            TargetInit::Closing => return None,
475        };
476
477        loop {
478            if self.init_state == TargetInit::Closing {
479                break None;
480            }
481
482            if let Some(frame) = self.frame_manager.main_frame() {
483                if frame.is_loaded() {
484                    while let Some(tx) = self.wait_for_frame_navigation.pop() {
485                        let _ = tx.send(frame.http_request().cloned());
486                    }
487                }
488            }
489
490            // Drain queued messages first.
491            if let Some(ev) = self.queued_events.pop_front() {
492                return Some(ev);
493            }
494
495            if let Some(handle) = self.page.as_mut() {
496                while let Poll::Ready(Some(msg)) = Pin::new(&mut handle.rx).poll_next(cx) {
497                    if self.init_state == TargetInit::Closing {
498                        break;
499                    }
500
501                    match msg {
502                        TargetMessage::Command(cmd) => {
503                            self.queued_events.push_back(TargetEvent::Command(cmd));
504                        }
505                        TargetMessage::MainFrame(tx) => {
506                            let _ =
507                                tx.send(self.frame_manager.main_frame().map(|f| f.id().clone()));
508                        }
509                        TargetMessage::AllFrames(tx) => {
510                            let _ = tx.send(
511                                self.frame_manager
512                                    .frames()
513                                    .map(|f| f.id().clone())
514                                    .collect(),
515                            );
516                        }
517                        TargetMessage::Url(req) => {
518                            let GetUrl { frame_id, tx } = req;
519                            let frame = if let Some(frame_id) = frame_id {
520                                self.frame_manager.frame(&frame_id)
521                            } else {
522                                self.frame_manager.main_frame()
523                            };
524                            let _ = tx.send(frame.and_then(|f| f.url().map(str::to_string)));
525                        }
526                        TargetMessage::Name(req) => {
527                            let GetName { frame_id, tx } = req;
528                            let frame = if let Some(frame_id) = frame_id {
529                                self.frame_manager.frame(&frame_id)
530                            } else {
531                                self.frame_manager.main_frame()
532                            };
533                            let _ = tx.send(frame.and_then(|f| f.name().map(str::to_string)));
534                        }
535                        TargetMessage::Parent(req) => {
536                            let GetParent { frame_id, tx } = req;
537                            let frame = self.frame_manager.frame(&frame_id);
538                            let _ = tx.send(frame.and_then(|f| f.parent_id().cloned()));
539                        }
540                        TargetMessage::WaitForNavigation(tx) => {
541                            if let Some(frame) = self.frame_manager.main_frame() {
542                                // TODO submit a navigation watcher: waitForFrameNavigation
543
544                                // TODO return the watchers navigationResponse
545                                if frame.is_loaded() {
546                                    let _ = tx.send(frame.http_request().cloned());
547                                } else {
548                                    self.wait_for_frame_navigation.push(tx);
549                                }
550                            } else {
551                                self.wait_for_frame_navigation.push(tx);
552                            }
553                        }
554                        TargetMessage::AddEventListener(req) => {
555                            // register a new listener
556                            self.event_listeners.add_listener(req);
557                        }
558                        TargetMessage::GetExecutionContext(ctx) => {
559                            let GetExecutionContext {
560                                dom_world,
561                                frame_id,
562                                tx,
563                            } = ctx;
564                            let frame = if let Some(frame_id) = frame_id {
565                                self.frame_manager.frame(&frame_id)
566                            } else {
567                                self.frame_manager.main_frame()
568                            };
569
570                            if let Some(frame) = frame {
571                                match dom_world {
572                                    DOMWorldKind::Main => {
573                                        let _ = tx.send(frame.main_world().execution_context());
574                                    }
575                                    DOMWorldKind::Secondary => {
576                                        let _ =
577                                            tx.send(frame.secondary_world().execution_context());
578                                    }
579                                }
580                            } else {
581                                let _ = tx.send(None);
582                            }
583                        }
584                        TargetMessage::Authenticate(credentials) => {
585                            self.network_manager.authenticate(credentials);
586                        }
587                    }
588                }
589            }
590
591            while let Some(event) = self.network_manager.poll() {
592                if self.init_state == TargetInit::Closing {
593                    break;
594                }
595                match event {
596                    NetworkEvent::SendCdpRequest((method, params)) => {
597                        // send a message to the browser
598                        self.queued_events.push_back(TargetEvent::Request(Request {
599                            method,
600                            session_id: self.session_id.clone().map(Into::into),
601                            params,
602                        }))
603                    }
604                    NetworkEvent::Request(_) => {}
605                    NetworkEvent::Response(_) => {}
606                    NetworkEvent::RequestFailed(request) => {
607                        self.frame_manager.on_http_request_finished(request);
608                    }
609                    NetworkEvent::RequestFinished(request) => {
610                        self.frame_manager.on_http_request_finished(request);
611                    }
612                }
613            }
614
615            while let Some(event) = self.frame_manager.poll(now) {
616                if self.init_state == TargetInit::Closing {
617                    break;
618                }
619                match event {
620                    FrameEvent::NavigationResult(res) => {
621                        self.queued_events
622                            .push_back(TargetEvent::NavigationResult(res));
623                    }
624                    FrameEvent::NavigationRequest(id, req) => {
625                        self.queued_events
626                            .push_back(TargetEvent::NavigationRequest(id, req));
627                    }
628                }
629            }
630
631            if self.queued_events.is_empty() {
632                return None;
633            }
634        }
635    }
636
637    /// Set the sender half of the channel who requested the creation of this
638    /// target
639    pub fn set_initiator(&mut self, tx: Sender<Result<Page>>) {
640        self.initiator = Some(tx);
641    }
642
643    pub(crate) fn page_init_commands(timeout: Duration) -> CommandChain {
644        CommandChain::new(INIT_COMMANDS_PARAMS.clone(), timeout)
645    }
646}
647
648#[derive(Debug, Clone)]
649pub struct TargetConfig {
650    pub ignore_https_errors: bool,
651    ///  Request timeout to use
652    pub request_timeout: Duration,
653    pub viewport: Option<Viewport>,
654    pub request_intercept: bool,
655    pub cache_enabled: bool,
656    pub ignore_visuals: bool,
657    pub ignore_javascript: bool,
658    pub ignore_analytics: bool,
659    pub ignore_stylesheets: bool,
660    pub only_html: bool,
661    pub service_worker_enabled: bool,
662    pub extra_headers: Option<std::collections::HashMap<String, String>>,
663    pub intercept_manager: NetworkInterceptManager,
664}
665
666impl Default for TargetConfig {
667    fn default() -> Self {
668        Self {
669            ignore_https_errors: true,
670            request_timeout: Duration::from_secs(REQUEST_TIMEOUT),
671            viewport: Default::default(),
672            request_intercept: false,
673            cache_enabled: true,
674            service_worker_enabled: true,
675            ignore_javascript: false,
676            ignore_visuals: false,
677            ignore_stylesheets: false,
678            ignore_analytics: true,
679            only_html: false,
680            extra_headers: Default::default(),
681            intercept_manager: NetworkInterceptManager::Unknown,
682        }
683    }
684}
685
686#[derive(Debug, Clone, Eq, PartialEq)]
687pub enum TargetType {
688    Page,
689    BackgroundPage,
690    ServiceWorker,
691    SharedWorker,
692    Other,
693    Browser,
694    Webview,
695    Unknown(String),
696}
697
698impl TargetType {
699    pub fn new(ty: &str) -> Self {
700        match ty {
701            "page" => TargetType::Page,
702            "background_page" => TargetType::BackgroundPage,
703            "service_worker" => TargetType::ServiceWorker,
704            "shared_worker" => TargetType::SharedWorker,
705            "other" => TargetType::Other,
706            "browser" => TargetType::Browser,
707            "webview" => TargetType::Webview,
708            s => TargetType::Unknown(s.to_string()),
709        }
710    }
711
712    pub fn is_page(&self) -> bool {
713        matches!(self, TargetType::Page)
714    }
715
716    pub fn is_background_page(&self) -> bool {
717        matches!(self, TargetType::BackgroundPage)
718    }
719
720    pub fn is_service_worker(&self) -> bool {
721        matches!(self, TargetType::ServiceWorker)
722    }
723
724    pub fn is_shared_worker(&self) -> bool {
725        matches!(self, TargetType::SharedWorker)
726    }
727
728    pub fn is_other(&self) -> bool {
729        matches!(self, TargetType::Other)
730    }
731
732    pub fn is_browser(&self) -> bool {
733        matches!(self, TargetType::Browser)
734    }
735
736    pub fn is_webview(&self) -> bool {
737        matches!(self, TargetType::Webview)
738    }
739}
740
741#[derive(Debug)]
742pub(crate) enum TargetEvent {
743    /// An internal request
744    Request(Request),
745    /// An internal navigation request
746    NavigationRequest(NavigationId, Request),
747    /// Indicates that a previous requested navigation has finished
748    NavigationResult(Result<NavigationOk, NavigationError>),
749    /// A new command arrived via a channel
750    Command(CommandMessage),
751}
752
753// TODO this can be moved into the classes?
754#[derive(Debug, PartialEq)]
755pub enum TargetInit {
756    InitializingFrame(CommandChain),
757    InitializingNetwork(CommandChain),
758    InitializingPage(CommandChain),
759    InitializingEmulation(CommandChain),
760    AttachToTarget,
761    Initialized,
762    Closing,
763}
764
765impl TargetInit {
766    fn commands_mut(&mut self) -> Option<&mut CommandChain> {
767        match self {
768            TargetInit::InitializingFrame(cmd) => Some(cmd),
769            TargetInit::InitializingNetwork(cmd) => Some(cmd),
770            TargetInit::InitializingPage(cmd) => Some(cmd),
771            TargetInit::InitializingEmulation(cmd) => Some(cmd),
772            TargetInit::AttachToTarget => None,
773            TargetInit::Initialized => None,
774            TargetInit::Closing => None,
775        }
776    }
777}
778
779#[derive(Debug)]
780pub struct GetExecutionContext {
781    /// For which world the execution context was requested
782    pub dom_world: DOMWorldKind,
783    /// The if of the frame to get the `ExecutionContext` for
784    pub frame_id: Option<FrameId>,
785    /// Sender half of the channel to send the response back
786    pub tx: Sender<Option<ExecutionContextId>>,
787}
788
789impl GetExecutionContext {
790    pub fn new(tx: Sender<Option<ExecutionContextId>>) -> Self {
791        Self {
792            dom_world: DOMWorldKind::Main,
793            frame_id: None,
794            tx,
795        }
796    }
797}
798
799#[derive(Debug)]
800pub struct GetUrl {
801    /// The id of the frame to get the url for (None = main frame)
802    pub frame_id: Option<FrameId>,
803    /// Sender half of the channel to send the response back
804    pub tx: Sender<Option<String>>,
805}
806
807impl GetUrl {
808    pub fn new(tx: Sender<Option<String>>) -> Self {
809        Self { frame_id: None, tx }
810    }
811}
812
813#[derive(Debug)]
814pub struct GetName {
815    /// The id of the frame to get the name for (None = main frame)
816    pub frame_id: Option<FrameId>,
817    /// Sender half of the channel to send the response back
818    pub tx: Sender<Option<String>>,
819}
820
821#[derive(Debug)]
822pub struct GetParent {
823    /// The id of the frame to get the parent for (None = main frame)
824    pub frame_id: FrameId,
825    /// Sender half of the channel to send the response back
826    pub tx: Sender<Option<FrameId>>,
827}
828
829#[derive(Debug)]
830pub enum TargetMessage {
831    /// Execute a command within the session of this target
832    Command(CommandMessage),
833    /// Return the main frame of this target's page
834    MainFrame(Sender<Option<FrameId>>),
835    /// Return all the frames of this target's page
836    AllFrames(Sender<Vec<FrameId>>),
837    /// Return the url if available
838    Url(GetUrl),
839    /// Return the name if available
840    Name(GetName),
841    /// Return the parent id of a frame
842    Parent(GetParent),
843    /// A Message that resolves when the frame finished loading a new url
844    WaitForNavigation(Sender<ArcHttpRequest>),
845    /// A request to submit a new listener that gets notified with every
846    /// received event
847    AddEventListener(EventListenerRequest),
848    /// Get the `ExecutionContext` if available
849    GetExecutionContext(GetExecutionContext),
850    Authenticate(Credentials),
851}