// 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 "chrome/browser/ui/qrcode_generator/qrcode_generator_bubble_controller.h"

#include "chrome/browser/browser_process.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/browser_window/public/global_browser_collection.h"
#include "chrome/browser/ui/qrcode_generator/qrcode_generator_bubble_view.h"
#include "chrome/browser/ui/sharing_hub/sharing_hub_bubble_controller.h"
#include "chrome/browser/ui/views/qrcode_generator/qrcode_window_controller.h"
#include "chrome/common/pref_names.h"
#include "components/prefs/pref_change_registrar.h"
#include "components/prefs/pref_service.h"
#include "content/public/browser/web_contents.h"
#include "url/gurl.h"

namespace {

PrefService* GetPrefService() {
  return g_browser_process ? g_browser_process->local_state() : nullptr;
}

// Returns whether the QR code generator is enabled by policy.
bool IsQRCodeGeneratorEnabledByPolicy() {
  const PrefService* prefs = GetPrefService();
  return !prefs || prefs->GetBoolean(prefs::kQRCodeGeneratorEnabled);
}

}  // namespace

namespace qrcode_generator {

QRCodeGeneratorBubbleController::~QRCodeGeneratorBubbleController() {
  HideBubble();
}

// static
bool QRCodeGeneratorBubbleController::IsGeneratorAvailable(const GURL& url) {
  if (!url.SchemeIsHTTPOrHTTPS()) {
    return false;
  }

  // Check policy.
  if (!IsQRCodeGeneratorEnabledByPolicy()) {
    return false;
  }

  return true;
}

// static
QRCodeGeneratorBubbleController* QRCodeGeneratorBubbleController::Get(
    content::WebContents* web_contents) {
  QRCodeGeneratorBubbleController::CreateForWebContents(web_contents);
  QRCodeGeneratorBubbleController* controller =
      QRCodeGeneratorBubbleController::FromWebContents(web_contents);
  return controller;
}

void QRCodeGeneratorBubbleController::ShowBubble(const GURL& url,
                                                 bool show_back_button) {
  // Ignore subsequent calls to open the dialog if it already is open.
  if (bubble_shown_) {
    return;
  }

  // Check policy.
  if (!IsQRCodeGeneratorEnabledByPolicy()) {
    return;
  }

  BrowserWindowInterface* browser =
      GlobalBrowserCollection::GetInstance()->FindBrowserWithTab(
          &GetWebContents());
  if (!browser || !browser->GetWindow()) {
    return;
  }

  bubble_shown_ = true;
  auto* qrcode_window_controller =
      qrcode_generator::QRCodeWindowController::From(browser);
  qrcode_generator_bubble_ = qrcode_window_controller
                                 ? qrcode_window_controller->ShowBubble(
                                       &GetWebContents(), url, show_back_button)
                                 : nullptr;

  // Start listening for policy pref changes.
  pref_change_registrar_ = std::make_unique<PrefChangeRegistrar>();
  pref_change_registrar_->Init(GetPrefService());
  pref_change_registrar_->Add(
      prefs::kQRCodeGeneratorEnabled,
      base::BindRepeating(&QRCodeGeneratorBubbleController::OnPolicyPrefChanged,
                          weak_ptr_factory_.GetWeakPtr()));
}

void QRCodeGeneratorBubbleController::HideBubble() {
  if (qrcode_generator_bubble_) {
    qrcode_generator_bubble_->Hide();
    qrcode_generator_bubble_ = nullptr;
  }
}

QRCodeGeneratorBubbleView*
QRCodeGeneratorBubbleController::qrcode_generator_bubble_view() const {
  return qrcode_generator_bubble_;
}

base::OnceClosure QRCodeGeneratorBubbleController::GetOnBubbleClosedCallback() {
  return base::BindOnce(&QRCodeGeneratorBubbleController::OnBubbleClosed,
                        weak_ptr_factory_.GetWeakPtr());
}

base::OnceClosure
QRCodeGeneratorBubbleController::GetOnBackButtonPressedCallback() {
  return base::BindOnce(&QRCodeGeneratorBubbleController::OnBackButtonPressed,
                        weak_ptr_factory_.GetWeakPtr());
}

void QRCodeGeneratorBubbleController::OnBubbleClosed() {
  bubble_shown_ = false;
  qrcode_generator_bubble_ = nullptr;

  // Stop listening for policy pref changes.
  pref_change_registrar_ = nullptr;
}

void QRCodeGeneratorBubbleController::OnBackButtonPressed() {
  sharing_hub::SharingHubBubbleController* controller =
      sharing_hub::SharingHubBubbleController::CreateOrGetFromWebContents(
          &GetWebContents());
  controller->ShowBubble(share::ShareAttempt(&GetWebContents()));
}

void QRCodeGeneratorBubbleController::OnPolicyPrefChanged() {
  if (!IsQRCodeGeneratorEnabledByPolicy()) {
    HideBubble();
  }
}

QRCodeGeneratorBubbleController::QRCodeGeneratorBubbleController(
    content::WebContents* web_contents)
    : content::WebContentsUserData<QRCodeGeneratorBubbleController>(
          *web_contents) {}

WEB_CONTENTS_USER_DATA_KEY_IMPL(QRCodeGeneratorBubbleController);

}  // namespace qrcode_generator