blob: bb80805aaeb2dcf351af107be439a68a2c057328 [file] [log] [blame]
// Copyright 2019 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 "ax_tree_manager_map.h"
#include "ax_enums.h"
#include "base/container_utils.h"
#include "base/logging.h"
namespace ui {
AXTreeManagerMap::AXTreeManagerMap() {}
AXTreeManagerMap::~AXTreeManagerMap() {}
AXTreeManagerMap& AXTreeManagerMap::GetInstance() {
static base::NoDestructor<AXTreeManagerMap> instance;
return *instance;
}
void AXTreeManagerMap::AddTreeManager(AXTreeID tree_id,
AXTreeManager* manager) {
if (tree_id != AXTreeIDUnknown())
map_[tree_id] = manager;
}
void AXTreeManagerMap::RemoveTreeManager(AXTreeID tree_id) {
if (tree_id != AXTreeIDUnknown())
map_.erase(tree_id);
}
AXTreeManager* AXTreeManagerMap::GetManager(AXTreeID tree_id) {
if (tree_id == AXTreeIDUnknown() || !base::Contains(map_, tree_id))
return nullptr;
return map_.at(tree_id);
}
AXTreeManager* AXTreeManagerMap::GetManagerForChildTree(
const AXNode& parent_node) {
if (!parent_node.data().HasStringAttribute(
ax::mojom::StringAttribute::kChildTreeId)) {
return nullptr;
}
AXTreeID child_tree_id =
AXTreeID::FromString(parent_node.data().GetStringAttribute(
ax::mojom::StringAttribute::kChildTreeId));
AXTreeManager* child_tree_manager =
AXTreeManagerMap::GetInstance().GetManager(child_tree_id);
// Some platforms do not use AXTreeManagers, so child trees don't exist in
// the browser process.
if (!child_tree_manager)
return nullptr;
BASE_DCHECK(child_tree_manager->GetParentNodeFromParentTreeAsAXNode()->id() ==
parent_node.id());
return child_tree_manager;
}
} // namespace ui