Correctly handle EOF in within 'arbitrary code' sections

Not handling these correctly caused the editor not to accept inputs in empty files and certain other situations (cherry picked from commit 3865577)
tsalzinger · Jul 21, 2019 · a3d3d0b · a3d3d0b
1 parent f8752e8
commit a3d3d0b
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/main/kotlin/me/salzinger/intellij/coco/Coco.flex b/src/main/kotlin/me/salzinger/intellij/coco/Coco.flex
@@ -163,6 +163,11 @@ ANY_CHAR=.
                            return EMBEDDED_CODE;
                           }
                        }
+  <<EOF>>              {
+                         yybegin(STATE_COMPILER);
+                         return EMBEDDED_CODE;
+                        }
+
   // everything is valid in here
   [^]                  { }
 }
@@ -186,6 +191,12 @@ ANY_CHAR=.
   "CHARACTERS"         { yypushback(yylength()); yybegin(STATE_COMPILER); return EMBEDDED_CODE; }
   {LINE_COMMENT}       { globalsStarted = true; }
   {BLOCK_COMMENT}      { globalsStarted = true; }
+
+  <<EOF>>              {
+                         yybegin(STATE_COMPILER);
+                         return EMBEDDED_CODE;
+                       }
+
   [^]                  { globalsStarted = true; }
 }