// 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.

#include "chrome/services/file_util/file_util_service.h"

#include <memory>
#include <utility>

#include "base/functional/bind.h"
#include "base/task/sequenced_task_runner.h"
#include "base/task/thread_pool.h"
#include "build/build_config.h"
#include "chrome/services/file_util/buildflags.h"
#include "components/safe_browsing/buildflags.h"
#include "mojo/public/cpp/bindings/self_owned_receiver.h"

#if BUILDFLAG(SAFE_BROWSING_DOWNLOAD_PROTECTION) && !BUILDFLAG(IS_ANDROID)
#include "chrome/services/file_util/safe_archive_analyzer.h"
#endif

#if BUILDFLAG(IS_CHROMEOS)
#include "chrome/services/file_util/zip_file_creator.h"
#endif

#if BUILDFLAG(ENABLE_EXTRACTORS)
#include "chrome/services/file_util/single_file_tar_file_extractor.h"
#include "chrome/services/file_util/single_file_tar_xz_file_extractor.h"
#endif

FileUtilService::FileUtilService(
    mojo::PendingReceiver<chrome::mojom::FileUtilService> receiver)
    : receiver_(this, std::move(receiver)) {}

FileUtilService::~FileUtilService() = default;

#if BUILDFLAG(IS_CHROMEOS)
void FileUtilService::BindZipFileCreator(
    mojo::PendingReceiver<chrome::mojom::ZipFileCreator> receiver) {
  new chrome::ZipFileCreator(std::move(receiver));  // self deleting
}
#endif

#if BUILDFLAG(SAFE_BROWSING_DOWNLOAD_PROTECTION) && !BUILDFLAG(IS_ANDROID)
void FileUtilService::BindSafeArchiveAnalyzer(
    mojo::PendingReceiver<chrome::mojom::SafeArchiveAnalyzer> receiver) {
  scoped_refptr<base::SequencedTaskRunner> runner =
      base::ThreadPool::CreateSequencedTaskRunner(
          {base::MayBlock(), base::TaskPriority::USER_VISIBLE,
           // CONTINUE_ON_SHUTDOWN will continue shutting down even if
           // tasks are running. This is the only appropriate shutdown
           // behavior for tasks you don't want blocking shutdown.
           base::TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN});
  runner->PostTask(
      FROM_HERE,
      base::BindOnce(
          [](mojo::PendingReceiver<chrome::mojom::SafeArchiveAnalyzer>
                 receiver) {
            mojo::MakeSelfOwnedReceiver(std::make_unique<SafeArchiveAnalyzer>(),
                                        std::move(receiver));
          },
          std::move(receiver)));
}
#endif

#if BUILDFLAG(ENABLE_EXTRACTORS)
void FileUtilService::BindSingleFileTarFileExtractor(
    mojo::PendingReceiver<chrome::mojom::SingleFileExtractor> receiver) {
  mojo::MakeSelfOwnedReceiver(std::make_unique<SingleFileTarFileExtractor>(),
                              std::move(receiver));
}
void FileUtilService::BindSingleFileTarXzFileExtractor(
    mojo::PendingReceiver<chrome::mojom::SingleFileExtractor> receiver) {
  mojo::MakeSelfOwnedReceiver(std::make_unique<SingleFileTarXzFileExtractor>(),
                              std::move(receiver));
}
#endif