From 6bc7add77c80f0aaff0f5a93d0a829c90a55bd7e Mon Sep 17 00:00:00 2001
From: Christian Tietze <christian.tietze@gmail.com>
Date: Wed, 27 Dec 2017 11:01:15 +0100
Subject: [PATCH] remove plural S from enum type names

---
 swift/cocoaconv.rb | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/swift/cocoaconv.rb b/swift/cocoaconv.rb
index cd29daf..064c4c3 100755
--- a/swift/cocoaconv.rb
+++ b/swift/cocoaconv.rb
@@ -93,12 +93,17 @@ class String
 end
 
 module NSEnum
-  def type_name
-    type.camelize
-  end
-  
   def self.type_names(type)
-    return "MMD6#{type.camelize}", type.camelize
+    type_name = type.camelize
+    # Remove plural "S"
+    type_name = if type_name == "TokenTypes"
+                  "TokenType"
+                elsif type_name == "ParserExtensions"
+                  "ParserExtension"
+                else
+                  type_name
+                end
+    return "MMD6#{type_name}", type_name
   end
 
   def ns_enum
@@ -107,7 +112,8 @@ module NSEnum
       .map { |line| NSEnum.case(type_name, line) }
       .join("\n")
       
-    %Q{typedef NS_ENUM(NSUInteger, #{type_name}) {
+    return %Q{
+typedef NS_ENUM(NSUInteger, #{type_name}) {
 #{ns_enum_cases}
 } NS_SWIFT_NAME(#{swift_type_name});}
   end
@@ -211,7 +217,7 @@ input.close
 
 
 result = if options[:mode] == :nsenum
-           enums.ns_enums.join("\n\n")
+           enums.ns_enums.join("\n")
          elsif options[:mode] == :swift
            enums.swift_descriptions.join("\n")
          else 
-- 
2.50.1