// Copyright 2026 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/glic/host/glic_guest_navigation_throttle.h"

#include "base/feature_list.h"
#include "chrome/browser/glic/host/glic_ui.h"
#include "chrome/browser/glic/host/glic_web_client_manager.h"
#include "chrome/browser/glic/host/guest_util.h"
#include "chrome/browser/glic/host/guest_util_internal.h"
#include "chrome/browser/glic/public/features.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/navigation_throttle_registry.h"

namespace glic {

// static
void GlicGuestNavigationThrottle::MaybeCreateAndAdd(
    content::NavigationThrottleRegistry& registry) {
  if (!base::FeatureList::IsEnabled(features::kGlicDisconnectedWebview)) {
    return;
  }
  content::NavigationHandle& handle = registry.GetNavigationHandle();
  if (handle.IsInMainFrame() && IsGlicGuest(handle.GetWebContents())) {
    registry.AddThrottle(
        std::make_unique<GlicGuestNavigationThrottle>(registry));
  }
}

GlicGuestNavigationThrottle::GlicGuestNavigationThrottle(
    content::NavigationThrottleRegistry& registry)
    : content::NavigationThrottle(registry) {}

GlicGuestNavigationThrottle::~GlicGuestNavigationThrottle() = default;

content::NavigationThrottle::ThrottleCheckResult
GlicGuestNavigationThrottle::WillStartRequest() {
  return HandleRequest();
}

content::NavigationThrottle::ThrottleCheckResult
GlicGuestNavigationThrottle::WillRedirectRequest() {
  return HandleRequest();
}

const char* GlicGuestNavigationThrottle::GetNameForLogging() {
  return "GlicGuestNavigationThrottle";
}

content::NavigationThrottle::ThrottleCheckResult
GlicGuestNavigationThrottle::HandleRequest() {
  content::WebContents* guest_contents = navigation_handle()->GetWebContents();
  GlicUI* glic_ui = GetGlicUiForGuest(guest_contents);
  if (!glic_ui || !glic_ui->web_client_manager()) {
    return PROCEED;
  }

  const GURL& url = navigation_handle()->GetURL();

  if (url.IsAboutBlank()) {
    return PROCEED;
  }

  if (IsAdminBlockedUrl(url)) {
    glic_ui->web_client_manager()->OnGuestNavigationBlocked(
        mojom::GuestPageType::kDisabledByAdmin);
    return CANCEL;
  }

  if (!IsGuestOriginAllowed(url::Origin::Create(url))) {
    glic_ui->web_client_manager()->OnGuestNavigationBlocked(
        mojom::GuestPageType::kLoadError);
    return CANCEL;
  }

  return PROCEED;
}

}  // namespace glic