Migrate instances when draining and verify hosts before enabling

src/lib/server/provisioning/agent-events.ts

@@ -5,10 +5,11 @@ import { recordProgress } from './progress.js';
 export function recordAgentReceipt(operationId: string, state: string, at: string): void {
     // Host operations also serve installations that have no provisioning schema yet.
     if (!db().prepare("SELECT 1 FROM sqlite_master WHERE type='table' AND name='native_provisioning'").get()) return;
-    const job = db().prepare(`SELECT d.id, d.phase_key, n.operation_id FROM native_provisioning n
+    const job = db().prepare(`SELECT d.id, d.phase_key, COALESCE(n.operation_id,m.remove_operation_id) AS operation_id FROM native_provisioning n
         JOIN server_deployments d ON d.id=n.deployment_id
-        WHERE (n.operation_id=? OR n.stage_operation_id=?) AND d.server_deleted_at IS NULL
-        AND d.state IN ('queued','waiting','running','awaiting-health','interrupted')`).get(operationId, operationId) as
+        LEFT JOIN ec_host_migrations m ON m.deployment_id=n.deployment_id
+        WHERE (n.operation_id=? OR n.stage_operation_id=? OR m.remove_operation_id=?) AND d.server_deleted_at IS NULL
+        AND d.state IN ('queued','waiting','running','awaiting-health','interrupted')`).get(operationId, operationId, operationId) as
         { id: string; phase_key: string | null; operation_id: string | null } | undefined;
     if (!job) return;
     const staging = job.operation_id !== operationId;