From 5886ded19a83ca96fc0e84c5cd4beb3d894b4fbe Mon Sep 17 00:00:00 2001
From: Nathan Woodburn <github@nathan.woodburn.au>
Date: Tue, 13 Feb 2024 13:42:02 +1100
Subject: [PATCH] feat: Add varo plugin

---
 plugins/varo.py | 135 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 135 insertions(+)
 create mode 100644 plugins/varo.py

diff --git a/plugins/varo.py b/plugins/varo.py
new file mode 100644
index 0000000..1ea52b8
--- /dev/null
+++ b/plugins/varo.py
@@ -0,0 +1,135 @@
+import json
+import account
+import requests
+import os
+import dotenv
+
+# Plugin Data
+info = {
+    "name": "Varo Functions",
+    "description": "Integration with Varo.",
+    "version": "1.0",
+    "author": "Nathan.Woodburn/"
+}
+
+# Functions
+functions = {
+    "status":{
+        "name": "Check connection",
+        "type": "dashboard",
+        "description": "You need tp set varo_instance to the ICANN domain of the chosen Varo instance and varo_api to your varo API key before you can connect",
+        "params": {},
+        "returns": {
+            "status": 
+            {
+                "name": "Status of varo connection",
+                "type": "text"
+            }
+        }
+    },
+    "addDomain":{
+        "name": "Add domain",
+        "type": "default",
+        "description": "Add a domain to Varo",
+        "params": {
+            "domain": {
+                "name":"Domain",
+                "type":"text"
+            }
+        },
+        "returns": {
+            "status": 
+            {
+                "name": "Status of the function",
+                "type": "text"
+            }
+        }
+    
+    }
+}
+
+def status(params, authentication):
+    # Try to connect to Varo
+    dotenv.load_dotenv()
+    api = os.getenv("varo_api")
+    instance = os.getenv("varo_instance")
+
+    if not api or not instance:
+        return {"status": "Missing Varo API or instance"}
+    
+    headers = {"Authorization": f"Bearer {api}"}
+    data = {
+        "action": "getInfo"
+    }
+    response = requests.post(f"https://{instance}/api", json=data, headers=headers)
+    if response.status_code != 200:
+        return {"status": "Error connecting to Varo"}
+    if response.json()["success"] != True:
+        return {"status": "Error connecting to Varo"}
+    return {"status": "Success"}
+
+def addDomain(params, authentication):
+    # Add a domain to Varo
+    domain = params["domain"]
+
+    dotenv.load_dotenv()
+    api = os.getenv("varo_api")
+    instance = os.getenv("varo_instance")
+
+    if not api or not instance:
+        return {"status": "Missing Varo API or instance"}
+    
+    headers = {"Authorization": f"Bearer {api}"}
+    data = {
+        "action": "getZones"
+    }
+    zones = requests.post(f"https://{instance}/api", json=data, headers=headers)
+    if zones.status_code != 200:
+        return {"status": "Error connecting to Varo"}
+    if zones.json()["success"] != True:
+        return {"status": "Error connecting to Varo"}
+    
+    zones = zones.json()["data"]
+    for zone in zones:
+        if zone["name"] == domain:
+            return {"status": "Domain already exists"}
+        
+    # Check domain is owned by user
+    wallet = authentication.split(":")[0]
+    owned = account.getDomains(wallet)
+    # Only keep owned domains ["name"]
+    ownedNames = [domain["name"] for domain in owned]
+    if domain not in ownedNames:
+        return {"status": "Domain not owned by user"}
+    
+    data = {
+        "action": "createZone",
+        "domain": domain
+    }
+    response = requests.post(f"https://{instance}/api", json=data, headers=headers)
+    if response.status_code != 200:
+        return {"status": "Error connecting to Varo"}
+    if response.json()["success"] != True:
+        return {"status": "Error connecting to Varo"}
+    zoneID = response.json()["data"]["zone"]
+    data = {
+        "action": "showZone",
+        "zone": zoneID
+    }
+    response = requests.post(f"https://{instance}/api", json=data, headers=headers)
+    if response.status_code != 200:
+        return {"status": "Error connecting to Varo"}
+    if response.json()["success"] != True:
+        return {"status": "Error connecting to Varo"}
+    zone = response.json()["data"]
+    
+    dns = []
+    for ns in zone['NS']:
+        dns.append({'type': 'NS', 'value': ns})
+    ds = zone['DS']
+    ds = ds.split(' ')
+    dns.append({'type': 'DS', 'keyTag': int(ds[0]), 'algorithm': int(ds[1]), 'digestType': int(ds[2]), 'digest': ds[3]})
+    dns = json.dumps(dns)
+    response = account.setDNS(authentication,domain,dns)
+
+    return {"status": "Success"}
\ No newline at end of file