From 78e29cd3fa25fe2a7c0d5c9fc87dfa8208dff9c9 Mon Sep 17 00:00:00 2001
From: Keiran S <keiran@gmail.com>
Date: Sun, 14 Oct 2018 06:03:30 +1100
Subject: [PATCH] Add AWS Route53 Dynamic DNS  support (#17072)

* Add AWS Route53 dynamic DNS support to Home Assistant

* Remove line breaks
---
 .coveragerc                         |   1 +
 homeassistant/components/route53.py | 113 ++++++++++++++++++++++++++++
 requirements_all.txt                |   4 +
 3 files changed, 118 insertions(+)
 create mode 100644 homeassistant/components/route53.py

diff --git a/.coveragerc b/.coveragerc
index cf56182fd6d..df73e2066b7 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -645,6 +645,7 @@ omit =
     homeassistant/components/remember_the_milk/__init__.py
     homeassistant/components/remote/harmony.py
     homeassistant/components/remote/itach.py
+    homeassistant/components/route53.py
     homeassistant/components/scene/hunterdouglas_powerview.py
     homeassistant/components/scene/lifx_cloud.py
     homeassistant/components/sensor/airvisual.py
diff --git a/homeassistant/components/route53.py b/homeassistant/components/route53.py
new file mode 100644
index 00000000000..f88a15b72b8
--- /dev/null
+++ b/homeassistant/components/route53.py
@@ -0,0 +1,113 @@
+"""
+Update the IP addresses of your Route53 DNS records.
+
+For more details about this component, please refer to the documentation at
+https://home-assistant.io/components/route53/
+"""
+from datetime import timedelta
+import logging
+
+import voluptuous as vol
+
+from homeassistant.const import CONF_DOMAIN, CONF_ZONE
+import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers.event import track_time_interval
+
+REQUIREMENTS = ['boto3==1.9.16', 'ipify==1.0.0']
+
+_LOGGER = logging.getLogger(__name__)
+
+CONF_ACCESS_KEY_ID = 'aws_access_key_id'
+CONF_SECRET_ACCESS_KEY = 'aws_secret_access_key'
+CONF_RECORDS = 'records'
+
+DOMAIN = 'route53'
+
+INTERVAL = timedelta(minutes=60)
+
+CONFIG_SCHEMA = vol.Schema({
+    DOMAIN: vol.Schema({
+        vol.Required(CONF_ACCESS_KEY_ID): cv.string,
+        vol.Required(CONF_DOMAIN): cv.string,
+        vol.Required(CONF_RECORDS): vol.All(cv.ensure_list, [cv.string]),
+        vol.Required(CONF_SECRET_ACCESS_KEY): cv.string,
+        vol.Required(CONF_ZONE): cv.string,
+    })
+}, extra=vol.ALLOW_EXTRA)
+
+
+def setup(hass, config):
+    """Set up the Route53 component."""
+    domain = config[DOMAIN][CONF_DOMAIN]
+    records = config[DOMAIN][CONF_RECORDS]
+    zone = config[DOMAIN][CONF_ZONE]
+    aws_access_key_id = config[DOMAIN][CONF_ACCESS_KEY_ID]
+    aws_secret_access_key = config[DOMAIN][CONF_SECRET_ACCESS_KEY]
+
+    def update_records_interval(now):
+        """Set up recurring update."""
+        _update_route53(
+            aws_access_key_id, aws_secret_access_key, zone, domain, records)
+
+    def update_records_service(now):
+        """Set up service for manual trigger."""
+        _update_route53(
+            aws_access_key_id, aws_secret_access_key, zone, domain, records)
+
+    track_time_interval(hass, update_records_interval, INTERVAL)
+
+    hass.services.register(DOMAIN, 'update_records', update_records_service)
+    return True
+
+
+def _update_route53(
+        aws_access_key_id, aws_secret_access_key, zone, domain, records):
+    import boto3
+    from ipify import get_ip
+    from ipify import exceptions
+
+    _LOGGER.debug("Starting update for zone %s", zone)
+
+    client = boto3.client(
+        DOMAIN,
+        aws_access_key_id=aws_access_key_id,
+        aws_secret_access_key=aws_secret_access_key,
+    )
+
+    # Get the IP Address and build an array of changes
+    try:
+        ipaddress = get_ip()
+
+    except exceptions.ConnectionError:
+        _LOGGER.warning("Unable to reach the ipify service")
+        return
+
+    except exceptions.ServiceError:
+        _LOGGER.warning("Unable to complete the ipfy request")
+        return
+
+    changes = []
+    for record in records:
+        _LOGGER.debug("Processing record: %s", record)
+
+        changes.append({
+            'Action': 'UPSERT',
+            'ResourceRecordSet': {
+                'Name': '{}.{}'.format(record, domain),
+                'Type': 'A',
+                'TTL': 300,
+                'ResourceRecords': [
+                    {'Value': ipaddress},
+                ],
+            }
+        })
+
+    _LOGGER.debug("Submitting the following changes to Route53")
+    _LOGGER.debug(changes)
+
+    response = client.change_resource_record_sets(
+        HostedZoneId=zone, ChangeBatch={'Changes': changes})
+    _LOGGER.debug("Response is %s", response)
+
+    if response['ResponseMetadata']['HTTPStatusCode'] != 200:
+        _LOGGER.warning(response)
diff --git a/requirements_all.txt b/requirements_all.txt
index ab7c2104874..5793dc4946b 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -199,6 +199,7 @@ blockchain==1.4.4
 # homeassistant.components.sensor.bme680
 # bme680==1.0.4
 
+# homeassistant.components.route53
 # homeassistant.components.notify.aws_lambda
 # homeassistant.components.notify.aws_sns
 # homeassistant.components.notify.aws_sqs
@@ -503,6 +504,9 @@ insteonplm==0.15.0
 # homeassistant.components.sensor.iperf3
 iperf3==0.1.10
 
+# homeassistant.components.route53
+ipify==1.0.0
+
 # homeassistant.components.verisure
 jsonpath==0.75