// Copyright 2017 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 CHROME_BROWSER_CHROMEOS_PRINTING_CUPS_PRINT_JOB_MANAGER_IMPL_H_
#define CHROME_BROWSER_CHROMEOS_PRINTING_CUPS_PRINT_JOB_MANAGER_IMPL_H_

#include <map>
#include <memory>
#include <string>
#include <vector>

#include "base/memory/ref_counted.h"
#include "base/memory/weak_ptr.h"
#include "base/sequence_checker.h"
#include "base/sequenced_task_runner.h"
#include "chrome/browser/chromeos/printing/cups_print_job.h"
#include "chrome/browser/chromeos/printing/cups_print_job_manager.h"
#include "chrome/browser/chromeos/printing/printers_manager.h"
#include "content/public/browser/notification_observer.h"
#include "content/public/browser/notification_registrar.h"
#include "printing/backend/cups_connection.h"

class Profile;

namespace chromeos {

struct QueryResult {
  QueryResult();
  QueryResult(const QueryResult& other);
  ~QueryResult();

  bool success;
  std::vector<::printing::QueueStatus> queues;
};

// A wrapper around the CUPS connection to ensure that it's always accessed on
// the same sequence.
class CupsWrapper {
 public:
  CupsWrapper();
  ~CupsWrapper();

  // Query CUPS for the current jobs for the given |printer_ids|.  Writes result
  // to |result|.
  void QueryCups(const std::vector<std::string>& printer_ids,
                 QueryResult* result);

  // Cancel the print job on the blocking thread.
  void CancelJobImpl(const std::string& printer_id, const int job_id);

 private:
  ::printing::CupsConnection cups_connection_;
  SEQUENCE_CHECKER(sequence_checker_);

  DISALLOW_COPY_AND_ASSIGN(CupsWrapper);
};

class CupsPrintJobManagerImpl : public CupsPrintJobManager,
                                public content::NotificationObserver {
 public:
  explicit CupsPrintJobManagerImpl(Profile* profile);
  ~CupsPrintJobManagerImpl() override;

  // CupsPrintJobManager overrides:
  void CancelPrintJob(CupsPrintJob* job) override;
  bool SuspendPrintJob(CupsPrintJob* job) override;
  bool ResumePrintJob(CupsPrintJob* job) override;

  // NotificationObserver overrides:
  void Observe(int type,
               const content::NotificationSource& source,
               const content::NotificationDetails& details) override;

 private:
  // Begin monitoring a print job for a given |printer_name| with the given
  // |title| with the pages |total_page_number|.
  bool CreatePrintJob(const std::string& printer_name,
                      const std::string& title,
                      int job_id,
                      int total_page_number);

  // Schedule a query of CUPS for print job status with the default delay.
  void ScheduleQuery();
  // Schedule a query of CUPS for print job status with a delay of |delay|.
  void ScheduleQuery(const base::TimeDelta& delay);

  // Schedule the CUPS query off the UI thread. Posts results back to UI thread
  // to UpdateJobs.
  void PostQuery();

  // Process jobs from CUPS and perform notifications.
  void UpdateJobs(std::unique_ptr<QueryResult> results);

  // Mark remaining jobs as errors and remove active jobs.
  void PurgeJobs();

  // Notify observers that a state update has occured for |job|.
  void NotifyJobStateUpdate(CupsPrintJob* job);

  // Ongoing print jobs.
  std::map<std::string, std::unique_ptr<CupsPrintJob>> jobs_;

  // Prevents multiple queries from being scheduled simultaneously.
  bool in_query_ = false;

  // Records the number of consecutive times the GetJobs query has failed.
  int retry_count_ = 0;

  content::NotificationRegistrar registrar_;
  // Task runner for queries to CUPS.
  scoped_refptr<base::SequencedTaskRunner> query_runner_;
  std::unique_ptr<CupsWrapper, base::OnTaskRunnerDeleter> cups_wrapper_;
  base::WeakPtrFactory<CupsPrintJobManagerImpl> weak_ptr_factory_;

  DISALLOW_COPY_AND_ASSIGN(CupsPrintJobManagerImpl);
};

}  // namespace chromeos

#endif  // CHROME_BROWSER_CHROMEOS_PRINTING_CUPS_PRINT_JOB_MANAGER_IMPL_H_