Initial project commit

app.py

@@ -0,0 +1,105 @@
+from flask import Flask, request
+import mmap
+import re
+
+app = Flask(__name__)
+
+
+@app.route('/')
+def hello_world():
+    return 'Hello World!'
+
+
+@app.route('/CMD_API_LOGIN_TEST')
+def login_test():
+    multi_dict = request.values
+    for key in multi_dict:
+        print(multi_dict.get(key))
+        print(multi_dict.getlist(key))
+    # print(request.values)
+    print(request.headers)
+    print(request.authorization)
+
+    return 'error=0&text=Login OK&details=none'
+
+
+@app.route('/CMD_API_DNS_ADMIN', methods=['GET', 'POST'])
+def domain_admin():
+    print(str(request.data, encoding="utf-8"))
+    print(request.values.get('action'))
+    action = request.values.get('action')
+    if action == 'exists':
+        # DirectAdmin is checking whether the domain is in the cluster
+        return 'result: exists=1'
+    if action == 'delete':
+        # Domain is being removed from the DNS
+        hostname = request.values.get('hostname')
+        username = request.values.get('username')
+        domain = request.values.get('select0')
+
+
+    if action == 'rawsave':
+        # DirectAdmin wants to add/update a domain
+        hostname = request.values.get('hostname')
+        username = request.values.get('username')
+        domain = request.values.get('domain')
+
+        if not check_zone_exists(str(domain)):
+            put_zone_index(str(domain))
+            write_zone_file(str(domain), request.data.decode("utf-8"))
+        else:
+            # Domain already exists
+            write_zone_file(str(domain), request.data.decode("utf-8"))
+
+
+def create_zone_index():
+    # Create an index of all zones present from zone definitions
+    regex = r"(?<=\")(?P<domain>.*)(?=\"\s)"
+
+    with open(zone_index_file, 'w+') as f:
+        with open(named_conf, 'r') as named_file:
+            while True:
+                # read line
+                line = named_file.readline()
+                if not line:
+                    # Reached end of file
+                    break
+                print(line)
+                hosted_domain = re.search(regex, line).group(0)
+                f.write(hosted_domain + "\n")
+
+
+def put_zone_index(zone_name):
+    # add a new zone to index
+    with open(zone_index_file, 'a+') as f:
+        # We are using append mode
+        f.write(zone_name)
+
+
+def write_zone_file(zone_name, data):
+    # Write the zone to file
+    with open(zones_dir + '/' + zone_name + '.db', 'w') as f:
+        f.write(data)
+
+
+def check_zone_exists(zone_name):
+    # Check if zone is present in the index
+    with open(zone_index_file, 'r') as f:
+        try:
+            s = mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_READ)
+            if s.find(bytes(zone_name, encoding='utf8')) != -1:
+                return True
+            else:
+                return False
+        except ValueError as e:
+            # File Empty?
+            return False
+
+
+if __name__ == '__main__':
+    zones_dir = "/etc/pdns/zones"
+    zone_index_file = "/etc/pdns/zones/.index"
+    named_conf = "/etc/pdns/named.conf"
+    create_zone_index()
+
+    app.run(host="0.0.0.0")