1 files changed, 17 insertions, 4 deletions

diff --git a/src/Lexical.kt b/src/Lexical.kt
index 7b92755..1655805 100644
--- a/src/Lexical.kt
+++ b/src/Lexical.kt
@@ -1,12 +1,16 @@
+import exceptions.UnknownType
+
 class Lexical {
     fun analyze(source: String): MutableList<Pair<String, TokenType>> {
         var buffer = ""
+        var stringMode = false
         val tokens = mutableListOf<Pair<String, TokenType>>()
         for (i in source.indices) {
             buffer += source[i]
-            val tokenType = getTokenType(buffer, if (source.length > i + 1) source[i + 1] else ' ')
+            val tokenType = getTokenType(buffer, if (source.length > i + 1) source[i + 1] else ' ', stringMode)
             if (tokenType != TokenType.Skip) {
                 tokens += buffer to tokenType
+                if (buffer == "\"") stringMode = true
                 buffer = ""
             }
         }
@@ -16,7 +20,7 @@ class Lexical {
     /**
      * Matches the tokens to a [TokenType]
      */
-    private fun getTokenType(token: String, next: Char): TokenType {
+    private fun getTokenType(token: String, next: Char, stringMode: Boolean): TokenType {
         return when {
             token + next in keyword -> TokenType.Skip
             token in keyword -> TokenType.Keyword
@@ -41,7 +45,12 @@ class Lexical {
 
             token in punctuation -> TokenType.Punctuation
 
-            token == " " -> TokenType.Empty
+            token in brackets -> TokenType.Bracket
+
+            token in classifier -> TokenType.Classifier
+
+            token.contains(" ") && token.length > 1 && !stringMode -> throw UnknownType(token)
+            token == " " && !stringMode -> TokenType.Empty
 
             else -> TokenType.Skip
         }
@@ -53,6 +62,8 @@ class Lexical {
     private val comparison = listOf("==", "!=", "<", "<=", ">", ">=")
     private val logical = listOf("&&", "||", "!")
     private val punctuation = listOf(",", ":", ".")
+    private val brackets = listOf("(", ")", "[", "]", "{", "}") // TODO: Use brackets for functions
+    private val classifier = listOf("\"")  // TODO: Add char mode e.g 'a'
 }
 
 enum class TokenType {
@@ -61,9 +72,11 @@ enum class TokenType {
     Arithmetic,
     Comparison,
     Logical,
-    Punctuation,
     Identifier,
     Constant,
+    Punctuation,
+    Bracket,
+    Classifier,
     Empty,
     Skip
 }
\ No newline at end of file