rune/ddns
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases 10 Wiki Activity

Inital 0.1 release

Browse Source
This commit is contained in:
rune 2023-03-07 17:49:38 +01:00
parent 81f5ef665c
commit 1cf21ea086
1 changed files with 245 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

303
ddns.py
View File

@ -9,6 +9,9 @@ import argparse
import requests import requests
import os import os
import time import time
from string import ascii_letters, digits
from rich import print
from argparse import RawTextHelpFormatter
homefilepath = Path.home() homefilepath = Path.home()
filepath = homefilepath.joinpath('.config/ddns') filepath = homefilepath.joinpath('.config/ddns')
@ -58,7 +61,9 @@ def connect_database():
c.execute('''CREATE TABLE IF NOT EXISTS subdomains c.execute('''CREATE TABLE IF NOT EXISTS subdomains
(id integer PRIMARY KEY, (id integer PRIMARY KEY,
main_id integer NOT NULL, main_id integer NOT NULL,
name text)''') name text,
current_ip4 text NOT NULL,
current_ip6 text NULL)''')
return conn return conn
@ -89,54 +94,103 @@ def api(api_value):
def add_domian(domain): def add_domian(domain):
apikey = get_api()
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(domain,)) cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(domain,))
count = cursor.fetchone()[0] count = cursor.fetchone()[0]
if count == 0: if count != 0:
cursor.execute('INSERT INTO domains values(?,?)', (None, domain,)) print('[red]Error:[/red] Domain name (%s) already in database!' % (domain))
print('The domain %s has been added to the DB' % (domain))
conn.commit()
else: else:
print('Error: Domain name (%s) already in database!' % (domain)) if apikey != None:
headers = {'Authorization': 'Bearer ' + apikey, "Content-Type": "application/json"}
response = requests.get('https://api.digitalocean.com/v2/domains/' + domain, headers=headers)
response_data = response.json()
if 'id' in response_data:
print('[red]Error: [/red]The domain does not exist in your DigitalOcean account.\nPlease add the domain from your control panel [b]https://cloud.digitalocean.com/networking/domains/[/b]')
else:
cursor.execute('INSERT INTO domains values(?,?)', (None, domain,))
print('The domain [b]%s[/b] has been added to the DB' % (domain))
conn.commit()
def add_subdomain(sub,top): def add_subdomain(domain):
apikey = get_api() if set(domain).difference(ascii_letters + '.' + digits):
if apikey == None: print('[red]Error:[/red] Give the domain name in simple form e.g. [b]test.domain.com[/b]')
print("Missing APIkey. Please add one!")
else: else:
ip = get_ip() parts = domain.split('.')
if ip == None or 'urlopen error' in ip: sub = parts[0]
print('Failed to get public IP. Do you have a typo in your URI? Error %s' % (ip)) top = parts[1] + '.' + parts[2]
apikey = get_api()
if apikey == None:
print("[red]Error:[/red] Missing APIkey. Please add one!")
else: else:
cursor = conn.cursor() ip = get_ip()
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(top,)) if ip == None or 'urlopen error' in ip:
print('[red]Error:[/red] Failed to get public IP. Do you have a typo in your URI? [red]Error %s.[/red]' % (ip))
else:
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(top,))
count = cursor.fetchone()[0]
if count == 0:
print('[red]Error:[/red] Top domain [bold]%s[/bold] does not exist in the DB. Please add it with [i]ddns -t %s[/i].' % (top,top))
else:
cursor.execute('SELECT id FROM domains WHERE name LIKE ?',(top,))
topdomain_id = cursor.fetchone()
topdomain_id = topdomain_id[0]
cursor.execute('SELECT count(*) FROM subdomains WHERE main_id LIKE ? AND name like ?',(topdomain_id,sub,))
count = cursor.fetchone()[0]
if count != 0:
print('[red]Error:[/red] [bold]%s[/bold] already exists.' % (domain))
else:
data = {'name': sub,'data': ip,'type': "A",'ttl': 3600}
headers = {'Authorization': 'Bearer ' + apikey, "Content-Type": "application/json"}
response = requests.post('https://api.digitalocean.com/v2/domains/' + top + '/records',
data=json.dumps(data), headers=headers)
if str(response) == '<Response [201]>':
if response != 'Fail':
response_data = response.json()
domainid = str(response_data['domain_record']['id'])
cursor.execute('INSERT INTO subdomains values(?,?,?,?,?)',(domainid,topdomain_id,sub,ip,None,))
conn.commit()
print('The domain %s has been added.' % (domain))
else:
return '[red]Error: %s [/red]' % (str(response))
def remove_subdomain(domain):
if set(domain).difference(ascii_letters + '.' + digits):
print('[red]Error:[/red] Give the domain name in simple form e.g. [b]test.domain.com[/b]')
else:
parts = domain.split('.')
sub = parts[0]
top = parts[1] + '.' + parts[2]
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(top,))
count = cursor.fetchone()[0]
if count == 0:
print('[red]Error:[/red] Top domain [bold]%s[/bold] does not exist in the DB. So I\'m giving up!.' % (top))
else:
cursor.execute('SELECT COUNT(*) FROM subdomains WHERE name like ? and main_id=(SELECT id from domains WHERE name=?)',(sub,top,))
count = cursor.fetchone()[0] count = cursor.fetchone()[0]
if count == 0: if count == 0:
print('Error: Top domain %s does not exist in the DB. Please add it with ddns -t %s ' % (top,top)) print('[red]Error:[/red] Domain [bold]%s[/bold] does not exist in the DB. So I\'m giving up!.' % (domain))
else: else:
cursor.execute('SELECT id FROM domains WHERE name LIKE ?',(top,)) apikey = get_api()
topdomain_id = cursor.fetchone() if apikey == None:
topdomain_id = topdomain_id[0] print("[red]Error:[/red] Missing APIkey. Please add one!")
cursor.execute('SELECT count(*) FROM subdomains WHERE main_id LIKE ? AND name like ?',(topdomain_id,sub,))
count = cursor.fetchone()[0]
if count != 0:
print('The subdomain %s already exists for the domain %s.' % (sub,top))
else: else:
data = {'name': sub,'data': ip,'type': "A",'ttl': 3600} cursor.execute('SELECT id FROM subdomains WHERE name like ? and main_id=(SELECT id from domains WHERE name=?)',(sub,top,))
subdomain_id = str(cursor.fetchone()[0])
headers = {'Authorization': 'Bearer ' + apikey, "Content-Type": "application/json"} headers = {'Authorization': 'Bearer ' + apikey, "Content-Type": "application/json"}
response = requests.post('https://api.digitalocean.com/v2/domains/' + top + '/records', response = requests.delete('https://api.digitalocean.com/v2/domains/'+top+'/records/' + subdomain_id, headers=headers)
data=json.dumps(data), headers=headers) if str(response) == '<Response [204]>':
if str(response) == '<Response [201]>': cursor.execute('DELETE from subdomains where id=?',(subdomain_id,))
if response != 'Fail': conn.commit()
response_data = response.json()
domainid = str(response_data['domain_record']['id'])
cursor.execute('INSERT INTO subdomains values(?,?,?)',(domainid,topdomain_id,sub,))
conn.commit()
print('The subdomain %s for the domain %s has been added.' % (sub,top))
else: else:
return 'Error: %s ' % (str(response)) print('[red]Error: [/red]An error occurred! Please try again later!')
def show_all_top_domains(): def show_all_top_domains():
@ -155,36 +209,86 @@ def show_all_top_domains():
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(k['name'],)) cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(k['name'],))
count = cursor.fetchone()[0] count = cursor.fetchone()[0]
if count != 0: if count != 0:
print('Name : '+k['name']+ ' *') print('Name : [bold]'+k['name']+ ' [*][/bold]')
else: else:
print('Name : '+k['name']) print('Name : '+k['name'])
else: else:
print("Missing APIkey. Please add one!") print("[red]Error:[/red] Missing APIkey. Please add one!")
def list_sub_domains(domain): def list_sub_domains(domain):
# TODO: Finish this :)
apikey = get_api() apikey = get_api()
cursor = conn.cursor() cursor = conn.cursor()
if apikey == None: if apikey == None:
print("Missing APIkey. Please add one!") print("[red]Error:[/red] Missing APIkey. Please add one!")
else: else:
cursor.execute('SELECT COUNT(*) FROM domains WHERE name LIKE ?',(domain,)) cursor.execute('SELECT COUNT(*) FROM domains WHERE name LIKE ?',(domain,))
count = cursor.fetchone()[0] count = cursor.fetchone()[0]
if count == 0: if count == 0:
print("Error: No such domain. Check spelling or use ddns -d to show all top domains.") print("[red]Error: [/red]No such domain. Check spelling or use ddns -d to show all top domains.")
else: else:
cursor.execute('SELECT * FROM domains WHERE name LIKE ?', (domain,)) print('\n\nCurrent sub domains for [b]%s[/b]' % (domain))
row = cursor.fetchone() print('=====================================================================')
print(row) cursor.execute('SELECT id FROM domains WHERE name LIKE ?', (domain,))
topdomain_id = cursor.fetchone()[0]
cursor.execute('SELECT COUNT(*) FROM subdomains WHERE main_id LIKE ?',(topdomain_id,))
count = cursor.fetchone()[0]
if count == 0:
print('[red]Error:[/red] No sub domains for [b]%s[/b]' % (domain))
else:
cursor.execute('SELECT name FROM subdomains WHERE main_id LIKE ?',(topdomain_id,) )
subdomains = cursor.fetchall()
for i in subdomains:
print(i[0]+'.'+domain)
print('\n')
def list_do_sub_domains(domain):
apikey = get_api()
cursor = conn.cursor()
if apikey == None:
print("[red]Error:[/red] Missing APIkey. Please add one!")
else:
req = urllib.request.Request('https://api.digitalocean.com/v2/domains/'+domain+'/records?type="A"/?per_page=200')
req.add_header('Content-Type', 'application/json')
req.add_header('Authorization', 'Bearer ' + apikey)
current = urllib.request.urlopen(req)
remote = current.read().decode('utf-8')
remoteData = json.loads(remote)
print('Domains in your DigitalOcean account not in ddns DB for [b]%s[/b]' % (domain))
print('============================================================================')
for k in remoteData["domain_records"]:
if k['type'] == 'A':
cursor.execute('SELECT COUNT(*) FROM subdomains WHERE id like ?',(str(k['id']),))
count = cursor.fetchone()[0]
if count == 0:
print(k['name']+'.'+domain+' ID : '+str(k['id']))
print("List -> " + domain)
def domaininfo(domain): def domaininfo(domain):
print("domaininfo") apikey = get_api()
local_ip = get_ip()
cursor = conn.cursor()
if set(domain).difference(ascii_letters + '.'):
print('[red]Error:[/red]. Give the domain name in simple form e.g. [bold]test.domain.com[/bold]')
else:
parts = domain.split('.')
topdomain = parts[1]+'.'+parts[2]
cursor.execute('SELECT id FROM domains WHERE name like ?', (topdomain,))
domainid = cursor.fetchone()[0]
cursor.execute('SELECT * FROM subdomains WHERE main_id like ?', (domainid,))
domains = cursor.fetchall()
if local_ip != domains[0][3]:
localip = '[red]%s[/red]' % (local_ip)
else:
localip = local_ip
print ('The domain [bold]%s[/bold] has the IP [bold]%s[/bold]. Your public IP is [bold]%s[/bold]' % (domain,domains[0][3],localip))
def show_current_info(): def show_current_info():
@ -194,19 +298,27 @@ def show_current_info():
cursor.execute('SELECT COUNT(ip4_server) FROM ipservers') cursor.execute('SELECT COUNT(ip4_server) FROM ipservers')
count = cursor.fetchone()[0] count = cursor.fetchone()[0]
if count == 0: if count == 0:
ipserver = 'No IP resolvers in DB' ipserver = '[red]Error:[/red] No IP resolvers in DB'
else: else:
cursor.execute('SELECT * FROM ipservers') cursor.execute('SELECT * FROM ipservers')
ipserver = cursor.fetchall()[0][1] ipserver = cursor.fetchall()[0][1]
if API == None: if API == None:
API = 'API key not stored in DB' API = '[red]Error:[/red] API key not stored in DB'
cursor.execute('SELECT COUNT(*) FROM domains')
topdomains = cursor.fetchone()[0]
cursor.execute('SELECT COUNT(*) FROM subdomains')
subdomains = cursor.fetchone()[0]
print('Current info.') print('Current info.')
print('==========================================') print('==========================================')
print('API key : %s' % (API)) print('API key : %s' % (API))
print('IP resolver : %s' % (ipserver)) print('IP resolver : %s' % (ipserver))
print('App version : %s' % (app_version)) print('App version : %s (https://gitlab.pm/rune/ddns)' % (app_version))
print('Top domains : %s' % (topdomains))
print('sub domains : %s' % (subdomains))
print('') print('')
print('IPv6 is not supported and not listed here.') print('IPv6 is not supported and not listed here.')
@ -239,16 +351,77 @@ def ip_server(ipserver, ip_type):
def updateip(): def updateip():
print("updateip") apikey = get_api()
current_ip = get_ip()
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM subdomains')
count = cursor.fetchone()[0]
if count == 0:
print('[red]Error: [/red]There are no dynamic domains active.'\
' Start by adding a new domain with [i]ddns -s test.example.com[/i]')
else:
cursor.execute('SELECT id FROM subdomains')
rows = cursor.fetchall()
for i in rows:
cursor.execute('SELECT name FROM domains WHERE id like (SELECT main_id from subdomains WHERE id = ?)',(i[0],))
domain_name = str(cursor.fetchone()[0])
subdomain_id = str(i[0])
data = {'type': 'A', 'data': current_ip}
#print(domain_name, subdomain_id,current_ip,data)
headers = {'Authorization': 'Bearer ' + apikey, "Content-Type": "application/json"}
response = requests.patch('https://api.digitalocean.com/v2/domains/'+domain_name+'/records/' + subdomain_id, data=json.dumps(data), headers=headers)
# response = response.json()
if str(response) != '<Response [200]>':
print('[red]Error: ' + str(response.json))
else:
cursor.execute('UPDATE subdomains SET current_ip4=? WHERE id = ?',(current_ip,subdomain_id,))
conn.commit()
def local_add_subdomain(domain,domainid):
if set(domain).difference(ascii_letters + '.' + digits + '-'):
print('[red]Error:[/red] Give the domain name in simple form e.g. [b]test.domain.com[/b]')
else:
parts = domain.split('.')
sub = parts[0]
top = parts[1] + '.' + parts[2]
apikey = get_api()
if apikey == None:
print("[red]Error:[/red] Missing APIkey. Please add one!")
else:
ip = get_ip()
if ip == None or 'urlopen error' in ip:
print('[red]Error:[/red] Failed to get public IP. Do you have a typo in your URI? [red]Error %s.[/red]' % (ip))
else:
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(top,))
count = cursor.fetchone()[0]
if count == 0:
print('[red]Error:[/red] Top domain [bold]%s[/bold] does not exist in the DB. Please add it with [i]ddns -t %s[/i].' % (top,top))
else:
cursor.execute('SELECT id FROM domains WHERE name LIKE ?',(top,))
topdomain_id = cursor.fetchone()
topdomain_id = topdomain_id[0]
cursor.execute('SELECT count(*) FROM subdomains WHERE main_id LIKE ? AND name like ?',(topdomain_id,sub,))
count = cursor.fetchone()[0]
if count != 0:
print('[red]Error:[/red] [bold]%s[/bold] already exists.' % (domain))
else:
cursor.execute('INSERT INTO subdomains values(?,?,?,?,?)',(domainid,topdomain_id,sub,ip,None,))
conn.commit()
print('The domain %s has been added.' % (domain))
# Commandline arguments # Commandline arguments
conn = connect_database() conn = connect_database()
parser = argparse.ArgumentParser(prog='ddns', parser = argparse.ArgumentParser(prog='ddns',
description='Application to use domains from DigitalOcean account as dynamic DNS domain(s).\r\nThe app only supports IP4. IPv6 is planned for a later release!', description='Application to use domains from DigitalOcean account as dynamic '\
'DNS domain(s).\nThe app only supports IP4. IPv6 is planned for a later release!'\
'\nYou\'ll always find the latest version on https://gitlab.pm/rune/ddns',
formatter_class=RawTextHelpFormatter,
epilog='Making Selfhosting easier...') epilog='Making Selfhosting easier...')
parser.add_argument('-a', '--api', help='Add/Change API key.', parser.add_argument('-a', '--api', help='Add/Change API key.',
@ -257,7 +430,10 @@ parser.add_argument('-a', '--api', help='Add/Change API key.',
parser.add_argument('-l', '--list', help='List subdomains for supplied domain.', parser.add_argument('-l', '--list', help='List subdomains for supplied domain.',
nargs=1, metavar=('domain'), required=False, action="append") nargs=1, metavar=('domain'), required=False, action="append")
parser.add_argument('-d', '--domains', help='List top domains on your DigitalOcean account.', parser.add_argument('-o', '--serverdomains', help='List subdomains for supplied domain not in ddns DB.',
nargs=1, metavar=('domain'), required=False, action="append")
parser.add_argument('-d', '--domains', help='List top domains in your DigitalOcean account.',
required=False, action="store_true") required=False, action="store_true")
parser.add_argument('-c', '--current', help='List the current IP address for the sub-domain given', parser.add_argument('-c', '--current', help='List the current IP address for the sub-domain given',
@ -266,8 +442,14 @@ parser.add_argument('-c', '--current', help='List the current IP address for the
parser.add_argument('-t', '--top', help='Add a new domain from your DigitalOcean account to use as a dynamic DNS domain', parser.add_argument('-t', '--top', help='Add a new domain from your DigitalOcean account to use as a dynamic DNS domain',
required=False, nargs=1, metavar=('domain'), action='append') required=False, nargs=1, metavar=('domain'), action='append')
parser.add_argument('-s', '--sub', help='Add a new subdomain to dynamic DNS DO domain', parser.add_argument('-s', '--sub', help='Add a new subdomain to your DigitalOcean account and use as dynamic DNS.\n',
required=False, nargs=2, metavar=('subdomain', 'domain'), action='append') required=False, nargs=1, metavar=('domain'), action='append')
parser.add_argument('-k', '--local', help='Add an existing DigitalOcean domain to your ddns DB and use as dynamic DNS. Get ',
required=False, nargs=2, metavar=('domain','domainid'), action='append')
parser.add_argument('-r', '--remove', help='Remove a subdomain from your DigitalOcean account and ddns.',
required=False, nargs=1, metavar=('domain'), action='append')
parser.add_argument('-i', '--info', help='Show current config info', parser.add_argument('-i', '--info', help='Show current config info',
required=False, action='store_true') required=False, action='store_true')
@ -280,23 +462,28 @@ if args['list']:
list_sub_domains(args['list'][0][0]) list_sub_domains(args['list'][0][0])
elif args['domains']: elif args['domains']:
show_all_top_domains() show_all_top_domains()
elif args['serverdomains']:
list_do_sub_domains(args['serverdomains'][0][0])
elif args['current']: elif args['current']:
domaininfo(args['current'][0][0]) domaininfo(args['current'][0][0])
elif args['top']: elif args['top']:
add_domian(args['top'][0][0]) add_domian(args['top'][0][0])
elif args['sub']: elif args['sub']:
add_subdomain(args['sub'][0][0],args['sub'][0][1]) add_subdomain(args['sub'][0][0])
#add_subdomain(args['sub'][0][0])
elif args['info']: elif args['info']:
show_current_info() show_current_info()
elif args['ipserver']: elif args['ipserver']:
ip_server(args['ipserver'][0][0],args['ipserver'][0][1]) ip_server(args['ipserver'][0][0],args['ipserver'][0][1])
elif args['api']: elif args['api']:
api(args['api'][0][0]) api(args['api'][0][0])
elif args['remove']:
remove_subdomain(args['remove'][0][0])
elif args['local']:
local_add_subdomain(args['local'][0][0],args['local'][0][1])
else: else:
get_ip() # get_ip()
# get_api() # get_api()
updateip()