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

#ifndef UI_ACCESSIBILITY_AX_NODE_DATA_H_
#define UI_ACCESSIBILITY_AX_NODE_DATA_H_

#include <stdint.h>

#include <memory>
#include <string>
#include <utility>
#include <vector>

#include "base/strings/string_split.h"
#include "base/strings/string_util.h"
#include "base/strings/utf_string_conversions.h"
#include "ui/accessibility/ax_attributes.h"
#include "ui/accessibility/ax_base_export.h"
#include "ui/accessibility/ax_bitset.h"
#include "ui/accessibility/ax_enums.mojom-forward.h"
#include "ui/accessibility/ax_node_id_forward.h"
#include "ui/accessibility/ax_relative_bounds.h"
#include "ui/accessibility/ax_states.h"
#include "ui/accessibility/ax_text_attributes.h"
#include "ui/gfx/geometry/rect_f.h"

namespace ui {

class AXTreeID;

// Return true if |attr| should be interpreted as the id of another node
// in the same tree.
AX_BASE_EXPORT bool IsNodeIdIntAttribute(ax::mojom::IntAttribute attr);

// Return true if |attr| should be interpreted as a list of ids of
// nodes in the same tree.
AX_BASE_EXPORT bool IsNodeIdIntListAttribute(ax::mojom::IntListAttribute attr);

// A compact representation of the accessibility information for a
// single accessible object, in a form that can be serialized and sent from
// one process to another.
struct AX_BASE_EXPORT AXNodeData final {
  // Defines the type used for AXNode IDs.
  using AXID = AXNodeID;

  // If a node is not yet or no longer valid, its ID should have a value of
  // kInvalidAXID.
  static constexpr AXID kInvalidAXID = kInvalidAXNodeID;

  static constexpr bool kDefaultBoolValue = false;
  static constexpr int kDefaultIntValue = 0;
  static constexpr int kDefaultFloatValue = 0;

  AXNodeData();
  ~AXNodeData();

  AXNodeData(const AXNodeData& other);
  AXNodeData(AXNodeData&& other);
  AXNodeData& operator=(const AXNodeData& other);

  // Accessing accessibility attributes:
  //
  // There are dozens of possible attributes for an accessibility node, but only
  // a few tend to apply to any one object, so we store them in sorted sparse
  // arrays of <attribute id, attribute value> pairs, organized by type (bool,
  // int, float, string, int list, string list).
  //
  // There are two accessors for each type of attribute: one that returns true
  // if the attribute is present and false if not, and another that returns the
  // default value for that type (e.g., false for a bool or an empty container
  // for a string or list) if the attribute is not present. In addition, strings
  // can be returned as either std::string or std::u16string, for convenience.

  bool HasBoolAttribute(ax::mojom::BoolAttribute attribute) const {
    return bool_attributes.Get(attribute).has_value();
  }
  bool GetBoolAttribute(ax::mojom::BoolAttribute attribute) const {
    return bool_attributes.Get(attribute).value_or(false);
  }
  bool HasFloatAttribute(ax::mojom::FloatAttribute attribute) const {
    return float_attributes.Has(attribute);
  }
  float GetFloatAttribute(ax::mojom::FloatAttribute attribute) const {
    return float_attributes.Get(attribute);
  }
  bool HasIntAttribute(ax::mojom::IntAttribute attribute) const {
    return int_attributes.Has(attribute);
  }
  int GetIntAttribute(ax::mojom::IntAttribute attribute) const {
    return int_attributes.Get(attribute);
  }
  bool HasStringAttribute(ax::mojom::StringAttribute attribute) const {
    return string_attributes.Has(attribute);
  }
  const std::string& GetStringAttribute(
      ax::mojom::StringAttribute attribute) const {
    return string_attributes.Get(attribute);
  }
  std::u16string GetString16Attribute(
      ax::mojom::StringAttribute attribute) const {
    return base::UTF8ToUTF16(GetStringAttribute(attribute));
  }
  bool HasIntListAttribute(ax::mojom::IntListAttribute attribute) const {
    return intlist_attributes.Has(attribute);
  }
  const std::vector<int32_t>& GetIntListAttribute(
      ax::mojom::IntListAttribute attribute) const {
    return intlist_attributes.Get(attribute);
  }
  bool HasStringListAttribute(ax::mojom::StringListAttribute attribute) const {
    return stringlist_attributes.Has(attribute);
  }
  const std::vector<std::string>& GetStringListAttribute(
      ax::mojom::StringListAttribute attribute) const {
    return stringlist_attributes.Get(attribute);
  }

