Do not extract enumeration types.

iamazeem · Sep 16, 2023 · ed22846 · ed22846
1 parent dc8e1b1
commit ed22846
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