Protobuf Team Bot
6 months ago
committed by
Copybara-Service
parent
9d835e65c6
commit
bf01e49273
10 changed files with 269 additions and 99 deletions
@ -0,0 +1,142 @@ |
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2008 Google Inc. All rights reserved.
|
||||
//
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file or at
|
||||
// https://developers.google.com/open-source/licenses/bsd
|
||||
|
||||
// Generates Kotlin code for a given .proto file.
|
||||
|
||||
#include "google/protobuf/compiler/kotlin/file.h" |
||||
|
||||
#include <memory> |
||||
#include <string> |
||||
#include <vector> |
||||
|
||||
#include "absl/strings/str_cat.h" |
||||
#include "google/protobuf/compiler/code_generator.h" |
||||
#include "google/protobuf/compiler/java/context.h" |
||||
#include "google/protobuf/compiler/java/generator_factory.h" |
||||
#include "google/protobuf/compiler/java/helpers.h" |
||||
#include "google/protobuf/compiler/java/full/generator_factory.h" |
||||
#include "google/protobuf/compiler/java/lite/generator_factory.h" |
||||
#include "google/protobuf/compiler/java/name_resolver.h" |
||||
#include "google/protobuf/compiler/java/names.h" |
||||
#include "google/protobuf/compiler/java/options.h" |
||||
#include "google/protobuf/descriptor.pb.h" |
||||
#include "google/protobuf/io/printer.h" |
||||
|
||||
namespace google { |
||||
namespace protobuf { |
||||
namespace compiler { |
||||
namespace kotlin { |
||||
|
||||
using google::protobuf::compiler::java::Context; |
||||
using google::protobuf::compiler::java::Options; |
||||
|
||||
// class LiteGeneratorFactory;
|
||||
|
||||
namespace { |
||||
std::unique_ptr<java::GeneratorFactory> CreateGeneratorFactory( |
||||
const FileDescriptor* file, const Options& options, Context* context) { |
||||
if (java::HasDescriptorMethods(file, context->EnforceLite())) { |
||||
return java::MakeImmutableGeneratorFactory(context); |
||||
} else { |
||||
return java::MakeImmutableLiteGeneratorFactory(context); |
||||
} |
||||
} |
||||
} // namespace
|
||||
|
||||
FileGenerator::FileGenerator(const FileDescriptor* file, const Options& options) |
||||
: file_(file), |
||||
java_package_(java::FileJavaPackage(file, options)), |
||||
message_generators_(file->message_type_count()), |
||||
context_(new Context(file, options)), |
||||
generator_factory_(CreateGeneratorFactory(file, options, context_.get())), |
||||
name_resolver_(context_->GetNameResolver()), |
||||
options_(options) { |
||||
for (int i = 0; i < file_->message_type_count(); ++i) { |
||||
message_generators_[i] = |
||||
generator_factory_->NewMessageGenerator(file_->message_type(i)); |
||||
} |
||||
} |
||||
|
||||
std::string FileGenerator::GetKotlinClassname() { |
||||
return absl::StrCat(name_resolver_->GetFileImmutableClassName(file_), "Kt"); |
||||
} |
||||
|
||||
void FileGenerator::Generate(io::Printer* printer) { |
||||
printer->Print( |
||||
"// Generated by the protocol buffer compiler. DO NOT EDIT!\n" |
||||
"// NO CHECKED-IN PROTOBUF " |
||||
// Intentional line breaker
|
||||
"GENCODE\n" |
||||
"// source: $filename$\n" |
||||
"\n", |
||||
"filename", file_->name()); |
||||
printer->Print( |
||||
"// Generated files should ignore deprecation warnings\n" |
||||
"@file:Suppress(\"DEPRECATION\")\n"); |
||||
if (!java_package_.empty()) { |
||||
printer->Print( |
||||
"package $package$;\n" |
||||
"\n", |
||||
"package", java::EscapeKotlinKeywords(java_package_)); |
||||
} |
||||
} |
||||
|
||||
void FileGenerator::GenerateSiblings( |
||||
const std::string& package_dir, GeneratorContext* context, |
||||
std::vector<std::string>* file_list, |
||||
std::vector<std::string>* annotation_list) { |
||||
for (int i = 0; i < file_->message_type_count(); i++) { |
||||
const Descriptor* descriptor = file_->message_type(i); |
||||
java::MessageGenerator* generator = message_generators_[i].get(); |
||||
auto open_file = [context](const std::string& filename) { |
||||
return std::unique_ptr<io::ZeroCopyOutputStream>(context->Open(filename)); |
||||
}; |
||||
std::string filename = |
||||
absl::StrCat(package_dir, descriptor->name(), "Kt.kt"); |
||||
file_list->push_back(filename); |
||||
std::string info_full_path = absl::StrCat(filename, ".pb.meta"); |
||||
GeneratedCodeInfo annotations; |
||||
io::AnnotationProtoCollector<GeneratedCodeInfo> annotation_collector( |
||||
&annotations); |
||||
auto output = open_file(filename); |
||||
io::Printer printer( |
||||
output.get(), '$', |
||||
options_.annotate_code ? &annotation_collector : nullptr); |
||||
|
||||
printer.Print( |
||||
"// Generated by the protocol buffer compiler. DO NOT EDIT!\n" |
||||
"// NO CHECKED-IN PROTOBUF " |
||||
// Intentional line breaker
|
||||
"GENCODE\n" |
||||
"// source: $filename$\n" |
||||
"\n", |
||||
"filename", descriptor->file()->name()); |
||||
printer.Print( |
||||
"// Generated files should ignore deprecation warnings\n" |
||||
"@file:Suppress(\"DEPRECATION\")\n"); |
||||
if (!java_package_.empty()) { |
||||
printer.Print( |
||||
"package $package$;\n" |
||||
"\n", |
||||
"package", java::EscapeKotlinKeywords(java_package_)); |
||||
} |
||||
|
||||
generator->GenerateKotlinMembers(&printer); |
||||
generator->GenerateTopLevelKotlinMembers(&printer); |
||||
|
||||
if (options_.annotate_code) { |
||||
auto info_output = open_file(info_full_path); |
||||
annotations.SerializeToZeroCopyStream(info_output.get()); |
||||
annotation_list->push_back(info_full_path); |
||||
} |
||||
} |
||||
} |
||||
|
||||
} // namespace kotlin
|
||||
} // namespace compiler
|
||||
} // namespace protobuf
|
||||
} // namespace google
|
||||
@ -0,0 +1,79 @@ |
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2008 Google Inc. All rights reserved.
|
||||
//
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file or at
|
||||
// https://developers.google.com/open-source/licenses/bsd
|
||||
|
||||
// Generates Kotlin code for a given .proto file.
|
||||
|
||||
#ifndef GOOGLE_PROTOBUF_COMPILER_KOTLIN_FILE_H__ |
||||
#define GOOGLE_PROTOBUF_COMPILER_KOTLIN_FILE_H__ |
||||
|
||||
#include <memory> |
||||
#include <string> |
||||
#include <vector> |
||||
|
||||
#include "google/protobuf/compiler/java/generator_factory.h" |
||||
#include "google/protobuf/compiler/java/options.h" |
||||
|
||||
namespace google { |
||||
namespace protobuf { |
||||
class FileDescriptor; // descriptor.h
|
||||
namespace io { |
||||
class Printer; // printer.h
|
||||
} |
||||
namespace compiler { |
||||
class GeneratorContext; // code_generator.h
|
||||
namespace java { |
||||
class Context; // context.h
|
||||
class MessageGenerator; // message.h
|
||||
class GeneratorFactory; // generator_factory.h
|
||||
class ExtensionGenerator; // extension.h
|
||||
class ClassNameResolver; // name_resolver.h
|
||||
} // namespace java
|
||||
} // namespace compiler
|
||||
} // namespace protobuf
|
||||
} // namespace google
|
||||
|
||||
namespace google { |
||||
namespace protobuf { |
||||
namespace compiler { |
||||
namespace kotlin { |
||||
|
||||
// TODO: b/366047913 - Move away from these generator classes and more towards
|
||||
// the "Context" model that the Kotlin Native & Rust code generators use:
|
||||
// http://google3/third_party/kotlin/protobuf/generator/native/file.cc;l=149-170;rcl=642292300
|
||||
class FileGenerator { |
||||
public: |
||||
FileGenerator(const FileDescriptor* file, const java::Options& options); |
||||
FileGenerator(const FileGenerator&) = delete; |
||||
FileGenerator& operator=(const FileGenerator&) = delete; |
||||
~FileGenerator() = default; |
||||
|
||||
std::string GetKotlinClassname(); |
||||
void Generate(io::Printer* printer); |
||||
void GenerateSiblings(const std::string& package_dir, |
||||
GeneratorContext* generator_context, |
||||
std::vector<std::string>* file_list, |
||||
std::vector<std::string>* annotation_list); |
||||
|
||||
const std::string& java_package() { return java_package_; } |
||||
|
||||
private: |
||||
const FileDescriptor* file_; |
||||
std::string java_package_; |
||||
|
||||
std::vector<std::unique_ptr<java::MessageGenerator>> message_generators_; |
||||
std::unique_ptr<java::Context> context_; |
||||
std::unique_ptr<java::GeneratorFactory> generator_factory_; |
||||
java::ClassNameResolver* name_resolver_; |
||||
const java::Options options_; |
||||
}; |
||||
|
||||
} // namespace kotlin
|
||||
} // namespace compiler
|
||||
} // namespace protobuf
|
||||
} // namespace google
|
||||
|
||||
#endif // GOOGLE_PROTOBUF_COMPILER_KOTLIN_FILE_H__
|
||||
Loading…
Reference in new issue