[Swift] Fix type name bugs

rouge-ruby · Jan 11, 2018 · 2b96a1a · 2b96a1a
1 parent d6c2cb7
commit 2b96a1a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/rouge/lexers/swift.rb b/lib/rouge/lexers/swift.rb
@@ -112,8 +112,8 @@ class deinit enum extension final func import init internal lazy let optional pr
           push :tuple
         end
 
-        rule /(?!\b(if|while|for|private|internal|unowned|switch|case)\b)\b#{id}(?=(\?|!)?\s*[({])/ do |m|
-          if m[1] == '(' && m[0] =~ /^[[:upper:]]/
+        rule /(?!\b(if|while|for|private|internal|unowned|switch|case|do|catch)\b)\b#{id}(?=(\?|!)?\s*[({])/ do |m|
+          if m[0] =~ /^[[:upper:]]/ || m[1] == '('
             token Keyword::Type
           else
             token Name::Function