grpc-rest 0.1.24 → 0.2.0

data/lib/grpc_rest.rb

@@ -1,10 +1,11 @@
+# frozen_string_literal: true
+
 require 'google/protobuf/well_known_types'
 require 'grpc'
 require 'grpc/core/status_codes'
 
 module GrpcRest
   class << self
-
     def underscore(s)
       GRPC::GenericService.underscore(s)
     end
@@ -39,7 +40,7 @@ module GrpcRest
 
     # https://stackoverflow.com/a/2158473/5199431
     def longest_common_substring(arr)
-      return "" if arr.empty?
+      return '' if arr.empty?
 
       result = 0
       first_val = arr[0]
@@ -48,6 +49,7 @@ module GrpcRest
         character = first_val[k]
         arr.each { |str| all_matched &= (character == str[k]) }
         break unless all_matched
+
         result += 1
       end
       first_val.slice(0, result)
@@ -64,14 +66,12 @@ module GrpcRest
     end
 
     def map_proto_type(descriptor, val)
-      if descriptor.subtype.is_a?(Google::Protobuf::EnumDescriptor)
-        return handle_enum_values(descriptor, val)
-      end
+      return handle_enum_values(descriptor, val) if descriptor.subtype.is_a?(Google::Protobuf::EnumDescriptor)
 
       case descriptor.type
-      when *%i(int32 int64 uint32 uint64 sint32 sint64 fixed32 fixed64 sfixed32 sfixed64)
+      when :int32, :int64, :uint32, :uint64, :sint32, :sint64, :fixed32, :fixed64, :sfixed32, :sfixed64
         return val.to_i
-      when *%i(float double)
+      when :float, :double
         return val.to_f
       when :bool
         return !!val
@@ -79,19 +79,18 @@ module GrpcRest
 
       case descriptor.subtype&.name
       when 'google.protobuf.Struct'
-        return Google::Protobuf::Struct.from_hash(val)
+        Google::Protobuf::Struct.from_hash(val)
       when 'google.protobuf.Timestamp'
-        if val.is_a?(Numeric)
-          return Google::Protobuf::Timestamp.from_time(Time.at(val))
-        else
-          return Google::Protobuf::Timestamp.from_time(Time.new(val))
-        end
+        return Google::Protobuf::Timestamp.from_time(Time.at(val)) if val.is_a?(Numeric)
+
+        Google::Protobuf::Timestamp.from_time(Time.new(val))
+
       when 'google.protobuf.Value'
-        return Google::Protobuf::Value.from_ruby(val)
+        Google::Protobuf::Value.from_ruby(val)
       when 'google.protobuf.ListValue'
-        return Google::Protobuf::ListValue.from_a(val)
+        Google::Protobuf::ListValue.from_a(val)
       else
-        return map_proto_record(descriptor.subtype, val)
+        map_proto_record(descriptor.subtype, val)
       end
     end
 
@@ -102,11 +101,16 @@ module GrpcRest
         next if val.nil?
 
         if descriptor.label == :repeated
-          params[field] = val.map { |v| map_proto_type(descriptor, v) }
+          # leave map entries as key => value
+          unless descriptor.subtype&.options&.to_h&.dig(:map_entry)
+            params[field] = val.map { |v| map_proto_type(descriptor, v) }
+          end
         else
           params[field] = map_proto_type(descriptor, val)
         end
       end
+
+
       params
     end
 
@@ -132,10 +136,10 @@ module GrpcRest
         end
         assign_value(request, entry[:name], value_to_use)
       end
-      if body_string.present? && body_string != '*'
-        # we need to "splat" the body parameters into the given sub-record rather than into the top-level.
-        sub_record = sub_field(request, body_string, parameters)
-      end
+      return unless body_string.present? && body_string != '*'
+
+      # we need to "splat" the body parameters into the given sub-record rather than into the top-level.
+      sub_field(request, body_string, parameters)
     end
 
     # Ported from https://github.com/grpc-ecosystem/grpc-gateway/blob/main/runtime/errors.go#L36
@@ -171,25 +175,26 @@ module GrpcRest
     end
 
     def error_msg(error)
