[shell mode] Fix number highlight at end of line.

zenorocha · Apr 30, 2014 · b6fb1fc · b6fb1fc
1 parent 097ef7e
commit b6fb1fc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/mode/shell/shell.js b/mode/shell/shell.js
@@ -64,7 +64,7 @@ CodeMirror.defineMode('shell', function() {
     }
     if (/\d/.test(ch)) {
       stream.eatWhile(/\d/);
-      if(!/\w/.test(stream.peek())) {
+      if(stream.eol() || !/\w/.test(stream.peek())) {
         return 'number';
       }
     }

diff --git a/mode/shell/test.js b/mode/shell/test.js
@@ -37,10 +37,11 @@
   MT("comment",
      "text [comment # Blurb]");
 
-  // TODO: numbers at EOL are not styled 'number'
+  MT("numbers",
+     "[number 0] [number 1] [number 2]");
   MT("keywords",
      "[keyword while] [atom true]; [keyword do]",
-     "  [builtin sleep] 3",
+     "  [builtin sleep] [number 3]",
      "[keyword done]");
   MT("options",
      "[builtin ls] [attribute -l] [attribute --human-readable]");