XmlIdCollector.cpp revision ce5e56e243d262a9b65459c3bd0bb9eaadd40628 
1/*
2 * Copyright (C) 2015 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include "compile/XmlIdCollector.h"
18
19#include <algorithm>
20#include <vector>
21
22#include "ResourceUtils.h"
23#include "ResourceValues.h"
24#include "xml/XmlDom.h"
25
26namespace aapt {
27
28namespace {
29
30static bool cmp_name(const SourcedResourceName& a, const ResourceNameRef& b) {
31  return a.name < b;
32}
33
34struct IdCollector : public xml::Visitor {
35 public:
36  using xml::Visitor::Visit;
37
38  explicit IdCollector(std::vector<SourcedResourceName>* out_symbols)
39      : out_symbols_(out_symbols) {}
40
41  void Visit(xml::Element* element) override {
42    for (xml::Attribute& attr : element->attributes) {
43      ResourceNameRef name;
44      bool create = false;
45      if (ResourceUtils::ParseReference(attr.value, &name, &create, nullptr)) {
46        if (create && name.type == ResourceType::kId) {
47          auto iter = std::lower_bound(out_symbols_->begin(),
48                                       out_symbols_->end(), name, cmp_name);
49          if (iter == out_symbols_->end() || iter->name != name) {
50            out_symbols_->insert(iter,
51                                 SourcedResourceName{name.ToResourceName(),
52                                                     element->line_number});
53          }
54        }
55      }
56    }
57
58    xml::Visitor::Visit(element);
59  }
60
61 private:
62  std::vector<SourcedResourceName>* out_symbols_;
63};
64
65}  // namespace
66
67bool XmlIdCollector::Consume(IAaptContext* context, xml::XmlResource* xmlRes) {
68  xmlRes->file.exported_symbols.clear();
69  IdCollector collector(&xmlRes->file.exported_symbols);
70  xmlRes->root->Accept(&collector);
71  return true;
72}
73
74}  // namespace aapt
75