Harden Gemini scripts with verified SSH trust

scripts/provision_wizard.py

@@ -15,10 +15,15 @@ import sys
 import time
 import argparse
 import requests
-import subprocess
 import json
 from typing import Optional, Dict, Any
 
+SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
+if SCRIPT_DIR not in sys.path:
+    sys.path.insert(0, SCRIPT_DIR)
+
+from ssh_trust import VerifiedSSHExecutor
+
 # --- CONFIGURATION ---
 DO_API_URL = "https://api.digitalocean.com/v2"
 # We expect DIGITALOCEAN_TOKEN to be set in the environment.
@@ -30,13 +35,14 @@ DEFAULT_IMAGE = "ubuntu-22-04-x64"
 LLAMA_CPP_REPO = "https://github.com/ggerganov/llama.cpp"
 
 class Provisioner:
-    def __init__(self, name: str, size: str, model: str, region: str = DEFAULT_REGION):
+    def __init__(self, name: str, size: str, model: str, region: str = DEFAULT_REGION, executor=None):
         self.name = name
         self.size = size
         self.model = model
         self.region = region
         self.droplet_id = None
         self.ip_address = None
+        self.executor = executor or VerifiedSSHExecutor(auto_enroll=True)
 
     def log(self, message: str):
         print(f"[*] {message}")
@@ -104,13 +110,8 @@ class Provisioner:
         self.log(f"Droplet IP: {self.ip_address}")
 
     def run_remote(self, command: str):
-        # Using subprocess to call ssh. Assumes local machine has the right private key.
-        ssh_cmd = [
-            "ssh", "-o", "StrictHostKeyChecking=no",
-            f"root@{self.ip_address}", command
-        ]
-        result = subprocess.run(ssh_cmd, capture_output=True, text=True)
-        return result
+        # Uses verified host trust. Brand-new nodes explicitly enroll on first contact.
+        return self.executor.run_script(self.ip_address, command, timeout=60)
 
     def setup_wizard(self):
         self.log("Starting remote setup...")