mojo/services/html_viewer/websockethandle_impl.h

// Copyright 2014 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.

#ifndef MOJO_SERVICES_HTML_VIEWER_WEBSOCKETHANDLE_IMPL_H_
#define MOJO_SERVICES_HTML_VIEWER_WEBSOCKETHANDLE_IMPL_H_

#include "base/macros.h"
#include "base/memory/scoped_ptr.h"
#include "mojo/common/handle_watcher.h"
#include "mojo/services/public/interfaces/network/web_socket.mojom.h"
#include "third_party/WebKit/public/platform/WebSocketHandle.h"

namespace mojo {
class NetworkService;
class WebSocketClientImpl;
class WebSocketWriteQueue;

// Implements WebSocketHandle by talking to the mojo WebSocket interface.
class WebSocketHandleImpl : public blink::WebSocketHandle {
 public:
  explicit WebSocketHandleImpl(NetworkService* network_service);

 private:
  friend class WebSocketClientImpl;

  virtual ~WebSocketHandleImpl();

  // blink::WebSocketHandle methods:
  virtual void connect(const blink::WebURL& url,
                       const blink::WebVector<blink::WebString>& protocols,
                       const blink::WebSerializedOrigin& origin,
                       blink::WebSocketHandleClient*) override;
  virtual void send(bool fin,
                    MessageType,
                    const char* data,
                    size_t size) override;
  virtual void flowControl(int64_t quota) override;
  virtual void close(unsigned short code,
                     const blink::WebString& reason) override;

  // Called when we finished writing to |send_stream_|.
  void DidWriteToSendStream(bool fin,
                            WebSocketHandle::MessageType type,
                            uint32_t num_bytes,
                            const char* data);

  // Called when the socket is closed.
  void Disconnect();

  WebSocketPtr web_socket_;
  scoped_ptr<WebSocketClientImpl> client_;
  ScopedDataPipeProducerHandle send_stream_;
  scoped_ptr<WebSocketWriteQueue> write_queue_;
  // True if close() was called.
  bool did_close_;

  DISALLOW_COPY_AND_ASSIGN(WebSocketHandleImpl);
};

}  // namespace mojo

#endif  // MOJO_SERVICES_HTML_VIEWER_WEBSOCKETHANDLE_IMPL_H_