add ? to custom operator list

Jintin · May 6, 2017 · 1f9442e · 1f9442e
1 parent 785281b
commit 1f9442e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/Parser/SwiftParser.swift b/Parser/SwiftParser.swift
@@ -23,7 +23,8 @@ let operatorList: [Character: [(String, Int)]] =
             (">->", 3), (">", 1)],
         "|": [("|||", 3), ("||=", 3), ("||", 2), ("|=", 2), ("|", 1)],
         "!": [("!==", 3), ("!=", 2)],
-        "=": [("===", 3), ("==", 2), ("=", 1)]
+        "=": [("===", 3), ("==", 2), ("=", 1)],
+        "?" : []
     ]
 
 fileprivate let negativeCheckSigns: [Character] =
@@ -91,6 +92,12 @@ class SwiftParser {
             }
             return space(with: operatorList[char]!)!
         case "?":
+            if let spaceChar = operatorList[char] {
+                if let index = space(with: spaceChar) {
+                    return index
+                }
+            }
+
             if isNext(char: "?") {
                 // MARK: check double optional or nil check
                 return add(string: "??", length: 2)