enabling network emulation in devtools

brightray/browser/net/devtools_network_controller.cc

@@ -0,0 +1,74 @@
+// Copyright (c) 2015 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-CHROMIUM file.
+
+#include "browser/net/devtools_network_controller.h"
+
+#include "browser/net/devtools_network_conditions.h"
+#include "browser/net/devtools_network_interceptor.h"
+#include "browser/net/devtools_network_transaction.h"
+
+#include "base/bind.h"
+#include "content/public/browser/browser_thread.h"
+
+using content::BrowserThread;
+
+namespace brightray {
+
+DevToolsNetworkController::DevToolsNetworkController()
+    : default_interceptor_(new DevToolsNetworkInterceptor) {
+}
+
+DevToolsNetworkController::~DevToolsNetworkController() {
+}
+
+void DevToolsNetworkController::SetNetworkState(
+    const std::string& client_id,
+    scoped_ptr<DevToolsNetworkConditions> conditions) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+
+  if (client_id.empty()) {
+    if (!conditions)
+      return;
+    default_interceptor_->UpdateConditions(conditions.Pass());
+    return;
+  }
+
+  auto interceptor = interceptors_.get(client_id);
+  if (!interceptor) {
+    if (!conditions)
+      return;
+    scoped_ptr<DevToolsNetworkInterceptor> new_interceptor(
+        new DevToolsNetworkInterceptor);
+    new_interceptor->UpdateConditions(conditions.Pass());
+    interceptors_.set(client_id, new_interceptor.Pass());
+  } else {
+    if (!conditions) {
+      scoped_ptr<DevToolsNetworkConditions> online_conditions(
+          new DevToolsNetworkConditions(false));
+      interceptor->UpdateConditions(online_conditions.Pass());
+      interceptors_.erase(client_id);
+    } else {
+      interceptor->UpdateConditions(conditions.Pass());
+    }
+  }
+}
+
+base::WeakPtr<DevToolsNetworkInterceptor>
+DevToolsNetworkController::GetInterceptor(DevToolsNetworkTransaction* transaction) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+  DCHECK(transaction->request());
+
+  if (!interceptors_.size())
+    return default_interceptor_->GetWeakPtr();
+
+  transaction->ProcessRequest();
+  auto& client_id = transaction->client_id();
+  auto interceptor = interceptors_.get(client_id);
+  if (!interceptor)
+    return default_interceptor_->GetWeakPtr();
+
+  return interceptor->GetWeakPtr();
+}
+
+}  // namespace brightray