  bool HasHtmlAttribute(const char* attribute) const;
  const std::string& GetHtmlAttribute(const char* attribute) const;
  std::u16string GetHtmlAttributeUTF16(const char* attribute) const;
  const std::string* FindHtmlAttribute(const char* attribute) const;

  //
  // Setting accessibility attributes.
  //
  // Replaces an attribute if present. This is safer than crashing via a DCHECK
  // or doing nothing, because most likely replacing is what the caller would
  // have wanted or what existing code already assumes.
  //

  void AddBoolAttribute(ax::mojom::BoolAttribute attribute, bool value) {
    bool_attributes.Set(attribute, value);
  }
  void AddChildTreeId(const AXTreeID& tree_id);
  void AddIntAttribute(ax::mojom::IntAttribute attribute, int32_t value) {
    int_attributes.Set(attribute, value);
  }
  void AddFloatAttribute(ax::mojom::FloatAttribute attribute, float value) {
    float_attributes.Set(attribute, value);
  }
  // This method cannot be used to set kChildTreeId due to a common
  // misuse of base::UnguessableToken serialization. Use AddChildTreeId instead.
  void AddStringAttribute(ax::mojom::StringAttribute attribute,
                          const std::string& value) {
    string_attributes.Set(attribute, value);
  }
  void AddIntListAttribute(ax::mojom::IntListAttribute attribute,
                           const std::vector<int32_t>& value) {
    intlist_attributes.Set(attribute, value);
  }
  void AddStringListAttribute(ax::mojom::StringListAttribute attribute,
                              const std::vector<std::string>& value) {
    stringlist_attributes.Set(attribute, value);
  }

  //
  // Removing accessibility attributes.
  //

  void RemoveBoolAttribute(ax::mojom::BoolAttribute attribute) {
    bool_attributes.Unset(attribute);
  }
  void RemoveIntAttribute(ax::mojom::IntAttribute attribute) {
    int_attributes.Remove(attribute);
  }
  void RemoveFloatAttribute(ax::mojom::FloatAttribute attribute) {
    float_attributes.Remove(attribute);
  }
  void RemoveStringAttribute(ax::mojom::StringAttribute attribute) {
    string_attributes.Remove(attribute);
  }
  void RemoveIntListAttribute(ax::mojom::IntListAttribute attribute) {
    intlist_attributes.Remove(attribute);
  }
  void RemoveStringListAttribute(ax::mojom::StringListAttribute attribute) {
    stringlist_attributes.Remove(attribute);
  }

  //
  // Text attributes, such as spelling markers and style information.
  //

  AXTextAttributes GetTextAttributes() const;

  //
  // Convenience functions.
  //

  // Return the DOMNodeID, if this object was associated with a DOM Node in
  // an HTML renderer, otherwise return 0.
  int GetDOMNodeId() const;

  // Adds the name attribute or replaces it if already present. Also sets the
  // NameFrom attribute if not already set.
  //
  // [[deprecated("Replaced by `SetNameChecked` and `SetNameExplicitlyEmpty`")]]
  // See `SetNameChecked` and `SetNameExplicitlyEmpty` which have DCHECKs for
  // conditions expected to be true, which in reality are not always true.
  // Tracked by crbug.com/1348081.
  void SetName(const std::string& name);
  // [[deprecated("Replaced by `SetNameChecked` and `SetNameExplicitlyEmpty`")]]
  // See `SetNameChecked` and `SetNameExplicitlyEmpty` which have DCHECKs for
  // conditions expected to be true, which in reality are not always true.
  // Tracked by crbug.com/1348081.
  void SetName(const std::u16string& name);

  // Adds the accessible name attribute or replaces it if already present, and
  // also sets the NameFrom attribute if not already set.
  //
  // The value of the accessible name is a localized, end-user-consumable string
  // which may be derived from visible information (e.g. the text on a button)
  // or invisible information (e.g. the alternative text describing an icon).
  // In the case of focusable objects, the name will be presented by the screen
  // reader when that object gains focus and is critical to understanding the
  // purpose of that object non-visually.
  //
  // Note that `SetNameChecked` must only be used to set a non-empty name, a
  // condition enforced by a DCHECK. This is done to prevent UI from
  // accidentally being given an empty name because, as a general rule, nameless
  // controls tend to be inaccessible. However, because there can be valid
  // reasons to remove or prevent naming of an item `SetNameExplicitlyEmpty`
  // provides a means for developers to do so.
  void SetNameChecked(const std::string& name);
  void SetNameChecked(const std::u16string& name);

