ansible: support local connections too

ansible_mitogen/connection/mitogen.py

@@ -47,14 +47,13 @@ class Connection(ansible.plugins.connection.ConnectionBase):
     def connected(self):
         return self.router is not None
 
-    def _connect(self):
-        if self.connected:
-            return
-
-        path = os.environ['LISTENER_SOCKET_PATH']
-        self.router, self.parent = mitogen.unix.connect(path)
+    def _connect_local(self):
+        return mitogen.service.call(self.parent, 500, {
+            'method': 'local',
+        }
 
-        host = mitogen.service.call(self.parent, 500, cast({
+    def _connect_ssh(self):
+        return mitogen.service.call(self.parent, 500, cast({
             'method': 'ssh',
             'hostname': self._play_context.remote_addr,
             'username': self._play_context.remote_user,
@@ -64,18 +63,33 @@ class Connection(ansible.plugins.connection.ConnectionBase):
             'ssh_path': self._play_context.ssh_executable,
         }))
 
-        if not self._play_context.become:
-            self.context = host
-        else:
-            self.context = mitogen.service.call(self.parent, 500, cast({
+    def _connect_sudo(self, via):
+        return mitogen.service.call(self.parent, 500, cast({
             'method': 'sudo',
             'username': self._play_context.become_user,
             'password': self._play_context.password,
             'python_path': '/usr/bin/python',
-                'via': host,
+            'via': via,
             'debug': True,
         }))
 
+    def _connect(self):
+        if self.connected:
+            return
+
+        path = os.environ['LISTENER_SOCKET_PATH']
+        self.router, self.parent = mitogen.unix.connect(path)
+
+        if self._play_context.connection == 'local':
+            host = self._connect_local()
+        else:
+            host = self._connect_ssh(self)
+
+        if not self._play_context.become:
+            self.context = host
+        else:
+            self.context = self._connect_sudo(via=host)
+
     def call_async(self, func, *args, **kwargs):
         self._connect()
         print[func, args, kwargs]