Match module with end in block matcher plugin

fonsp · Nov 13, 2023 · 9ac9c88 · 9ac9c88
1 parent 794a5f4
commit 9ac9c88
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/frontend/components/CellInput/block_matcher_plugin.js b/frontend/components/CellInput/block_matcher_plugin.js
@@ -223,6 +223,16 @@ function match_block(node) {
         return match_try_node(node)
     }
 
+    if (node.name === "module" || node.name === "baremodule") {
+        let possibly_end = node.parent.lastChild
+        let did_match = possibly_end.name === "end"
+        if (!did_match) return null
+
+        return [
+            { from: node.from, to: node.to },
+            { from: possibly_end.from, to: possibly_end.to },
+        ]
+    }
 
     return null
 }