// Copyright (c) 2014 The Chromium Embedded Framework Authors.
// Portions copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef LIBCEF_BROWSER_PRINTING_PRINT_DIALOG_LINUX_H_
#define LIBCEF_BROWSER_PRINTING_PRINT_DIALOG_LINUX_H_

#include "base/files/file_path.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/ref_counted.h"
#include "base/task/sequenced_task_runner_helpers.h"
#include "cef/include/cef_print_handler.h"
#include "cef/libcef/browser/browser_host_base.h"
#include "content/public/browser/browser_thread.h"
#include "printing/buildflags/buildflags.h"
#include "printing/print_dialog_linux_interface.h"
#include "ui/linux/linux_ui.h"

namespace printing {
class MetafilePlayer;
class PrintSettings;
}  // namespace printing

using printing::PrintingContextLinux;

class CefPrintingContextLinuxDelegate
    : public ui::PrintingContextLinuxDelegate {
 public:
  CefPrintingContextLinuxDelegate();

  CefPrintingContextLinuxDelegate(const CefPrintingContextLinuxDelegate&) =
      delete;
  CefPrintingContextLinuxDelegate& operator=(
      const CefPrintingContextLinuxDelegate&) = delete;

  printing::PrintDialogLinuxInterface* CreatePrintDialog(
      printing::PrintingContextLinux* context) override;
  gfx::Size GetPdfPaperSize(printing::PrintingContextLinux* context) override;

  void SetDefaultDelegate(ui::PrintingContextLinuxDelegate* delegate);

 private:
  raw_ptr<ui::PrintingContextLinuxDelegate> default_delegate_ = nullptr;
};

// Needs to be freed on the UI thread to clean up its member variables.
class CefPrintDialogLinux : public printing::PrintDialogLinuxInterface,
                            public base::RefCountedThreadSafe<
                                CefPrintDialogLinux,
                                content::BrowserThread::DeleteOnUIThread> {
 public:
  CefPrintDialogLinux(const CefPrintDialogLinux&) = delete;
  CefPrintDialogLinux& operator=(const CefPrintDialogLinux&) = delete;

  // PrintDialogLinuxInterface implementation.
  void UseDefaultSettings() override;
  void UpdateSettings(
      std::unique_ptr<printing::PrintSettings> settings) override;
#if BUILDFLAG(ENABLE_OOP_PRINTING_NO_OOP_BASIC_PRINT_DIALOG)
  void LoadPrintSettings(const printing::PrintSettings& settings) override;
#endif
  void ShowDialog(
      gfx::NativeView parent_view,
      bool has_selection,
      PrintingContextLinux::PrintSettingsCallback callback) override;
  void PrintDocument(const printing::MetafilePlayer& metafile,
                     const std::u16string& document_name) override;
  void ReleaseDialog() override;

 private:
  friend class base::DeleteHelper<CefPrintDialogLinux>;
  friend class base::RefCountedThreadSafe<
      CefPrintDialogLinux,
      content::BrowserThread::DeleteOnUIThread>;
  friend struct content::BrowserThread::DeleteOnThread<
      content::BrowserThread::UI>;
  friend class CefPrintDialogCallbackImpl;
  friend class CefPrintJobCallbackImpl;
  friend class CefPrintingContextLinuxDelegate;

  CefPrintDialogLinux(PrintingContextLinux* context,
                      CefRefPtr<CefBrowserHostBase> browser,
                      CefRefPtr<CefPrintHandler> handler);
  ~CefPrintDialogLinux() override;

  bool UpdateSettings(std::unique_ptr<printing::PrintSettings> settings,
                      bool get_defaults);

  // Prints document named |document_name|.
  void SendDocumentToPrinter(const std::u16string& document_name);

  // Handles print dialog response.
  void OnPrintContinue(CefRefPtr<CefPrintSettings> settings);
  void OnPrintCancel();

  // Handles print job response.
  void OnJobCompleted();

  // Printing dialog callback.
  PrintingContextLinux::PrintSettingsCallback callback_;

  raw_ptr<PrintingContextLinux> context_;
  CefRefPtr<CefBrowserHostBase> browser_;
  CefRefPtr<CefPrintHandler> handler_;

  base::FilePath path_to_pdf_;
};

#endif  // LIBCEF_BROWSER_PRINTING_PRINT_DIALOG_LINUX_H_