134 lines
3.2 KiB
Python
134 lines
3.2 KiB
Python
from ipaddress import ip_interface
|
|
from json import dumps
|
|
|
|
|
|
defaults = {
|
|
'apt': {
|
|
'packages': {
|
|
'bind9': {},
|
|
},
|
|
},
|
|
'bind': {
|
|
'zones': {},
|
|
'slaves': {},
|
|
},
|
|
'telegraf': {
|
|
'config': {
|
|
'inputs': {
|
|
'bind': [{
|
|
'urls': ['http://localhost:8053/xml/v3'],
|
|
'gather_memory_contexts': False,
|
|
'gather_views': True,
|
|
}],
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
|
|
@metadata_reactor.provides(
|
|
'bind/type',
|
|
)
|
|
def type(metadata):
|
|
return {
|
|
'bind': {
|
|
'type': 'slave' if metadata.get('bind/master_node', None) else 'master',
|
|
}
|
|
}
|
|
|
|
|
|
@metadata_reactor.provides(
|
|
'dns',
|
|
)
|
|
def dns(metadata):
|
|
return {
|
|
'dns': {
|
|
metadata.get('bind/hostname'): repo.libs.dns.get_a_records(metadata),
|
|
}
|
|
}
|
|
|
|
|
|
@metadata_reactor.provides(
|
|
'bind/zones',
|
|
)
|
|
def collect_records(metadata):
|
|
if metadata.get('bind/type') == 'slave':
|
|
return {}
|
|
|
|
zones = {}
|
|
|
|
for other_node in repo.nodes:
|
|
for fqdn, records in other_node.metadata.get('dns', {}).items():
|
|
matching_zones = sorted(
|
|
filter(
|
|
lambda potential_zone: fqdn.endswith(potential_zone),
|
|
metadata.get('bind/zones').keys()
|
|
),
|
|
key=len,
|
|
)
|
|
if matching_zones:
|
|
zone = matching_zones[-1]
|
|
else:
|
|
continue
|
|
|
|
name = fqdn[0:-len(zone) - 1]
|
|
|
|
for type, values in records.items():
|
|
for value in values:
|
|
entry = {'name': name, 'type': type, 'value': value}
|
|
zones\
|
|
.setdefault(zone, {})\
|
|
.update({
|
|
str(hash(dumps(entry))): entry,
|
|
})
|
|
|
|
return {
|
|
'bind': {
|
|
'zones': zones,
|
|
},
|
|
}
|
|
|
|
|
|
@metadata_reactor.provides(
|
|
'bind/zones',
|
|
)
|
|
def ns_records(metadata):
|
|
if metadata.get('bind/type') == 'slave':
|
|
return {}
|
|
|
|
nameservers = [
|
|
node.metadata.get('bind/hostname'),
|
|
*[
|
|
repo.get_node(slave).metadata.get('bind/hostname')
|
|
for slave in node.metadata.get('bind/slaves')
|
|
]
|
|
]
|
|
return {
|
|
'bind': {
|
|
'zones': {
|
|
zone: {
|
|
# FIXME: bw currently cant handle lists of dicts :(
|
|
str(hash(dumps({'name': '@', 'type': 'NS', 'value': f"{nameserver}."}))): {'name': '@', 'type': 'NS', 'value': f"{nameserver}."}
|
|
for nameserver in nameservers
|
|
} for zone in metadata.get('bind/zones').keys()
|
|
},
|
|
},
|
|
}
|
|
|
|
|
|
@metadata_reactor.provides(
|
|
'bind/slaves',
|
|
)
|
|
def slaves(metadata):
|
|
if metadata.get('bind/type') == 'slave':
|
|
return {}
|
|
|
|
return {
|
|
'bind': {
|
|
'slaves': [
|
|
other_node.name
|
|
for other_node in repo.nodes
|
|
if other_node.has_bundle('bind') and other_node.metadata.get('bind/master_node', None) == node.name
|
|
],
|
|
},
|
|
}
|