// 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 "components/webapps/browser/installable/installable_metrics.h"

#include <atomic>
#include <ostream>

#include "base/check.h"
#include "base/metrics/histogram_base.h"
#include "base/metrics/histogram_functions.h"
#include "base/notreached.h"
#include "components/webapps/browser/webapps_client.h"

namespace webapps {

std::ostream& operator<<(std::ostream& os, WebappInstallSource source) {
  switch (source) {
    case WebappInstallSource::MENU_BROWSER_TAB:
      return os << "menu browser tab";
    case WebappInstallSource::MENU_CUSTOM_TAB:
      return os << "menu custom tab";
    case WebappInstallSource::AUTOMATIC_PROMPT_BROWSER_TAB:
      return os << "automatic prompt browser tab";
    case WebappInstallSource::AUTOMATIC_PROMPT_CUSTOM_TAB:
      return os << "automatic prompt custom tab";
    case WebappInstallSource::API_BROWSER_TAB:
      return os << "api browser tab";
    case WebappInstallSource::API_CUSTOM_TAB:
      return os << "api custom tab";
    case WebappInstallSource::DEVTOOLS:
      return os << "devtools";
    case WebappInstallSource::MANAGEMENT_API:
      return os << "management api";
    case WebappInstallSource::AMBIENT_BADGE_BROWSER_TAB:
      return os << "ambient badge browser tab";
    case WebappInstallSource::AMBIENT_BADGE_CUSTOM_TAB:
      return os << "ambient badge custom tab";
    case WebappInstallSource::ARC:
      return os << "arc";
    case WebappInstallSource::INTERNAL_DEFAULT:
      return os << "internal default";
    case WebappInstallSource::EXTERNAL_DEFAULT:
      return os << "external default";
    case WebappInstallSource::EXTERNAL_POLICY:
      return os << "external policy";
    case WebappInstallSource::SYSTEM_DEFAULT:
      return os << "system default";
    case WebappInstallSource::OMNIBOX_INSTALL_ICON:
      return os << "omnibox install icon";
    case WebappInstallSource::SYNC:
      return os << "sync";
    case WebappInstallSource::MENU_CREATE_SHORTCUT:
      return os << "menu create shortcut";
    case WebappInstallSource::SUB_APP:
      return os << "sub app";
    case WebappInstallSource::CHROME_SERVICE:
      return os << "chrome service";
    case WebappInstallSource::RICH_INSTALL_UI_WEBLAYER:
      return os << "rich install ui weblayer";
    case WebappInstallSource::KIOSK:
      return os << "kiosk";
    case WebappInstallSource::IWA_GRAPHICAL_INSTALLER:
      return os << "isolated web app: graphical installer";
    case WebappInstallSource::IWA_DEV_COMMAND_LINE:
      return os << "isolated web app: dev command line";
    case WebappInstallSource::IWA_DEV_UI:
      return os << "isolated web app: dev UI";
    case WebappInstallSource::IWA_EXTERNAL_POLICY:
      return os << "isolated web app: external policy";
    case WebappInstallSource::IWA_SHIMLESS_RMA:
      return os << "isolated web app: shimless RMA";
    case WebappInstallSource::EXTERNAL_LOCK_SCREEN:
      return os << "external lock screen";
    case WebappInstallSource::PRELOADED_OEM:
      return os << "preloaded oem";
    case WebappInstallSource::MICROSOFT_365_SETUP:
      return os << "microsoft 365 setup";
    case WebappInstallSource::PROFILE_MENU:
      return os << "profile menu";
    case WebappInstallSource::ML_PROMOTION:
      return os << "ml promotion";
    case WebappInstallSource::PRELOADED_DEFAULT:
      return os << "preloaded default";
    case WebappInstallSource::ALMANAC_INSTALL_APP_URI:
      return os << "app install uri";
    case WebappInstallSource::WEBAPK_RESTORE:
      return os << "webapk restore";
    case WebappInstallSource::OOBE_APP_RECOMMENDATIONS:
      return os << "oobe app recommendations";
    case WebappInstallSource::WEB_INSTALL:
      return os << "web install";
    case WebappInstallSource::CHROMEOS_HELP_APP:
      return os << "chromeos help app";
    case WebappInstallSource::MIGRATION:
      return os << "migration";
  }
}

std::ostream& operator<<(std::ostream& os, WebappUninstallSource source) {
  switch (source) {
    case webapps::WebappUninstallSource::kUnknown:
      return os << "Unknown";
    case webapps::WebappUninstallSource::kAppMenu:
      return os << "AppMenu";
    case webapps::WebappUninstallSource::kAppsPage:
      return os << "AppsPage";
    case webapps::WebappUninstallSource::kOsSettings:
      return os << "OS Settings";
    case webapps::WebappUninstallSource::kSync:
      return os << "Sync";
    case webapps::WebappUninstallSource::kAppManagement:
      return os << "App Management";
    case webapps::WebappUninstallSource::kUninstallAndReplaceMigration:
      return os << "Uninstall and Replace migration";
    case webapps::WebappUninstallSource::kAppList:
      return os << "App List";
    case webapps::WebappUninstallSource::kShelf:
      return os << "Shelf";
    case webapps::WebappUninstallSource::kInternalPreinstalled:
      return os << "Internal Preinstalled";
    case webapps::WebappUninstallSource::kExternalPreinstalled:
      return os << "External Preinstalled";
    case webapps::WebappUninstallSource::kExternalPolicy:
      return os << "External Policy";
    case webapps::WebappUninstallSource::kSystemPreinstalled:
      return os << "System Preinstalled";
    case webapps::WebappUninstallSource::kPlaceholderReplacement:
      return os << "Placeholder Replacement";
    case webapps::WebappUninstallSource::kArc:
      return os << "Arc";
    case webapps::WebappUninstallSource::kSubApp:
      return os << "SubApp";
    case webapps::WebappUninstallSource::kStartupCleanup:
      return os << "Startup Cleanup";
    case webapps::WebappUninstallSource::kParentUninstall:
      return os << "Parent App Uninstalled";
    case webapps::WebappUninstallSource::kExternalLockScreen:
      return os << "External Lock Screen";
    case webapps::WebappUninstallSource::kTestCleanup:
      return os << "Test cleanup";
    case webapps::WebappUninstallSource::kInstallUrlDeduping:
      return os << "Install URL deduping";
    case webapps::WebappUninstallSource::kHealthcareUserInstallCleanup:
      return os << "Healthcare User Install Cleanup";
    case webapps::WebappUninstallSource::kIwaEnterprisePolicy:
      return os << "Isolated Web Apps Enterprise Policy";
    case webapps::WebappUninstallSource::kDevtools:
      return os << "Devtools";
    case webapps::WebappUninstallSource::kIwaBlocklisted:
      return os << "Isolated Web App Blocklisted";
    case webapps::WebappUninstallSource::kAppMigration:
      return os << "AppMigration";
    case webapps::WebappUninstallSource::kToolbarPostInstall:
      return os << "ToolbarPostFirstInstall";
  }
}

bool IsUserUninstall(WebappUninstallSource source) {
  switch (source) {
    case webapps::WebappUninstallSource::kSync:
    case webapps::WebappUninstallSource::kUninstallAndReplaceMigration:
    case webapps::WebappUninstallSource::kInternalPreinstalled:
    case webapps::WebappUninstallSource::kExternalPreinstalled:
    case webapps::WebappUninstallSource::kExternalPolicy:
    case webapps::WebappUninstallSource::kSystemPreinstalled:
    case webapps::WebappUninstallSource::kPlaceholderReplacement:
    case webapps::WebappUninstallSource::kArc:
    case webapps::WebappUninstallSource::kSubApp:
    case webapps::WebappUninstallSource::kStartupCleanup:
    case webapps::WebappUninstallSource::kParentUninstall:
    case webapps::WebappUninstallSource::kTestCleanup:
    case webapps::WebappUninstallSource::kInstallUrlDeduping:
    case webapps::WebappUninstallSource::kHealthcareUserInstallCleanup:
    case webapps::WebappUninstallSource::kIwaEnterprisePolicy:
    case webapps::WebappUninstallSource::kIwaBlocklisted:
      return false;
    case webapps::WebappUninstallSource::kUnknown:
    case webapps::WebappUninstallSource::kAppMenu:
    case webapps::WebappUninstallSource::kAppsPage:
    case webapps::WebappUninstallSource::kOsSettings:
    case webapps::WebappUninstallSource::kAppManagement:
    case webapps::WebappUninstallSource::kAppList:
    case webapps::WebappUninstallSource::kShelf:
    case webapps::WebappUninstallSource::kExternalLockScreen:
    case webapps::WebappUninstallSource::kDevtools:
    case webapps::WebappUninstallSource::kAppMigration:
    case webapps::WebappUninstallSource::kToolbarPostInstall:
      return true;
  }
}

// static
void InstallableMetrics::TrackInstallEvent(WebappInstallSource source) {
  DCHECK(IsReportableInstallSource(source));
  base::UmaHistogramEnumeration("Webapp.Install.InstallEvent", source);
}

// static
bool InstallableMetrics::IsReportableInstallSource(WebappInstallSource source) {
  switch (source) {
    case WebappInstallSource::AMBIENT_BADGE_BROWSER_TAB:
    case WebappInstallSource::AMBIENT_BADGE_CUSTOM_TAB:
    case WebappInstallSource::API_BROWSER_TAB:
    case WebappInstallSource::API_CUSTOM_TAB:
    case WebappInstallSource::ARC:
    case WebappInstallSource::AUTOMATIC_PROMPT_BROWSER_TAB:
    case WebappInstallSource::AUTOMATIC_PROMPT_CUSTOM_TAB:
    case WebappInstallSource::CHROME_SERVICE:
    case WebappInstallSource::DEVTOOLS:
    case WebappInstallSource::EXTERNAL_DEFAULT:
    case WebappInstallSource::EXTERNAL_LOCK_SCREEN:
    case WebappInstallSource::EXTERNAL_POLICY:
    case WebappInstallSource::INTERNAL_DEFAULT:
    case WebappInstallSource::KIOSK:
    case WebappInstallSource::MENU_BROWSER_TAB:
    case WebappInstallSource::MENU_CREATE_SHORTCUT:
    case WebappInstallSource::MENU_CUSTOM_TAB:
    case WebappInstallSource::MICROSOFT_365_SETUP:
    case WebappInstallSource::ML_PROMOTION:
    case WebappInstallSource::OMNIBOX_INSTALL_ICON:
    case WebappInstallSource::PRELOADED_OEM:
    case WebappInstallSource::PROFILE_MENU:
    case WebappInstallSource::RICH_INSTALL_UI_WEBLAYER:
    case WebappInstallSource::SYSTEM_DEFAULT:
    case WebappInstallSource::PRELOADED_DEFAULT:
    case WebappInstallSource::ALMANAC_INSTALL_APP_URI:
    case WebappInstallSource::WEBAPK_RESTORE:
    case WebappInstallSource::OOBE_APP_RECOMMENDATIONS:
    case WebappInstallSource::WEB_INSTALL:
    case WebappInstallSource::CHROMEOS_HELP_APP:
      return true;
    case WebappInstallSource::IWA_GRAPHICAL_INSTALLER:
    case WebappInstallSource::IWA_DEV_UI:
    case WebappInstallSource::IWA_DEV_COMMAND_LINE:
    case WebappInstallSource::IWA_EXTERNAL_POLICY:
    case WebappInstallSource::IWA_SHIMLESS_RMA:
    case WebappInstallSource::MANAGEMENT_API:
    case WebappInstallSource::MIGRATION:
    case WebappInstallSource::SUB_APP:
    case WebappInstallSource::SYNC:
      return false;
  }
}

// static
bool InstallableMetrics::IsInstallSurfaceConsideredTrusted(
    WebappInstallSource source) {
  switch (source) {
    case WebappInstallSource::SYSTEM_DEFAULT:
    case WebappInstallSource::EXTERNAL_DEFAULT:
    case WebappInstallSource::INTERNAL_DEFAULT:
    case WebappInstallSource::EXTERNAL_POLICY:
    case WebappInstallSource::SUB_APP:
    case WebappInstallSource::MICROSOFT_365_SETUP:
    case WebappInstallSource::EXTERNAL_LOCK_SCREEN:
    case WebappInstallSource::KIOSK:
      return true;
    case WebappInstallSource::AMBIENT_BADGE_BROWSER_TAB:
    case WebappInstallSource::AMBIENT_BADGE_CUSTOM_TAB:
    case WebappInstallSource::API_BROWSER_TAB:
    case WebappInstallSource::API_CUSTOM_TAB:
    case WebappInstallSource::ARC:
    case WebappInstallSource::AUTOMATIC_PROMPT_BROWSER_TAB:
    case WebappInstallSource::AUTOMATIC_PROMPT_CUSTOM_TAB:
    case WebappInstallSource::CHROME_SERVICE:
    case WebappInstallSource::DEVTOOLS:
    case WebappInstallSource::MENU_BROWSER_TAB:
    case WebappInstallSource::MENU_CREATE_SHORTCUT:
    case WebappInstallSource::MENU_CUSTOM_TAB:
    case WebappInstallSource::ML_PROMOTION:
    case WebappInstallSource::OMNIBOX_INSTALL_ICON:
    case WebappInstallSource::PRELOADED_OEM:
    case WebappInstallSource::PROFILE_MENU:
    case WebappInstallSource::RICH_INSTALL_UI_WEBLAYER:
    case WebappInstallSource::PRELOADED_DEFAULT:
    case WebappInstallSource::ALMANAC_INSTALL_APP_URI:
    case WebappInstallSource::WEBAPK_RESTORE:
    case WebappInstallSource::OOBE_APP_RECOMMENDATIONS:
    case WebappInstallSource::WEB_INSTALL:
    case WebappInstallSource::CHROMEOS_HELP_APP:
    case WebappInstallSource::IWA_GRAPHICAL_INSTALLER:
    case WebappInstallSource::IWA_DEV_UI:
    case WebappInstallSource::IWA_DEV_COMMAND_LINE:
    case WebappInstallSource::IWA_EXTERNAL_POLICY:
    case WebappInstallSource::IWA_SHIMLESS_RMA:
    case WebappInstallSource::MANAGEMENT_API:
    case WebappInstallSource::MIGRATION:
    case WebappInstallSource::SYNC:
      return false;
  }
}

// static
WebappInstallSource InstallableMetrics::GetInstallSource(
    content::WebContents* web_contents,
    InstallTrigger trigger) {
  return WebappsClient::Get()->GetInstallSource(web_contents, trigger);
}

// static
void InstallableMetrics::TrackUninstallEvent(WebappUninstallSource source) {
  base::UmaHistogramEnumeration("Webapp.Install.UninstallEvent", source);
}

// static
void InstallableMetrics::TrackInstallResult(bool result,
                                            WebappInstallSource source) {
  base::UmaHistogramBoolean("WebApp.Install.Result", result);
  if (IsReportableInstallSource(source)) {
    base::UmaHistogramEnumeration(result ? "WebApp.Install.Source.Success"
                                         : "WebApp.Install.Source.Failure",
                                  source);
  }
}
}  // namespace webapps