components/content_settings/android/cookie_controls_bridge.cc

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

#include "components/content_settings/android/cookie_controls_bridge.h"

#include <memory>

#include "components/content_settings/core/browser/cookie_settings.h"
#include "components/content_settings/core/common/cookie_controls_state.h"
#include "components/permissions/permissions_client.h"
#include "content/public/browser/android/browser_context_handle.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/web_contents.h"

// Must come after all headers that specialize FromJniType() / ToJniType().
#include "components/content_settings/android/content_settings_jni_headers/CookieControlsBridge_jni.h"

namespace content_settings {

using base::android::JavaRef;
using base::android::ScopedJavaLocalRef;

CookieControlsBridge::CookieControlsBridge(
    JNIEnv* env,
    const JavaRef<jobject>& obj,
    const JavaRef<jobject>& jweb_contents_android,
    const JavaRef<jobject>& joriginal_browser_context_handle,
    bool is_incognito_branded)
    : jobject_(obj) {
  UpdateWebContents(env, jweb_contents_android,
                    joriginal_browser_context_handle, is_incognito_branded);
}

void CookieControlsBridge::UpdateWebContents(
    JNIEnv* env,
    const JavaRef<jobject>& jweb_contents_android,
    const JavaRef<jobject>& joriginal_browser_context_handle,
    bool is_incognito_branded) {
  content::WebContents* web_contents =
      content::WebContents::FromJavaWebContents(jweb_contents_android);

  content::BrowserContext* original_context =
      content::BrowserContextFromJavaHandle(joriginal_browser_context_handle);

  content::BrowserContext* context = web_contents->GetBrowserContext();
  auto* permissions_client = permissions::PermissionsClient::Get();

  observation_.Reset();

  controller_ = std::make_unique<CookieControlsController>(
      permissions_client->GetCookieSettings(context),
      original_context ? permissions_client->GetCookieSettings(original_context)
                       : nullptr,
      permissions_client->GetSettingsMap(context),
      is_incognito_branded);

  observation_.Observe(controller_.get());
  controller_->Update(web_contents);
}

void CookieControlsBridge::OnStatusChanged(
    CookieControlsState controls_state,
    CookieControlsEnforcement enforcement,
    base::Time expiration) {
  // Only invoke the callback when there is a change.
  if (controls_state_ == controls_state && enforcement_ == enforcement &&
      expiration_ == expiration) {
    return;
  }
  controls_state_ = controls_state;
  enforcement_ = enforcement;
  expiration_ = expiration;
  JNIEnv* env = base::android::AttachCurrentThread();

  Java_CookieControlsBridge_onStatusChanged(
      env, jobject_, static_cast<int>(controls_state_),
      static_cast<int>(enforcement_),
      expiration.InMillisecondsSinceUnixEpoch());
}

void CookieControlsBridge::SetThirdPartyCookieBlockingEnabledForSite(
    JNIEnv* env,
    bool block_cookies) {
  controller_->OnCookieBlockingEnabledForSite(block_cookies);
}

void CookieControlsBridge::OnUiClosing(JNIEnv* env) {
  controller_->OnUiClosing();
}

CookieControlsBridge::~CookieControlsBridge() = default;

void CookieControlsBridge::Destroy(JNIEnv* env) {
  delete this;
}

static bool JNI_CookieControlsBridge_IsCookieControlsEnabled(
    JNIEnv* env,
    const JavaRef<jobject>& jbrowser_context_handle) {
  content::BrowserContext* context =
      content::BrowserContextFromJavaHandle(jbrowser_context_handle);
  return permissions::PermissionsClient::Get()
      ->GetCookieSettings(context)
      ->ShouldBlockThirdPartyCookies();
}

static int64_t JNI_CookieControlsBridge_Init(
    JNIEnv* env,
    const JavaRef<jobject>& obj,
    const JavaRef<jobject>& jweb_contents_android,
    const JavaRef<jobject>& joriginal_browser_context_handle,
    bool is_incognito_branded) {
  return reinterpret_cast<intptr_t>(new CookieControlsBridge(
      env, obj, jweb_contents_android, joriginal_browser_context_handle,
      is_incognito_branded));
}

}  // namespace content_settings

DEFINE_JNI(CookieControlsBridge)