From 9b803a22953e3b31cda6fe560e4cdf2110e4d484 Mon Sep 17 00:00:00 2001
From: Raymond Hill <rhill@raymondhill.net>
Date: Mon, 5 Mar 2018 11:25:45 -0500
Subject: [PATCH] fine tuning syntax highlighter

---
 src/css/codemirror.css                    |  9 +--
 src/js/codemirror/ubo-static-filtering.js | 69 +++++++++++++++++------
 2 files changed, 57 insertions(+), 21 deletions(-)

diff --git a/src/css/codemirror.css b/src/css/codemirror.css
index 6c92a2dab..9f929738d 100644
--- a/src/css/codemirror.css
+++ b/src/css/codemirror.css
@@ -18,10 +18,11 @@
     height: 100%;
     }
 
-.cm-s-default .cm-comment {color: #777;}
-.cm-staticext {color: #008;}
-.cm-staticnet.cm-block {color: #800;}
-.cm-staticnet.cm-allow {color: #004f00;}
+.cm-s-default .cm-comment { color: #777; }
+.cm-staticext { color: #008; }
+.cm-staticextA { font-weight: bold; }
+.cm-staticnet.cm-block { color: #800; }
+.cm-staticnet.cm-allow { color: #004f00; }
 
 .cm-search-widget {
     align-items: center;
diff --git a/src/js/codemirror/ubo-static-filtering.js b/src/js/codemirror/ubo-static-filtering.js
index 7509e4860..24a871400 100644
--- a/src/js/codemirror/ubo-static-filtering.js
+++ b/src/js/codemirror/ubo-static-filtering.js
@@ -24,29 +24,64 @@
 'use strict';
 
 CodeMirror.defineMode("ubo-static-filtering", function() {
-    var reNotSpace = /\S/;
-    var reComment = /^(?:!|#[^#@]|#@[^#])/;
-    var reExt = /^[^#]*#(?:#[^#]|##[^#]|@#)/;
-    var reNetAllow = /^@@/;
+    var reComment1 = /^\s*!/;
+    var reComment2 = /^\s*#/;
+    var reExt = /^(\s*[^#]*)(#(?:#|@#|\$#|@\$#|\?#|@\?#))(.+)$/;
+    var reNetAllow = /^\s*@@/;
+    var lineStyle = null;
+    var lineMatches = null;
+
+    var lineStyles = {
+        staticext: [ '', 'staticextA', '' ]
+    };
+
+    var styleFromStream = function(stream) {
+        for ( var i = 1, l = 0; i < lineMatches.length; i++ ) {
+            l += lineMatches[i].length;
+            if ( stream.pos < l ) {
+                stream.pos = l;
+                var style = lineStyle;
+                var xstyle = lineStyles[style][i-1];
+                if ( xstyle !== '' ) { style += ' ' + xstyle; }
+                return style;
+            }
+        }
+        stream.skipToEnd();
+        return '';
+    };
 
     return {
         token: function(stream) {
-            var style = null;
-            var pos = stream.string.search(reNotSpace);
-            if ( pos !== -1 ) {
-                var s = stream.string.slice(pos);
-                if ( reComment.test(s) ) {
-                    style = 'comment';
-                } else if ( reExt.test(s) ) {
-                    style = 'staticext';
-                } else if ( reNetAllow.test(s) ) {
-                    style = 'staticnet allow';
-                } else {
-                    style = 'staticnet block';
+            if ( stream.sol() ) {
+                lineStyle = null;
+                lineMatches = null;
+            } else if ( lineStyle !== null ) {
+                return styleFromStream(stream);
+            }
+            if ( reComment1.test(stream.string) ) {
+                stream.skipToEnd();
+                return 'comment';
+            }
+            if ( stream.string.indexOf('#') !== -1 ) {
+                lineMatches = reExt.exec(stream.string);
+                if (
+                    lineMatches !== null &&
+                    lineMatches[3].startsWith('##') === false
+                ) {
+                    lineStyle = 'staticext';
+                    return styleFromStream(stream);
+                }
+                if ( reComment2.test(stream.string) ) {
+                    stream.skipToEnd();
+                    return 'comment';
                 }
             }
+            if ( reNetAllow.test(stream.string) ) {
+                stream.skipToEnd();
+                return 'staticnet allow';
+            }
             stream.skipToEnd();
-            return style;
+            return 'staticnet block';
         }
     };
 });