src/compiler/csharp_plugin.cc

   1 /*
   2  *
   3  * Copyright 2015 gRPC authors.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  */
  18
  19 // Generates C# gRPC service interface out of Protobuf IDL.
  20
  21 #include <memory>
  22
  23 #include "src/compiler/config.h"
  24 #include "src/compiler/csharp_generator.h"
  25 #include "src/compiler/csharp_generator_helpers.h"
  26
  27 class CSharpGrpcGenerator : public grpc::protobuf::compiler::CodeGenerator {
  28  public:
  29   CSharpGrpcGenerator() {}
  30   ~CSharpGrpcGenerator() {}
  31
  32   bool Generate(const grpc::protobuf::FileDescriptor* file,
  33                 const grpc::string& parameter,
  34                 grpc::protobuf::compiler::GeneratorContext* context,
  35                 grpc::string* error) const {
  36     std::vector<std::pair<grpc::string, grpc::string> > options;
  37     grpc::protobuf::compiler::ParseGeneratorParameter(parameter, &options);
  38
  39     bool generate_client = true;
  40     bool generate_server = true;
  41     bool internal_access = false;
  42     bool lite_client = false;
  43     for (size_t i = 0; i < options.size(); i++) {
  44       if (options[i].first == "no_client") {
  45         generate_client = false;
  46       } else if (options[i].first == "no_server") {
  47         generate_server = false;
  48       } else if (options[i].first == "internal_access") {
  49         internal_access = true;
  50       } else if (options[i].first == "lite_client") {
  51         // will only be used if generate_client is true.
  52         // NOTE: experimental option, can be removed in future release
  53         lite_client = true;
  54       } else {
  55         *error = "Unknown generator option: " + options[i].first;
  56         return false;
  57       }
  58     }
  59
  60     grpc::string code = grpc_csharp_generator::GetServices(
  61         file, generate_client, generate_server, internal_access, lite_client);
  62     if (code.size() == 0) {
  63       return true;  // don't generate a file if there are no services
  64     }
  65
  66     // Get output file name.
  67     grpc::string file_name;
  68     if (!grpc_csharp_generator::ServicesFilename(file, &file_name)) {
  69       return false;
  70     }
  71     std::unique_ptr<grpc::protobuf::io::ZeroCopyOutputStream> output(
  72         context->Open(file_name));
  73     grpc::protobuf::io::CodedOutputStream coded_out(output.get());
  74     coded_out.WriteRaw(code.data(), code.size());
  75     return true;
  76   }
  77 };
  78
  79 int main(int argc, char* argv[]) {
  80   CSharpGrpcGenerator generator;
  81   return grpc::protobuf::compiler::PluginMain(argc, argv, &generator);
  82 }