// Copyright (c) 2019 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_WEB_APPLICATIONS_TEST_WEB_APP_ICON_TEST_UTILS_H_
#define CHROME_BROWSER_WEB_APPLICATIONS_TEST_WEB_APP_ICON_TEST_UTILS_H_

#include <map>
#include <vector>

#include "base/containers/span.h"
#include "base/files/file_path.h"
#include "chrome/browser/web_applications/web_app_id.h"
#include "chrome/browser/web_applications/web_app_install_utils.h"
#include "chrome/browser/web_applications/web_application_info.h"
#include "third_party/blink/public/common/manifest/manifest.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "third_party/skia/include/core/SkColor.h"

class GURL;
class Profile;

namespace gfx {
class ImageSkia;
}

namespace web_app {

class FileUtilsWrapper;
class WebAppIconManager;

SkBitmap CreateSquareIcon(int size_px, SkColor solid_color);

void AddGeneratedIcon(std::map<SquareSizePx, SkBitmap>* icon_bitmaps,
                      int size_px,
                      SkColor solid_color);

void AddIconToIconsMap(const GURL& icon_url,
                       int size_px,
                       SkColor solid_color,
                       IconsMap* icons_map);

bool AreColorsEqual(SkColor expected_color,
                    SkColor actual_color,
                    int threshold);

base::FilePath GetAppIconsAnyDir(Profile* profile, const AppId& app_id);

base::FilePath GetAppIconsMaskableDir(Profile* profile, const AppId& app_id);

base::FilePath GetOtherIconsDir(Profile* profile, const AppId& app_id);

// Performs blocking IO and decompression.
bool ReadBitmap(FileUtilsWrapper* utils,
                const base::FilePath& file_path,
                SkBitmap* bitmap);

base::span<const int> GetIconSizes();

bool ContainsOneIconOfEachSize(
    const std::map<SquareSizePx, SkBitmap>& icon_bitmaps);

void ExpectImageSkiaRep(const gfx::ImageSkia& image_skia,
                        float scale,
                        SquareSizePx size_px,
                        SkColor color);

blink::Manifest::ImageResource CreateSquareImageResource(
    const GURL& src,
    int size_px,
    const std::vector<IconPurpose>& purposes);

// Performs blocking IO and decompression.
std::map<SquareSizePx, SkBitmap> ReadPngsFromDirectory(
    FileUtilsWrapper* file_utils,
    const base::FilePath& icons_dir);

struct GeneratedIconsInfo {
  GeneratedIconsInfo();
  GeneratedIconsInfo(const GeneratedIconsInfo&);
  GeneratedIconsInfo(IconPurpose purpose,
                     std::vector<SquareSizePx> sizes_px,
                     std::vector<SkColor> colors);
  ~GeneratedIconsInfo();

  IconPurpose purpose;
  std::vector<SquareSizePx> sizes_px;
  std::vector<SkColor> colors;
};

apps::IconInfo CreateIconInfo(const GURL& icon_base_url,
                              IconPurpose purpose,
                              SquareSizePx size_px);

void AddIconsToWebApplicationInfo(
    WebApplicationInfo* web_application_info,
    const GURL& icons_base_url,
    const std::vector<GeneratedIconsInfo>& icons_info);

void IconManagerWriteGeneratedIcons(
    WebAppIconManager& icon_manager,
    const AppId& app_id,
    const std::vector<GeneratedIconsInfo>& icons_info);

// Favicons are read on WebAppIconManager startup, awaits
// WebAppIconManager::favicon_read_callback_ synchronously.
void IconManagerStartAndAwaitFaviconAny(WebAppIconManager& icon_manager,
                                        const AppId& app_id);

// Monochrome favicons are read on WebAppIconManager startup, awaits
// WebAppIconManager::favicon_monochrome_read_callback_ synchronously.
void IconManagerStartAndAwaitFaviconMonochrome(WebAppIconManager& icon_manager,
                                               const AppId& app_id);

// Synchronous read of an app icon pixel.
SkColor IconManagerReadAppIconPixel(const WebAppIconManager& icon_manager,
                                    const AppId& app_id,
                                    SquareSizePx size_px,
                                    int x = 0,
                                    int y = 0);

}  // namespace web_app

#endif  // CHROME_BROWSER_WEB_APPLICATIONS_TEST_WEB_APP_ICON_TEST_UTILS_H_
