Cast: Add GetStats() extensions API.

chrome/renderer/extensions/cast_streaming_native_handler.cc

 // Copyright 2013 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 "chrome/renderer/extensions/cast_streaming_native_handler.h"
 
 #include <functional>
+#include <iterator>
 
 #include "base/base64.h"
 #include "base/logging.h"
 #include "base/message_loop/message_loop.h"
 #include "chrome/common/extensions/api/cast_streaming_rtp_stream.h"
 #include "chrome/common/extensions/api/cast_streaming_udp_transport.h"
 #include "chrome/renderer/extensions/chrome_v8_context.h"
 #include "chrome/renderer/media/cast_rtp_stream.h"
 #include "chrome/renderer/media/cast_session.h"
 #include "chrome/renderer/media/cast_udp_transport.h"
@@ skipping 98 matching lines @@
     linked_ptr<CodecSpecificParams> ext_codec_params(
         new CodecSpecificParams());
     FromCastCodecSpecificParams(cast_params.codec_specific_params[i],
                                 ext_codec_params.get());
     ext_params->codec_specific_params.push_back(ext_codec_params);
   }
 }
 
 void FromCastRtpParams(const CastRtpParams& cast_params,
                        RtpParams* ext_params) {
-  std::copy(cast_params.rtcp_features.begin(), cast_params.rtcp_features.end(),
-            ext_params->rtcp_features.begin());
+  std::copy(cast_params.rtcp_features.begin(),
+            cast_params.rtcp_features.end(),
+            std::back_inserter(ext_params->rtcp_features));
   FromCastRtpPayloadParams(cast_params.payload, &ext_params->payload);
 }
 
 bool ToCastRtpParamsOrThrow(v8::Isolate* isolate,
                             const RtpParams& ext_params,
                             CastRtpParams* cast_params) {
-  std::copy(ext_params.rtcp_features.begin(), ext_params.rtcp_features.end(),
-            cast_params->rtcp_features.begin());
+  std::copy(ext_params.rtcp_features.begin(),
+            ext_params.rtcp_features.end(),
+            std::back_inserter(cast_params->rtcp_features));
   if (!ToCastRtpPayloadParamsOrThrow(isolate,
                                      ext_params.payload,
                                      &cast_params->payload)) {
     return false;
   }
   return true;
 }
 
 }  // namespace
 
@@ skipping 306 matching lines @@
       base::Bind(&CastStreamingNativeHandler::CallGetRawEventsCallback,
                  weak_factory_.GetWeakPtr(),
                  transport_id));
 }
 
 void CastStreamingNativeHandler::GetStats(
     const v8::FunctionCallbackInfo<v8::Value>& args) {
   CHECK_EQ(2, args.Length());
   CHECK(args[0]->IsInt32());
   CHECK(args[1]->IsFunction());
+  const int transport_id = args[0]->ToInt32()->Value();
+  CastRtpStream* transport = GetRtpStreamOrThrow(transport_id);
+  if (!transport)
+    return;
 
-  // TODO(imcheng): Implement this.
+  linked_ptr<extensions::ScopedPersistent<v8::Function> > callback(
+      new extensions::ScopedPersistent<v8::Function>);
+  callback->reset(args[1].As<v8::Function>());
+  get_stats_callbacks_.insert(std::make_pair(transport_id, callback));
+
+  transport->GetStats(
+      base::Bind(&CastStreamingNativeHandler::CallGetStatsCallback,
+                 weak_factory_.GetWeakPtr(),
+                 transport_id));
 }
 
 void CastStreamingNativeHandler::CallGetRawEventsCallback(
     int transport_id,
     scoped_ptr<std::string> raw_events) {
   v8::Isolate* isolate = context()->isolate();
   v8::HandleScope handle_scope(isolate);
   v8::Context::Scope context_scope(context()->v8_context());
 
   RtpStreamCallbackMap::iterator it =
       get_raw_events_callbacks_.find(transport_id);
-  if (it != get_raw_events_callbacks_.end()) {
-    v8::Handle<v8::Value> callback_args[1];
-    callback_args[0] = v8::String::NewFromUtf8(isolate,
-                                               raw_events->data(),
-                                               v8::String::kNormalString,
-                                               raw_events->size());
-    context()->CallFunction(it->second->NewHandle(isolate), 1, callback_args);
-    get_raw_events_callbacks_.erase(it);
-  }
+  if (it == get_raw_events_callbacks_.end())
+    return;
+  v8::Handle<v8::Value> callback_args[1];
+  callback_args[0] = v8::String::NewFromUtf8(isolate,
+                                             raw_events->data(),
+                                             v8::String::kNormalString,
+                                             raw_events->size());
+  context()->CallFunction(it->second->NewHandle(isolate), 1, callback_args);
+  get_raw_events_callbacks_.erase(it);
+}
+
+void CastStreamingNativeHandler::CallGetStatsCallback(
+    int transport_id,
+    scoped_ptr<base::DictionaryValue> stats) {
+  v8::Isolate* isolate = context()->isolate();
+  v8::HandleScope handle_scope(isolate);
+  v8::Context::Scope context_scope(context()->v8_context());
+
+  RtpStreamCallbackMap::iterator it = get_stats_callbacks_.find(transport_id);
+  if (it == get_stats_callbacks_.end())
+    return;
+
+  scoped_ptr<V8ValueConverter> converter(V8ValueConverter::create());
+  v8::Handle<v8::Value> callback_args[1];
+  callback_args[0] = converter->ToV8Value(stats.get(), context()->v8_context());
+  context()->CallFunction(it->second->NewHandle(isolate), 1, callback_args);
+  get_stats_callbacks_.erase(it);
 }
 
 CastRtpStream* CastStreamingNativeHandler::GetRtpStreamOrThrow(
     int transport_id) const {
   RtpStreamMap::const_iterator iter = rtp_stream_map_.find(
       transport_id);
   if (iter != rtp_stream_map_.end())
     return iter->second.get();
   v8::Isolate* isolate = context()->v8_context()->GetIsolate();
   isolate->ThrowException(v8::Exception::RangeError(v8::String::NewFromUtf8(
       isolate, kRtpStreamNotFound)));
   return NULL;
 }
 
 CastUdpTransport* CastStreamingNativeHandler::GetUdpTransportOrThrow(
     int transport_id) const {
   UdpTransportMap::const_iterator iter = udp_transport_map_.find(
       transport_id);
   if (iter != udp_transport_map_.end())
     return iter->second.get();
   v8::Isolate* isolate = context()->v8_context()->GetIsolate();
   isolate->ThrowException(v8::Exception::RangeError(
       v8::String::NewFromUtf8(isolate, kUdpTransportNotFound)));
   return NULL;
 }
 
 }  // namespace extensions