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 } 58
