#ifndef CHROME_BROWSER_EXTENSIONS_API_COOKIES_COOKIES_API_H_
#define CHROME_BROWSER_EXTENSIONS_API_COOKIES_COOKIES_API_H_
#include <string>
#include "base/compiler_specific.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "chrome/browser/extensions/chrome_extension_function.h"
#include "chrome/browser/net/chrome_cookie_notification_details.h"
#include "chrome/common/extensions/api/cookies.h"
#include "content/public/browser/notification_observer.h"
#include "content/public/browser/notification_registrar.h"
#include "extensions/browser/browser_context_keyed_api_factory.h"
#include "extensions/browser/event_router.h"
#include "net/cookies/canonical_cookie.h"
#include "url/gurl.h"
namespace net {
class URLRequestContextGetter;
}
namespace extensions {
class CookiesEventRouter : public content::NotificationObserver {
public:
explicit CookiesEventRouter(content::BrowserContext* context);
virtual ~CookiesEventRouter();
private:
virtual void Observe(int type,
const content::NotificationSource& source,
const content::NotificationDetails& details) OVERRIDE;
void CookieChanged(Profile* profile, ChromeCookieDetails* details);
void DispatchEvent(content::BrowserContext* context,
const std::string& event_name,
scoped_ptr<base::ListValue> event_args,
GURL& cookie_domain);
content::NotificationRegistrar registrar_;
Profile* profile_;
DISALLOW_COPY_AND_ASSIGN(CookiesEventRouter);
};
class CookiesFunction : public ChromeAsyncExtensionFunction {
protected:
virtual ~CookiesFunction() {}
bool ParseUrl(const std::string& url_string, GURL* url,
bool check_host_permissions);
bool ParseStoreContext(std::string* store_id,
net::URLRequestContextGetter** context);
};
class CookiesGetFunction : public CookiesFunction {
public:
DECLARE_EXTENSION_FUNCTION("cookies.get", COOKIES_GET)
CookiesGetFunction();
protected:
virtual ~CookiesGetFunction();
virtual bool RunImpl() OVERRIDE;
private:
void GetCookieOnIOThread();
void RespondOnUIThread();
void GetCookieCallback(const net::CookieList& cookie_list);
GURL url_;
scoped_refptr<net::URLRequestContextGetter> store_browser_context_;
scoped_ptr<extensions::api::cookies::Get::Params> parsed_args_;
};
class CookiesGetAllFunction : public CookiesFunction {
public:
DECLARE_EXTENSION_FUNCTION("cookies.getAll", COOKIES_GETALL)
CookiesGetAllFunction();
protected:
virtual ~CookiesGetAllFunction();
virtual bool RunImpl() OVERRIDE;
private:
void GetAllCookiesOnIOThread();
void RespondOnUIThread();
void GetAllCookiesCallback(const net::CookieList& cookie_list);
GURL url_;
scoped_refptr<net::URLRequestContextGetter> store_browser_context_;
scoped_ptr<extensions::api::cookies::GetAll::Params> parsed_args_;
};
class CookiesSetFunction : public CookiesFunction {
public:
DECLARE_EXTENSION_FUNCTION("cookies.set", COOKIES_SET)
CookiesSetFunction();
protected:
virtual ~CookiesSetFunction();
virtual bool RunImpl() OVERRIDE;
private:
void SetCookieOnIOThread();
void RespondOnUIThread();
void PullCookie(bool set_cookie_);
void PullCookieCallback(const net::CookieList& cookie_list);
GURL url_;
bool success_;
scoped_refptr<net::URLRequestContextGetter> store_browser_context_;
scoped_ptr<extensions::api::cookies::Set::Params> parsed_args_;
};
class CookiesRemoveFunction : public CookiesFunction {
public:
DECLARE_EXTENSION_FUNCTION("cookies.remove", COOKIES_REMOVE)
CookiesRemoveFunction();
protected:
virtual ~CookiesRemoveFunction();
virtual bool RunImpl() OVERRIDE;
private:
void RemoveCookieOnIOThread();
void RespondOnUIThread();
void RemoveCookieCallback();
GURL url_;
scoped_refptr<net::URLRequestContextGetter> store_browser_context_;
scoped_ptr<extensions::api::cookies::Remove::Params> parsed_args_;
};
class CookiesGetAllCookieStoresFunction : public CookiesFunction {
public:
DECLARE_EXTENSION_FUNCTION("cookies.getAllCookieStores",
COOKIES_GETALLCOOKIESTORES)
protected:
virtual ~CookiesGetAllCookieStoresFunction() {}
virtual void Run() OVERRIDE;
virtual bool RunImpl() OVERRIDE;
};
class CookiesAPI : public BrowserContextKeyedAPI,
public extensions::EventRouter::Observer {
public:
explicit CookiesAPI(content::BrowserContext* context);
virtual ~CookiesAPI();
virtual void Shutdown() OVERRIDE;
static BrowserContextKeyedAPIFactory<CookiesAPI>* GetFactoryInstance();
virtual void OnListenerAdded(const extensions::EventListenerInfo& details)
OVERRIDE;
private:
friend class BrowserContextKeyedAPIFactory<CookiesAPI>;
content::BrowserContext* browser_context_;
static const char* service_name() {
return "CookiesAPI";
}
static const bool kServiceIsNULLWhileTesting = true;
scoped_ptr<CookiesEventRouter> cookies_event_router_;
DISALLOW_COPY_AND_ASSIGN(CookiesAPI);
};
}
#endif