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 "java/JavaClassGenerator.h"
18
19 #include <algorithm>
20 #include <ostream>
21 #include <set>
22 #include <sstream>
23 #include <tuple>
24
25 #include "android-base/errors.h"
26 #include "android-base/logging.h"
27 #include "android-base/stringprintf.h"
28 #include "androidfw/StringPiece.h"
29
30 #include "NameMangler.h"
31 #include "Resource.h"
32 #include "ResourceTable.h"
33 #include "ResourceValues.h"
34 #include "SdkConstants.h"
35 #include "ValueVisitor.h"
36 #include "java/AnnotationProcessor.h"
37 #include "java/ClassDefinition.h"
38 #include "process/SymbolTable.h"
39
40 using ::aapt::io::OutputStream;
41 using ::aapt::text::Printer;
42 using ::android::StringPiece;
43 using ::android::base::StringPrintf;
44
45 namespace aapt {
46
47 static const std::set<StringPiece> sJavaIdentifiers = {
48 "abstract", "assert", "boolean", "break", "byte",
49 "case", "catch", "char", "class", "const",
50 "continue", "default", "do", "double", "else",
51 "enum", "extends", "final", "finally", "float",
52 "for", "goto", "if", "implements", "import",
53 "instanceof", "int", "interface", "long", "native",
54 "new", "package", "private", "protected", "public",
55 "return", "short", "static", "strictfp", "super",
56 "switch", "synchronized", "this", "throw", "throws",
57 "transient", "try", "void", "volatile", "while",
58 "true", "false", "null"};
59
IsValidSymbol(const StringPiece & symbol)60 static bool IsValidSymbol(const StringPiece& symbol) {
61 return sJavaIdentifiers.find(symbol) == sJavaIdentifiers.end();
62 }
63
64 // Java symbols can not contain . or -, but those are valid in a resource name.
65 // Replace those with '_'.
TransformToFieldName(const StringPiece & symbol)66 std::string JavaClassGenerator::TransformToFieldName(const StringPiece& symbol) {
67 std::string output = symbol.to_string();
68 for (char& c : output) {
69 if (c == '.' || c == '-') {
70 c = '_';
71 }
72 }
73 return output;
74 }
75
76 // Transforms an attribute in a styleable to the Java field name:
77 //
78 // <declare-styleable name="Foo">
79 // <attr name="android:bar" />
80 // <attr name="bar" />
81 // </declare-styleable>
82 //
83 // Foo_android_bar
84 // Foo_bar
TransformNestedAttr(const ResourceNameRef & attr_name,const std::string & styleable_class_name,const StringPiece & package_name_to_generate)85 static std::string TransformNestedAttr(const ResourceNameRef& attr_name,
86 const std::string& styleable_class_name,
87 const StringPiece& package_name_to_generate) {
88 std::string output = styleable_class_name;
89
90 // We may reference IDs from other packages, so prefix the entry name with
91 // the package.
92 if (!attr_name.package.empty() &&
93 package_name_to_generate != attr_name.package) {
94 output += "_" + JavaClassGenerator::TransformToFieldName(attr_name.package);
95 }
96 output += "_" + JavaClassGenerator::TransformToFieldName(attr_name.entry);
97 return output;
98 }
99
AddAttributeFormatDoc(AnnotationProcessor * processor,Attribute * attr)100 static void AddAttributeFormatDoc(AnnotationProcessor* processor, Attribute* attr) {
101 const uint32_t type_mask = attr->type_mask;
102 if (type_mask & android::ResTable_map::TYPE_REFERENCE) {
103 processor->AppendComment(
104 "<p>May be a reference to another resource, in the form\n"
105 "\"<code>@[+][<i>package</i>:]<i>type</i>/<i>name</i></code>\" or a "
106 "theme\n"
107 "attribute in the form\n"
108 "\"<code>?[<i>package</i>:]<i>type</i>/<i>name</i></code>\".");
109 }
110
111 if (type_mask & android::ResTable_map::TYPE_STRING) {
112 processor->AppendComment(
113 "<p>May be a string value, using '\\\\;' to escape characters such as\n"
114 "'\\\\n' or '\\\\uxxxx' for a unicode character;");
115 }
116
117 if (type_mask & android::ResTable_map::TYPE_INTEGER) {
118 processor->AppendComment(
119 "<p>May be an integer value, such as \"<code>100</code>\".");
120 }
121
122 if (type_mask & android::ResTable_map::TYPE_BOOLEAN) {
123 processor->AppendComment(
124 "<p>May be a boolean value, such as \"<code>true</code>\" or\n"
125 "\"<code>false</code>\".");
126 }
127
128 if (type_mask & android::ResTable_map::TYPE_COLOR) {
129 processor->AppendComment(
130 "<p>May be a color value, in the form of "
131 "\"<code>#<i>rgb</i></code>\",\n"
132 "\"<code>#<i>argb</i></code>\", \"<code>#<i>rrggbb</i></code>\", or \n"
133 "\"<code>#<i>aarrggbb</i></code>\".");
134 }
135
136 if (type_mask & android::ResTable_map::TYPE_FLOAT) {
137 processor->AppendComment(
138 "<p>May be a floating point value, such as \"<code>1.2</code>\".");
139 }
140
141 if (type_mask & android::ResTable_map::TYPE_DIMENSION) {
142 processor->AppendComment(
143 "<p>May be a dimension value, which is a floating point number "
144 "appended with a\n"
145 "unit such as \"<code>14.5sp</code>\".\n"
146 "Available units are: px (pixels), dp (density-independent pixels),\n"
147 "sp (scaled pixels based on preferred font size), in (inches), and\n"
148 "mm (millimeters).");
149 }
150
151 if (type_mask & android::ResTable_map::TYPE_FRACTION) {
152 processor->AppendComment(
153 "<p>May be a fractional value, which is a floating point number "
154 "appended with\n"
155 "either % or %p, such as \"<code>14.5%</code>\".\n"
156 "The % suffix always means a percentage of the base size;\n"
157 "the optional %p suffix provides a size relative to some parent "
158 "container.");
159 }
160
161 if (type_mask &
162 (android::ResTable_map::TYPE_FLAGS | android::ResTable_map::TYPE_ENUM)) {
163 if (type_mask & android::ResTable_map::TYPE_FLAGS) {
164 processor->AppendComment(
165 "<p>Must be one or more (separated by '|') of the following "
166 "constant values.</p>");
167 } else {
168 processor->AppendComment(
169 "<p>Must be one of the following constant values.</p>");
170 }
171
172 processor->AppendComment(
173 "<table>\n<colgroup align=\"left\" />\n"
174 "<colgroup align=\"left\" />\n"
175 "<colgroup align=\"left\" />\n"
176 "<tr><th>Constant</th><th>Value</th><th>Description</th></tr>\n");
177 for (const Attribute::Symbol& symbol : attr->symbols) {
178 std::stringstream line;
179 line << "<tr><td>" << symbol.symbol.name.value().entry << "</td>"
180 << "<td>" << std::hex << symbol.value << std::dec << "</td>"
181 << "<td>" << util::TrimWhitespace(symbol.symbol.GetComment())
182 << "</td></tr>";
183 processor->AppendComment(line.str());
184 }
185 processor->AppendComment("</table>");
186 }
187 }
188
JavaClassGenerator(IAaptContext * context,ResourceTable * table,const JavaClassGeneratorOptions & options)189 JavaClassGenerator::JavaClassGenerator(IAaptContext* context,
190 ResourceTable* table,
191 const JavaClassGeneratorOptions& options)
192 : context_(context), table_(table), options_(options) {}
193
SkipSymbol(Visibility::Level level)194 bool JavaClassGenerator::SkipSymbol(Visibility::Level level) {
195 switch (options_.types) {
196 case JavaClassGeneratorOptions::SymbolTypes::kAll:
197 return false;
198 case JavaClassGeneratorOptions::SymbolTypes::kPublicPrivate:
199 return level == Visibility::Level::kUndefined;
200 case JavaClassGeneratorOptions::SymbolTypes::kPublic:
201 return level != Visibility::Level::kPublic;
202 }
203 return true;
204 }
205
206 // Whether or not to skip writing this symbol.
SkipSymbol(const Maybe<SymbolTable::Symbol> & symbol)207 bool JavaClassGenerator::SkipSymbol(const Maybe<SymbolTable::Symbol>& symbol) {
208 return !symbol || (options_.types == JavaClassGeneratorOptions::SymbolTypes::kPublic &&
209 !symbol.value().is_public);
210 }
211
212 struct StyleableAttr {
213 const Reference* attr_ref = nullptr;
214 std::string field_name;
215 Maybe<SymbolTable::Symbol> symbol;
216 };
217
operator <(const StyleableAttr & lhs,const StyleableAttr & rhs)218 static bool operator<(const StyleableAttr& lhs, const StyleableAttr& rhs) {
219 const ResourceId lhs_id = lhs.attr_ref->id.value_or_default(ResourceId(0));
220 const ResourceId rhs_id = rhs.attr_ref->id.value_or_default(ResourceId(0));
221 if (lhs_id == rhs_id) {
222 return lhs.attr_ref->name.value() < rhs.attr_ref->name.value();
223 }
224 return cmp_ids_dynamic_after_framework(lhs_id, rhs_id);
225 }
226
GetRFieldReference(const ResourceName & name,StringPiece fallback_package_name)227 static FieldReference GetRFieldReference(const ResourceName& name,
228 StringPiece fallback_package_name) {
229 const std::string package_name =
230 name.package.empty() ? fallback_package_name.to_string() : name.package;
231 const std::string entry = JavaClassGenerator::TransformToFieldName(name.entry);
232 return FieldReference(
233 StringPrintf("%s.R.%s.%s", package_name.c_str(), to_string(name.type).data(), entry.c_str()));
234 }
235
ProcessStyleable(const ResourceNameRef & name,const ResourceId & id,const Styleable & styleable,const StringPiece & package_name_to_generate,ClassDefinition * out_class_def,MethodDefinition * out_rewrite_method,Printer * r_txt_printer)236 bool JavaClassGenerator::ProcessStyleable(const ResourceNameRef& name, const ResourceId& id,
237 const Styleable& styleable,
238 const StringPiece& package_name_to_generate,
239 ClassDefinition* out_class_def,
240 MethodDefinition* out_rewrite_method,
241 Printer* r_txt_printer) {
242 const std::string array_field_name = TransformToFieldName(name.entry);
243 std::unique_ptr<ResourceArrayMember> array_def =
244 util::make_unique<ResourceArrayMember>(array_field_name);
245
246 // The array must be sorted by resource ID.
247 std::vector<StyleableAttr> sorted_attributes;
248 sorted_attributes.reserve(styleable.entries.size());
249 for (const auto& attr : styleable.entries) {
250 // If we are not encoding final attributes, the styleable entry may have no
251 // ID if we are building a static library.
252 CHECK(!options_.use_final || attr.id) << "no ID set for Styleable entry";
253 CHECK(bool(attr.name)) << "no name set for Styleable entry";
254
255 // We will need the unmangled, transformed name in the comments and the field,
256 // so create it once and cache it in this StyleableAttr data structure.
257 StyleableAttr styleable_attr;
258 styleable_attr.attr_ref = &attr;
259
260 // The field name for this attribute is prefixed by the name of this styleable and
261 // the package it comes from.
262 styleable_attr.field_name =
263 TransformNestedAttr(attr.name.value(), array_field_name, package_name_to_generate);
264
265 Reference ref = attr;
266 if (attr.name.value().package.empty()) {
267
268 // If the resource does not have a package name, set the package to the unmangled package name
269 // of the styleable declaration because attributes without package names would have been
270 // declared in the same package as the styleable.
271 ref.name = ResourceName(package_name_to_generate, ref.name.value().type,
272 ref.name.value().entry);
273 }
274
275 // Look up the symbol so that we can write out in the comments what are possible legal values
276 // for this attribute.
277 const SymbolTable::Symbol* symbol = context_->GetExternalSymbols()->FindByReference(ref);
278
279 if (symbol && symbol->attribute) {
280 // Copy the symbol data structure because the returned instance can be destroyed.
281 styleable_attr.symbol = *symbol;
282 }
283 sorted_attributes.push_back(std::move(styleable_attr));
284 }
285
286 // Sort the attributes by ID.
287 std::sort(sorted_attributes.begin(), sorted_attributes.end());
288
289 // Build the JavaDoc comment for the Styleable array. This has references to child attributes
290 // and what possible values can be used for them.
291 const size_t attr_count = sorted_attributes.size();
292 if (out_class_def != nullptr && attr_count > 0) {
293 std::stringstream styleable_comment;
294 if (!styleable.GetComment().empty()) {
295 styleable_comment << styleable.GetComment() << "\n";
296 } else {
297 // Apply a default intro comment if the styleable has no comments of its own.
298 styleable_comment << "Attributes that can be used with a " << array_field_name << ".\n";
299 }
300
301 styleable_comment << "<p>Includes the following attributes:</p>\n"
302 "<table>\n"
303 "<colgroup align=\"left\" />\n"
304 "<colgroup align=\"left\" />\n"
305 "<tr><th>Attribute</th><th>Description</th></tr>\n";
306
307 // Removed and hidden attributes are public but hidden from the documentation, so don't emit
308 // them as part of the class documentation.
309 std::vector<StyleableAttr> documentation_attrs = sorted_attributes;
310 auto documentation_remove_iter = std::remove_if(documentation_attrs.begin(),
311 documentation_attrs.end(),
312 [&](StyleableAttr entry) -> bool {
313 if (SkipSymbol(entry.symbol)) {
314 return true;
315 }
316 const StringPiece attr_comment_line = entry.symbol.value().attribute->GetComment();
317 return attr_comment_line.contains("@removed") || attr_comment_line.contains("@hide");
318 });
319 documentation_attrs.erase(documentation_remove_iter, documentation_attrs.end());
320
321 // Build the table of attributes with their links and names.
322 for (const StyleableAttr& entry : documentation_attrs) {
323 const ResourceName& attr_name = entry.attr_ref->name.value();
324 styleable_comment << "<tr><td><code>{@link #" << entry.field_name << " "
325 << (!attr_name.package.empty() ? attr_name.package
326 : package_name_to_generate)
327 << ":" << attr_name.entry << "}</code></td>";
328
329 // Only use the comment up until the first '.'. This is to stay compatible with
330 // the way old AAPT did it (presumably to keep it short and to avoid including
331 // annotations like @hide which would affect this Styleable).
332 StringPiece attr_comment_line = entry.symbol.value().attribute->GetComment();
333 styleable_comment << "<td>" << AnnotationProcessor::ExtractFirstSentence(attr_comment_line)
334 << "</td></tr>\n";
335 }
336 styleable_comment << "</table>\n";
337
338 // Generate the @see lines for each attribute.
339 for (const StyleableAttr& entry : documentation_attrs) {
340 styleable_comment << "@see #" << entry.field_name << "\n";
341 }
342
343 array_def->GetCommentBuilder()->AppendComment(styleable_comment.str());
344 }
345
346 if (r_txt_printer != nullptr) {
347 r_txt_printer->Print("int[] styleable ").Print(array_field_name).Print(" {");
348 }
349
350 // Add the ResourceIds to the array member.
351 for (size_t i = 0; i < attr_count; i++) {
352 const StyleableAttr& attr = sorted_attributes[i];
353 std::string r_txt_contents;
354 if (attr.symbol && attr.symbol.value().is_dynamic) {
355 if (!attr.attr_ref->name) {
356 error_ = "unable to determine R.java field name of dynamic resource";
357 return false;
358 }
359
360 const FieldReference field_name =
361 GetRFieldReference(attr.attr_ref->name.value(), package_name_to_generate);
362 array_def->AddElement(field_name);
363 r_txt_contents = field_name.ref;
364 } else {
365 const ResourceId attr_id = attr.attr_ref->id.value_or_default(ResourceId(0));
366 array_def->AddElement(attr_id);
367 r_txt_contents = to_string(attr_id);
368 }
369
370 if (r_txt_printer != nullptr) {
371 if (i != 0) {
372 r_txt_printer->Print(",");
373 }
374 r_txt_printer->Print(" ").Print(r_txt_contents);
375 }
376 }
377
378 if (r_txt_printer != nullptr) {
379 r_txt_printer->Println(" }");
380 }
381
382 // Add the Styleable array to the Styleable class.
383 if (out_class_def != nullptr) {
384 out_class_def->AddMember(std::move(array_def));
385 }
386
387 // Now we emit the indices into the array.
388 for (size_t i = 0; i < attr_count; i++) {
389 const StyleableAttr& styleable_attr = sorted_attributes[i];
390 if (SkipSymbol(styleable_attr.symbol)) {
391 continue;
392 }
393
394 if (out_class_def != nullptr) {
395 StringPiece comment = styleable_attr.attr_ref->GetComment();
396 if (styleable_attr.symbol.value().attribute && comment.empty()) {
397 comment = styleable_attr.symbol.value().attribute->GetComment();
398 }
399
400 if (comment.contains("@removed")) {
401 // Removed attributes are public but hidden from the documentation, so
402 // don't emit them as part of the class documentation.
403 continue;
404 }
405
406 const ResourceName& attr_name = styleable_attr.attr_ref->name.value();
407
408 StringPiece package_name = attr_name.package;
409 if (package_name.empty()) {
410 package_name = package_name_to_generate;
411 }
412
413 std::unique_ptr<IntMember> index_member =
414 util::make_unique<IntMember>(sorted_attributes[i].field_name, static_cast<uint32_t>(i));
415
416 AnnotationProcessor* attr_processor = index_member->GetCommentBuilder();
417
418 if (!comment.empty()) {
419 attr_processor->AppendComment("<p>\n@attr description");
420 attr_processor->AppendComment(comment);
421 } else {
422 std::stringstream default_comment;
423 default_comment << "<p>This symbol is the offset where the "
424 << "{@link " << package_name << ".R.attr#"
425 << TransformToFieldName(attr_name.entry) << "}\n"
426 << "attribute's value can be found in the "
427 << "{@link #" << array_field_name << "} array.";
428 attr_processor->AppendComment(default_comment.str());
429 }
430
431 attr_processor->AppendNewLine();
432 AddAttributeFormatDoc(attr_processor, styleable_attr.symbol.value().attribute.get());
433 attr_processor->AppendNewLine();
434 attr_processor->AppendComment(
435 StringPrintf("@attr name %s:%s", package_name.data(), attr_name.entry.data()));
436
437 out_class_def->AddMember(std::move(index_member));
438 }
439
440 if (r_txt_printer != nullptr) {
441 r_txt_printer->Println(
442 StringPrintf("int styleable %s %zd", sorted_attributes[i].field_name.c_str(), i));
443 }
444 }
445
446 return true;
447 }
448
ProcessResource(const ResourceNameRef & name,const ResourceId & id,const ResourceEntry & entry,ClassDefinition * out_class_def,MethodDefinition * out_rewrite_method,text::Printer * r_txt_printer)449 void JavaClassGenerator::ProcessResource(const ResourceNameRef& name, const ResourceId& id,
450 const ResourceEntry& entry, ClassDefinition* out_class_def,
451 MethodDefinition* out_rewrite_method,
452 text::Printer* r_txt_printer) {
453 ResourceId real_id = id;
454 if (context_->GetMinSdkVersion() < SDK_O && name.type == ResourceType::kId &&
455 id.package_id() > kAppPackageId) {
456 // Workaround for feature splits using package IDs > 0x7F.
457 // See b/37498913.
458 real_id = ResourceId(kAppPackageId, id.package_id(), id.entry_id());
459 }
460
461 const std::string field_name = TransformToFieldName(name.entry);
462 if (out_class_def != nullptr) {
463 auto resource_member =
464 util::make_unique<ResourceMember>(field_name, real_id, entry.visibility.staged_api);
465
466 // Build the comments and annotations for this entry.
467 AnnotationProcessor* processor = resource_member->GetCommentBuilder();
468
469 // Add the comments from any <public> tags.
470 if (entry.visibility.level != Visibility::Level::kUndefined) {
471 processor->AppendComment(entry.visibility.comment);
472 }
473
474 // Add the comments from all configurations of this entry.
475 for (const auto& config_value : entry.values) {
476 processor->AppendComment(config_value->value->GetComment());
477 }
478
479 // If this is an Attribute, append the format Javadoc.
480 if (!entry.values.empty()) {
481 if (Attribute* attr = ValueCast<Attribute>(entry.values.front()->value.get())) {
482 // We list out the available values for the given attribute.
483 AddAttributeFormatDoc(processor, attr);
484 }
485 }
486
487 out_class_def->AddMember(std::move(resource_member));
488 }
489
490 if (r_txt_printer != nullptr) {
491 r_txt_printer->Print("int ")
492 .Print(to_string(name.type))
493 .Print(" ")
494 .Print(field_name)
495 .Print(" ")
496 .Println(real_id.to_string());
497 }
498
499 if (out_rewrite_method != nullptr) {
500 const StringPiece& type_str = to_string(name.type);
501 out_rewrite_method->AppendStatement(
502 StringPrintf("%s.%s = (%s.%s & 0x00ffffff) | packageIdBits;", type_str.data(),
503 field_name.data(), type_str.data(), field_name.data()));
504 }
505 }
506
UnmangleResource(const StringPiece & package_name,const StringPiece & package_name_to_generate,const ResourceEntry & entry)507 Maybe<std::string> JavaClassGenerator::UnmangleResource(const StringPiece& package_name,
508 const StringPiece& package_name_to_generate,
509 const ResourceEntry& entry) {
510 if (SkipSymbol(entry.visibility.level)) {
511 return {};
512 }
513
514 std::string unmangled_package;
515 std::string unmangled_name = entry.name;
516 if (NameMangler::Unmangle(&unmangled_name, &unmangled_package)) {
517 // The entry name was mangled, and we successfully unmangled it.
518 // Check that we want to emit this symbol.
519 if (package_name_to_generate != unmangled_package) {
520 // Skip the entry if it doesn't belong to the package we're writing.
521 return {};
522 }
523 } else if (package_name_to_generate != package_name) {
524 // We are processing a mangled package name,
525 // but this is a non-mangled resource.
526 return {};
527 }
528 return {std::move(unmangled_name)};
529 }
530
ProcessType(const StringPiece & package_name_to_generate,const ResourceTablePackage & package,const ResourceTableType & type,ClassDefinition * out_type_class_def,MethodDefinition * out_rewrite_method_def,Printer * r_txt_printer)531 bool JavaClassGenerator::ProcessType(const StringPiece& package_name_to_generate,
532 const ResourceTablePackage& package,
533 const ResourceTableType& type,
534 ClassDefinition* out_type_class_def,
535 MethodDefinition* out_rewrite_method_def,
536 Printer* r_txt_printer) {
537 for (const auto& entry : type.entries) {
538 const Maybe<std::string> unmangled_name =
539 UnmangleResource(package.name, package_name_to_generate, *entry);
540 if (!unmangled_name) {
541 continue;
542 }
543
544 // Create an ID if there is one (static libraries don't need one).
545 ResourceId id;
546 if (entry->id) {
547 id = entry->id.value();
548 }
549
550 // We need to make sure we hide the fact that we are generating kAttrPrivate attributes.
551 const ResourceNameRef resource_name(
552 package_name_to_generate,
553 type.type == ResourceType::kAttrPrivate ? ResourceType::kAttr : type.type,
554 unmangled_name.value());
555
556 // Check to see if the unmangled name is a valid Java name (not a keyword).
557 if (!IsValidSymbol(unmangled_name.value())) {
558 std::stringstream err;
559 err << "invalid symbol name '" << resource_name << "'";
560 error_ = err.str();
561 return false;
562 }
563
564 if (resource_name.type == ResourceType::kStyleable) {
565 CHECK(!entry->values.empty());
566 const auto styleable = reinterpret_cast<const Styleable*>(entry->values.front()->value.get());
567 if (!ProcessStyleable(resource_name, id, *styleable, package_name_to_generate,
568 out_type_class_def, out_rewrite_method_def, r_txt_printer)) {
569 return false;
570 }
571 } else {
572 ProcessResource(resource_name, id, *entry, out_type_class_def, out_rewrite_method_def,
573 r_txt_printer);
574 }
575 }
576 return true;
577 }
578
Generate(const StringPiece & package_name_to_generate,OutputStream * out,OutputStream * out_r_txt)579 bool JavaClassGenerator::Generate(const StringPiece& package_name_to_generate, OutputStream* out,
580 OutputStream* out_r_txt) {
581 return Generate(package_name_to_generate, package_name_to_generate, out, out_r_txt);
582 }
583
AppendJavaDocAnnotations(const std::vector<std::string> & annotations,AnnotationProcessor * processor)584 static void AppendJavaDocAnnotations(const std::vector<std::string>& annotations,
585 AnnotationProcessor* processor) {
586 for (const std::string& annotation : annotations) {
587 std::string proper_annotation = "@";
588 proper_annotation += annotation;
589 processor->AppendComment(proper_annotation);
590 }
591 }
592
Generate(const StringPiece & package_name_to_generate,const StringPiece & out_package_name,OutputStream * out,OutputStream * out_r_txt)593 bool JavaClassGenerator::Generate(const StringPiece& package_name_to_generate,
594 const StringPiece& out_package_name, OutputStream* out,
595 OutputStream* out_r_txt) {
596 ClassDefinition r_class("R", ClassQualifier::kNone, true);
597 std::unique_ptr<MethodDefinition> rewrite_method;
598
599 std::unique_ptr<Printer> r_txt_printer;
600 if (out_r_txt != nullptr) {
601 r_txt_printer = util::make_unique<Printer>(out_r_txt);
602 }
603 // Generate an onResourcesLoaded() callback if requested.
604 if (out != nullptr && options_.rewrite_callback_options) {
605 rewrite_method =
606 util::make_unique<MethodDefinition>("public static void onResourcesLoaded(int p)");
607 for (const std::string& package_to_callback :
608 options_.rewrite_callback_options.value().packages_to_callback) {
609 rewrite_method->AppendStatement(
610 StringPrintf("%s.R.onResourcesLoaded(p);", package_to_callback.data()));
611 }
612 rewrite_method->AppendStatement("final int packageIdBits = p << 24;");
613 }
614
615 const bool is_public = (options_.types == JavaClassGeneratorOptions::SymbolTypes::kPublic);
616
617 for (const auto& package : table_->packages) {
618 for (const auto& type : package->types) {
619 if (type->type == ResourceType::kAttrPrivate || type->type == ResourceType::kMacro) {
620 // We generate kAttrPrivate as part of the kAttr type, so skip them here.
621 // Macros are not actual resources, so skip them as well.
622 continue;
623 }
624
625 // Stay consistent with AAPT and generate an empty type class if the R class is public.
626 const bool force_creation_if_empty = is_public;
627
628 std::unique_ptr<ClassDefinition> class_def;
629 if (out != nullptr) {
630 class_def = util::make_unique<ClassDefinition>(
631 to_string(type->type), ClassQualifier::kStatic, force_creation_if_empty);
632 }
633
634 if (!ProcessType(package_name_to_generate, *package, *type, class_def.get(),
635 rewrite_method.get(), r_txt_printer.get())) {
636 return false;
637 }
638
639 if (type->type == ResourceType::kAttr) {
640 // Also include private attributes in this same class.
641 if (const ResourceTableType* priv_type = package->FindType(ResourceType::kAttrPrivate)) {
642 if (!ProcessType(package_name_to_generate, *package, *priv_type, class_def.get(),
643 rewrite_method.get(), r_txt_printer.get())) {
644 return false;
645 }
646 }
647 }
648
649 if (out != nullptr && type->type == ResourceType::kStyleable && is_public) {
650 // When generating a public R class, we don't want Styleable to be part
651 // of the API. It is only emitted for documentation purposes.
652 class_def->GetCommentBuilder()->AppendComment("@doconly");
653 }
654
655 if (out != nullptr) {
656 AppendJavaDocAnnotations(options_.javadoc_annotations, class_def->GetCommentBuilder());
657 r_class.AddMember(std::move(class_def));
658 }
659 }
660 }
661
662 if (rewrite_method != nullptr) {
663 r_class.AddMember(std::move(rewrite_method));
664 }
665
666 if (out != nullptr) {
667 AppendJavaDocAnnotations(options_.javadoc_annotations, r_class.GetCommentBuilder());
668 ClassDefinition::WriteJavaFile(&r_class, out_package_name, options_.use_final, !is_public, out);
669 }
670 return true;
671 }
672
673 } // namespace aapt
674