+      Rails.logger.error("#{error.message}, backtrace: #{error.backtrace.join("\n")}")
       if error.respond_to?(:code)
-      {
+        {
           code: error.code,
           message: error.message,
           details: [
-                    {
-                      backtrace: error.backtrace
-                    }
+            {
+              backtrace: error.backtrace
+            }
           ]
-      }
+        }
       else
         {
-            code: 3,
-            message: "InvalidArgument: #{error.message}",
-            details: [
-                      {
-                        backtrace: error.backtrace
-                      }
-            ]
+          code: 3,
+          message: "InvalidArgument: #{error.message}",
+          details: [
+            {
+              backtrace: error.backtrace
+            }
+          ]
         }
       end
     end
@@ -228,14 +233,12 @@ module GrpcRest
         klass = ::Gruf::Controllers::Base.subclasses.find do |k|
           k.bound_service == service_obj
         end
-        if klass
-          return send_gruf_request(klass, service_obj, method, request)
-        end
+        return send_gruf_request(klass, service_obj, method, request) if klass
       end
       send_grpc_request(service, method, request)
     end
 
-    def send_request(service, method, request, options={})
+    def send_request(service, method, request, options = {})
       response = get_response(service, method, request)
       if options[:emit_defaults]
         response.to_json(emit_defaults: true)
@@ -244,5 +247,4 @@ module GrpcRest
       end
     end
   end
-
 end

data/{protoc-gen-rails/testdata/base/app/controllers/my_service_controller.rb → spec/__snapshots__/service.snap}

@@ -1,4 +1,4 @@
-
+spec/app/app/controllers/my_service_controller.rb:
 ####### THIS FILE IS AUTOMATICALLY GENERATED BY protoc-gen-rails. DO NOT MODIFY. #######
 
 require 'grpc_rest'
@@ -14,54 +14,60 @@ class MyServiceController < ActionController::Base
   rescue_from ActionDispatch::Http::Parameters::ParseError, Google::Protobuf::TypeError do |e|
     render json: GrpcRest.error_msg(e), status: :bad_request
   end
+  METHOD_PARAM_MAP = {"test"=>[{:name=>"foobar", :val=>"=blah/*", :split_name=>["foobar"]}],
+ "test2"=>[],
+ "test3"=>
+  [{:name=>"sub_record.sub_id",
+    :val=>"",
+    :split_name=>["sub_record", "sub_id"]}],
+ "test4"=>[]}
+.freeze
 
-  METHOD_PARAM_MAP = {
-
-    "test" => [
-       {name: "foobar", val: "blah/*", split_name:["foobar"]},
-			 ],
-
-    "test_2" => [
-       ],
-
-    "test_3" => [
-       {name: "sub_record.sub_id", val: nil, split_name:["sub_record","sub_id"]},
-			 ],
-
-    "test_4" => [
-       ],
-}.freeze
 
 	def test
     fields = Testdata::TestRequest.descriptor.to_a.map(&:name)
     parameters = request.parameters.to_h.deep_transform_keys(&:underscore)
     grpc_request = GrpcRest.init_request(Testdata::TestRequest, parameters.slice(*fields))
     GrpcRest.assign_params(grpc_request, METHOD_PARAM_MAP["test"], "", request.parameters)
-    render json: GrpcRest.send_request("Testdata::MyService", "test", grpc_request, {emit_defaults: true})
+    render json: GrpcRest.send_request("Testdata::MyService", "test", grpc_request, {:emit_defaults=>true})
   end
 
-	def test_2
+	def test2
     fields = Testdata::TestRequest.descriptor.to_a.map(&:name)
     parameters = request.parameters.to_h.deep_transform_keys(&:underscore)
     grpc_request = GrpcRest.init_request(Testdata::TestRequest, parameters.slice(*fields))
-    GrpcRest.assign_params(grpc_request, METHOD_PARAM_MAP["test_2"], "second_record", request.parameters)
-    render json: GrpcRest.send_request("Testdata::MyService", "test_2", grpc_request, {})
+    GrpcRest.assign_params(grpc_request, METHOD_PARAM_MAP["test2"], "second_record", request.parameters)
+    render json: GrpcRest.send_request("Testdata::MyService", "test2", grpc_request, {})
   end
 
