Support nested comments

tomaz1502 · Feb 5, 2025 · 22b7540 · 22b7540
1 parent 96511fe
commit 22b7540
Showing 1 changed file with 7 additions and 13 deletions.
diff --git a/mode/lean/lean.js b/mode/lean/lean.js
@@ -16,25 +16,19 @@
       token: function(stream, state) {
         if (stream.eatSpace()) return null;
 
-        if (state.inComment) {
-          if (stream.match(/.*?-\//)) {
-            state.inComment = false;
+        if (state.commentLevel > 0) {
+          if (stream.match(/.*?\/-/)) {
+            state.commentLevel += 1;
+          } else if (stream.match(/.*?-\//)) {
+            state.commentLevel -= 1;
           } else {
             stream.skipToEnd();
           }
           return "comment";
         }
 
-
-        if (state.inComment) {
-          if (stream.match(/-\//)) {
-            state.inComment = false;
-          }
-          return "comment";
-        }
-
         if (stream.match(/\/-/)) {
-          state.inComment = true;
+          state.commentLevel += 1;
           return "comment";
         }
 
@@ -99,7 +93,7 @@
         return null;
       },
       startState: function() {
-        return { inDefinition: false, inComment: false };
+        return { inDefinition: false, commentLevel: 0 };
       }
     };
   });