Update isValidKeyAlgorithm impl

Infisical · Dec 18, 2024 · 31a21a4 · 31a21a4
1 parent 765280e
commit 31a21a4
Showing 1 changed file with 2 additions and 6 deletions.
diff --git 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 {