Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 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 | #ifndef AAPT_JAVA_CLASSDEFINITION_H |
| 18 | #define AAPT_JAVA_CLASSDEFINITION_H |
| 19 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 20 | #include <ostream> |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 21 | #include <set> |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 22 | #include <string> |
| 23 | |
| 24 | #include "android-base/macros.h" |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 25 | #include "androidfw/StringPiece.h" |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 26 | |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 27 | #include "Resource.h" |
| 28 | #include "java/AnnotationProcessor.h" |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 29 | #include "util/Util.h" |
| 30 | |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 31 | namespace aapt { |
| 32 | |
| 33 | // The number of attributes to emit per line in a Styleable array. |
| 34 | constexpr static size_t kAttribsPerLine = 4; |
| 35 | constexpr static const char* kIndent = " "; |
| 36 | |
| 37 | class ClassMember { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 38 | public: |
| 39 | virtual ~ClassMember() = default; |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 40 | |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 41 | AnnotationProcessor* GetCommentBuilder() { |
| 42 | return &processor_; |
| 43 | } |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 44 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 45 | virtual bool empty() const = 0; |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 46 | |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 47 | virtual const std::string& GetName() const = 0; |
| 48 | |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 49 | // Writes the class member to the out stream. Subclasses should derive this method |
| 50 | // to write their own data. Call this base method from the subclass to write out |
| 51 | // this member's comments/annotations. |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 52 | virtual void WriteToStream(const android::StringPiece& prefix, bool final, |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 53 | std::ostream* out) const; |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 54 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 55 | private: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 56 | AnnotationProcessor processor_; |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 57 | }; |
| 58 | |
| 59 | template <typename T> |
| 60 | class PrimitiveMember : public ClassMember { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 61 | public: |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 62 | PrimitiveMember(const android::StringPiece& name, const T& val) |
| 63 | : name_(name.to_string()), val_(val) {} |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 64 | |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 65 | bool empty() const override { |
| 66 | return false; |
| 67 | } |
| 68 | |
| 69 | const std::string& GetName() const override { |
| 70 | return name_; |
| 71 | } |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 72 | |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 73 | void WriteToStream(const android::StringPiece& prefix, bool final, |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 74 | std::ostream* out) const override { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 75 | ClassMember::WriteToStream(prefix, final, out); |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 76 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 77 | *out << prefix << "public static " << (final ? "final " : "") << "int " |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 78 | << name_ << "=" << val_ << ";"; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 79 | } |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 80 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 81 | private: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 82 | std::string name_; |
| 83 | T val_; |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 84 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 85 | DISALLOW_COPY_AND_ASSIGN(PrimitiveMember); |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 86 | }; |
| 87 | |
| 88 | /** |
| 89 | * Specialization for strings so they get the right type and are quoted with "". |
| 90 | */ |
| 91 | template <> |
| 92 | class PrimitiveMember<std::string> : public ClassMember { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 93 | public: |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 94 | PrimitiveMember(const android::StringPiece& name, const std::string& val) |
| 95 | : name_(name.to_string()), val_(val) {} |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 96 | |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 97 | bool empty() const override { |
| 98 | return false; |
| 99 | } |
| 100 | |
| 101 | const std::string& GetName() const override { |
| 102 | return name_; |
| 103 | } |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 104 | |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 105 | void WriteToStream(const android::StringPiece& prefix, bool final, |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 106 | std::ostream* out) const override { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 107 | ClassMember::WriteToStream(prefix, final, out); |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 108 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 109 | *out << prefix << "public static " << (final ? "final " : "") << "String " |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 110 | << name_ << "=\"" << val_ << "\";"; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 111 | } |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 112 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 113 | private: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 114 | std::string name_; |
| 115 | std::string val_; |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 116 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 117 | DISALLOW_COPY_AND_ASSIGN(PrimitiveMember); |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 118 | }; |
| 119 | |
| 120 | using IntMember = PrimitiveMember<uint32_t>; |
| 121 | using ResourceMember = PrimitiveMember<ResourceId>; |
| 122 | using StringMember = PrimitiveMember<std::string>; |
| 123 | |
| 124 | template <typename T> |
| 125 | class PrimitiveArrayMember : public ClassMember { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 126 | public: |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 127 | explicit PrimitiveArrayMember(const android::StringPiece& name) : name_(name.to_string()) {} |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 128 | |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 129 | void AddElement(const T& val) { |
| 130 | elements_.push_back(val); |
| 131 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 132 | |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 133 | bool empty() const override { |
| 134 | return false; |
| 135 | } |
| 136 | |
| 137 | const std::string& GetName() const override { |
| 138 | return name_; |
| 139 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 140 | |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 141 | void WriteToStream(const android::StringPiece& prefix, bool final, |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 142 | std::ostream* out) const override { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 143 | ClassMember::WriteToStream(prefix, final, out); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 144 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 145 | *out << prefix << "public static final int[] " << name_ << "={"; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 146 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 147 | const auto begin = elements_.begin(); |
| 148 | const auto end = elements_.end(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 149 | for (auto current = begin; current != end; ++current) { |
| 150 | if (std::distance(begin, current) % kAttribsPerLine == 0) { |
| 151 | *out << "\n" << prefix << kIndent << kIndent; |
| 152 | } |
| 153 | |
| 154 | *out << *current; |
| 155 | if (std::distance(current, end) > 1) { |
| 156 | *out << ", "; |
| 157 | } |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 158 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 159 | *out << "\n" << prefix << kIndent << "};"; |
| 160 | } |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 161 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 162 | private: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 163 | std::string name_; |
| 164 | std::vector<T> elements_; |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 165 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 166 | DISALLOW_COPY_AND_ASSIGN(PrimitiveArrayMember); |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 167 | }; |
| 168 | |
| 169 | using ResourceArrayMember = PrimitiveArrayMember<ResourceId>; |
| 170 | |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 171 | // Represents a method in a class. |
| 172 | class MethodDefinition : public ClassMember { |
| 173 | public: |
| 174 | // Expected method signature example: 'public static void onResourcesLoaded(int p)'. |
| 175 | explicit MethodDefinition(const android::StringPiece& signature) |
| 176 | : signature_(signature.to_string()) {} |
| 177 | |
| 178 | // Appends a single statement to the method. It should include no newlines or else |
| 179 | // formatting may be broken. |
| 180 | void AppendStatement(const android::StringPiece& statement); |
| 181 | |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 182 | // Not quite the same as a name, but good enough. |
| 183 | const std::string& GetName() const override { |
| 184 | return signature_; |
| 185 | } |
| 186 | |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 187 | // Even if the method is empty, we always want to write the method signature. |
| 188 | bool empty() const override { return false; } |
| 189 | |
| 190 | void WriteToStream(const android::StringPiece& prefix, bool final, |
| 191 | std::ostream* out) const override; |
| 192 | |
| 193 | private: |
| 194 | std::string signature_; |
| 195 | std::vector<std::string> statements_; |
| 196 | }; |
| 197 | |
| 198 | enum class ClassQualifier { kNone, kStatic }; |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 199 | |
| 200 | class ClassDefinition : public ClassMember { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 201 | public: |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 202 | static bool WriteJavaFile(const ClassDefinition* def, const android::StringPiece& package, |
| 203 | bool final, std::ostream* out); |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 204 | |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 205 | ClassDefinition(const android::StringPiece& name, ClassQualifier qualifier, bool createIfEmpty) |
| 206 | : name_(name.to_string()), qualifier_(qualifier), create_if_empty_(createIfEmpty) {} |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 207 | |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 208 | enum class Result { |
| 209 | kAdded, |
| 210 | kOverridden, |
| 211 | }; |
| 212 | |
| 213 | Result AddMember(std::unique_ptr<ClassMember> member); |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 214 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 215 | bool empty() const override; |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 216 | |
| 217 | const std::string& GetName() const override { |
| 218 | return name_; |
| 219 | } |
| 220 | |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 221 | void WriteToStream(const android::StringPiece& prefix, bool final, |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 222 | std::ostream* out) const override; |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 223 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 224 | private: |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 225 | struct ClassMemberCompare { |
| 226 | using T = std::unique_ptr<ClassMember>; |
| 227 | bool operator()(const T& a, const T& b) const { |
| 228 | return a->GetName() < b->GetName(); |
| 229 | } |
| 230 | }; |
| 231 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 232 | std::string name_; |
| 233 | ClassQualifier qualifier_; |
| 234 | bool create_if_empty_; |
Adam Lesinski | 6e241e7 | 2017-08-18 19:49:58 -0700 | [diff] [blame^] | 235 | std::set<std::unique_ptr<ClassMember>, ClassMemberCompare> members_; |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 236 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 237 | DISALLOW_COPY_AND_ASSIGN(ClassDefinition); |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 238 | }; |
| 239 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 240 | } // namespace aapt |
Adam Lesinski | 6cbfb1d | 2016-03-31 13:33:02 -0700 | [diff] [blame] | 241 | |
| 242 | #endif /* AAPT_JAVA_CLASSDEFINITION_H */ |