Do not extract enumeration types. (#10)

iamazeem · Sep 16, 2023 · eb44e6e · eb44e6e
1 parent 315de37
commit eb44e6e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bin/proto-convert b/bin/proto-convert
@@ -70,7 +70,8 @@ end
 def valid_msgtype?(compiled_proto, msg_type)
   msg_types = []
   File.foreach(compiled_proto) do |line|
-    if line.include?('::Google::Protobuf::DescriptorPool.generated_pool.lookup')
+    line.strip!
+    if line.include?('::Google::Protobuf::DescriptorPool.generated_pool.lookup') && line.end_with?('.msgclass')
       extracted_msg_type = line[/"([^"]*)"/, 1].freeze
       msg_types.push(extracted_msg_type) unless extracted_msg_type.nil?
     end