59 lines
2.0 KiB
C++
59 lines
2.0 KiB
C++
// 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
|
|
|
|
#ifndef GOOGLE_PROTOBUF_COMPILER_KOTLIN_MESSAGE_H__
|
|
#define GOOGLE_PROTOBUF_COMPILER_KOTLIN_MESSAGE_H__
|
|
|
|
#include "absl/container/btree_map.h"
|
|
#include "google/protobuf/compiler/java/context.h"
|
|
#include "google/protobuf/compiler/java/generator_common.h"
|
|
#include "google/protobuf/compiler/java/full/field_generator.h"
|
|
#include "google/protobuf/compiler/java/lite/field_generator.h"
|
|
#include "google/protobuf/compiler/java/name_resolver.h"
|
|
#include "google/protobuf/descriptor.h"
|
|
|
|
namespace google {
|
|
namespace protobuf {
|
|
namespace compiler {
|
|
namespace kotlin {
|
|
|
|
class MessageGenerator {
|
|
public:
|
|
MessageGenerator(const Descriptor* descriptor, java::Context* context);
|
|
MessageGenerator(const MessageGenerator&) = delete;
|
|
MessageGenerator& operator=(const MessageGenerator&) = delete;
|
|
virtual ~MessageGenerator() = default;
|
|
|
|
void Generate(io::Printer* printer) const;
|
|
void GenerateMembers(io::Printer* printer) const;
|
|
void GenerateTopLevelMembers(io::Printer* printer) const;
|
|
|
|
private:
|
|
java::Context* context_;
|
|
java::ClassNameResolver* name_resolver_;
|
|
const Descriptor* descriptor_;
|
|
absl::btree_map<int, const OneofDescriptor*> oneofs_;
|
|
bool lite_;
|
|
bool jvm_dsl_;
|
|
|
|
// TODO: b/366047913 - These can be simplified once lite and full field
|
|
// generators are unified.
|
|
java::FieldGeneratorMap<java::ImmutableFieldLiteGenerator>
|
|
lite_field_generators_;
|
|
java::FieldGeneratorMap<java::ImmutableFieldGenerator> field_generators_;
|
|
|
|
void GenerateExtensions(io::Printer* printer) const;
|
|
void GenerateOrNull(io::Printer* printer) const;
|
|
void GenerateFieldMembers(io::Printer* printer) const;
|
|
};
|
|
} // namespace kotlin
|
|
} // namespace compiler
|
|
} // namespace protobuf
|
|
} // namespace google
|
|
|
|
#endif // GOOGLE_PROTOBUF_COMPILER_KOTLIN_MESSAGE_H__
|