// Copyright 2016 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.

#ifndef COMPONENTS_UI_DEVTOOLS_VIEWS_DOM_AGENT_VIEWS_H_
#define COMPONENTS_UI_DEVTOOLS_VIEWS_DOM_AGENT_VIEWS_H_

#include "components/ui_devtools/DOM.h"
#include "components/ui_devtools/dom_agent.h"

namespace ui_devtools {

class DOMAgentViews : public DOMAgent {
 public:
  ~DOMAgentViews() override;
  static std::unique_ptr<DOMAgentViews> Create();

 protected:
  DOMAgentViews();

  virtual std::unique_ptr<protocol::DOM::Node> BuildTreeForWindow(
      UIElement* window_element_root) = 0;
  std::unique_ptr<protocol::DOM::Node> BuildTreeForRootWidget(
      UIElement* widget_element);
  std::unique_ptr<protocol::DOM::Node> BuildTreeForView(
      UIElement* view_element);

  std::unique_ptr<protocol::DOM::Node> BuildTreeForUIElement(
      UIElement* ui_element) override;

 private:
  DISALLOW_COPY_AND_ASSIGN(DOMAgentViews);
};

}  // namespace ui_devtools

#endif  // COMPONENTS_UI_DEVTOOLS_VIEWS_DOM_AGENT_VIEWS_H_