@take-out/scripts 0.1.16 → 0.1.18

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@take-out/scripts",
-  "version": "0.1.16",
+  "version": "0.1.18",
   "type": "module",
   "main": "./src/run.ts",
   "sideEffects": false,

package/src/helpers/ssh.ts

@@ -9,15 +9,38 @@ export async function checkSSHKey(sshKeyPath: string): Promise<void> {
   }
 }
 
-export async function testSSHConnection(host: string, sshKey: string): Promise<void> {
+async function sleep(ms: number) {
+  return new Promise((resolve) => setTimeout(resolve, ms))
+}
+
+export async function testSSHConnection(
+  host: string,
+  sshKey: string,
+  options?: { retries?: number; retryDelayMs?: number }
+): Promise<void> {
+  const maxRetries = options?.retries ?? 3
+  const retryDelay = options?.retryDelayMs ?? 5000
+
   console.info('\n🔑 testing ssh connection...\n')
 
-  try {
-    await run(
-      `ssh -i ${sshKey} -o StrictHostKeyChecking=no -o ConnectTimeout=5 ${host} "echo 'SSH connection successful'"`
-    )
-    console.info('✅ ssh connection verified')
-  } catch {
-    throw new Error(`cannot connect to ${host} - check ssh key: ${sshKey}`)
+  let lastError: Error | null = null
+
+  for (let attempt = 1; attempt <= maxRetries; attempt++) {
+    try {
+      await run(
+        `ssh -i ${sshKey} -o StrictHostKeyChecking=no -o ConnectTimeout=10 ${host} "echo 'SSH connection successful'"`,
+        { silent: attempt > 1 }
+      )
+      console.info('✅ ssh connection verified')
+      return
+    } catch (err) {
+      lastError = err as Error
+      if (attempt < maxRetries) {
+        console.info(`  ssh attempt ${attempt}/${maxRetries} failed, retrying in ${retryDelay / 1000}s...`)
+        await sleep(retryDelay)
+      }
+    }
   }
+
+  throw new Error(`cannot connect to ${host} after ${maxRetries} attempts - check ssh key: ${sshKey}`)
 }