// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_BROWSER_ACCESSIBILITY_ACCESSIBILITY_TREE_FORMATTER_BLINK_H_
#define CONTENT_BROWSER_ACCESSIBILITY_ACCESSIBILITY_TREE_FORMATTER_BLINK_H_

#include <string>
#include <vector>

#include "content/common/content_export.h"
#include "ui/accessibility/platform/inspect/ax_tree_formatter_base.h"

namespace ui {
class BrowserAccessibility;
}

namespace content {

class CONTENT_EXPORT AccessibilityTreeFormatterBlink
    : public ui::AXTreeFormatterBase {
 public:
  explicit AccessibilityTreeFormatterBlink();
  ~AccessibilityTreeFormatterBlink() override;

  base::DictValue BuildTree(ui::AXPlatformNodeDelegate* root) const override;
  base::DictValue BuildTreeForSelector(
      const AXTreeSelector& selector) const override;
  base::DictValue BuildTreeForNode(ui::AXNode* node) const override;
  std::string DumpInternalAccessibilityTree(
      ui::AXTreeID tree_id,
      const std::vector<AXPropertyFilter>& property_filters) override;

 protected:
  void AddDefaultFilters(
      std::vector<AXPropertyFilter>* property_filters) override;

 private:
  void RecursiveBuildTree(const ui::BrowserAccessibility& node,
                          base::DictValue* dict) const;

  void RecursiveBuildTree(const ui::AXNode& node, base::DictValue* dict) const;

  base::DictValue BuildNode(ui::AXPlatformNodeDelegate* node) const override;

  void AddProperties(const ui::BrowserAccessibility& node,
                     base::DictValue* dict) const;

  void AddProperties(const ui::AXNode& node, base::DictValue* dict) const;

  std::string ProcessTreeForOutput(const base::DictValue& node) const override;
};

}  // namespace content

#endif  // CONTENT_BROWSER_ACCESSIBILITY_ACCESSIBILITY_TREE_FORMATTER_BLINK_H_