-	def test_3
+	def test3
     fields = Testdata::TestRequest.descriptor.to_a.map(&:name)
     parameters = request.parameters.to_h.deep_transform_keys(&:underscore)
     grpc_request = GrpcRest.init_request(Testdata::TestRequest, parameters.slice(*fields))
-    GrpcRest.assign_params(grpc_request, METHOD_PARAM_MAP["test_3"], "", request.parameters)
-    render json: GrpcRest.send_request("Testdata::MyService", "test_3", grpc_request, {})
+    GrpcRest.assign_params(grpc_request, METHOD_PARAM_MAP["test3"], "", request.parameters)
+    render json: GrpcRest.send_request("Testdata::MyService", "test3", grpc_request, {})
   end
 
-	def test_4
+	def test4
     fields = Testdata::TestRequest.descriptor.to_a.map(&:name)
     parameters = request.parameters.to_h.deep_transform_keys(&:underscore)
     grpc_request = GrpcRest.init_request(Testdata::TestRequest, parameters.slice(*fields))
-    GrpcRest.assign_params(grpc_request, METHOD_PARAM_MAP["test_4"], "*", request.parameters)
-    render json: GrpcRest.send_request("Testdata::MyService", "test_4", grpc_request, {})
+    GrpcRest.assign_params(grpc_request, METHOD_PARAM_MAP["test4"], "*", request.parameters)
+    render json: GrpcRest.send_request("Testdata::MyService", "test4", grpc_request, {})
   end
 
 end
+
+
+spec/app/config/routes/grpc.rb:
+# frozen_string_literal: true
+
+####### THIS FILE IS AUTOMATICALLY GENERATED BY protoc-gen-rails. DO NOT MODIFY. #######
+
+
+get '/test/*foobar' => 'my_service#test' 
+post '/test2' => 'my_service#test2' 
+post '/test3/*sub_id' => 'my_service#test3' 
+post '/test4' => 'my_service#test4' 
+
+

data/spec/base_interceptor_spec.rb

@@ -5,7 +5,7 @@ require 'src/proto/grpc/testing/test_services_pb'
 
 RSpec.describe GrpcRest::BaseInterceptor, type: :class do
   let(:rpc_service) { Grpc::Testing::TestService::Service.new }
-  let(:rpc_desc) { Grpc::Testing::TestService::Service.rpc_descs.values.first}
+  let(:rpc_desc) { Grpc::Testing::TestService::Service.rpc_descs.values.first }
   let(:message) { Grpc::Testing::SimpleRequest.new }
 
   describe '#fail!' do
@@ -18,10 +18,13 @@ RSpec.describe GrpcRest::BaseInterceptor, type: :class do
         service: rpc_service,
         rpc_desc: rpc_desc,
         active_call: nil,
-        message: message)
+        message: message
+      )
       interceptor = GrpcRest::BaseInterceptor.new(request, Gruf::Error.new)
 
-      expect{ interceptor.fail!(error_code, error_code, error_message) }.to raise_error(GRPC::InvalidArgument) do |error|
+      expect do
+        interceptor.fail!(error_code, error_code, error_message)
+      end.to raise_error(GRPC::InvalidArgument) do |error|
         expect(error.message).to match(error_message)
         expect(error.code).to eq(GRPC::Core::StatusCodes::INVALID_ARGUMENT)
       end

data/spec/{protoc-gen-openapiv2 → gen/protoc-gen-openapiv2}/options/annotations_pb.rb

@@ -1,4 +1,5 @@
 # frozen_string_literal: true
+
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: protoc-gen-openapiv2/options/annotations.proto
 
@@ -15,7 +16,7 @@ pool = Google::Protobuf::DescriptorPool.generated_pool
 
 begin
   pool.add_serialized_file(descriptor_data)
-rescue TypeError => e
+rescue TypeError
   # Compatibility code: will be removed in the next major version.
   require 'google/protobuf/descriptor_pb'
   parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)
