From e1458421a1b0e575d126e329e7934c936130f23a Mon Sep 17 00:00:00 2001
From: Michael Gregson <gregson.michael@gmail.com>
Date: Mon, 30 Sep 2013 20:50:22 -0600
Subject: [PATCH] Add basic domain handling to digital_ocean cloud action
---
cloud/digital_ocean | 140 +++++++++++++++++++++++++++++++++++++++++++-
1 file changed, 137 insertions(+), 3 deletions(-)
diff --git a/cloud/digital_ocean b/cloud/digital_ocean
index e5d24608fa7..fe304e0151e 100644
--- a/cloud/digital_ocean
+++ b/cloud/digital_ocean
@@ -27,7 +27,7 @@ options:
description:
- Which target you want to operate on.
default: droplet
- choices: ['droplet', 'ssh']
+ choices: ['droplet', 'ssh', 'domain']
state:
description:
- Indicate desired state of the target.
@@ -44,7 +44,7 @@ options:
- Numeric, the droplet id you want to operate on.
name:
description:
- - String, this is the name of the droplet - must be formatted by hostname rules, or the name of a SSH key.
+ - String, this is the name of the droplet - must be formatted by hostname rules, or the name of a SSH key, or the name of a domain.
unique_name:
description:
- Bool, require unique hostnames. By default, digital ocean allows multiple hosts with the same name. Setting this to "yes" allows only one host per name. Useful for idempotence.
@@ -75,6 +75,9 @@ options:
ssh_pub_key:
description:
- The public SSH key you want to add to your account.
+ ip:
+ description:
+ - The IP address to point a domain at.
notes:
- Two environment variables can be used, DO_CLIENT_ID and DO_API_KEY.
@@ -141,6 +144,31 @@ EXAMPLES = '''
size_id=1
region_id=2
image_id=3
+
+# Create a domain record
+
+- digital_ocean: >
+ state=present
+ command=domain
+ name=my.digitalocean.domain
+ ip=127.0.0.1
+
+# Create a droplet and a corresponding domain record
+
+- digital_cean: >
+ state=present
+ command=droplet
+ name=test_droplet
+ size_id=1
+ region_id=2
+ image_id=3
+ register: test_droplet
+
+- digital_ocean: >
+ state=present
+ command=domain
+ name={{ test_droplet.name }}.my.domain
+ ip={{ test_droplet.ip_address }}
'''
import sys
@@ -275,6 +303,72 @@ class SSH(JsonfyMixIn):
json = cls.manager.new_ssh_key(name, key_pub)
return cls(json)
+class DomainRecord(JsonfyMixIn):
+ manager = None
+
+ def __init__(self, json):
+ self.__dict__.update(json)
+ update_attr = __init__
+
+ def update(self, data = None, record_type = None):
+ json = self.manager.edit_domain_record(self.domain_id,
+ self.id,
+ record_type if record_type is not None else self.record_type,
+ data if data is not None else self.data)
+ self.__dict__.update(json)
+ return self
+
+ def destroy(self):
+ json = self.manager.destroy_domain_record(self.domain_id, self.id)
+ return json
+
+class Domain(JsonfyMixIn):
+ manager = None
+
+ def __init__(self, domain_json):
+ self.__dict__.update(domain_json)
+
+ def destroy(self):
+ self.manager.destroy_domain(self.id)
+
+ def records(self):
+ json = self.manager.all_domain_records(self.id)
+ return map(DomainRecord, json)
+
+ @classmethod
+ def add(cls, name, ip):
+ json = cls.manager.new_domain(name, ip)
+ return cls(json)
+
+ @classmethod
+ def setup(cls, client_id, api_key):
+ cls.manager = DoManager(client_id, api_key)
+ DomainRecord.manager = cls.manager
+
+ @classmethod
+ def list_all(cls):
+ domains = cls.manager.all_domains()
+ return map(cls, domains)
+
+ @classmethod
+ def find(cls, name=None, id=None):
+ if name is None and id is None:
+ return False
+
+ domains = Domain.list_all()
+
+ if id is not None:
+ for domain in domains:
+ if domain.id == id:
+ return domain
+
+ if name is not None:
+ for domain in domains:
+ if domain.name == name:
+ return domain
+
+ return False
+
def core(module):
def getkeyordie(k):
v = module.params[k]
@@ -361,11 +455,50 @@ def core(module):
key.destroy()
module.exit_json(changed=True)
+ elif command == 'domain':
+ Domain.setup(client_id, api_key)
+ if state in ('present', 'active'):
+ domain = Domain.find(id=module.params["id"])
+
+ if not domain:
+ domain = Domain.find(name=getkeyordie("name"))
+
+ if not domain:
+ domain = Domain.add(getkeyordie("name"),
+ getkeyordie("ip"))
+ module.exit_json(changed=True, domain=domain.to_json())
+ else:
+ records = domain.records()
+ at_record = None
+ for record in records:
+ if record.name == "@":
+ at_record = record
+
+ if not at_record.data == getkeyordie("ip"):
+ record.update(data=getkeyordie("ip"), record_type='A')
+ module.exit_json(changed=True, domain=Domain.find(id=record.domain_id).to_json())
+
+ module.exit_json(changed=False, domain=domain.to_json())
+
+ elif state in ('absent', 'deleted'):
+ domain = None
+ if "id" in module.params:
+ domain = Domain.find(id=module.params["id"])
+
+ if not domain and "name" in module.params:
+ domain = Domain.find(name=module.params["name"])
+
+ if not domain:
+ module.exit_json(changed=False, msg="Domain not found.")
+
+ event_json = domain.destroy()
+ module.exit_json(changed=True, event=event_json)
+
def main():
module = AnsibleModule(
argument_spec = dict(
- command = dict(choices=['droplet', 'ssh'], default='droplet'),
+ command = dict(choices=['droplet', 'ssh', 'domain'], default='droplet'),
state = dict(choices=['active', 'present', 'absent', 'deleted'], default='present'),
client_id = dict(aliases=['CLIENT_ID'], no_log=True),
api_key = dict(aliases=['API_KEY'], no_log=True),
@@ -379,6 +512,7 @@ def main():
wait = dict(type='bool', choices=BOOLEANS, default='yes'),
wait_timeout = dict(default=300, type='int'),
ssh_pub_key = dict(type='str'),
+ ip = dict(type='str'),
),
required_together = (
['size_id', 'image_id', 'region_id'],