ssh-keygen: add -b arg completions according to specified -t

scop · Apr 29, 2020 · 69b477b · 69b477b
1 parent edc4f9b
commit 69b477b
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/completions/ssh-keygen b/completions/ssh-keygen
@@ -6,7 +6,23 @@ _ssh_keygen()
     _init_completion -n = || return
 
     case $prev in
-        -*[abCIJjMNPSVWz])
+        -*[aCIJjMNPSVWz])
+            return
+            ;;
+        -*b)
+            local -a sizes
+            case "${words[*]}" in
+                *" -t dsa"?( *))
+                    sizes=(1024)
+                    ;;
+                *" -t ecdsa"?( *))
+                    sizes=(256 384 521)
+                    ;;
+                *" -t rsa"?( *))
+                    sizes=(1024 2048 3072 4096)
+                    ;;
+            esac
+            COMPREPLY=( $(compgen -W '${sizes[@]}' -- "$cur") )
             return
             ;;
         -*E)

diff --git a/test/t/test_ssh_keygen.py b/test/t/test_ssh_keygen.py
@@ -35,3 +35,11 @@ def test_r_without_Y(self, completion):
     @pytest.mark.complete("ssh-keygen -Y foo -r ")
     def test_r_with_Y(self, completion):
         assert "ssh/" in completion
+
+    @pytest.mark.complete("ssh-keygen -t ecdsa -b ")
+    def test_ecdsa_b(self, completion):
+        assert completion
+
+    @pytest.mark.complete("ssh-keygen -t ecdsa-sk -b ")
+    def test_ecdsa_sk_b(self, completion):
+        assert not completion