diff options
3 files changed, 15 insertions, 1 deletions
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/parser/TokenType.java b/src/main/java/org/zwobble/hobgoblin/compiler/parser/TokenType.java index 8277653..df86ae6 100644 --- a/src/main/java/org/zwobble/hobgoblin/compiler/parser/TokenType.java +++ b/src/main/java/org/zwobble/hobgoblin/compiler/parser/TokenType.java @@ -6,6 +6,7 @@ enum TokenType { IDENTIFIER, WHITESPACE, + KEYWORD_ENUM, KEYWORD_FIELD, KEYWORD_FROM, KEYWORD_IMPORT, diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/parser/Tokenizer.java b/src/main/java/org/zwobble/hobgoblin/compiler/parser/Tokenizer.java index 8acde49..c0cf716 100644 --- a/src/main/java/org/zwobble/hobgoblin/compiler/parser/Tokenizer.java +++ b/src/main/java/org/zwobble/hobgoblin/compiler/parser/Tokenizer.java @@ -122,7 +122,9 @@ class Tokenizer { var sourceRange = start.to(end); TokenType tokenType; - if (CharSequence.compare(sourceRange.charSequence(), "field") == 0) { + if (CharSequence.compare(sourceRange.charSequence(), "enum") == 0) { + tokenType = TokenType.KEYWORD_ENUM; + } else if (CharSequence.compare(sourceRange.charSequence(), "field") == 0) { tokenType = TokenType.KEYWORD_FIELD; } else if (CharSequence.compare(sourceRange.charSequence(), "from") == 0) { tokenType = TokenType.KEYWORD_FROM; diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/parser/TokenizerTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/parser/TokenizerTests.java index 0440719..848738a 100644 --- a/src/test/java/org/zwobble/hobgoblin/compiler/parser/TokenizerTests.java +++ b/src/test/java/org/zwobble/hobgoblin/compiler/parser/TokenizerTests.java @@ -19,6 +19,17 @@ public class TokenizerTests { } @Test + public void keywordEnumIsTokenized() { + var sourceText = SourceText.fromString("<string>", "enum"); + + var tokens = Tokenizer.tokenize(sourceText); + + assertThat(tokens.tokens(), isSequence( + equalTo(new Token(TokenType.KEYWORD_ENUM, sourceRange(sourceText, 0, 4))) + )); + } + + @Test public void keywordFieldIsTokenized() { var sourceText = SourceText.fromString("<string>", "field"); |