screen: add serial device basic arg (non)completion

Author: scop

completions/screen

@@ -39,6 +39,14 @@ _screen()
     fi
 
     case ${words[1]} in
+        /dev*)
+            if ((cword == 2)); then
+                COMPREPLY=( $(compgen -W '110 300 600 1200 2400 4800 9600 \
+                14400 19200 38400 57600 115200 128000 256000' -- "$cur") )
+                # TODO more, comma separated options
+            fi
+            return
+            ;;
         //telnet)
             ((cword == 2)) && _known_hosts_real -- "$cur"
             return

test/t/test_screen.py

@@ -38,3 +38,11 @@ def test_telnet_first_arg(self, completion):
     @pytest.mark.complete("screen //telnet foo ", env=dict(HOME="$PWD/shared"))
     def test_telnet_other_args(self, completion):
         assert not completion
+
+    @pytest.mark.complete("screen /dev/ttyUSB0 ")
+    def test_serial_2nd_arg(self, completion):
+        assert "19200" in completion
+
+    @pytest.mark.complete("screen /dev/ttyUSB0 9600 ")
+    def test_serial_3rdplus_arg(self, completion):
+        assert not completion