  // Indicates this object should not have an accessible name. One use case is
  // to prevent screen readers from speaking redundant information, for instance
  // if the parent View has the same name as this View, causing the screen
  // reader to speak the name twice. This function can also be used to allow
  // focusable nameless objects to pass accessibility checks in tests, a
  // practice that should not be applied in production code.
  void SetNameExplicitlyEmpty();

  // Adds the description attribute or replaces it if already present. Also
  // sets the DescriptionFrom attribute if not already set. Note that
  // `SetDescription` must only be used to set a non-empty description, a
  // condition enforced by a DCHECK. If an object should not have an accessible
  // description in order to improve the user experience, use
  // `SetDescriptionExplicitlyEmpty`.
  void SetDescription(const std::string& description);
  void SetDescription(const std::u16string& description);

  // Indicates this object should not have an accessible description. One use
  // case is to prevent screen readers from speaking redundant information, for
  // instance if a View's description comes from a tooltip whose content is
  // similar to that View's accessible name, the screen reader presentation may
  // be overly verbose.
  void SetDescriptionExplicitlyEmpty();

  // Adds the value attribute or replaces it if already present.
  void SetValue(const std::string& value);
  void SetValue(const std::u16string& value);

  // Returns the field of state bits.
  AXStates GetStates() const { return state; }

  // Returns true if the given enum bit is 1.
  bool HasState(ax::mojom::State a_state) const {
    return ui::HasState(state, a_state);
  }
  bool HasAction(ax::mojom::Action action) const;
  bool HasTextStyle(ax::mojom::TextStyle text_style) const;

  // Set or remove bits in the given enum's corresponding bitfield.
  void AddState(ax::mojom::State a_state) { ui::AddState(state, a_state); }
  void RemoveState(ax::mojom::State a_state) {
    ui::RemoveState(state, a_state);
  }
  void AddAction(ax::mojom::Action action);
  void RemoveAction(ax::mojom::Action action);
  void AddTextStyle(ax::mojom::TextStyle text_style);

  // Helper functions to get or set some common int attributes with some
  // specific enum types. To remove an attribute, set it to None.
  //
  // Please keep in alphabetic order.
  ax::mojom::CheckedState GetCheckedState() const;
  void SetCheckedState(ax::mojom::CheckedState checked_state);
  bool HasCheckedState() const;
  ax::mojom::DefaultActionVerb GetDefaultActionVerb() const;
  void SetDefaultActionVerb(ax::mojom::DefaultActionVerb default_action_verb);
  ax::mojom::HasPopup GetHasPopup() const;
  void SetHasPopup(ax::mojom::HasPopup has_popup);
  ax::mojom::IsPopup GetIsPopup() const;
  void SetIsPopup(ax::mojom::IsPopup is_popup);
  ax::mojom::InvalidState GetInvalidState() const;
  void SetInvalidState(ax::mojom::InvalidState invalid_state);
  ax::mojom::NameFrom GetNameFrom() const;
  void SetNameFrom(ax::mojom::NameFrom name_from);
  ax::mojom::DescriptionFrom GetDescriptionFrom() const;
  void SetDescriptionFrom(ax::mojom::DescriptionFrom description_from);
  ax::mojom::DetailsFrom GetDetailsFrom() const;
  void SetDetailsFrom(ax::mojom::DetailsFrom details_from);
  ax::mojom::TextPosition GetTextPosition() const;
  void SetTextPosition(ax::mojom::TextPosition text_position);
  ax::mojom::Restriction GetRestriction() const;
  void SetRestriction(ax::mojom::Restriction restriction);
  ax::mojom::ListStyle GetListStyle() const;
  void SetListStyle(ax::mojom::ListStyle list_style);
  int GetPaintOrder() const;
  ax::mojom::TextAlign GetTextAlign() const;
  void SetTextAlign(ax::mojom::TextAlign text_align);
  ax::mojom::WritingDirection GetTextDirection() const;
  void SetTextDirection(ax::mojom::WritingDirection text_direction);
  ax::mojom::ImageAnnotationStatus GetImageAnnotationStatus() const;
  void SetImageAnnotationStatus(ax::mojom::ImageAnnotationStatus status);

  // Helper to determine if the data belongs to a node that gains focus when
  // clicked, such as a text field or a native HTML list box.
  bool IsActivatable() const;

  // Helper to determine if the data belongs to a node that is at the root of an
  // ARIA live region that is active, i.e. its status is not set to "off".
  bool IsActiveLiveRegionRoot() const;

  // Helper to determine if the data belongs to a node that is a native button
  // or ARIA role="button" in a pressed state.
  bool IsButtonPressed() const;

