feat: :ReevaluateAllCells command (#85)

benlubas · Dec 14, 2023 · 6da3d19 · 6da3d19
1 parent 1ba4023
commit 6da3d19
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/rplugin/python3/molten/__init__.py b/rplugin/python3/molten/__init__.py
@@ -587,6 +587,17 @@ def cmd(kernel_id: str):
             """
             self.nvim.exec_lua(lua, async_=False)
 
+    @pynvim.command("MoltenReevaluateAll", nargs=0, sync=True)  # type: ignore
+    @nvimui  # type: ignore
+    def command_reevaluate_all(self) -> None:
+        self._initialize_if_necessary()
+
+        molten_kernels = self._get_current_buf_kernels(True)
+        assert molten_kernels is not None
+
+        for kernel in molten_kernels:
+            kernel.reevaluate_all()
+
     @pynvim.command("MoltenReevaluateCell", nargs=0, sync=True)  # type: ignore
     @nvimui  # type: ignore
     def command_evaluate_cell(self) -> None:

diff --git a/rplugin/python3/molten/moltenbuffer.py b/rplugin/python3/molten/moltenbuffer.py
@@ -115,6 +115,11 @@ def run_code(self, code: str, span: CodeCell) -> None:
 
         self._check_if_done_running()
 
+    def reevaluate_all(self) -> None:
+        for span in sorted(self.outputs.keys(), key=lambda s: s.begin):
+            code = span.get_text(self.nvim)
+            self.run_code(code, span)
+
     def reevaluate_cell(self) -> bool:
         self.selected_cell = self._get_selected_span()
         if self.selected_cell is None: