chrome/browser/ui/webui/welcome/ntp_background_handler.cc

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

#include "chrome/browser/ui/webui/welcome/ntp_background_handler.h"

#include <memory>
#include <utility>

#include "base/bind.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/utf_string_conversions.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/search/background/ntp_backgrounds.h"
#include "chrome/browser/search/instant_service.h"
#include "chrome/browser/search/instant_service_factory.h"
#include "chrome/grit/generated_resources.h"
#include "chrome/grit/welcome_resources.h"
#include "components/strings/grit/components_strings.h"
#include "content/public/browser/web_ui.h"
#include "ui/base/l10n/l10n_util.h"
#include "ui/base/resource/resource_bundle.h"

namespace welcome {

enum class NtpBackgrounds {
  kArt = 0,
  kCityscape = 1,
  kEarth = 2,
  kGeometricShapes = 3,
  kLandscape = 4,
};

NtpBackgroundHandler::NtpBackgroundHandler() {}
NtpBackgroundHandler::~NtpBackgroundHandler() {}

void NtpBackgroundHandler::RegisterMessages() {
  web_ui()->RegisterMessageCallback(
      "clearBackground",
      base::BindRepeating(&NtpBackgroundHandler::HandleClearBackground,
                          base::Unretained(this)));

  web_ui()->RegisterMessageCallback(
      "getBackgrounds",
      base::BindRepeating(&NtpBackgroundHandler::HandleGetBackgrounds,
                          base::Unretained(this)));

  web_ui()->RegisterMessageCallback(
      "setBackground",
      base::BindRepeating(&NtpBackgroundHandler::HandleSetBackground,
                          base::Unretained(this)));
}

void NtpBackgroundHandler::HandleClearBackground(const base::ListValue* args) {
  InstantService* instant_service =
      InstantServiceFactory::GetForProfile(Profile::FromWebUI(web_ui()));
  instant_service->ResetCustomBackgroundInfo();
}

void NtpBackgroundHandler::HandleGetBackgrounds(const base::ListValue* args) {
  AllowJavascript();
  CHECK_EQ(1U, args->GetSize());
  const base::Value* callback_id;
  CHECK(args->Get(0, &callback_id));

  base::ListValue list_value;
  std::array<GURL, kNtpBackgroundsCount> NtpBackgrounds = GetNtpBackgrounds();
  const std::string kUrlPrefix = "preview-background.jpg?";

  auto element = std::make_unique<base::DictionaryValue>();
  int id = static_cast<int>(NtpBackgrounds::kEarth);
  element->SetInteger("id", id);
  element->SetString("title", l10n_util::GetStringUTF8(
                                  IDS_WELCOME_NTP_BACKGROUND_EARTH_TITLE));
  element->SetString("imageUrl", kUrlPrefix + base::NumberToString(id));
  element->SetString("thumbnailClass", "earth");
  list_value.Append(std::move(element));

  element = std::make_unique<base::DictionaryValue>();
  id = static_cast<int>(NtpBackgrounds::kCityscape);
  element->SetInteger("id", id);
  element->SetString("title", l10n_util::GetStringUTF8(
                                  IDS_WELCOME_NTP_BACKGROUND_CITYSCAPE_TITLE));
  element->SetString("imageUrl", kUrlPrefix + base::NumberToString(id));
  element->SetString("thumbnailClass", "cityscape");
  list_value.Append(std::move(element));

  element = std::make_unique<base::DictionaryValue>();
  id = static_cast<int>(NtpBackgrounds::kLandscape);
  element->SetInteger("id", id);
  element->SetString("title", l10n_util::GetStringUTF8(
                                  IDS_WELCOME_NTP_BACKGROUND_LANDSCAPE_TITLE));
  element->SetString("imageUrl", kUrlPrefix + base::NumberToString(id));
  element->SetString("thumbnailClass", "landscape");
  list_value.Append(std::move(element));

  element = std::make_unique<base::DictionaryValue>();
  id = static_cast<int>(NtpBackgrounds::kArt);
  element->SetInteger("id", id);
  element->SetString(
      "title", l10n_util::GetStringUTF8(IDS_WELCOME_NTP_BACKGROUND_ART_TITLE));
  element->SetString("imageUrl", kUrlPrefix + base::NumberToString(id));
  element->SetString("thumbnailClass", "art");
  list_value.Append(std::move(element));

  element = std::make_unique<base::DictionaryValue>();
  id = static_cast<int>(NtpBackgrounds::kGeometricShapes);
  element->SetInteger("id", id);
  element->SetString("title",
                     l10n_util::GetStringUTF8(
                         IDS_WELCOME_NTP_BACKGROUND_GEOMETRIC_SHAPES_TITLE));
  element->SetString("imageUrl", kUrlPrefix + base::NumberToString(id));
  element->SetString("thumbnailClass", "geometric-shapes");
  list_value.Append(std::move(element));

  ResolveJavascriptCallback(*callback_id, list_value);
}

void NtpBackgroundHandler::HandleSetBackground(const base::ListValue* args) {
  CHECK_EQ(1U, args->GetSize());
  int backgroundIndex;
  args->GetInteger(0, &backgroundIndex);

  std::array<GURL, kNtpBackgroundsCount> NtpBackgrounds = GetNtpBackgrounds();
  InstantService* instant_service =
      InstantServiceFactory::GetForProfile(Profile::FromWebUI(web_ui()));

  switch (backgroundIndex) {
    case static_cast<int>(NtpBackgrounds::kArt):
      instant_service->SetCustomBackgroundInfo(
          NtpBackgrounds[backgroundIndex], "Universe Cosmic Vacum",
          "Philipp Rietz — Walli",
          GURL("https://walli.shanga.co/image/view/?id=370"), "");
      break;
    case static_cast<int>(NtpBackgrounds::kCityscape):
      instant_service->SetCustomBackgroundInfo(
          NtpBackgrounds[backgroundIndex],
          l10n_util::GetStringFUTF8(IDS_WELCOME_NTP_BACKGROUND_PHOTO_BY_LABEL,
                                    u"Ev Tchebotarev"),
          "",
          GURL("https://500px.com/photo/135751035/"
               "soulseek-by-%E5%B0%A4%E9%87%91%E5%B0%BC-ev-tchebotarev"),
          "");
      break;
    case static_cast<int>(NtpBackgrounds::kEarth):
      instant_service->SetCustomBackgroundInfo(
          NtpBackgrounds[backgroundIndex],
          l10n_util::GetStringFUTF8(IDS_WELCOME_NTP_BACKGROUND_PHOTO_BY_LABEL,
                                    u"NASA Image Library"),
          "", GURL("https://www.google.com/sky/"), "");
      break;
    case static_cast<int>(NtpBackgrounds::kGeometricShapes):
      instant_service->SetCustomBackgroundInfo(
          NtpBackgrounds[backgroundIndex], "Tessellation 15",
          "Justin Prno — Walli",
          GURL("https://walli.shanga.co/image/view/?id=1375"), "");
      break;
    case static_cast<int>(NtpBackgrounds::kLandscape):
      instant_service->SetCustomBackgroundInfo(
          NtpBackgrounds[backgroundIndex],
          l10n_util::GetStringFUTF8(IDS_WELCOME_NTP_BACKGROUND_PHOTO_BY_LABEL,
                                    u"Giulio Rosso Chioso"),
          "",
          GURL("https://500px.com/photo/41149196/"
               "le-piscine-sunset-by-giulio-rosso-chioso"),
          "");
      break;
  }
}

}  // namespace welcome