h3artbl33d/chatmail
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cmdeploy run: don't run if crucial DNS entries are missing

Browse Source
This commit is contained in:
missytake 2023-12-13 00:07:41 +01:00
parent 49848ec01e
commit d0f5d08443
2 changed files with 18 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
cmdeploy/src/cmdeploy/cmdeploy.py
View File

@ -15,7 +15,7 @@ from pathlib import Path
from termcolor import colored
from chatmaild.config import read_config, write_initial_config
import cmdeploy.dns
from cmdeploy.dns import resolve
#
@ -38,8 +38,8 @@ def init_cmd(args, out):
else:
write_initial_config(args.inipath, args.chatmail_domain)
out.green(f"created config file for {args.chatmail_domain} in {args.inipath}")
ipaddress, _ = cmdeploy.dns.resolve(args.chatmail_domain)
mta_ipadress, _ = cmdeploy.dns.resolve("mta-sts." + args.chatmail_domain)
ipaddress = resolve(args.chatmail_domain)
mta_ipadress = resolve("mta-sts." + args.chatmail_domain)
entries = 0
to_print = ["Now you should add %dnsentry% at your DNS provider:\n"]
if not ipaddress:
@ -47,7 +47,9 @@ def init_cmd(args, out):
to_print.append(f"\tA\t{args.chatmail_domain}\t\t<your server's IPv4 address>")
if not mta_ipadress:
entries += 1
to_print.append(f"\tCNAME\tmta-sts.{args.chatmail_domain}\t{args.chatmail_domain}.")
to_print.append(
f"\tCNAME\tmta-sts.{args.chatmail_domain}\t{args.chatmail_domain}."
)
if entries == 1:
singular = "this entry"
elif entries == 2:
@ -57,6 +59,7 @@ def init_cmd(args, out):
to_print[0] = to_print[0].replace("%dnsentry%", singular)
for line in to_print:
print(line)
print()
def run_cmd_options(parser):
@ -76,6 +79,12 @@ def run_cmd(args, out):
deploy_path = importlib.resources.files(__package__).joinpath("deploy.py").resolve()
pyinf = "pyinfra --dry" if args.dry_run else "pyinfra"
cmd = f"{pyinf} --ssh-user root {args.config.mail_domain} {deploy_path}"
mail_domain = args.config.mail_domain
if not resolve(mail_domain) or not resolve(f"mta-sts.{mail_domain}"):
out.red("DNS entries missing. Show instructions with:\n")
print(f"\tcmdeploy init {mail_domain}\n")
sys.exit(1)
out.check_call(cmd, env=env)

10
cmdeploy/src/cmdeploy/dns.py
View File

@ -7,15 +7,15 @@ dns_types = {
}
def resolve(domain: str) -> (str, str):
result, typ = get("A", domain), "A"
def resolve(domain: str) -> str:
result = get("A", domain)
if not result:
result = get("CNAME", domain)
if result:
result, typ = get("A", result[:-1]), "A"
result = get("A", result[:-1])
if not result:
result, typ = get("AAAA", domain), "AAAA"
return result, typ
result = get("AAAA", domain)
return result
def get(typ: str, domain: str) -> str: