protect context upon load

brownplt · Sep 13, 2021 · 9032c99 · 9032c99
1 parent be1a95a
commit 9032c99
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/web/js/beforePyret.js b/src/web/js/beforePyret.js
@@ -360,6 +360,7 @@ $(function() {
           CM.clearGutter("help-gutter");
         }
       }
+
       CM.on("change", function(change) {
         function doesNotChangeFirstLine(c) { return c.from.line !== 0; }
         if(change.curOp.changeObjs.every(doesNotChangeFirstLine)) { return; }
@@ -369,6 +370,14 @@ $(function() {
           namespacemark = CM.markText({line: 0, ch: 0}, {line: 1, ch: 0}, { attributes: { useline: true }, className: "useline", atomic: true, inclusiveLeft: true, inclusiveRight: false });
         }
       });
+
+      CM.on("swapDoc", function(oldDoc) {
+        var hasNamespace = firstLineIsNamespace();
+        if(hasNamespace) {
+          if(namespacemark) { namespacemark.clear(); }
+          namespacemark = CM.markText({line: 0, ch: 0}, {line: 1, ch: 0}, { attributes: { useline: true }, className: "useline", atomic: true, inclusiveLeft: true, inclusiveRight: false });
+        }
+      });
     }
     if (useLineNumbers) {
       CM.display.wrapper.appendChild(mkWarningUpper()[0]);