Initial commit

1 files changed, 41 insertions, 0 deletions

diff --git a/simpleddns.py b/simpleddns.py
new file mode 100755
index 0000000..19196ce
--- /dev/null
+++ b/simpleddns.py
@@ -0,0 +1,41 @@
+#!/usr/bin/env python3
+import os
+import sys
+import requests
+import stat
+import subprocess
+from pathlib import Path
+
+simpleddns_config_dir = os.getenv('SIMPLEDDNS_CONFIG_DIR')
+xdg_config_dir = os.getenv('XDG_CONFIG_DIR')
+if simpleddns_config_dir:
+	config_dir = Path(simpleddns_config_dir)
+elif xdg_config_dir:
+	config_dir = Path(xdg_config_dir, 'simpleddns')
+else:
+	config_dir = Path('~/.config/simpleddns').expanduser()
+os.makedirs(config_dir, exist_ok = True)
+
+def fatal_error(message):
+	sys.stderr.write(f'Fatal error: {message}\n')
+	exit(1)
+
+get_ip_path = Path(config_dir, 'getip')
+
+if not get_ip_path.exists():
+	print(f'Warning: {get_ip_path} does not exist. Putting default of "curl ifconfig.co" there')
+	with open(get_ip_path, 'w') as f:
+		f.write('curl --no-progress-meter ifconfig.co')
+
+def get_ip():
+	if os.name == 'posix' and (get_ip_path.stat().st_mode & 0o100):
+		command = str(get_ip_path)
+	else:
+		command = open(get_ip_path).read()
+	result = subprocess.run(command, shell=True, stdout=subprocess.PIPE)
+	if result.returncode:
+		print(f'WARNING: {command} failed (exit code {result.returncode})')
+		return None
+	return result.stdout.decode(errors='replace').strip()
+
+print(get_ip())