@@ -24,7 +25,7 @@ rescue TypeError => e
   file = pool.add_serialized_file(serialized)
   warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
   imports = [
-    ["grpc.gateway.protoc_gen_openapiv2.options.Swagger", "protoc-gen-openapiv2/options/openapiv2.proto"],
+    ['grpc.gateway.protoc_gen_openapiv2.options.Swagger', 'protoc-gen-openapiv2/options/openapiv2.proto']
   ]
   imports.each do |type_name, expected_filename|
     import_file = pool.lookup(type_name).file_descriptor
@@ -32,8 +33,8 @@ rescue TypeError => e
       warn "- #{file.name} imports #{expected_filename}, but that import was loaded as #{import_file.name}"
     end
   end
-  warn "Each proto file must use a consistent fully-qualified name."
-  warn "This will become an error in the next major version."
+  warn 'Each proto file must use a consistent fully-qualified name.'
+  warn 'This will become an error in the next major version.'
 end
 
 module Grpc

data/spec/{protoc-gen-openapiv2 → gen/protoc-gen-openapiv2}/options/openapiv2_pb.rb

@@ -1,4 +1,5 @@
 # frozen_string_literal: true
+
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: protoc-gen-openapiv2/options/openapiv2.proto
 
@@ -14,7 +15,7 @@ pool = Google::Protobuf::DescriptorPool.generated_pool
 
 begin
   pool.add_serialized_file(descriptor_data)
-rescue TypeError => e
+rescue TypeError
   # Compatibility code: will be removed in the next major version.
   require 'google/protobuf/descriptor_pb'
   parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)
@@ -23,7 +24,7 @@ rescue TypeError => e
   file = pool.add_serialized_file(serialized)
   warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
   imports = [
-    ["google.protobuf.Value", "google/protobuf/struct.proto"],
+    ['google.protobuf.Value', 'google/protobuf/struct.proto']
   ]
   imports.each do |type_name, expected_filename|
     import_file = pool.lookup(type_name).file_descriptor
@@ -31,39 +32,39 @@ rescue TypeError => e
       warn "- #{file.name} imports #{expected_filename}, but that import was loaded as #{import_file.name}"
     end
   end
-  warn "Each proto file must use a consistent fully-qualified name."
-  warn "This will become an error in the next major version."
+  warn 'Each proto file must use a consistent fully-qualified name.'
+  warn 'This will become an error in the next major version.'
 end
 
 module Grpc
   module Gateway
     module ProtocGenOpenapiv2
       module Options
-        Swagger = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.Swagger").msgclass
-        Operation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.Operation").msgclass
-        Parameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.Parameters").msgclass
-        HeaderParameter = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.HeaderParameter").msgclass
-        HeaderParameter::Type = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.HeaderParameter.Type").enummodule
-        Header = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.Header").msgclass
-        Response = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.Response").msgclass
-        Info = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.Info").msgclass
-        Contact = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.Contact").msgclass
-        License = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.License").msgclass
-        ExternalDocumentation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.ExternalDocumentation").msgclass
-        Schema = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.Schema").msgclass
-        JSONSchema = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.JSONSchema").msgclass
-        JSONSchema::FieldConfiguration = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.JSONSchema.FieldConfiguration").msgclass
-        JSONSchema::JSONSchemaSimpleTypes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.JSONSchema.JSONSchemaSimpleTypes").enummodule
-        Tag = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.Tag").msgclass
-        SecurityDefinitions = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.SecurityDefinitions").msgclass
-        SecurityScheme = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.SecurityScheme").msgclass
-        SecurityScheme::Type = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.SecurityScheme.Type").enummodule
-        SecurityScheme::In = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.SecurityScheme.In").enummodule
-        SecurityScheme::Flow = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.SecurityScheme.Flow").enummodule
-        SecurityRequirement = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.SecurityRequirement").msgclass
-        SecurityRequirement::SecurityRequirementValue = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.SecurityRequirement.SecurityRequirementValue").msgclass
-        Scopes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.Scopes").msgclass
-        Scheme = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.gateway.protoc_gen_openapiv2.options.Scheme").enummodule
+        Swagger = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.Swagger').msgclass
+        Operation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.Operation').msgclass
+        Parameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.Parameters').msgclass
+        HeaderParameter = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.HeaderParameter').msgclass
+        HeaderParameter::Type = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.HeaderParameter.Type').enummodule
+        Header = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.Header').msgclass
+        Response = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.Response').msgclass
+        Info = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.Info').msgclass
+        Contact = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.Contact').msgclass
+        License = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.License').msgclass
+        ExternalDocumentation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.ExternalDocumentation').msgclass
+        Schema = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.Schema').msgclass
+        JSONSchema = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.JSONSchema').msgclass
+        JSONSchema::FieldConfiguration = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.JSONSchema.FieldConfiguration').msgclass
+        JSONSchema::JSONSchemaSimpleTypes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.JSONSchema.JSONSchemaSimpleTypes').enummodule
+        Tag = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.Tag').msgclass
+        SecurityDefinitions = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.SecurityDefinitions').msgclass
+        SecurityScheme = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.SecurityScheme').msgclass
+        SecurityScheme::Type = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.SecurityScheme.Type').enummodule
+        SecurityScheme::In = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.SecurityScheme.In').enummodule
+        SecurityScheme::Flow = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.SecurityScheme.Flow').enummodule
+        SecurityRequirement = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.SecurityRequirement').msgclass
+        SecurityRequirement::SecurityRequirementValue = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.SecurityRequirement.SecurityRequirementValue').msgclass
+        Scopes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.Scopes').msgclass
+        Scheme = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('grpc.gateway.protoc_gen_openapiv2.options.Scheme').enummodule
       end
     end
   end

