viewpoint_core/context/page_management/
mod.rs

1//! Page creation and management within a browser context.
2
3use tracing::{debug, info, instrument};
4
5use viewpoint_cdp::protocol::target_domain::{GetTargetsParams, GetTargetsResult};
6
7use crate::context::page_factory;
8use crate::error::ContextError;
9use crate::page::Page;
10
11use super::{BrowserContext, PageInfo};
12
13impl BrowserContext {
14    /// Create a new page in this context.
15    ///
16    /// # Errors
17    ///
18    /// Returns an error if page creation fails.
19    #[instrument(level = "info", skip(self), fields(context_id = %self.context_id))]
20    pub async fn new_page(&self) -> Result<Page, ContextError> {
21        if self.closed {
22            return Err(ContextError::Closed);
23        }
24
25        info!("Creating new page");
26
27        // Create target and attach to it
28        let (create_result, attach_result) =
29            page_factory::create_and_attach_target(&self.connection, &self.context_id).await?;
30
31        let target_id = &create_result.target_id;
32        let session_id = &attach_result.session_id;
33
34        // Enable required CDP domains on the page
35        page_factory::enable_page_domains(&self.connection, session_id).await?;
36
37        // Apply emulation settings (viewport, touch, locale, etc.)
38        page_factory::apply_emulation_settings(&self.connection, session_id, &self.options).await?;
39
40        // Get the main frame ID
41        let frame_id = page_factory::get_main_frame_id(&self.connection, session_id).await?;
42
43        // Track the page
44        page_factory::track_page(
45            &self.pages,
46            create_result.target_id.clone(),
47            attach_result.session_id.clone(),
48        )
49        .await;
50
51        // Apply context-level init scripts to the new page
52        if let Err(e) = self.apply_init_scripts_to_session(session_id).await {
53            debug!("Failed to apply init scripts: {}", e);
54        }
55
56        info!(target_id = %target_id, session_id = %session_id, frame_id = %frame_id, "Page created successfully");
57
58        // Get the test ID attribute from context
59        let test_id_attr = self.test_id_attribute.read().await.clone();
60
61        // Convert context HTTP credentials to network auth credentials
62        let http_credentials = page_factory::convert_http_credentials(&self.options);
63
64        // Create page with or without video recording
65        let page = page_factory::create_page_instance(
66            self.connection.clone(),
67            create_result,
68            attach_result,
69            frame_id,
70            &self.options,
71            test_id_attr,
72            self.route_registry.clone(),
73            http_credentials,
74        )
75        .await;
76
77        // Enable Fetch domain if there are context-level routes
78        // This ensures requests are intercepted for context routes
79        if let Err(e) = page.enable_fetch_for_context_routes().await {
80            debug!("Failed to enable Fetch for context routes: {}", e);
81        }
82
83        // Emit page event to registered handlers
84        self.event_manager.emit_page(page.clone_internal()).await;
85
86        Ok(page)
87    }
88
89    /// Get all pages in this context.
90    ///
91    /// # Errors
92    ///
93    /// Returns an error if querying targets fails.
94    pub async fn pages(&self) -> Result<Vec<PageInfo>, ContextError> {
95        if self.closed {
96            return Err(ContextError::Closed);
97        }
98
99        let result: GetTargetsResult = self
100            .connection
101            .send_command("Target.getTargets", Some(GetTargetsParams::default()), None)
102            .await?;
103
104        let pages: Vec<PageInfo> = result
105            .target_infos
106            .into_iter()
107            .filter(|t| {
108                // For the default context (empty string ID), match targets with no context ID
109                // or with an empty context ID
110                let matches_context = if self.context_id.is_empty() {
111                    // Default context: match targets without a context ID or with empty context ID
112                    t.browser_context_id.as_deref().is_none()
113                        || t.browser_context_id.as_deref() == Some("")
114                } else {
115                    // Named context: exact match
116                    t.browser_context_id.as_deref() == Some(&self.context_id)
117                };
118                matches_context && t.target_type == "page"
119            })
120            .map(|t| PageInfo {
121                target_id: t.target_id,
122                session_id: String::new(), // Would need to track sessions
123            })
124            .collect();
125
126        Ok(pages)
127    }
128}
viewpoint_core/context/page_management/mod.rs

viewpoint_core/context/page_management/
mod.rs
