From 75064228bde9880e3aab4b14369c963b6c4a9e87 Mon Sep 17 00:00:00 2001
From: Mariusz Jurowicz <mariusz.jurowicz@xsolve.pl>
Date: Fri, 13 Oct 2017 16:09:40 +0200
Subject: [PATCH] #6137 add commenting feature to newly created cells

---
 beakerx/js/src/extension.js | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/beakerx/js/src/extension.js b/beakerx/js/src/extension.js
index 776c90d353..4bf86d918c 100644
--- a/beakerx/js/src/extension.js
+++ b/beakerx/js/src/extension.js
@@ -126,16 +126,15 @@ define([
           }
         });
       });
-
-      Jupyter.notebook.get_cells().map(function(cell, i) {
-        if (cell.cell_type === 'code'){
-          setCodeMirrorLineComment(cell.code_mirror);
-        }
-      });
     });
     Jupyter.notebook.events.on('kernel_interrupting.Kernel', function() {
       interrupt();
     });
+    Jupyter.notebook.get_cells().map(function(cell, i) {
+      if (cell.cell_type === 'code'){
+        setCodeMirrorLineComment(cell.code_mirror);
+      }
+    });
   }
 
   installKernelHandler();
@@ -326,6 +325,9 @@ define([
             callback_notebook_loaded();
           }
           events.on('notebook_loaded.Notebook', callback_notebook_loaded);
+          events.on('create.Cell', function(data, cell) {
+            setCodeMirrorLineComment(cell.cell.code_mirror);
+          });
         }).catch(function (reason) {
           console.error(log_prefix, 'unhandled error:', reason);
         });