[chromedriver] Implement SwitchToFrame command.

chrome/test/chromedriver/dom_tracker.cc

+// 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.
+
+#include "chrome/test/chromedriver/dom_tracker.h"
+
+#include <utility>
+
+#include "base/json/json_writer.h"
+#include "base/logging.h"
+#include "base/values.h"
+#include "chrome/test/chromedriver/status.h"
+
+DomTracker::DomTracker() {}
+
+DomTracker::~DomTracker() {}
+
+Status DomTracker::GetFrameIdForNode(
+    int node_id, std::string* frame_id) {
+  if (node_to_frame_map_.count(node_id) == 0)
+    return Status(kUnknownError, "element is not a frame");
+  *frame_id = node_to_frame_map_[node_id];
+  return Status(kOk);
+}
+
+Status DomTracker::GetContextIdForFrame(
+    const std::string& frame_id, int* context_id) {
+  if (frame_to_context_map_.count(frame_id) == 0)
+    return Status(kUnknownError, "frame does not have execution context");
+  *context_id = frame_to_context_map_[frame_id];
+  return Status(kOk);
+}
+
+void DomTracker::OnEvent(const std::string& method,
+                         const base::DictionaryValue& params) {
+  if (method == "DOM.setChildNodes") {
+    const base::Value* nodes;
+    if (!params.Get("nodes", &nodes)) {
+      LOG(ERROR) << "DOM.setChildNodes missing 'nodes'";
+      return;
+    }
+    if (!ProcessNodeList(nodes)) {
+      std::string json;
+      base::JSONWriter::Write(nodes, &json);
+      LOG(ERROR) << "DOM.setChildNodes has invalid 'nodes': " << json;
+    }
+  } else if (method == "Runtime.executionContextCreated") {
+    const base::DictionaryValue* context;
+    if (!params.GetDictionary("context", &context)) {
+      LOG(ERROR) << "Runtime.executionContextCreated missing dict 'context'";
+      return;
+    }
+    int context_id;
+    std::string frame_id;
+    if (!context->GetInteger("id", &context_id) ||
+        !context->GetString("frameId", &frame_id)) {
+      std::string json;
+      base::JSONWriter::Write(context, &json);
+      LOG(ERROR) << "Runtime.executionContextCreated has invalid 'context': "
+                 << json;
+      return;
+    }
+    frame_to_context_map_.insert(std::make_pair(frame_id, context_id));
+  } else if (method == "DOM.documentUpdated") {
+    node_to_frame_map_.clear();
+    frame_to_context_map_.clear();
+  }
+}
+
+bool DomTracker::ProcessNodeList(const base::Value* nodes) {
+  const base::ListValue* nodes_list;
+  if (!nodes->GetAsList(&nodes_list))
+    return false;
+  for (size_t i = 0; i < nodes_list->GetSize(); ++i) {
+    const base::Value* node;
+    if (!nodes_list->Get(i, &node))
+      return false;
+    if (!ProcessNode(node))
+      return false;
+  }
+  return true;
+}
+
+bool DomTracker::ProcessNode(const base::Value* node) {
+  const base::DictionaryValue* dict;
+  if (!node->GetAsDictionary(&dict))
+    return false;
+  int node_id;
+  if (!dict->GetInteger("nodeId", &node_id))
+    return false;
+  std::string frame_id;
+  if (dict->GetString("frameId", &frame_id))
+    node_to_frame_map_.insert(std::make_pair(node_id, frame_id));
+
+  const base::Value* children;
+  if (dict->Get("children", &children))
+    return ProcessNodeList(children);
+  return true;
+}