data/spec/gen/test_pb.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: test.proto
+
+require 'google/protobuf'
+
+descriptor_data = "\n\ntest.proto\x12\x04test\"\x14\n\x04Test\x12\x0c\n\x04test\x18\x01 \x01(\tb\x06proto3"
+
+pool = Google::Protobuf::DescriptorPool.generated_pool
+pool.add_serialized_file(descriptor_data)
+
+module Test
+  Test = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('test.Test').msgclass
+end

data/spec/gen/test_service_pb.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: test_service.proto
+
+require 'google/protobuf'
+
+require 'google/api/annotations_pb'
+require 'google/protobuf/struct_pb'
+require 'google/protobuf/timestamp_pb'
+require 'protoc-gen-openapiv2/options/annotations_pb'
+
+descriptor_data = "\n\x12test_service.proto\x12\x08testdata\x1a\x1cgoogle/api/annotations.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a.protoc-gen-openapiv2/options/annotations.proto\"\xd7\x03\n\x0bTestRequest\x12\x0f\n\x07test_id\x18\x01 \x01(\t\x12\x0e\n\x06\x66oobar\x18\x02 \x01(\t\x12\x17\n\x0frepeated_string\x18\x03 \x03(\t\x12\'\n\nsub_record\x18\x04 \x01(\x0b\x32\x13.testdata.SubRecord\x12*\n\rsecond_record\x18\x05 \x01(\x0b\x32\x13.testdata.SubRecord\x12-\n\x0cstruct_field\x18\x06 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x33\n\x0ftimestamp_field\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nlist_value\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.ListValue\x12*\n\nbare_value\x18\t \x01(\x0b\x32\x16.google.protobuf.Value\x12(\n\x0bsub_records\x18\n \x03(\x0b\x32\x13.testdata.SubRecord\x12\x10\n\x08some_int\x18\x0b \x01(\x05\x12%\n\tsome_enum\x18\x0c \x01(\x0e\x32\x12.testdata.TestEnum\x12\x16\n\x0erepeated_float\x18\r \x03(\x02\"/\n\tSubRecord\x12\x0e\n\x06sub_id\x18\x01 \x01(\t\x12\x12\n\nanother_id\x18\x02 \x01(\t\"L\n\x0cTestResponse\x12\x10\n\x08some_int\x18\x01 \x01(\x05\x12\x15\n\rfull_response\x18\x02 \x01(\t\x12\x13\n\x0bignored_key\x18\x03 \x01(\t*K\n\x08TestEnum\x12\x19\n\x15TEST_ENUM_UNSPECIFIED\x10\x00\x12\x11\n\rTEST_ENUM_FOO\x10\x01\x12\x11\n\rTEST_ENUM_BAR\x10\x02\x32\x83\x03\n\tMyService\x12x\n\x04Test\x12\x15.testdata.TestRequest\x1a\x16.testdata.TestResponse\"A\x92\x41!j\x1f\n\x19x-grpc-rest-emit-defaults\x12\x02 \x01\x82\xd3\xe4\x93\x02\x17\x12\x15/test/{foobar=blah/*}\x12U\n\x05Test2\x12\x15.testdata.TestRequest\x1a\x16.testdata.TestResponse\"\x1d\x82\xd3\xe4\x93\x02\x17\"\x06/test2:\rsecond_record\x12Z\n\x05Test3\x12\x15.testdata.TestRequest\x1a\x16.testdata.TestResponse\"\"\x82\xd3\xe4\x93\x02\x1c\"\x1a/test3/{sub_record.sub_id}\x12I\n\x05Test4\x12\x15.testdata.TestRequest\x1a\x16.testdata.TestResponse\"\x11\x82\xd3\xe4\x93\x02\x0b\"\x06/test4:\x01*b\x06proto3"
+
+pool = Google::Protobuf::DescriptorPool.generated_pool
+pool.add_serialized_file(descriptor_data)
+
+module Testdata
+  TestRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('testdata.TestRequest').msgclass
+  SubRecord = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('testdata.SubRecord').msgclass
+  TestResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('testdata.TestResponse').msgclass
+  TestEnum = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('testdata.TestEnum').enummodule
+end

