screen: add //telnet completion

Author: scop

completions/screen

@@ -25,13 +25,26 @@ _screen()
     local cur prev words cword
     _init_completion || return
 
-    if [[ $cword -eq 1 && $cur == /dev* ]]; then
-        COMPREPLY=( $(compgen -W "$(shopt -s nullglob; printf '%s\n' \
-            /dev/serial/by-id/* /dev/ttyUSB* /dev/ttyACM* 2>/dev/null)" \
-            -- "$cur") )
-        return
+    if ((cword == 1)); then
+        if [[ $cur == /dev* ]]; then
+            COMPREPLY=( $(compgen -W "$(shopt -s nullglob; printf '%s\n' \
+                /dev/serial/by-id/* /dev/ttyUSB* /dev/ttyACM* 2>/dev/null)" \
+                -- "$cur") )
+            return
+        fi
+        if [[ $cur == //* ]]; then
+            COMPREPLY=( $(compgen -W '//telnet' -- "$cur") )
+            return
+        fi
     fi
 
+    case ${words[1]} in
+        //telnet)
+            ((cword == 2)) && _known_hosts_real -- "$cur"
+            return
+            ;;
+    esac
+
     if ((cword > 2)); then
         case ${words[cword-2]} in
             -*[dD])

test/fixtures/shared/.ssh/known_hosts

@@ -0,0 +1 @@
+bash-completion-canary-host.local

test/t/test_screen.py

@@ -22,3 +22,19 @@ def test_4(self, completion):
     @pytest.mark.complete("screen -T foo cat")
     def test_5(self, completion):
         assert completion
+
+    @pytest.mark.complete("screen //")
+    def test_telnet(self, completion):
+        assert completion == "//telnet"
+
+    @pytest.mark.complete("screen cat //")
+    def test_not_telnet(self, completion):
+        assert completion != "//telnet"
+
+    @pytest.mark.complete("screen //telnet ", env=dict(HOME="$PWD/shared"))
+    def test_telnet_first_arg(self, completion):
+        assert "bash-completion-canary-host.local" in completion
+
+    @pytest.mark.complete("screen //telnet foo ", env=dict(HOME="$PWD/shared"))
+    def test_telnet_other_args(self, completion):
+        assert not completion