grpc-extended-cpp-plugin/extended_cpp_plugin.h at main · EVerest/grpc-extended-cpp-plugin · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#ifndef GRPC_EXTENDED_CPP_PLUGIN_H
#define GRPC_EXTENDED_CPP_PLUGIN_H

#include "src/compiler/cpp_plugin.h"

#include "extended_cpp_generator.h"

class ExtendedCppGrpcGenerator : public CppGrpcGenerator {
    public:
        ExtendedCppGrpcGenerator() {}
        virtual ~ExtendedCppGrpcGenerator() {}
        virtual bool Generate(const grpc::protobuf::FileDescriptor* file,
                              const std::string& parameter,
                              grpc::protobuf::compiler::GeneratorContext* generator_context,
                              std::string* error) const override {
            bool result = CppGrpcGenerator::Generate(file, parameter, generator_context, error);
            if (!result) {
                return false;
            }
            std::string file_name = grpc_generator::StripProto(file->name());

            // extend existing generated files
            this->Insert(
                generator_context,
                file_name + ".pb.h",
                "namespace_scope",
                extended_cpp_generator::GetHeaderCreateFunctions(file)
            );
            this->Insert(
                generator_context,
                file_name + ".pb.cc",
                "namespace_scope",
                extended_cpp_generator::GetSourceCreateFunctions(file)
            );

            // create new files
            std::string header_code =
                extended_cpp_generator::GetHeaderPrologue(file) +
                extended_cpp_generator::GetHeaderIncludes(file) +
                extended_cpp_generator::GetHeaderServices(file) +
                extended_cpp_generator::GetHeaderEpilogue(file);
            std::unique_ptr<grpc::protobuf::io::ZeroCopyOutputStream> header_output(
                generator_context->Open(file_name + ".grpc-ext.pb.h")
            );
            grpc::protobuf::io::CodedOutputStream header_coded_out(header_output.get());
            header_coded_out.WriteRaw(header_code.data(), header_code.size());

            std::string source_code =
                extended_cpp_generator::GetSourcePrologue(file) +
                extended_cpp_generator::GetSourceIncludes(file) +
                extended_cpp_generator::GetSourceServices(file) +
                extended_cpp_generator::GetSourceEpilogue(file);
            std::unique_ptr<grpc::protobuf::io::ZeroCopyOutputStream> source_output(
                generator_context->Open(file_name + ".grpc-ext.pb.cc")
            );
            grpc::protobuf::io::CodedOutputStream source_coded_out(source_output.get());
            source_coded_out.WriteRaw(source_code.data(), source_code.size());

            return true;
        }
    private:
          // Insert the given code into the given file at the given insertion point.
        void Insert(grpc::protobuf::compiler::GeneratorContext* context,
                    const std::string& filename,
                    const std::string& insertion_point,
                    const std::string& code) const {
            std::unique_ptr<grpc::protobuf::io::ZeroCopyOutputStream> output(
            context->OpenForInsert(filename, insertion_point));
            grpc::protobuf::io::CodedOutputStream coded_out(output.get());
            coded_out.WriteRaw(code.data(), code.size());
        }
};


#endif  // GRPC_EXTENDED_CPP_PLUGIN_H