From ba4838cde57b71a205ac4601e360f8697e32f1fb Mon Sep 17 00:00:00 2001
From: James Cammarata <jimi@sngx.net>
Date: Tue, 1 Apr 2014 22:14:35 -0500
Subject: [PATCH] Adding wait/wait_timeout parameters to the ec2_key module

Fixes #6455
---
 library/cloud/ec2_key | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/library/cloud/ec2_key b/library/cloud/ec2_key
index 289deb6c9d6..9e94d22a0b3 100644
--- a/library/cloud/ec2_key
+++ b/library/cloud/ec2_key
@@ -70,6 +70,20 @@ options:
     default: null
     aliases: []
     version_added: "1.6"
+  wait:
+    description:
+      - Wait for the specified action to complete before returning.
+    required: false
+    default: false
+    aliases: []
+    version_added: "1.6"
+  wait_timeout:
+    description:
+      - How long before wait gives up, in seconds
+    required: false
+    default: 300
+    aliases: []
+    version_added: "1.6"
 
 requirements: [ "boto" ]
 author: Vincent Viallet
@@ -124,6 +138,8 @@ def main():
             name=dict(required=True),
             key_material=dict(required=False),
             state = dict(default='present', choices=['present', 'absent']),
+            wait = dict(type='bool', default=False),
+            wait_timeout = dict(default=300),
         )
     )
     module = AnsibleModule(
@@ -134,6 +150,8 @@ def main():
     name = module.params['name']
     state = module.params.get('state')
     key_material = module.params.get('key_material')
+    wait = module.params.get('wait')
+    wait_timeout = int(module.params.get('wait_timeout'))
 
     changed = False
 
@@ -148,6 +166,16 @@ def main():
             '''found a match, delete it'''
             try:
                 key.delete()
+                if wait:
+                    start = time.time()
+                    action_complete = False
+                    while (time.time() - start) < wait_timeout:
+                        if not ec2.get_key_pair(name):
+                            action_complete = True
+                            break
+                        time.sleep(1)
+                    if not action_complete:
+                        module.fail_json(msg="timed out while waiting for the key to be removed")
             except Exception, e:
                 module.fail_json(msg="Unable to delete key pair '%s' - %s" % (key, e))
             else:
@@ -178,6 +206,18 @@ def main():
                     retrieve the private key
                     '''
                     key = ec2.create_key_pair(name)
+
+                if wait:
+                    start = time.time()
+                    action_complete = False
+                    while (time.time() - start) < wait_timeout:
+                        if ec2.get_key_pair(name):
+                            action_complete = True
+                            break
+                        time.sleep(1)
+                    if not action_complete:
+                        module.fail_json(msg="timed out while waiting for the key to be created")
+
             changed = True
 
     if key: