feat: refactors

lua/cp/runner/execute.lua

@@ -142,7 +142,7 @@ function M.run(cmd, stdin, timeout_ms, memory_mb, on_complete)
       or lower:find('enomem', 1, true)
     local near_cap = peak_mb >= (0.90 * memory_mb)
 
-    local mled = (peak_mb >= memory_mb) or near_cap or (oom_hint and not tled)
+    local mled = (peak_mb >= memory_mb) or near_cap or (oom_hint ~= nil and not tled)
 
     if tled then
       logger.log(('Execution timed out in %.1fms.'):format(dt))

lua/cp/runner/run.lua

@@ -101,7 +101,7 @@ end
 
 ---@param test_case RanTestCase
 ---@param debug boolean?
----@param on_complete fun(result: { status: "pass"|"fail"|"tle"|"mle", actual: string, actual_highlights: Highlight[], error: string, stderr: string, time_ms: number, code: integer, ok: boolean, signal: string, tled: boolean, mled: boolean, rss_mb: number })
+---@param on_complete fun(result: { status: "pass"|"fail"|"tle"|"mle", actual: string, actual_highlights: Highlight[], error: string, stderr: string, time_ms: number, code: integer, ok: boolean, signal: string?, tled: boolean, mled: boolean, rss_mb: number })
 local function run_single_test_case(test_case, debug, on_complete)
   local source_file = state.get_source_file()
 

lua/cp/ui/views.lua

@@ -81,15 +81,26 @@ function M.toggle_interactive(interactor_cmd)
 
   local execute = require('cp.runner.execute')
   local run = require('cp.runner.run')
-  local compile_result = execute.compile_problem()
+
+  local function restore_session()
+    if state.saved_interactive_session then
+      vim.cmd.source(state.saved_interactive_session)
+      vim.fn.delete(state.saved_interactive_session)
+      state.saved_interactive_session = nil
+    end
+  end
+
+  execute.compile_problem(false, function(compile_result)
     if not compile_result.success then
       run.handle_compilation_failure(compile_result.output)
+      restore_session()
       return
     end
 
     local binary = state.get_binary_file()
     if not binary or binary == '' then
       logger.log('No binary produced.', vim.log.levels.ERROR)
+      restore_session()
       return
     end
 
@@ -104,14 +115,11 @@ function M.toggle_interactive(interactor_cmd)
           ("Interactor '%s' is not executable."):format(interactor_cmd),
           vim.log.levels.ERROR
         )
-      if state.saved_interactive_session then
+        restore_session()
-        vim.cmd.source(state.saved_interactive_session)
-        vim.fn.delete(state.saved_interactive_session)
-        state.saved_interactive_session = nil
-      end
         return
       end
-    local orchestrator = vim.fn.fnamemodify(utils.get_plugin_path() .. '/scripts/interact.py', ':p')
+      local orchestrator =
+        vim.fn.fnamemodify(utils.get_plugin_path() .. '/scripts/interact.py', ':p')
       cmdline = table.concat({
         'uv',
         'run',
@@ -139,11 +147,7 @@ function M.toggle_interactive(interactor_cmd)
           pcall(vim.fn.jobstop, job)
         end
       end
-    if state.saved_interactive_session then
+      restore_session()
-      vim.cmd.source(state.saved_interactive_session)
-      vim.fn.delete(state.saved_interactive_session)
-      state.saved_interactive_session = nil
-    end
       state.interactive_buf = nil
       state.interactive_win = nil
       state.set_active_panel(nil)
@@ -189,6 +193,7 @@ function M.toggle_interactive(interactor_cmd)
     state.interactive_buf = term_buf
     state.interactive_win = term_win
     state.set_active_panel('interactive')
+  end)
 end
 
 ---@return integer, integer