components/html_viewer/web_url_loader_impl.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef COMPONENTS_HTML_VIEWER_WEB_URL_LOADER_IMPL_H_
   6 #define COMPONENTS_HTML_VIEWER_WEB_URL_LOADER_IMPL_H_
   7
   8 #include "base/macros.h"
   9 #include "base/memory/weak_ptr.h"
  10 #include "components/html_viewer/mock_web_blob_registry_impl.h"
  11 #include "mojo/message_pump/handle_watcher.h"
  12 #include "mojo/services/network/public/interfaces/url_loader.mojom.h"
  13 #include "third_party/WebKit/public/platform/WebBlobData.h"
  14 #include "third_party/WebKit/public/platform/WebReferrerPolicy.h"
  15 #include "third_party/WebKit/public/platform/WebURLLoader.h"
  16 #include "third_party/WebKit/public/platform/WebURLRequest.h"
  17
  18 namespace mojo {
  19 class URLLoaderFactory;
  20 }
  21
  22 namespace html_viewer {
  23
  24 // The concrete type of WebURLRequest::ExtraData.
  25 class WebURLRequestExtraData : public blink::WebURLRequest::ExtraData {
  26  public:
  27   WebURLRequestExtraData();
  28   virtual ~WebURLRequestExtraData();
  29
  30   mojo::URLResponsePtr synthetic_response;
  31 };
  32
  33 class WebURLLoaderImpl : public blink::WebURLLoader {
  34  public:
  35   explicit WebURLLoaderImpl(mojo::URLLoaderFactory* url_loader_factory,
  36                             MockWebBlobRegistryImpl* web_blob_registry);
  37
  38  private:
  39   virtual ~WebURLLoaderImpl();
  40
  41   // blink::WebURLLoader methods:
  42   virtual void loadSynchronously(const blink::WebURLRequest& request,
  43                                  blink::WebURLResponse& response,
  44                                  blink::WebURLError& error,
  45                                  blink::WebData& data);
  46   virtual void loadAsynchronously(const blink::WebURLRequest&,
  47                                   blink::WebURLLoaderClient* client);
  48   virtual void cancel();
  49   virtual void setDefersLoading(bool defers_loading);
  50
  51   void OnReceivedResponse(const blink::WebURLRequest& request,
  52                           mojo::URLResponsePtr response);
  53   void OnReceivedError(mojo::URLResponsePtr response);
  54   void OnReceivedRedirect(const blink::WebURLRequest& request,
  55                           mojo::URLResponsePtr response);
  56   void OnReceiveWebBlobData(
  57       const blink::WebURLRequest& request,
  58       const blink::WebVector<blink::WebBlobData::Item*>& items);
  59   void ReadMore();
  60   void WaitToReadMore();
  61   void OnResponseBodyStreamReady(MojoResult result);
  62
  63   blink::WebURLLoaderClient* client_;
  64   MockWebBlobRegistryImpl* web_blob_registry_;
  65   GURL url_;
  66   blink::WebReferrerPolicy referrer_policy_;
  67   mojo::URLLoaderPtr url_loader_;
  68   mojo::ScopedDataPipeConsumerHandle response_body_stream_;
  69   mojo::common::HandleWatcher handle_watcher_;
  70
  71   base::WeakPtrFactory<WebURLLoaderImpl> weak_factory_;
  72
  73   DISALLOW_COPY_AND_ASSIGN(WebURLLoaderImpl);
  74 };
  75
  76 }  // namespace html_viewer
  77
  78 #endif  // COMPONENTS_HTML_VIEWER_WEB_URL_LOADER_IMPL_H_