// Copyright 2021 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/browser/apps/app_service/web_contents_app_id_utils.h"

#include "chrome/browser/apps/app_service/app_service_proxy.h"
#include "chrome/browser/apps/app_service/app_service_proxy_factory.h"
#include "chrome/browser/apps/app_service/launch_utils.h"
#include "chrome/browser/extensions/app_tab_helper.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/web_applications/web_app_tab_helper.h"
#include "chrome/browser/web_applications/web_app_utils.h"
#include "components/services/app_service/public/cpp/app_types.h"
#include "components/services/app_service/public/cpp/app_update.h"
#include "components/webapps/common/web_app_id.h"
#include "content/public/browser/web_contents.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/common/extension.h"

namespace apps {

namespace {

bool IsAppReady(Profile* profile, const std::string& app_id) {
  if (!apps::AppServiceProxyFactory::IsAppServiceAvailableForProfile(profile)) {
    return false;
  }
  auto* proxy = apps::AppServiceProxyFactory::GetForProfile(profile);
  bool app_installed = false;
  proxy->AppRegistryCache().ForOneApp(
      app_id, [&app_installed](const apps::AppUpdate& update) {
        app_installed = update.Readiness() == apps::Readiness::kReady;
      });
  return app_installed;
}

}  // namespace

std::string GetAppIdForWebContents(content::WebContents* web_contents) {
  const webapps::AppId* app_id =
      web_app::WebAppTabHelper::GetAppId(web_contents);
  if (app_id) {
    return *app_id;
  }

  extensions::AppTabHelper* extensions_app_tab_helper =
      extensions::AppTabHelper::FromWebContents(web_contents);
  // extensions_app_tab_helper is nullptr in some tests.
  return extensions_app_tab_helper
             ? extensions_app_tab_helper->GetExtensionAppId()
             : std::string();
}

void SetAppIdForWebContents(Profile* profile,
                            content::WebContents* web_contents,
                            const std::string& app_id) {
  if (!web_app::AreWebAppsEnabled(profile)) {
    return;
  }
  auto* extensions_app_tab_helper =
      extensions::AppTabHelper::FromWebContents(web_contents);
  auto* web_app_helper =
      web_app::WebAppTabHelper::FromWebContents(web_contents);
  if (!extensions_app_tab_helper || !web_app_helper) {
    return;
  }

  const extensions::Extension* extension =
      extensions::ExtensionRegistry::Get(profile)->GetInstalledExtension(
          app_id);
  if (extension) {
    DCHECK(extension->is_app());
    web_app_helper->SetAppId(std::nullopt);
    extensions_app_tab_helper->SetExtensionAppById(app_id);
  } else {
    bool app_installed = IsAppReady(profile, app_id);
    web_app_helper->SetAppId(
        app_installed ? std::optional<webapps::AppId>(app_id) : std::nullopt);
    extensions_app_tab_helper->SetExtensionAppById(std::string());
  }
}

bool IsInstalledApp(Profile* profile, const std::string& app_id) {
  const extensions::Extension* extension =
      extensions::ExtensionRegistry::Get(profile)->GetInstalledExtension(
          app_id);
  if (extension) {
    return true;
  }
  return IsAppReady(profile, app_id);
}

}  // namespace apps