electron/shell/browser/atom_navigation_throttle.cc

// Copyright (c) 2015 GitHub, Inc.
// Use of this source code is governed by the MIT license that can be
// found in the LICENSE file.

#include "shell/browser/atom_navigation_throttle.h"

#include "content/public/browser/navigation_handle.h"
#include "shell/browser/api/atom_api_web_contents.h"

namespace electron {

AtomNavigationThrottle::AtomNavigationThrottle(
    content::NavigationHandle* navigation_handle)
    : content::NavigationThrottle(navigation_handle) {}

AtomNavigationThrottle::~AtomNavigationThrottle() = default;

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

content::NavigationThrottle::ThrottleCheckResult
AtomNavigationThrottle::WillRedirectRequest() {
  auto* handle = navigation_handle();
  auto* contents = handle->GetWebContents();
  if (!contents) {
    NOTREACHED();
    return PROCEED;
  }

  auto api_contents =
      electron::api::WebContents::From(v8::Isolate::GetCurrent(), contents);
  if (api_contents.IsEmpty()) {
    // No need to emit any event if the WebContents is not available in JS.
    return PROCEED;
  }

  if (api_contents->EmitNavigationEvent("will-redirect", handle)) {
    return CANCEL;
  }
  return PROCEED;
}

}  // namespace electron