API addition to set a cookie for a given URL.

sirikata · Jun 12, 2012 · c11cef7 · c11cef7
1 parent 4fb5458
commit c11cef7
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 1 deletion.
diff --git a/include/berkelium/Berkelium.hpp b/include/berkelium/Berkelium.hpp
@@ -105,6 +105,11 @@ void BERKELIUM_EXPORT update();
 void BERKELIUM_EXPORT runUntilStopped();
 void BERKELIUM_EXPORT stopRunning();
 
+/** Sets a cookie for a given URL. 
+ *  \param url The URL to set the cookie on.
+ *  \param cookieString	The cookie string.
+ */
+void BERKELIUM_EXPORT setCookie (URLString url, WideString cookieString);
 }
 
 #endif
diff --git a/src/Berkelium.cpp b/src/Berkelium.cpp
@@ -72,5 +72,7 @@ void stopRunning() {
 void setErrorHandler (ErrorDelegate *errorHandler) {
     Root::getSingleton().setErrorHandler(errorHandler);
 }
-
+void setCookie (URLString url, WideString cookieString) {
+    Root::getSingleton().setCookie(url, cookieString);
+}
 }
diff --git a/src/Root.cpp b/src/Root.cpp
@@ -100,6 +100,8 @@
 #include <iostream>
 
 #include "chrome/browser/ui/webui/chrome_url_data_manager_backend.h"
+#include "chrome/common/net/url_request_context_getter.h"	
+#include "base/utf_string_conversions.h"
 
 #if !defined(OS_WIN)
 extern "C"
@@ -613,5 +615,10 @@ Root::~Root(){
     mTempProfileDir.reset(); // Delete the profile if necessary.
 }
 
+void Root::setCookie (URLString url, WideString cookieString) {
+     if (getDefaultRequestContext() )
+         if (net::CookieStore * cs = getDefaultRequestContext () -> GetCookieStore ())
+             cs -> SetCookie (GURL(url.get<std::string>()), WideToUTF8(cookieString.get<std::wstring>()));
+}
 
 }
diff --git a/src/Root.hpp b/src/Root.hpp
@@ -123,6 +123,8 @@ class Root : public Singleton<Root> {
     void addWindow(WindowImpl* w);
     void removeWindow(WindowImpl* w);
     WindowList getWindows();
+
+    void setCookie (URLString url, WideString cookieString);
 };
 
 }