f5db2a8cfd9df7990823003ec65a1a3017362a6c
[zonemaker.git] / zonemaker / zone.py
1 from typing import List, Dict, Any
2 import ipaddress, re
3
4 second = 1
5 minute = 60*second
6 hour = 60*minute
7 day = 24*hour
8
9 def hostname(name: str) -> str:
10     # check hostname for validity
11     label = r'[a-zA-Z90-9]([a-zA-Z90-9-]{0,61}[a-zA-Z90-9])?' # must not start or end with hyphen
12     pattern = r'^{0}(\.{0})*\.?'.format(label)
13     print(pattern)
14     if re.match(pattern, name):
15         return name
16     raise Exception(name+" is not a valid hostname")
17
18 class Address:
19     def __init__(self, IPv4 = None, IPv6 = None):
20         self._IPv4 = None if IPv4 is None else ipaddress.IPv4Address(IPv4)
21         self._IPv6 = None if IPv6 is None else ipaddress.IPv6Address(IPv6)
22     
23     def IPv4(self):
24         return Address(IPv4 = self._IPv4)
25     
26     def IPv6(self):
27         return Address(IPv6 = self._IPv6)
28
29 class Name:
30     def __init__(self, address = None, MX = None, TCP = None, UDP = None):
31         self._address = address
32
33 class Service:
34     def __init__(self, SRV = None, TLSA=None):
35         self._SRV = SRV
36         self._TLSA = TLSA
37
38 class CName:
39     def __init__(self, name):
40         self._name = name
41
42 class Delegation():
43     def __init__(self, NS, DS = None):
44         pass
45
46 class Zone:
47     def __init__(self, name: str, mail: str, NS: List[str],
48                  secondary_refresh: int, secondary_retry: int, secondary_discard: int,
49                  NX_TTL: int = None, A_TTL: int = None, other_TTL: int = None,
50                  domains: Dict[str, Any] = {}) -> None:
51         self._name = hostname(name)
52         if not mail.endswith('.'): raise Exception("Mail must be absolute, end with a dot")
53         atpos = mail.find('@')
54         if atpos < 0 or atpos > mail.find('.'): raise Exception("Mail must contain an @ before the first dot")
55         self._mail = hostname(mail.replace('@', '.', 1))
56         self._NS = list(map(hostname, NS))
57         
58         self._secondary_refresh = secondary_refresh
59         self._secondary_retry = secondary_retry
60         self._secondary_discard = secondary_discard
61         
62         assert other_TTL is not None
63         self._NX_TTL = other_TTL if NX_TTL is None else NX_TTL
64         self._A_TTL = other_TTL if A_TTL is None else A_TTL
65         self._other_TTL = other_TTL
66     
67     def write(self, file):
68         raise NotImplementedError()