electron/atom/browser/net/atom_network_delegate.h

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

#ifndef ATOM_BROWSER_NET_ATOM_NETWORK_DELEGATE_H_
#define ATOM_BROWSER_NET_ATOM_NETWORK_DELEGATE_H_

#include <map>
#include <set>

#include "brightray/browser/network_delegate.h"
#include "base/callback.h"
#include "base/values.h"
#include "extensions/common/url_pattern.h"
#include "net/base/net_errors.h"
#include "net/http/http_request_headers.h"
#include "net/http/http_response_headers.h"

namespace extensions {
class URLPattern;
}

namespace atom {

using URLPatterns = std::set<extensions::URLPattern>;

class AtomNetworkDelegate : public brightray::NetworkDelegate {
 public:
  struct BlockingResponse;
  using Listener =
      base::Callback<BlockingResponse(const base::DictionaryValue&)>;

  enum EventType {
    kOnBeforeRequest,
    kOnBeforeSendHeaders,
    kOnSendHeaders,
    kOnHeadersReceived,
    kOnBeforeRedirect,
    kOnResponseStarted,
    kOnCompleted,
    kOnErrorOccurred,
  };

  struct ListenerInfo {
    URLPatterns url_patterns;
    AtomNetworkDelegate::Listener callback;
  };

  struct BlockingResponse {
    BlockingResponse() : cancel(false) {}
    ~BlockingResponse() {}

    int code() const {
      return cancel ? net::ERR_BLOCKED_BY_CLIENT : net::OK;
    }

    bool cancel;
    GURL redirect_url;
    net::HttpRequestHeaders request_headers;
    scoped_refptr<net::HttpResponseHeaders> response_headers;
  };

  AtomNetworkDelegate();
  ~AtomNetworkDelegate() override;

  void SetListenerInIO(EventType type,
                       const URLPatterns& patterns,
                       const Listener& callback);

 protected:
  // net::NetworkDelegate:
  int OnBeforeURLRequest(net::URLRequest* request,
                         const net::CompletionCallback& callback,
                         GURL* new_url) override;
  int OnBeforeSendHeaders(net::URLRequest* request,
                          const net::CompletionCallback& callback,
                          net::HttpRequestHeaders* headers) override;
  void OnSendHeaders(net::URLRequest* request,
                     const net::HttpRequestHeaders& headers) override;
  int OnHeadersReceived(
      net::URLRequest* request,
      const net::CompletionCallback& callback,
      const net::HttpResponseHeaders* original_response_headers,
      scoped_refptr<net::HttpResponseHeaders>* override_response_headers,
      GURL* allowed_unsafe_redirect_url) override;
  void OnBeforeRedirect(net::URLRequest* request,
                        const GURL& new_location) override;
  void OnResponseStarted(net::URLRequest* request) override;
  void OnCompleted(net::URLRequest* request, bool started) override;

  void OnErrorOccurred(net::URLRequest* request);

 private:
  std::map<EventType, ListenerInfo> event_listener_map_;

  DISALLOW_COPY_AND_ASSIGN(AtomNetworkDelegate);
};

}   // namespace atom

#endif  // ATOM_BROWSER_NET_ATOM_NETWORK_DELEGATE_H_