data/spec/gen/test_service_services_pb.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# Source: test_service.proto for package 'testdata'
+
+require 'grpc'
+require 'test_service_pb'
+
+module Testdata
+  module MyService
+    class Service
+      include ::GRPC::GenericService
+
+      self.marshal_class_method = :encode
+      self.unmarshal_class_method = :decode
+      self.service_name = 'testdata.MyService'
+
+      rpc :Test, ::Testdata::TestRequest, ::Testdata::TestResponse
+      rpc :Test2, ::Testdata::TestRequest, ::Testdata::TestResponse
+      rpc :Test3, ::Testdata::TestRequest, ::Testdata::TestResponse
+      rpc :Test4, ::Testdata::TestRequest, ::Testdata::TestResponse
+    end
+
+    Stub = Service.rpc_stub_class
+  end
+end

data/spec/generator_spec.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+# This is used to check the generated files all at once.
+class MultiFileSerializer
+  def process_string(s)
+    # Ruby 3.4 changes how hashes are printed
+    if Gem::Version.new(RUBY_VERSION) > Gem::Version.new('3.4.0')
+      s.gsub(/{"(.*)" => /, '{"\1"=>')
+    else
+      s
+    end
+  end
+
+  def dump(value)
+    value.keys.sort.map { |k| "#{k}:\n#{process_string(value[k])}\n" }.join("\n")
+  end
+end
+
+# The easiest way to test a protoc plugin is by actually running protoc. Here we are specifying
+# the binary to use for the rails plugin and passing in the dependency paths and the
+# place to find the generated Ruby files (which we generated in advance and live inside `spec/gen`.)
+def protoc(files)
+  `bundle exec protoc \
+      --proto_path=#{__dir__} \
+     --proto_path=#{__dir__}/google-deps \
+     --plugin=protoc-gen-rails=#{__dir__}/../bin/protoc-gen-rails \
+     --rails_out=#{__dir__}/app \
+     --rails_opt=require=#{__dir__}/gen #{files.join(' ')}`
+end
+
+RSpec.describe 'protoc-gen-rails' do
+  let(:files) { Dir['spec/app/**/*.rb'].map { |f| [f, File.read(f)] }.to_h }
+  before(:each) do
+    FileUtils.mkdir('spec/app') unless File.exist?('spec/app')
+  end
+  after(:each) do
+    FileUtils.rm_rf('spec/app') if File.exist?('spec/app')
+  end
+
+  it 'should generate for a service' do
+    protoc(%w[testdata/test.proto testdata/test_service.proto])
+    expect(files).to match_snapshot('service', snapshot_serializer: MultiFileSerializer)
+  end
+
+  it 'should not generate if no services' do
+    protoc(%w[testdata/test.proto])
+    expect(files).to match_snapshot('no_service', snapshot_serializer: MultiFileSerializer)
+  end
+end