src/compiler/ruby_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 Ruby gRPC service interface out of Protobuf IDL.
  20
  21 #include <memory>
  22
  23 #include "src/compiler/config.h"
  24 #include "src/compiler/ruby_generator.h"
  25 #include "src/compiler/ruby_generator_helpers-inl.h"
  26
  27 class RubyGrpcGenerator : public grpc::protobuf::compiler::CodeGenerator {
  28  public:
  29   RubyGrpcGenerator() {}
  30   ~RubyGrpcGenerator() {}
  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     grpc::string code = grpc_ruby_generator::GetServices(file);
  37     if (code.size() == 0) {
  38       return true;  // don't generate a file if there are no services
  39     }
  40
  41     // Get output file name.
  42     grpc::string file_name;
  43     if (!grpc_ruby_generator::ServicesFilename(file, &file_name)) {
  44       return false;
  45     }
  46     std::unique_ptr<grpc::protobuf::io::ZeroCopyOutputStream> output(
  47         context->Open(file_name));
  48     grpc::protobuf::io::CodedOutputStream coded_out(output.get());
  49     coded_out.WriteRaw(code.data(), code.size());
  50     return true;
  51   }
  52 };
  53
  54 int main(int argc, char* argv[]) {
  55   RubyGrpcGenerator generator;
  56   return grpc::protobuf::compiler::PluginMain(argc, argv, &generator);
  57 }