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

#ifndef COMPONENTS_DOWNLOAD_INTERNAL_BACKGROUND_SERVICE_FILE_MONITOR_IMPL_H_
#define COMPONENTS_DOWNLOAD_INTERNAL_BACKGROUND_SERVICE_FILE_MONITOR_IMPL_H_

#include <memory>
#include <set>
#include <vector>

#include "base/files/file_path.h"
#include "base/memory/scoped_refptr.h"
#include "base/memory/weak_ptr.h"
#include "base/task/sequenced_task_runner.h"
#include "base/time/time.h"
#include "components/download/internal/background_service/driver_entry.h"
#include "components/download/internal/background_service/file_monitor.h"
#include "components/download/internal/background_service/model.h"
#include "components/download/internal/background_service/stats.h"

namespace download {

// An utility class containing various file cleanup methods.
class FileMonitorImpl : public FileMonitor {
 public:
  FileMonitorImpl(
      const base::FilePath& download_file_dir,
      const scoped_refptr<base::SequencedTaskRunner>& file_thread_task_runner);

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

  ~FileMonitorImpl() override;

  // FileMonitor implementation.
  void Initialize(InitCallback callback) override;
  void DeleteUnknownFiles(const Model::EntryList& known_entries,
                          const std::vector<DriverEntry>& known_driver_entries,
                          base::OnceClosure completion_callback) override;
  void CleanupFilesForCompletedEntries(
      const Model::EntryList& entries,
      base::OnceClosure completion_callback) override;
  void DeleteFiles(const std::set<base::FilePath>& files_to_remove,
                   stats::FileCleanupReason reason) override;
  void HardRecover(InitCallback callback) override;

 private:
  const base::FilePath download_file_dir_;
  const base::TimeDelta file_keep_alive_time_;

  scoped_refptr<base::SequencedTaskRunner> file_thread_task_runner_;
  base::WeakPtrFactory<FileMonitorImpl> weak_factory_{this};
};

}  // namespace download

#endif  // COMPONENTS_DOWNLOAD_INTERNAL_BACKGROUND_SERVICE_FILE_MONITOR_IMPL_H_