Merge pull request #1571 from skvidal/ec2

add ec2 module

ec2

@@ -0,0 +1,187 @@
+#!/usr/bin/python -tt
+# This file is part of Ansible
+#
+# Ansible is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Ansible is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Ansible.  If not, see <http://www.gnu.org/licenses/>.
+
+DOCUMENTATION = '''
+---
+module: ec2
+short_description: create an instance in ec2, return instanceid
+description:
+     - creates ec2 instances and optionally waits for it to be 'running'
+version_added: "0.9"
+options:
+  keypair:
+    description:
+      - keypair to use on the instance
+    required: true
+    default: null
+    aliases: []
+  group:
+    description:
+      - security group to use on the instance
+    required: false
+    default: 'default'
+    aliases: []
+  instance_type:
+    description:
+      - instance type to use for the instance
+    required: true
+    default: null
+    aliases: []
+  image:
+    description:
+       - emi (or ami) to use for the instance
+    required: true
+    default: null
+    aliases: []
+  kernel:
+    description:
+      - kernel eki to use for the instance
+    required: false
+    default: null
+    aliases: []
+  ramdisk:
+    description:
+      - ramdisk eri to use for the instance
+    required: false
+    default: null
+    aliases: []
+  wait:
+    description:
+      - wait for the instance to be in state 'running' before returning
+    required: False
+    default: False
+    aliases: []
+  ec2_url:
+    description:
+      - url to use to connect to ec2 or your cloud
+    required: False
+    default: null
+    aliases: []
+  ec2_secret_key:
+    description:
+      - ec2 secret key
+    required: False
+    default: null
+    aliases: []
+  ec2_access_key:
+    description:
+      - ec2 access key
+    required: False
+    default: null
+    aliases: []
+  
+examples:
+   - code: "local_action: ec2 keypair=admin instance_type=m1.large image=emi-40603AD1 wait=true group=webserver"
+     description: "Examples from Ansible Playbooks"
+   - code: 
+requirements: [ "euca2ools" ]
+author: Seth Vidal
+
+'''
+
+import euca2ools.commands.euca.runinstances
+import time
+
+def _run(cmd):
+    # returns (rc, stdout, stderr) from shell command
+    process = subprocess.Popen(cmd, stdout=subprocess.PIPE,
+                               stderr=subprocess.PIPE, shell=True)
+    stdout, stderr = process.communicate()
+    return (process.returncode, stdout, stderr)
+
+
+def main():
+    module = AnsibleModule(
+        argument_spec = dict(
+            keypair = dict(required=True),
+            group = dict(default='default'),
+            instance_type = dict(aliases=['type']),
+            image = dict(required=True),
+            kernel = dict(),
+            #count = dict(default='1'), # maybe someday
+            ramdisk = dict(),
+            wait = dict(choices=BOOLEANS, default=False),
+            ec2_url = dict(aliases=['EC2_URL']),
+            ec2_secret_key = dict(aliases=['EC2_SECRET_KEY']),
+            ec2_access_key = dict(aliases=['EC2_ACCESS_KEY']),
+        )
+    )
+
+    keypair = module.params.get('keypair')
+    group = module.params.get('group')
+    instance_type = module.params.get('instance_type')
+    image = module.params.get('image')
+    #count = module.params.get('count') 
+    kernel = module.params.get('kernel')
+    ramdisk = module.params.get('ramdisk')
+    wait = module.params.get('wait')
+    ec2_url = module.params.get('ec2_url')
+    ec2_secret_key = module.params.get('ec2_secret_key')
+    ec2_access_key = module.params.get('ec2_access_key')
+    
+    if ec2_url:
+        os.environ['EC2_URL'] = ec2_url
+    if ec2_secret_key:
+        os.environ['EC2_SECRET_KEY'] = ec2_secret_key
+    if ec2_access_key:
+        os.environ['EC2_ACCESS_KEY'] = ec2_access_key
+
+        
+    # yes I recognize how hacky this is - but it is easier than rewriting
+    # all the try/except's myself.
+    sys.argv.append(image)
+    eri = euca2ools.commands.euca.runinstances.RunInstances()
+    conn = eri.make_connection()
+    res = eri.make_request_cli(conn, 'run_instances',
+                               image_id=image,
+                               min_count=1,
+                               max_count=1,
+                               key_name=keypair,
+                               security_groups=[group],
+                               instance_type=instance_type,
+                               kernel_id=kernel,
+                               ramdisk_id=ramdisk)
+
+    instids = [ i.id for i in res.instances ]
+
+    res_list = res.connection.get_all_instances(instids)
+    this_res = res_list[0]
+    if wait:
+        # wait here until the instances are up
+        num_running = 0
+        while num_running != len(instids):
+            res_list = res.connection.get_all_instances(instids)
+            this_res = res_list[0]
+            num_running = len([ i for i in this_res.instances if i.state=='running' ])
+            time.sleep(2)
+    
+    # there's only one - but maybe one day there could be more
+    instances = []
+    for inst in this_res.instances:
+        d = {
+           'id': inst.id,
+           'public_ip': inst.ip_address,
+            }
+        instances.append(d)
+
+    result = {"changed": True,
+              "instances": instances }
+    module.exit_json(**result)
+
+# this is magic, see lib/ansible/module_common.py
+#<<INCLUDE_ANSIBLE_MODULE_COMMON>>
+
+main()