Simplify VideoReader and VideoWriter

remoting/protocol/video_writer.h

 // Copyright (c) 2012 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.
 
-// VideoWriter is a generic interface used by ConnectionToClient to
-// write into the video stream. ProtobufVideoWriter implements this
-// interface for protobuf video streams.
-
 #ifndef REMOTING_PROTOCOL_VIDEO_WRITER_H_
 #define REMOTING_PROTOCOL_VIDEO_WRITER_H_
 
-#include "base/basictypes.h"
-#include "base/callback.h"
+#include <string>
+
+#include "base/compiler_specific.h"
+#include "remoting/protocol/buffered_socket_writer.h"
+#include "remoting/protocol/channel_dispatcher_base.h"
 #include "remoting/protocol/video_stub.h"
 
 namespace remoting {
 namespace protocol {
 
-class Session;
-class SessionConfig;
-
-class VideoWriter : public VideoStub {
+class VideoWriter : public ChannelDispatcherBase, public VideoStub {
  public:
+  VideoWriter();
   virtual ~VideoWriter();
 
-  // The callback is called when initialization is finished. The
-  // parameter is set to true on success.
-  typedef base::Callback<void(bool)> InitializedCallback;
-
-  static scoped_ptr<VideoWriter> Create(const SessionConfig& config);
-
-  // Initializes the writer.
-  virtual void Init(Session* session, const InitializedCallback& callback) = 0;
-
-  // Stops writing. Must be called on the network thread before this
-  // object is destroyed.
-  virtual void Close() = 0;
-
-  // Returns true if the channel is connected.
-  virtual bool is_connected() = 0;
-
- protected:
-  VideoWriter() { }
+  // VideoStub interface.
+  virtual void ProcessVideoPacket(scoped_ptr<VideoPacket> packet,
+                                  const base::Closure& done) OVERRIDE;
 
  private:
+  void OnInitialized();
+
+  BufferedSocketWriter buffered_writer_;
+
   DISALLOW_COPY_AND_ASSIGN(VideoWriter);
 };
 
 }  // namespace protocol
 }  // namespace remoting
 
 #endif  // REMOTING_PROTOCOL_VIDEO_WRITER_H_