external/mailinabox
1
0
Fork 0
mirror of https://github.com/mail-in-a-box/mailinabox.git synced 2024-11-22 02:17:26 +00:00
Code Issues Releases Wiki Activity

let the user override some DNS records in a different way

Browse Source 
Moved the configuration to a single YAML file, rather than one per domain, to be clearer.

re-does 33f06f29c1
This commit is contained in:
Joshua Tauberer 2014-06-22 19:33:30 +00:00
parent 45e93f7dcc
commit 5aa09c3f9b
1 changed files with 22 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
management/dns_update.py
View File

@ -52,13 +52,19 @@ def do_dns_update(env):
domains = get_dns_domains(env) domains = get_dns_domains(env)
zonefiles = get_dns_zones(env) zonefiles = get_dns_zones(env)
# Custom records to add to zones.
try:
additional_records = rtyaml.load(open(os.path.join(env['STORAGE_ROOT'], 'dns/custom.yaml')))
except:
additional_records = { }
# Write zone files. # Write zone files.
os.makedirs('/etc/nsd/zones', exist_ok=True) os.makedirs('/etc/nsd/zones', exist_ok=True)
updated_domains = [] updated_domains = []
for i, (domain, zonefile) in enumerate(zonefiles): for i, (domain, zonefile) in enumerate(zonefiles):
# Build the records to put in the zone. # Build the records to put in the zone.
subdomains = [d for d in domains if d.endswith("." + domain)] subdomains = [d for d in domains if d.endswith("." + domain)]
records = build_zone(domain, zonefile, subdomains, env) records = build_zone(domain, subdomains, additional_records, env)
# See if the zone has changed, and if so update the serial number # See if the zone has changed, and if so update the serial number
# and write the zone file. # and write the zone file.
@ -119,7 +125,7 @@ def do_dns_update(env):
######################################################################## ########################################################################
def build_zone(domain, zonefile, subdomains, env, with_ns=True): def build_zone(domain, subdomains, additional_records, env, with_ns=True):
records = [] records = []
# For top-level zones, define ourselves as the authoritative name server. # For top-level zones, define ourselves as the authoritative name server.
@ -134,7 +140,7 @@ def build_zone(domain, zonefile, subdomains, env, with_ns=True):
# in the zone. # in the zone.
for subdomain in subdomains: for subdomain in subdomains:
subdomain_qname = subdomain[0:-len("." + domain)] subdomain_qname = subdomain[0:-len("." + domain)]
for child_qname, child_rtype, child_value in build_zone(subdomain, None, [], env, with_ns=False): for child_qname, child_rtype, child_value in build_zone(subdomain, [], {}, env, with_ns=False):
if child_qname == None: if child_qname == None:
child_qname = subdomain_qname child_qname = subdomain_qname
else: else:
@ -157,17 +163,19 @@ def build_zone(domain, zonefile, subdomains, env, with_ns=True):
return False return False
# The user may set other records that don't conflict with our settings. # The user may set other records that don't conflict with our settings.
custom_zone_file = os.path.join(env['STORAGE_ROOT'], 'dns/custom', zonefile.replace(".txt", ".yaml")) if zonefile else None for qname, value in additional_records.items():
if zonefile and os.path.exists(custom_zone_file): if qname != domain and not qname.endswith("." + domain): continue
custom_zone = rtyaml.load(open(custom_zone_file)) if qname == domain:
for qname, value in custom_zone.items(): qname = None
if has_rec(qname, value): continue else:
if isinstance(value, str): qname = qname[0:len(qname)-len("." + domain)]
records.append((qname, "A", value)) if has_rec(qname, value): continue
elif isinstance(value, dict): if isinstance(value, str):
for rtype, value2 in value.items(): records.append((qname, "A", value))
if rtype == "TXT": value2 = "\"" + value2 + "\"" elif isinstance(value, dict):
records.append((qname, rtype, value2)) for rtype, value2 in value.items():
if rtype == "TXT": value2 = "\"" + value2 + "\""
records.append((qname, rtype, value2))
# Add defaults if not overridden by the user's custom settings. # Add defaults if not overridden by the user's custom settings.
if not has_rec(None, "A"): records.append((None, "A", env["PUBLIC_IP"])) if not has_rec(None, "A"): records.append((None, "A", env["PUBLIC_IP"]))