From 367c391a61091817cf34475f91a23ca286eec334 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Isma=C3=ABl=20Bouya?= <ismael.bouya@normalesup.org>
Date: Sat, 18 Aug 2018 13:09:29 +0200
Subject: Add hetzner server scripts

---
 python/get_initial_configuration_hetzner_server.py | 21 ++++++++
 python/hetzner_helper.py                           | 59 ++++++++++++++++++++++
 python/list_servers.py                             |  6 +++
 python/reboot_hetzner_server.py                    | 18 +++++++
 4 files changed, 104 insertions(+)
 create mode 100644 python/get_initial_configuration_hetzner_server.py
 create mode 100644 python/hetzner_helper.py
 create mode 100644 python/reboot_hetzner_server.py

(limited to 'python')

diff --git a/python/get_initial_configuration_hetzner_server.py b/python/get_initial_configuration_hetzner_server.py
new file mode 100644
index 0000000..71583ff
--- /dev/null
+++ b/python/get_initial_configuration_hetzner_server.py
@@ -0,0 +1,21 @@
+import sys
+import json
+import hetzner_helper
+
+instance = sys.argv[-1]
+instance = hetzner_helper.get("servers/{}".format(instance))[1]["server"]
+
+infos = {}
+infos["ips"] = {
+        "v4": {
+            "ipAddress": instance["public_net"]["ipv4"]["ip"],
+            "gateway":   "172.31.1.1",
+            },
+        "v6": {
+            "ipAddress": instance["public_net"]["ipv6"]["ip"].split("/")[0],
+            "gateway":   "fe80::1",
+            "mask":      instance["public_net"]["ipv6"]["ip"].split("/")[1],
+            }
+        }
+
+print(json.dumps(infos))
diff --git a/python/hetzner_helper.py b/python/hetzner_helper.py
new file mode 100644
index 0000000..496e14d
--- /dev/null
+++ b/python/hetzner_helper.py
@@ -0,0 +1,59 @@
+import json
+import requests
+import os
+
+from configparser import RawConfigParser, NoSectionError, NoOptionError
+
+class AuthenticationException(Exception):
+    pass
+
+class RateLimitExceeded(Exception):
+    pass
+
+class InternalServer(Exception):
+    pass
+
+class HetznerConfig:
+    def __init__(self):
+        config = RawConfigParser()
+        config.read([os.path.expanduser('~/.hetzner.conf')])
+
+        self.api_key = config.get("default", "api_key")
+
+config = HetznerConfig()
+
+def call(endpoint, url_params=None, body=None, method="GET"):
+    api = "https://api.hetzner.cloud/v1/{}".format(endpoint)
+    headers = {"Authorization": "Bearer {}".format(config.api_key)}
+    data = json.dumps(body) if body is not None else None
+
+    if method == "GET":
+        request = requests.get(api, headers=headers, params=url_params)
+    elif method == "POST" or (method == "GET" and body is not None):
+        request = requests.post(api, data=data, headers=headers, params=url_params)
+    elif method == "DELETE":
+        request = requests.delete(api, headers=headers)
+    elif method == "PUT":
+        request = requests.put(api, headers=headers, data=data)
+
+    if request.status_code == 401 or request.status_code == 403:
+        raise AuthenticationException()
+
+    if request.status_code == 429:
+        raise RateLimitExceeded()
+
+    if request.status_code == 500:
+        raise InternalServer(request.text)
+
+    if not request.text:
+        return request.status_code, ""
+
+    js = request.json()
+
+    return request.status_code, js
+
+def get(*args, **kwargs):
+    return call(*args, method="GET", **kwargs)
+
+def post(*args, **kwargs):
+    return call(*args, method="POST", **kwargs)
diff --git a/python/list_servers.py b/python/list_servers.py
index e7bd2af..6d22a56 100644
--- a/python/list_servers.py
+++ b/python/list_servers.py
@@ -21,3 +21,9 @@ print("OVH VPS SSD servers:")
 for vps in vps_list:
     print("\t{}".format(vps))
 
+import hetzner_helper
+
+print("Hetzner VPS servers:")
+return_code, json = hetzner_helper.call("servers")
+for server in json["servers"]:
+    print("\t{}: {}".format(server["name"], server["id"]))
diff --git a/python/reboot_hetzner_server.py b/python/reboot_hetzner_server.py
new file mode 100644
index 0000000..7452afe
--- /dev/null
+++ b/python/reboot_hetzner_server.py
@@ -0,0 +1,18 @@
+import sys
+import hetzner_helper
+
+instance = sys.argv[-1]
+actions = []
+if "--rescue" in sys.argv:
+    actions.append("enable_rescue")
+elif "--local" in sys.argv:
+    actions.append("disable_rescue")
+
+if "--hard" in sys.argv:
+    actions.append("reset")
+else:
+    actions.append("reboot")
+
+for action in actions:
+    result = hetzner_helper.post("servers/{}/actions/{}".format(instance, action))
+    print(result)
-- 
cgit v1.2.3