chrome/browser/android/preferences/cookie_controls_service_bridge.cc

// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/android/preferences/cookie_controls_service_bridge.h"

#include <memory>
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/cookie_controls/cookie_controls_service.h"
#include "chrome/browser/ui/cookie_controls/cookie_controls_service_factory.h"
#include "components/content_settings/core/common/cookie_controls_enforcement.h"

// Must come after all headers that specialize FromJniType() / ToJniType().
#include "chrome/android/chrome_jni_headers/CookieControlsServiceBridge_jni.h"

using base::android::JavaParamRef;

CookieControlsServiceBridge::CookieControlsServiceBridge(
    JNIEnv* env,
    const JavaParamRef<jobject>& obj,
    Profile* profile)
    : jobject_(obj), profile_(profile) {}

CookieControlsServiceBridge::~CookieControlsServiceBridge() = default;

void CookieControlsServiceBridge::UpdateServiceIfNecessary() {
  CookieControlsService* new_service =
      CookieControlsServiceFactory::GetForProfile(profile_);
  // Update the service only if it is for a new profile
  if (new_service != service_) {
    cookie_controls_service_obs_.Reset();
    service_ = new_service;
    cookie_controls_service_obs_.Observe(service_);
    SendCookieControlsUIChanges();
  }
}

void CookieControlsServiceBridge::HandleCookieControlsToggleChanged(
    JNIEnv* env,
    jboolean checked) {
  UpdateServiceIfNecessary();
  service_->HandleCookieControlsToggleChanged(checked);
}

void CookieControlsServiceBridge::UpdateServiceIfNecessary(JNIEnv* env) {
  UpdateServiceIfNecessary();
}

void CookieControlsServiceBridge::SendCookieControlsUIChanges() {
  bool checked = service_->GetToggleCheckedValue();
  CookieControlsEnforcement enforcement =
      service_->GetCookieControlsEnforcement();
  JNIEnv* env = jni_zero::AttachCurrentThread();
  Java_CookieControlsServiceBridge_sendCookieControlsUiChanges(
      env, jobject_, checked, static_cast<int>(enforcement));
}

void CookieControlsServiceBridge::OnThirdPartyCookieBlockingPrefChanged() {
  SendCookieControlsUIChanges();
}

void CookieControlsServiceBridge::OnThirdPartyCookieBlockingPolicyChanged() {
  SendCookieControlsUIChanges();
}

void CookieControlsServiceBridge::Destroy(JNIEnv* env,
                                          const JavaParamRef<jobject>& obj) {
  delete this;
}

static jlong JNI_CookieControlsServiceBridge_Init(
    JNIEnv* env,
    const JavaParamRef<jobject>& obj,
    Profile* profile) {
  return reinterpret_cast<intptr_t>(
      new CookieControlsServiceBridge(env, obj, profile));
}