From f1d4c04c35baabe90a23f308abad6802c4f1e795 Mon Sep 17 00:00:00 2001
From: novelhawk <novelhawk@gmail.com>
Date: Sat, 17 Aug 2024 00:31:56 +0200
Subject: [PATCH 1/2] Add basic support for cookie manager

---
 src/lib.rs | 110 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 110 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 3482d26..0376875 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1054,6 +1054,116 @@ impl WebView {
         ICoreWebView2WindowCloseRequestedEventHandler
     );
     remove_event_handler!(remove_window_close_requested);
+    pub fn get_webview2(&self) -> Result<WebView_2> {
+        let inner = self
+            .inner
+            .get_interface::<dyn ICoreWebView2_2>()
+            .ok_or_else(|| Error::new(E_NOINTERFACE))?;
+        Ok(WebView_2 { inner })
+    }
+}
+
+impl WebView_2 {
+    pub fn get_cookie_manager(&self) -> Result<CookieManager> {
+        let mut ppv: *mut *mut ICoreWebView2CookieManagerVTable = ptr::null_mut();
+        check_hresult(unsafe { self.inner.get_cookie_manager(&mut ppv) })?;
+        Ok(CookieManager {
+            inner: unsafe { add_ref_to_rc(ppv) },
+        })
+    }
+}
+
+impl CookieManager {
+    pub fn create_cookie(
+        &self,
+        name: &str,
+        value: &str,
+        domain: &str,
+        path: &str,
+    ) -> Result<Cookie> {
+        let mut ppv: *mut *mut ICoreWebView2CookieVTable = ptr::null_mut();
+        let name = WideCString::from_str(name)?;
+        let value = WideCString::from_str(value)?;
+        let domain = WideCString::from_str(domain)?;
+        let path = WideCString::from_str(path)?;
+        check_hresult(unsafe {
+            self.inner.create_cookie(
+                name.as_ptr(),
+                value.as_ptr(),
+                domain.as_ptr(),
+                path.as_ptr(),
+                &mut ppv,
+            )
+        })?;
+        Ok(Cookie {
+            inner: unsafe { add_ref_to_rc(ppv) },
+        })
+    }
+
+    pub fn get_cookies(
+        &self,
+        uri: &str,
+        callback: impl FnOnce(CookieList) -> Result<()> + 'static,
+    ) -> Result<()> {
+        let uri = WideCString::from_str(uri)?;
+        let callback = Cell::new(Some(callback));
+
+        let handler = callback!(
+            ICoreWebView2GetCookiesCompletedHandler,
+            move |error_code: HRESULT, list: *mut *mut ICoreWebView2CookieListVTable| -> HRESULT {
+                to_hresult(check_hresult(error_code).and_then(|_| {
+                    let list = CookieList {
+                        inner: unsafe { add_ref_to_rc(list) },
+                    };
+
+                    if let Some(callback) = callback.take() {
+                        callback(list)
+                    } else {
+                        Ok(())
+                    }
+                }))
+            }
+        );
+
+        check_hresult(unsafe { self.inner.get_cookies(uri.as_ptr(), handler.as_raw()) })
+    }
+
+    call!(delete_all_cookies);
+}
+
+impl CookieList {
+    get!(get_count, u32);
+    pub fn get_value_at_index(&self, index: u32) -> Result<Cookie> {
+        let mut ppv: *mut *mut ICoreWebView2CookieVTable = ptr::null_mut();
+        check_hresult(unsafe { self.inner.get_value_at_index(index, &mut ppv) })?;
+        Ok(Cookie {
+            inner: unsafe { add_ref_to_rc(ppv) },
+        })
+    }
+}
+
+impl Cookie {
+    get_string!(get_name);
+
+    get_string!(get_value);
+    put_string!(put_value);
+
+    get_string!(get_domain);
+    get_string!(get_path);
+
+    get!(get_expires, f64);
+    put!(put_expires, expires: f64);
+
+    get_bool!(get_is_http_only);
+    put_bool!(put_is_http_only);
+
+    get!(get_same_site, CookieSameSiteKind);
+    put!(put_same_site, same_site: CookieSameSiteKind);
+
+    get_bool!(get_is_secure);
+    put_bool!(put_is_secure);
+
+    get_bool!(get_is_session);
 }
 
 impl Settings {

From 30ca4edef9f559fa3770d26efd3df674de902e9e Mon Sep 17 00:00:00 2001
From: novelhawk <novelhawk@gmail.com>
Date: Sat, 17 Aug 2024 00:47:14 +0200
Subject: [PATCH 2/2] Used get_interface! instead of manual implementation

---
 src/lib.rs | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 0376875..2a7a7e6 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1064,13 +1064,7 @@ impl WebView {
 }
 
 impl WebView_2 {
-    pub fn get_cookie_manager(&self) -> Result<CookieManager> {
-        let mut ppv: *mut *mut ICoreWebView2CookieManagerVTable = ptr::null_mut();
-        check_hresult(unsafe { self.inner.get_cookie_manager(&mut ppv) })?;
-        Ok(CookieManager {
-            inner: unsafe { add_ref_to_rc(ppv) },
-        })
-    }
+    get_interface!(get_cookie_manager, CookieManager);
 }
 
 impl CookieManager {