DevTools: Add geolocation override in browser

content/browser/devtools/protocol/page_handler.cc

 // 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.
 
 #include "content/browser/devtools/protocol/page_handler.h"
 
 #include <string>
 
 #include "base/base64.h"
 #include "base/bind.h"
 #include "base/strings/string16.h"
 #include "base/strings/utf_string_conversions.h"
 #include "content/browser/devtools/protocol/color_picker.h"
+#include "content/browser/geolocation/geolocation_dispatcher_host.h"
 #include "content/browser/renderer_host/render_view_host_impl.h"
 #include "content/browser/renderer_host/render_widget_host_view_base.h"
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/common/view_messages.h"
 #include "content/public/browser/javascript_dialog_manager.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/web_contents_delegate.h"
 #include "content/public/common/referrer.h"
 #include "content/public/common/url_constants.h"
@@ skipping 172 matching lines @@
   for (int i = 0; i != controller.GetEntryCount(); ++i) {
     if (controller.GetEntryAtIndex(i)->GetUniqueID() == entry_id) {
       controller.GoToIndex(i);
       return Response::OK();
     }
   }
 
   return Response::InvalidParams("No entry with passed id");
 }
 
+Response PageHandler::SetGeolocationOverride(double* latitude,
+                                             double* longitude,
+                                             double* accuracy) {
+  WebContentsImpl* web_contents = static_cast<WebContentsImpl*>(
+      WebContents::FromRenderViewHost(host_));
+  if (!web_contents)
+    return Response::InternalError("No WebContents to override");
+  GeolocationDispatcherHost* geolocation_host =
+      web_contents->geolocation_dispatcher_host();
+  scoped_ptr<Geoposition> geoposition(new Geoposition());
+  if (latitude && longitude && accuracy) {
+    geoposition->latitude = *latitude;
+    geoposition->longitude = *longitude;
+    geoposition->accuracy = *accuracy;
+    geoposition->timestamp = base::Time::Now();
+  } else {
+    geoposition->error_code = Geoposition::ERROR_CODE_POSITION_UNAVAILABLE;
+  }
+  geolocation_host->SetOverride(geoposition.Pass());
+  return Response::OK();
+}
+
+Response PageHandler::ClearGeolocationOverride() {
+  WebContentsImpl* web_contents = static_cast<WebContentsImpl*>(
+      WebContents::FromRenderViewHost(host_));
+  if (!web_contents)
+    return Response::InternalError("No WebContents to override");
+  GeolocationDispatcherHost* geolocation_host =
+      web_contents->geolocation_dispatcher_host();
+  geolocation_host->ClearOverride();
+  return Response::OK();
+}
+
+
 Response PageHandler::SetTouchEmulationEnabled(bool enabled) {
   touch_emulation_enabled_ = enabled;
   UpdateTouchEventEmulationState();
   return Response::FallThrough();
 }
 
 scoped_refptr<DevToolsProtocol::Response> PageHandler::CaptureScreenshot(
     scoped_refptr<DevToolsProtocol::Command> command) {
   if (!host_ || !host_->GetView())
     return command->InternalErrorResponse("Could not connect to view");
@@ skipping 287 matching lines @@
   color.set_b(b);
   color.set_a(a);
   ColorPickedParams params;
   params.set_color(color);
   client_->ColorPicked(params);
 }
 
 }  // namespace page
 }  // namespace devtools
 }  // namespace content