From 31a21a432d0fad9a5e6dd61b2c265f2beddb6adf Mon Sep 17 00:00:00 2001
From: Tuan Dang <dangtony98@gmail.com>
Date: Tue, 17 Dec 2024 20:40:03 -0800
Subject: [PATCH] Update isValidKeyAlgorithm impl

---
 cli/packages/cmd/ssh.go | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/cli/packages/cmd/ssh.go b/cli/packages/cmd/ssh.go
index 63160b8bb..d611e3b8d 100644
--- a/cli/packages/cmd/ssh.go
+++ b/cli/packages/cmd/ssh.go
@@ -52,12 +52,8 @@ var algoToFileName = map[infisicalSdkUtil.CertKeyAlgorithm]string{
 }
 
 func isValidKeyAlgorithm(algo infisicalSdkUtil.CertKeyAlgorithm) bool {
-	switch algo {
-	case infisicalSdkUtil.RSA2048, infisicalSdkUtil.RSA4096, infisicalSdkUtil.ECDSAP256, infisicalSdkUtil.ECDSAP384:
-		return true
-	default:
-		return false
-	}
+    _, exists := algoToFileName[algo]
+    return exists
 }
 
 func isValidCertType(certType infisicalSdkUtil.SshCertType) bool {