Initial Release

crates/obscura-cdp/src/domains/storage.rs

@@ -0,0 +1,117 @@
+use serde_json::{json, Value};
+
+use crate::dispatch::CdpContext;
+
+pub async fn handle(
+    method: &str,
+    params: &Value,
+    ctx: &mut CdpContext,
+    _session_id: &Option<String>,
+) -> Result<Value, String> {
+    match method {
+        "getCookies" => {
+            let cookies = ctx.default_context.cookie_jar.get_all_cookies();
+            let cdp_cookies: Vec<Value> = cookies
+                .iter()
+                .map(|c| {
+                    json!({
+                        "name": c.name,
+                        "value": c.value,
+                        "domain": c.domain,
+                        "path": c.path,
+                        "expires": -1,
+                        "size": c.name.len() + c.value.len(),
+                        "httpOnly": c.http_only,
+                        "secure": c.secure,
+                        "session": true,
+                        "priority": "Medium",
+                        "sameParty": false,
+                        "sourceScheme": if c.secure { "Secure" } else { "NonSecure" },
+                        "sourcePort": 80,
+                    })
+                })
+                .collect();
+            Ok(json!({ "cookies": cdp_cookies }))
+        }
+        "setCookies" => {
+            if let Some(cookies) = params.get("cookies").and_then(|v| v.as_array()) {
+                let now = std::time::SystemTime::now()
+                    .duration_since(std::time::UNIX_EPOCH)
+                    .unwrap_or_default()
+                    .as_secs_f64();
+
+                for c in cookies {
+                    let name = match c.get("name").and_then(|v| v.as_str()) {
+                        Some(n) => n.to_string(),
+                        None => continue,
+                    };
+                    let value = c.get("value").and_then(|v| v.as_str()).unwrap_or("").to_string();
+
+                    let domain = c
+                        .get("domain")
+                        .and_then(|v| v.as_str())
+                        .map(|s| s.to_string())
+                        .or_else(|| {
+                            c.get("url")
+                                .and_then(|v| v.as_str())
+                                .and_then(|u| url::Url::parse(u).ok())
+                                .and_then(|u| u.host_str().map(|h| h.to_string()))
+                        })
+                        .unwrap_or_default();
+
+                    let expires = c.get("expires").and_then(|v| v.as_f64());
+                    if let Some(exp) = expires {
+                        if exp > 0.0 && exp < now {
+                            ctx.default_context.cookie_jar.delete_cookie(&name, &domain);
+                            continue;
+                        }
+                    }
+
+                    let path = c
+                        .get("path")
+                        .and_then(|v| v.as_str())
+                        .unwrap_or("/")
+                        .to_string();
+                    let secure =
+                        c.get("secure").and_then(|v| v.as_bool()).unwrap_or(false);
+                    let http_only =
+                        c.get("httpOnly").and_then(|v| v.as_bool()).unwrap_or(false);
+
+                    ctx.default_context.cookie_jar.set_cookies_from_cdp(vec![
+                        obscura_net::CookieInfo {
+                            name,
+                            value,
+                            domain,
+                            path,
+                            secure,
+                            http_only,
+                        },
+                    ]);
+                }
+            }
+            Ok(json!({}))
+        }
+        "deleteCookies" => {
+            let name = params.get("name").and_then(|v| v.as_str()).unwrap_or("");
+            let domain_from_url = params
+                .get("url")
+                .and_then(|v| v.as_str())
+                .and_then(|u| url::Url::parse(u).ok())
+                .and_then(|u| u.host_str().map(|h| h.to_string()));
+            let domain = params
+                .get("domain")
+                .and_then(|v| v.as_str())
+                .map(|s| s.to_string())
+                .or(domain_from_url)
+                .unwrap_or_default();
+
+            if !name.is_empty() {
+                ctx.default_context
+                    .cookie_jar
+                    .delete_cookie(name, &domain);
+            }
+            Ok(json!({}))
+        }
+        _ => Ok(json!({})),
+    }
+}