93 lines
4.7 KiB
Python
93 lines
4.7 KiB
Python
import argparse
|
|
from .azure import get_dns_nameservers
|
|
from . import miab
|
|
|
|
def azure_list_command(args):
|
|
nameservers = get_dns_nameservers(args.resource_group, args.zone_name)
|
|
print(f"Nameservers for zone {args.zone_name}:")
|
|
for ns in nameservers:
|
|
print(f" - {ns}")
|
|
|
|
# Mail-in-a-Box API wrapper functions
|
|
def miab_set_command(args):
|
|
return miab.set_custom_dns(args.domain_name, args.record_type, args.record_value)
|
|
|
|
def miab_delete_command(args):
|
|
return miab.delete_custom_dns(args.domain_name, args.record_type)
|
|
|
|
def miab_add_command(args):
|
|
return miab.add_custom_dns(args.domain_name, args.record_type, args.record_value)
|
|
|
|
def miab_list_command(args):
|
|
records = miab.list_custom_dns(args.record_type)
|
|
if args.record_type:
|
|
print(f"Custom {args.record_type} records:")
|
|
else:
|
|
print("Custom DNS records:")
|
|
|
|
for record in records:
|
|
print(f" - {record['name']} ({record['type']}): {record['value']}")
|
|
|
|
def miab_configure_command(args):
|
|
# Get a list of nameservers for the specified Azure DNS zone
|
|
nameservers = get_dns_nameservers(args.resource_group, args.zone_name)
|
|
|
|
if not nameservers:
|
|
print("No nameservers found. Aborting configuration.")
|
|
return
|
|
|
|
# Add each nameserver as an NS record in the Mail-in-a-Box DNS server
|
|
print(f"Configuring Mail-in-a-Box with nameservers from Azure DNS zone {args.zone_name}...")
|
|
for ns in nameservers:
|
|
print(f"Adding NS record for {ns}...")
|
|
success = miab.add_custom_dns(args.zone_name, "NS", ns)
|
|
if success:
|
|
print(f"Successfully added NS record for {ns}")
|
|
else:
|
|
print(f"Failed to add NS record for {ns}")
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="DNS Configuration Tool")
|
|
subparsers = parser.add_subparsers()
|
|
|
|
# list command - lists nameservers for a given Azure DNS zone
|
|
list_azure_parser = subparsers.add_parser("list-azure", help="List nameservers for an Azure DNS zone")
|
|
list_azure_parser.add_argument("--resource-group", required=True, help="Resource group name")
|
|
list_azure_parser.add_argument("--zone-name", required=True, help="DNS zone name")
|
|
list_azure_parser.set_defaults(func=azure_list_command)
|
|
|
|
# list-miab command - lists custom DNS records in the Mail-in-a-Box DNS server
|
|
list_miab_parser = subparsers.add_parser("list-miab", help="List custom DNS records in the Mail-in-a-Box DNS server")
|
|
list_miab_parser.add_argument("--record-type", help="Filter by DNS record type (e.g., A, CNAME, MX)")
|
|
list_miab_parser.set_defaults(func=miab_list_command)
|
|
|
|
# set command - sets a record in the Mail-in-a-Box DNS server
|
|
set_parser = subparsers.add_parser("set", help="Set a DNS record in the Mail-in-a-Box DNS server")
|
|
set_parser.add_argument("--domain-name", required=True, help="Domain name to set the record for")
|
|
set_parser.add_argument("--record-type", required=True, help="DNS record type (e.g., A, CNAME, MX)")
|
|
set_parser.add_argument("--record-value", required=True, help="Value for the DNS record")
|
|
set_parser.set_defaults(func=miab_set_command)
|
|
|
|
# delete command - deletes a record from the Mail-in-a-Box DNS server
|
|
delete_parser = subparsers.add_parser("delete", help="Delete a DNS record from the Mail-in-a-Box DNS server")
|
|
delete_parser.add_argument("--domain-name", required=True, help="Domain name to delete the record for")
|
|
delete_parser.add_argument("--record-type", required=True, help="DNS record type (e.g., A, CNAME, MX)")
|
|
delete_parser.add_argument("--record-value", help="Value for the DNS record")
|
|
delete_parser.set_defaults(func=miab_delete_command)
|
|
|
|
# add command - adds a record to the Mail-in-a-Box DNS server
|
|
add_parser = subparsers.add_parser("add", help="Add a DNS record to the Mail-in-a-Box DNS server")
|
|
add_parser.add_argument("--domain-name", required=True, help="Domain name to add the record for")
|
|
add_parser.add_argument("--record-type", required=True, help="DNS record type (e.g., A, CNAME, MX)")
|
|
add_parser.add_argument("--record-value", required=True, help="Value for the DNS record")
|
|
add_parser.set_defaults(func=miab_add_command)
|
|
|
|
# configure-miab - retrieves nameservers for an Azure DNS zone and configures them in the Mail-in-a-Box DNS server
|
|
configure_miab_parser = subparsers.add_parser("configure-miab", help="Configure Mail-in-a-Box DNS server with nameservers from an Azure DNS zone")
|
|
configure_miab_parser.add_argument("--resource-group", required=True, help="Azure Resource group name containing the DNS zone")
|
|
configure_miab_parser.add_argument("--zone-name", required=True, help="DNS zone name")
|
|
configure_miab_parser.set_defaults(func=miab_configure_command)
|
|
|
|
args = parser.parse_args()
|
|
args.func(args)
|