Merge pull request #6 from OpacityLabs/oscar/sdk-160-support-for-extracting-cookies-on-android

Add support for get cookies for current url

Author: ospfranco

OpacityCore/src/main/assets/extension/background.js

@@ -1,5 +1,9 @@
 browser.webRequest.onHeadersReceived.addListener(
   function (details) {
+    let url = details.url;
+
+    let domain = new URL(url).hostname;
+
     let cookiesHeaders = details.responseHeaders.filter(
       (header) => header.name.toLowerCase() === "set-cookie"
     );
@@ -30,36 +34,40 @@ browser.webRequest.onHeadersReceived.addListener(
     browser.runtime.sendNativeMessage("gecko", {
       event: "cookies",
       cookies: cookieDict,
+      domain: domain,
     });
   },
   { urls: ["<all_urls>"] }, // Intercept all URLs
   ["responseHeaders"]
 );
 
 browser.webNavigation.onDOMContentLoaded.addListener(function (details) {
-  if (details.frameId === 0) { // Ensure it's the top-level frame
+  if (details.frameId === 0) {
+    // Ensure it's the top-level frame
     // Inject a script to fetch the outerHTML of the current document
-    browser.tabs.executeScript(details.tabId, {
-      code: "document.documentElement.outerHTML"
-    }).then(result => {
-      if (result && result.length > 0) {
-        const htmlBody = result[0]; // The outerHTML of the page
-
-        // Send the HTML content and cookies to the native app
-        browser.runtime.sendNativeMessage("gecko", {
-          event: "html_body",
-          html: htmlBody
-        });
+    browser.tabs
+      .executeScript(details.tabId, {
+        code: "document.documentElement.outerHTML",
+      })
+      .then((result) => {
+        if (result && result.length > 0) {
+          const htmlBody = result[0]; // The outerHTML of the page
 
-      } else {
+          // Send the HTML content and cookies to the native app
+          browser.runtime.sendNativeMessage("gecko", {
+            event: "html_body",
+            html: htmlBody,
+          });
+        } else {
+          browser.runtime.sendNativeMessage("gecko", {
+            event: "No results when getting outerHTML",
+          });
+        }
+      })
+      .catch((err) => {
         browser.runtime.sendNativeMessage("gecko", {
-          event: "No results when getting outerHTML",
+          event: "Execute script error: " + err.toString(),
         });
-      }
-    }).catch(err => {
-      browser.runtime.sendNativeMessage("gecko", {
-        event: "Execute script error: " + err.toString(),
       });
-    })
   }
-});
+});

OpacityCore/src/main/cpp/OpacityCore.cpp

@@ -198,6 +198,41 @@ extern "C" void android_close_webview() {
   env->CallVoidMethod(java_object, method);
 }
 
+extern "C" const char *android_get_browser_cookies_for_current_url() {
+  JNIEnv *env = GetJniEnv();
+
+  jclass jOpacityCore = env->GetObjectClass(java_object);
+
+  jmethodID method = env->GetMethodID(
+      jOpacityCore, "getBrowserCookiesForCurrentUrl", "()Ljava/lang/String;");
+  auto res = (jstring)env->CallObjectMethod(java_object, method);
+
+  if (res == nullptr) {
+    return nullptr;
+  }
+
+  const char *val_str = env->GetStringUTFChars(res, nullptr);
+  return val_str;
+}
+
+extern "C" const char *
+android_get_browser_cookies_for_domain(const char *domain) {
+  JNIEnv *env = GetJniEnv();
+
+  jclass jOpacityCore = env->GetObjectClass(java_object);
+  jmethodID method =
+      env->GetMethodID(jOpacityCore, "getBrowserCookiesForDomain",
+                       "(Ljava/lang/String;)Ljava/lang/String;");
+  jstring jdomain = env->NewStringUTF(domain);
+  auto res = (jstring)env->CallObjectMethod(java_object, method, jdomain);
+  if (res == nullptr) {
+    return nullptr;
+  }
+  const char *val_str = env->GetStringUTFChars(res, nullptr);
+  //    opacity_core::free_string(domain); // Free the domain string
+  return val_str; // Caller must free this memory
+}
+
 extern "C" JNIEXPORT jint JNICALL
 Java_com_opacitylabs_opacitycore_OpacityCore_init(
     JNIEnv *env, jobject thiz, jstring api_key, jboolean dry_run,

OpacityCore/src/main/kotlin/com/opacitylabs/opacitycore/CookieResultReceiver.kt

@@ -0,0 +1,40 @@
+package com.opacitylabs.opacitycore
+
+
+import android.os.Parcel
+import android.os.Parcelable
+import org.json.JSONObject
+import java.util.concurrent.CountDownLatch
+import java.util.concurrent.TimeUnit
+
+
+class CookieResultReceiver() : Parcelable {
+    private var cookies: JSONObject? = null
+    private val latch = CountDownLatch(1)
+
+    constructor(parcel: Parcel) : this()
+
+    fun onReceiveResult(cookies: JSONObject?) {
+        this.cookies = cookies
+        latch.countDown()
+    }
+
+    fun waitForResult(timeoutMs: Long): JSONObject? {
+        latch.await(timeoutMs, TimeUnit.MILLISECONDS)
+        return cookies
+    }
+
+    override fun writeToParcel(parcel: Parcel, flags: Int) {}
+
+    override fun describeContents(): Int = 0
+
+    companion object CREATOR : Parcelable.Creator<CookieResultReceiver> {
+        override fun createFromParcel(parcel: Parcel): CookieResultReceiver {
+            return CookieResultReceiver(parcel)
+        }
+
+        override fun newArray(size: Int): Array<CookieResultReceiver?> {
+            return arrayOfNulls(size)
+        }
+    }
+}