fix parsing without module keyword

atsky · Feb 25, 2015 · fabf5dd · fabf5dd
1 parent 9dc94e8
commit fabf5dd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/plugin/src/org/jetbrains/haskell/parser/LexerState.kt b/plugin/src/org/jetbrains/haskell/parser/LexerState.kt
@@ -113,7 +113,11 @@ public fun getCachedTokens(builder: PsiBuilder): CachedTokens {
 }
 
 public fun newLexerState(tokens: CachedTokens): LexerState {
-    return LexerState(tokens, 0, 0, null, null)
+    if (tokens.tokens.firstOrNull() == HaskellLexerTokens.MODULE) {
+        return LexerState(tokens, 0, 0, null, null)
+    } else {
+        return LexerState(tokens, 0, 0, null, IntStack(0, null))
+    }
 }
 
 public class CachedTokens(val tokens: List<IElementType>,