Redraw completed interactive prompt

1 files changed, 32 insertions, 3 deletions

diff --git a/lua/clever_f/acquisition_service.lua b/lua/clever_f/acquisition_service.lua
index e86bd73..3708ca4 100644
--- a/lua/clever_f/acquisition_service.lua
+++ b/lua/clever_f/acquisition_service.lua
@@ -191,7 +191,7 @@ local scope_metatable = {
   __metatable = "clever_f.acquisition_service.TemporaryResourceScope",
 }
 
-function TemporaryResourceScope.new(request, feedback)
+function TemporaryResourceScope.new(request, feedback, host)
   if not AcquisitionRequest.is(request) then
     fail("temporary resource scope requires an AcquisitionRequest", 2)
   end
@@ -199,7 +199,12 @@ function TemporaryResourceScope.new(request, feedback)
   scope_records[scope] = {
     request = request,
     feedback = feedback,
+    host = host,
     active = true,
+    interactive = not request.macro_state.executing,
+    prompt_shown = false,
+    input_completed = false,
+    acquisition_completed = false,
     cursor_marker = nil,
     direct_marker = nil,
     cursor_presentation_lease = nil,
@@ -247,6 +252,18 @@ function TemporaryResourceScope:set_input_packet(packet)
   return set_scope_resource(self, "input_packet", packet)
 end
 
+function TemporaryResourceScope:mark_prompt_shown()
+  return set_scope_resource(self, "prompt_shown", true)
+end
+
+function TemporaryResourceScope:mark_input_completed()
+  return set_scope_resource(self, "input_completed", true)
+end
+
+function TemporaryResourceScope:mark_acquisition_completed()
+  return set_scope_resource(self, "acquisition_completed", true)
+end
+
 function TemporaryResourceScope:set_acquired_target(target)
   return set_scope_resource(self, "acquired_target", target)
 end
@@ -304,6 +321,13 @@ function TemporaryResourceScope:release()
   if record == nil then
     fail("temporary resource scope is invalid", 2)
   end
+  if record.interactive
+    and record.prompt_shown
+    and record.input_completed
+    and record.acquisition_completed
+  then
+    record.host:redraw("full")
+  end
   record.active = false
 end
 
@@ -466,7 +490,7 @@ end
 function AcquisitionService:start_temporary_scope(request)
   request = AcquisitionRequest.new(request)
   local record = service_records[self]
-  local scope = TemporaryResourceScope.new(request, record.feedback)
+  local scope = TemporaryResourceScope.new(request, record.feedback, record.host)
   record.last_scope = scope
   record.started_scope_count = record.started_scope_count + 1
   return scope
@@ -702,9 +726,11 @@ function AcquisitionService:acquire(descriptor, context, position, count, macro_
   end
   if acquisition.show_prompt and not request.macro_state.executing then
     record.host:show_prompt(M.PROMPT)
+    scope:mark_prompt_shown()
   end
   record.transitions:BeginAcquisition(request.context, request.descriptor)
   local packet = scope:set_input_packet(read_input_packet(record.host))
+  scope:mark_input_completed()
   if M.is_escape(packet) then
     local outcome = scope:set_outcome(domain.ActionOutcome.escape(request.position))
     scope:release()
@@ -789,7 +815,7 @@ function AcquisitionService:acquire(descriptor, context, position, count, macro_
       window = window,
     })
   end
-  return AcquisitionResult.new(request, {
+  local result = AcquisitionResult.new(request, {
     target = target,
     target_plan = target_plan,
     motion_plan = motion_plan,
@@ -800,6 +826,9 @@ function AcquisitionService:acquire(descriptor, context, position, count, macro_
     cached_target = scope.cached_target,
     missing_previous_input = scope.missing_previous_input,
   })
+  scope:mark_acquisition_completed()
+  scope:release()
+  return result
 end
 
 function M.new(options, dependencies)