Sample primary restoration policy

1 files changed, 11 insertions, 0 deletions

diff --git a/lua/clever_f/feedback_service.lua b/lua/clever_f/feedback_service.lua
index ac5f54b..4262e1b 100644
--- a/lua/clever_f/feedback_service.lua
+++ b/lua/clever_f/feedback_service.lua
@@ -662,6 +662,17 @@ local function has_target_overlay(record, window)
   return false
 end
 
+function FeedbackService:primary_restoration_active(context, window)
+  local record = service_records[self]
+  context = domain.ModeContext.from_full_mode(context)
+  local mark_char = record.policy:sample_markers().mark_char
+  if not mark_char or not M.persistent_context_eligible(context) then
+    return false
+  end
+  window = window or record.host:read_window()
+  return not has_target_overlay(record, window)
+end
+
 function FeedbackService:migrate_command(request)
   local reason = M.command_migration_reason(request)
   local record = service_records[self]