#!/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())