  // Helper to determine if the data belongs to a node that can respond to
  // clicks.
  bool IsClickable() const;

  // Helper to determine if the data belongs to a node that is part of an active
  // ARIA live region, and for which live announcements should be made.
  bool IsContainedInActiveLiveRegion() const;

  // Helper to determine if the object is selectable.
  bool IsSelectable() const;

  // Helper to determine if the data has the ignored state or ignored role.
  bool IsIgnored() const;

  // Helper to determine if the data has the invisible state.
  bool IsInvisible() const;

  // Helper to determine if the data has the ignored state, the invisible state
  // or the ignored role.
  bool IsInvisibleOrIgnored() const;

  // Helper to determine if the data belongs to a node that is invocable.
  bool IsInvocable() const;

  // Helper to determine if the data belongs to a node that is a menu button.
  bool IsMenuButton() const;

  // This data belongs to a text field. This is any widget in which the user
  // should be able to enter and edit text.
  //
  // Examples include <input type="text">, <input type="password">, <textarea>,
  // <div contenteditable="true">, <div role="textbox">, <div role="searchbox">
  // and <div role="combobox">. Note that when an ARIA role that indicates that
  // the widget is editable is used, such as "role=textbox", the element doesn't
  // need to be contenteditable for this method to return true, as in theory
  // JavaScript could be used to implement editing functionality. In practice,
  // this situation should be rare.
  bool IsTextField() const;

  // This data belongs to a text field that is used for entering passwords.
  bool IsPasswordField() const;

  // This data belongs to an atomic text field. An atomic text field does not
  // expose its internal implementation to assistive software, appearing as a
  // single leaf node in the accessibility tree. Examples include: An <input> or
  // a <textarea> on the Web, a text field in a PDF form, a Views-based text
  // field, or a native Android one.
  bool IsAtomicTextField() const;

  // This data belongs to a text field whose value is exposed both on the field
  // itself as well as on descendant nodes which are expose to platform
  // accessibility APIs. A non-native text field also exposes stylistic and
  // document marker information on descendant nodes. Examples include fields
  // created using the CSS "user-modify" property, or the "contenteditable"
  // attribute.
  bool IsNonAtomicTextField() const;

  // Any element that has `spinbutton` set on the root editable element should
  // be treated as a SpinnerTextField.
  // For example, <input type="text" role=spinbutton> is a spinner text field.
  // Richly editable elements should be treated as spinners when they have
  // their roles set to `spinbutton` and when they are not the descendant of a
  // <contenteditable> element.
  bool IsSpinnerTextField() const;

  // Helper to determine if the data belongs to a node that supports
  // range-based values.
  bool IsRangeValueSupported() const;

  // Helper to determine if the data belongs to a node that supports
  // expand/collapse.
  bool SupportsExpandCollapse() const;

  bool HasChildTreeID() const;
  std::optional<AXTreeID> GetChildTreeID() const;

  // Return a string representation of this data, for debugging.
  std::string ToString(bool verbose = true) const;

  // Returns the approximate size in bytes.
  size_t ByteSize() const;

  struct AX_BASE_EXPORT AXNodeDataSize {
    size_t int_attribute_size = 0;
    size_t float_attribute_size = 0;
    size_t bool_attribute_size = 0;
    size_t string_attribute_size = 0;
    size_t int_list_attribhute_size = 0;
    size_t string_list_attribute_size = 0;
    size_t html_attribute_size = 0;
    size_t child_ids_size = 0;

    size_t ByteSize() const;
  };
  void AccumulateSize(AXNodeDataSize& node_data_size) const;

  // Return a string representation of |aria-dropeffect| values, for testing
  // and debugging.
  // aria-dropeffect is deprecated in WAI-ARIA 1.1.
  std::string DropeffectBitfieldToString() const;

  // As much as possible this should behave as a simple, serializable,
  // copyable struct.
  AXNodeID id = kInvalidAXNodeID;
  ax::mojom::Role role;
  AXStates state{0U};
  uint64_t actions = 0ULL;
  AXStringAttributes string_attributes;
  AXIntAttributes int_attributes;
  AXFloatAttributes float_attributes;
  AXBitset<ax::mojom::BoolAttribute> bool_attributes;
  AXIntListAttributes intlist_attributes;
  AXStringListAttributes stringlist_attributes;
  base::StringPairs html_attributes;
  std::vector<int32_t> child_ids;

  AXRelativeBounds relative_bounds;
};

}  // namespace ui

#endif  // UI_ACCESSIBILITY_AX_NODE_DATA_H_