components/services/app_service/public/cpp/file_handler.cc

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

#include "components/services/app_service/public/cpp/file_handler.h"

#include <tuple>

namespace apps {

FileHandler::FileHandler() = default;
FileHandler::~FileHandler() = default;
FileHandler::FileHandler(const FileHandler& file_handler) = default;

FileHandler::AcceptEntry::AcceptEntry() = default;
FileHandler::AcceptEntry::~AcceptEntry() = default;
FileHandler::AcceptEntry::AcceptEntry(const AcceptEntry& accept_entry) =
    default;

base::Value FileHandler::AcceptEntry::AsDebugValue() const {
  base::Value root(base::Value::Type::DICTIONARY);

  root.SetStringKey("mime_type", mime_type);
  base::Value& file_extensions_json =
      *root.SetKey("file_extensions", base::Value(base::Value::Type::LIST));
  for (const std::string& file_extension : file_extensions)
    file_extensions_json.Append(file_extension);

  return root;
}

base::Value FileHandler::AsDebugValue() const {
  base::Value root(base::Value::Type::DICTIONARY);

  base::Value& accept_json =
      *root.SetKey("accept", base::Value(base::Value::Type::LIST));
  for (const AcceptEntry& entry : accept)
    accept_json.Append(entry.AsDebugValue());
  root.SetStringKey("action", action.spec());
  base::Value& icons_json =
      *root.SetKey("downloaded_icons", base::Value(base::Value::Type::LIST));
  for (const IconInfo& entry : downloaded_icons)
    icons_json.Append(entry.AsDebugValue());
  root.SetStringKey("name", display_name);
  root.SetStringKey("launch_type", launch_type == LaunchType::kSingleClient
                                       ? "kSingleClient"
                                       : "kMultipleClients");

  return root;
}

std::set<std::string> GetMimeTypesFromFileHandlers(
    const FileHandlers& file_handlers) {
  std::set<std::string> mime_types;
  for (const auto& file_handler : file_handlers) {
    std::set<std::string> file_handler_mime_types =
        GetMimeTypesFromFileHandler(file_handler);
    mime_types.insert(file_handler_mime_types.begin(),
                      file_handler_mime_types.end());
  }
  return mime_types;
}

std::set<std::string> GetMimeTypesFromFileHandler(
    const FileHandler& file_handler) {
  std::set<std::string> mime_types;
  for (const auto& accept_entry : file_handler.accept)
    mime_types.insert(accept_entry.mime_type);
  return mime_types;
}

std::set<std::string> GetFileExtensionsFromFileHandlers(
    const FileHandlers& file_handlers) {
  std::set<std::string> file_extensions;
  for (const auto& file_handler : file_handlers) {
    std::set<std::string> file_handler_file_extensions =
        GetFileExtensionsFromFileHandler(file_handler);
    file_extensions.insert(file_handler_file_extensions.begin(),
                           file_handler_file_extensions.end());
  }
  return file_extensions;
}

std::set<std::string> GetFileExtensionsFromFileHandler(
    const FileHandler& file_handler) {
  std::set<std::string> file_extensions;
  for (const auto& accept_entry : file_handler.accept)
    file_extensions.insert(accept_entry.file_extensions.begin(),
                           accept_entry.file_extensions.end());
  return file_extensions;
}

bool operator==(const FileHandler::AcceptEntry& accept_entry1,
                const FileHandler::AcceptEntry& accept_entry2) {
  return std::tie(accept_entry1.mime_type, accept_entry1.file_extensions) ==
         std::tie(accept_entry2.mime_type, accept_entry2.file_extensions);
}

bool operator==(const FileHandler& file_handler1,
                const FileHandler& file_handler2) {
  return std::tie(file_handler1.action, file_handler1.accept,
                  file_handler1.display_name, file_handler1.launch_type) ==
         std::tie(file_handler2.action, file_handler2.accept,
                  file_handler2.display_name, file_handler2.launch_type);
}

bool operator!=(const FileHandler::AcceptEntry& accept_entry1,
                const FileHandler::AcceptEntry& accept_entry2) {
  return !(accept_entry1 == accept_entry2);
}

bool operator!=(const FileHandler& file_handler1,
                const FileHandler& file_handler2) {
  return !(file_handler1 == file_handler2);
}

}  // namespace apps