// Copyright 2015 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 "components/web_restrictions/browser/web_restrictions_resource_throttle.h"

#include "base/bind.h"
#include "components/web_restrictions/browser/web_restrictions_client.h"
#include "content/public/browser/resource_controller.h"
#include "net/base/net_errors.h"
#include "net/url_request/redirect_info.h"
#include "net/url_request/url_request.h"

namespace web_restrictions {

WebRestrictionsResourceThrottle::WebRestrictionsResourceThrottle(
    WebRestrictionsClient* provider,
    const GURL& request_url,
    bool is_main_frame)
    : provider_(provider),
      request_url_(request_url),
      is_main_frame_(is_main_frame),
      weak_ptr_factory_(this) {}

WebRestrictionsResourceThrottle::~WebRestrictionsResourceThrottle() {}

void WebRestrictionsResourceThrottle::WillStartRequest(bool* defer) {
  *defer = ShouldDefer(request_url_);
}

void WebRestrictionsResourceThrottle::WillRedirectRequest(
    const net::RedirectInfo& redirect_info,
    bool* defer) {
  *defer = ShouldDefer(redirect_info.new_url);
}

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

bool WebRestrictionsResourceThrottle::ShouldDefer(const GURL& url) {
  // For requests to function correctly, we need to allow subresources.
  if (provider_->SupportsRequest() && !is_main_frame_)
    return false;
  UrlAccess access = provider_->ShouldProceed(
      is_main_frame_, url,
      base::Bind(&WebRestrictionsResourceThrottle::OnCheckResult,
                 weak_ptr_factory_.GetWeakPtr()));
  if (access == DISALLOW)
    OnCheckResult(false);
  return access == PENDING;
}

void WebRestrictionsResourceThrottle::OnCheckResult(const bool should_proceed) {
  if (should_proceed) {
    controller()->Resume();
  } else {
    controller()->CancelWithError(net::ERR_BLOCKED_BY_ADMINISTRATOR);
  }
}

}  // namespace web_restrictions