1a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// Protocol Buffers - Google's data interchange format 2a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// Copyright 2008 Google Inc. All rights reserved. 3afb4b72037e3f13db208590fc782c4bc8e27f862Jeff Davidson// https://developers.google.com/protocol-buffers/ 4a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// 5a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// Redistribution and use in source and binary forms, with or without 6a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// modification, are permitted provided that the following conditions are 7a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// met: 8a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// 9a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// * Redistributions of source code must retain the above copyright 10a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// notice, this list of conditions and the following disclaimer. 11a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// * Redistributions in binary form must reproduce the above 12a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// copyright notice, this list of conditions and the following disclaimer 13a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// in the documentation and/or other materials provided with the 14a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// distribution. 15a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// * Neither the name of Google Inc. nor the names of its 16a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// contributors may be used to endorse or promote products derived from 17a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// this software without specific prior written permission. 18a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// 19a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 21a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 22a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 23a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 24a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 25a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 29a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 31a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson#include <google/protobuf/compiler/java/java_context.h> 32a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 33a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson#include <google/protobuf/compiler/java/java_field.h> 34a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson#include <google/protobuf/compiler/java/java_helpers.h> 35a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson#include <google/protobuf/compiler/java/java_name_resolver.h> 36a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson#include <google/protobuf/descriptor.h> 37a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson#include <google/protobuf/stubs/strutil.h> 38a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson#include <google/protobuf/stubs/map_util.h> 39a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 40a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidsonnamespace google { 41a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidsonnamespace protobuf { 42a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidsonnamespace compiler { 43a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidsonnamespace java { 44a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 45a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff DavidsonContext::Context(const FileDescriptor* file) 46a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson : name_resolver_(new ClassNameResolver) { 47a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson InitializeFieldGeneratorInfo(file); 48a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} 49a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 50a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff DavidsonContext::~Context() { 51a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} 52a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 53a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff DavidsonClassNameResolver* Context::GetNameResolver() { 54a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson return name_resolver_.get(); 55a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} 56a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 57a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidsonnamespace { 58a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// Whether two fields have conflicting accessors (assuming name1 and name2 59a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// are different). name1 and name2 are field1 and field2's camel-case name 60a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson// respectively. 61a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidsonbool IsConflicting(const FieldDescriptor* field1, const string& name1, 62a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const FieldDescriptor* field2, const string& name2, 63a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson string* info) { 64a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson if (field1->is_repeated()) { 65a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson if (field2->is_repeated()) { 66a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // Both fields are repeated. 67a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson return false; 68a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } else { 69a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // field1 is repeated, and field2 is not. 70a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson if (name1 + "Count" == name2) { 71a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson *info = "both repeated field \"" + field1->name() + "\" and singular " + 72a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson "field \"" + field2->name() + "\" generates the method \"" + 73a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson "get" + name1 + "Count()\""; 74a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson return true; 75a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 76a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson if (name1 + "List" == name2) { 77a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson *info = "both repeated field \"" + field1->name() + "\" and singular " + 78a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson "field \"" + field2->name() + "\" generates the method \"" + 79a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson "get" + name1 + "List()\""; 80a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson return true; 81a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 82a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // Well, there are obviously many more conflicting cases, but it probably 83a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // doesn't worth the effort to exhaust all of them because they rarely 84a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // happen and as we are continuing adding new methods/changing existing 85a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // methods the number of different conflicting cases will keep growing. 86a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // We can just add more cases here when they are found in the real world. 87a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson return false; 88a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 89a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } else { 90a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson if (field2->is_repeated()) { 91a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson return IsConflicting(field2, name2, field1, name1, info); 92a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } else { 93a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // None of the two fields are repeated. 94a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson return false; 95a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 96a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 97a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} 98a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} // namespace 99a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 100a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidsonvoid Context::InitializeFieldGeneratorInfo(const FileDescriptor* file) { 101a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson for (int i = 0; i < file->message_type_count(); ++i) { 102a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson InitializeFieldGeneratorInfoForMessage(file->message_type(i)); 103a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 104a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} 105a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 106a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidsonvoid Context::InitializeFieldGeneratorInfoForMessage( 107a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const Descriptor* message) { 108a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson for (int i = 0; i < message->nested_type_count(); ++i) { 109a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson InitializeFieldGeneratorInfoForMessage(message->nested_type(i)); 110a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 111a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson vector<const FieldDescriptor*> fields; 112a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson for (int i = 0; i < message->field_count(); ++i) { 113a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson fields.push_back(message->field(i)); 114a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 115a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson InitializeFieldGeneratorInfoForFields(fields); 116a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 117a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson for (int i = 0; i < message->oneof_decl_count(); ++i) { 118a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const OneofDescriptor* oneof = message->oneof_decl(i); 119a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson OneofGeneratorInfo info; 120a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson info.name = UnderscoresToCamelCase(oneof->name(), false); 121a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson info.capitalized_name = UnderscoresToCamelCase(oneof->name(), true); 122a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson oneof_generator_info_map_[oneof] = info; 123a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 124a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} 125a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 126a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidsonvoid Context::InitializeFieldGeneratorInfoForFields( 127a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const vector<const FieldDescriptor*>& fields) { 128a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // Find out all fields that conflict with some other field in the same 129a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // message. 130a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson vector<bool> is_conflict(fields.size()); 131a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson vector<string> conflict_reason(fields.size()); 132a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson for (int i = 0; i < fields.size(); ++i) { 133a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const FieldDescriptor* field = fields[i]; 134a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const string& name = UnderscoresToCapitalizedCamelCase(field); 135a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson for (int j = i + 1; j < fields.size(); ++j) { 136a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const FieldDescriptor* other = fields[j]; 137a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const string& other_name = UnderscoresToCapitalizedCamelCase(other); 138a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson if (name == other_name) { 139a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson is_conflict[i] = is_conflict[j] = true; 140a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson conflict_reason[i] = conflict_reason[j] = 141a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson "capitalized name of field \"" + field->name() + 142a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson "\" conflicts with field \"" + other->name() + "\""; 143a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } else if (IsConflicting(field, name, other, other_name, 144a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson &conflict_reason[j])) { 145a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson is_conflict[i] = is_conflict[j] = true; 146a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson conflict_reason[i] = conflict_reason[j]; 147a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 148a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 149a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson if (is_conflict[i]) { 150a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson GOOGLE_LOG(WARNING) << "field \"" << field->full_name() << "\" is conflicting " 151a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson << "with another field: " << conflict_reason[i]; 152a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 153a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 154a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson for (int i = 0; i < fields.size(); ++i) { 155a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const FieldDescriptor* field = fields[i]; 156a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson FieldGeneratorInfo info; 157a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson info.name = UnderscoresToCamelCase(field); 158a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson info.capitalized_name = UnderscoresToCapitalizedCamelCase(field); 159a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // For fields conflicting with some other fields, we append the field 160a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson // number to their field names in generated code to avoid conflicts. 161a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson if (is_conflict[i]) { 162a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson info.name += SimpleItoa(field->number()); 163a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson info.capitalized_name += SimpleItoa(field->number()); 164a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson info.disambiguated_reason = conflict_reason[i]; 165a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 166a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson field_generator_info_map_[field] = info; 167a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 168a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} 169a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 170a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidsonconst FieldGeneratorInfo* Context::GetFieldGeneratorInfo( 171a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const FieldDescriptor* field) const { 172a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const FieldGeneratorInfo* result = 173a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson FindOrNull(field_generator_info_map_, field); 174a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson if (result == NULL) { 175a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson GOOGLE_LOG(FATAL) << "Can not find FieldGeneratorInfo for field: " 176a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson << field->full_name(); 177a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 178a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson return result; 179a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} 180a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 181a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidsonconst OneofGeneratorInfo* Context::GetOneofGeneratorInfo( 182a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const OneofDescriptor* oneof) const { 183a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson const OneofGeneratorInfo* result = 184a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson FindOrNull(oneof_generator_info_map_, oneof); 185a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson if (result == NULL) { 186a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson GOOGLE_LOG(FATAL) << "Can not find OneofGeneratorInfo for oneof: " 187a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson << oneof->name(); 188a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson } 189a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson return result; 190a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} 191a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson 192a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} // namespace java 193a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} // namespace compiler 194a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} // namespace protobuf 195a3b2a6da25a76f17c73d31def3952feb0fd2296eJeff Davidson} // namespace google 196