66 lines
2 KiB
C++
66 lines
2 KiB
C++
// Copyright (c) 2018 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "shell/browser/cookie_change_notifier.h"
|
|
|
|
#include <utility>
|
|
|
|
#include "base/bind.h"
|
|
#include "content/public/browser/browser_thread.h"
|
|
#include "content/public/browser/storage_partition.h"
|
|
#include "net/cookies/canonical_cookie.h"
|
|
#include "shell/browser/atom_browser_context.h"
|
|
|
|
using content::BrowserThread;
|
|
|
|
namespace electron {
|
|
|
|
CookieChangeNotifier::CookieChangeNotifier(AtomBrowserContext* browser_context)
|
|
: browser_context_(browser_context), receiver_(this) {
|
|
StartListening();
|
|
}
|
|
|
|
CookieChangeNotifier::~CookieChangeNotifier() = default;
|
|
|
|
std::unique_ptr<
|
|
base::CallbackList<void(const net::CookieChangeInfo& change)>::Subscription>
|
|
CookieChangeNotifier::RegisterCookieChangeCallback(
|
|
const base::Callback<void(const net::CookieChangeInfo& change)>& cb) {
|
|
DCHECK_CURRENTLY_ON(BrowserThread::UI);
|
|
|
|
return cookie_change_sub_list_.Add(cb);
|
|
}
|
|
|
|
void CookieChangeNotifier::StartListening() {
|
|
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
|
|
DCHECK(!receiver_.is_bound());
|
|
|
|
network::mojom::CookieManager* cookie_manager =
|
|
content::BrowserContext::GetDefaultStoragePartition(browser_context_)
|
|
->GetCookieManagerForBrowserProcess();
|
|
|
|
// Cookie manager should be created whenever network context is created,
|
|
// if this fails then there is something wrong with our context creation
|
|
// cycle.
|
|
CHECK(cookie_manager);
|
|
|
|
cookie_manager->AddGlobalChangeListener(receiver_.BindNewPipeAndPassRemote());
|
|
receiver_.set_disconnect_handler(base::BindOnce(
|
|
&CookieChangeNotifier::OnConnectionError, base::Unretained(this)));
|
|
}
|
|
|
|
void CookieChangeNotifier::OnConnectionError() {
|
|
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
|
|
|
|
receiver_.reset();
|
|
StartListening();
|
|
}
|
|
|
|
void CookieChangeNotifier::OnCookieChange(const net::CookieChangeInfo& change) {
|
|
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
|
|
|
|
cookie_change_sub_list_.Notify(change);
|
|
}
|
|
|
|
} // namespace electron
|