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