From d5eca3a6bf3ac91bf5432d4dbb0f9953ff6c12e8 Mon Sep 17 00:00:00 2001
From: Julien Nicoulaud <julien.nicoulaud@gmail.com>
Date: Tue, 9 Aug 2011 14:30:50 +0200
Subject: [PATCH] #40 adb: Complete connect and disconnect commands

---
 _adb | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/_adb b/_adb
index cae4b96..623ff7b 100644
--- a/_adb
+++ b/_adb
@@ -67,11 +67,8 @@ _adb() {
         (help|version|devices|jdwp|bugreport|wait-for-device|start-server|kill-server|get-state|get-serialno|status-window|remount|reboot-bootloader|root|usb)
           _message 'no more arguments' && ret=0
         ;;
-        (connect)
-          _message '<host>[:<port>]' && ret=0 # TODO Not implemented
-        ;;
-        (disconnect)
-          _message '[<host>[:<port>]]' && ret=0 # TODO Not implemented
+        (connect|disconnect)
+          _arguments '1: :_adb_host_colon_ports' && ret=0
         ;;
         (push)
           _message '<local> <remote>' && ret=0 # TODO Not implemented
@@ -175,4 +172,15 @@ _adb_scripting_cmds() {
   _describe -t scripting-commands 'scripting command' commands "$@"
 }
 
+(( $+functions[_adb_host_colon_ports] )) ||
+_adb_host_colon_ports() {
+  local ret=1
+  if compset -P '*:'; then
+    _message 'port' && ret=0
+  else
+    _wanted hosts expl 'host' _hosts -qS: && ret=0
+  fi
+  return ret
+}
+
 _adb "$@"