Merge pull request #3 from mpdevilleres/replace-yaml-to-toml

replace yaml to toml

README.md

@@ -24,7 +24,7 @@ This can be useful for:
 
 ## Installation
 
-1) Install Python 3.10 or newer.
+1) Install Python 3.11 or newer.
 2) Install PyYAML:
 ```
 pip3 install pyyaml
@@ -33,7 +33,7 @@ pip3 install pyyaml
 
 ## Running PolarDNS
 
-Make sure you are using Python 3.10 or newer to run the PolarDNS server:
+Make sure you are using Python 3.11 or newer to run the PolarDNS server:
 ```
 python polardns.py
 ```

polardns.py

@@ -8,16 +8,33 @@
 import time
 import sys
 import os
-import yaml
+import tomllib
 
 version = "1.0"
 stamp = str(time.time()).ljust(18, "0")
 print("%s | PolarDNS %s server starting up" % (stamp, version))
 
 ################################
 
-config = yaml.safe_load(open("polardns.yml"))
+### PARSE CONFIGURATION
+# note: this has some transformation as not to break the existing usage of config variable
+# please update the usage to simplify this transformation in future iteration`
+with open("polardns.toml", "rb") as f:
+    _config = tomllib.load(f)
+config = {k:v for k,v in _config['main'].items() if k != 'known_servers'}
+
+known_servers = {}
+for line in _config['main']['known_servers'].split('\n'):
+    if not line:
+        continue
+    host, ip_address = line.split()
+    known_servers[host] = ip_address
+
+config['known_servers'] = known_servers
 debug = config['debug']
+
+### END PARSE CONFIGURATION
+
 globalttl = int(config['ttl'])
 globalsleep = float(config['sleep'])
 

polardns.toml

@@ -0,0 +1,27 @@
+[main]
+domain = 'yourdomain.com'
+ns1 = '127.0.0.1'
+ns2 = '127.0.0.1'
+
+# server address
+listen_addr = '0.0.0.0:53'
+
+# debug (0 or 1)
+debug = 0
+
+# default TTL
+ttl = 60
+
+# default latency (in ms)
+sleep = 0
+
+# a 3rdparty domain which we don't own
+a3rdparty_domain = 'whatever.com'
+
+known_servers = """
+127.0.0.1   localhost
+1.2.3.4     powerdns1
+1.2.3.5     powerdns2
+2.3.4.5     bind-test1
+4.5.6.7     msdns-test1
+"""