tree-sitter-kdl 1.0.2 → 1.0.3

package/src/scanner.c

@@ -2,7 +2,7 @@
 #include "wctype.h"
 #include <ctype.h>
 
-enum { EOF };
+enum { EOF, MULTI_LINE_COMMENT };
 
 void *tree_sitter_kdl_external_scanner_create() { return NULL; }
 
@@ -17,14 +17,60 @@ void tree_sitter_kdl_external_scanner_deserialize(void *payload,
                                                   const char *buffer,
                                                   unsigned length) {}
 
+static void advance(TSLexer *lexer) { lexer->advance(lexer, false); }
+
 bool tree_sitter_kdl_external_scanner_scan(void *payload, TSLexer *lexer,
                                            const bool *valid_symbols) {
-  // check for eof
-  if (lexer->lookahead == 0) {
+  // check for End-of-file
+  if (valid_symbols[EOF] && lexer->lookahead == 0) {
     lexer->result_symbol = EOF;
-    lexer->advance(lexer, true);
+    advance(lexer);
     return true;
   }
 
+  // multi-line-comment := '/*' commented-block
+  if (lexer->lookahead == '/') {
+    advance(lexer);
+    if (lexer->lookahead != '*')
+      return false;
+    advance(lexer);
+
+    bool after_star = false;
+    unsigned nesting_depth = 1;
+    // commented-block := '*/' | (multi-line-comment | '*' | '/' | [^*/]+) commented-block
+    for (;;) {
+      switch (lexer->lookahead) {
+      case '\0':
+        return false;
+      case '*':
+        advance(lexer);
+        after_star = true;
+        break;
+      case '/':
+        if (after_star) {
+          advance(lexer);
+          after_star = false;
+          nesting_depth--;
+          if (nesting_depth == 0) {
+            lexer->result_symbol = MULTI_LINE_COMMENT;
+            return true;
+          }
+        } else {
+          advance(lexer);
+          after_star = false;
+          if (lexer->lookahead == '*') {
+            nesting_depth++;
+            advance(lexer);
+          }
+        }
+        break;
+      default:
+        advance(lexer);
+        after_star = false;
+        break;
+      }
+    }
+  }
+
   return false;
 }