mirror of
https://github.com/github/octodns.git
synced 2024-05-11 05:55:00 +00:00
100 lines
3.5 KiB
Python
100 lines
3.5 KiB
Python
#
|
|
#
|
|
#
|
|
|
|
from __future__ import absolute_import, division, print_function, \
|
|
unicode_literals
|
|
|
|
from collections import defaultdict
|
|
|
|
from ..provider.plan import Plan
|
|
from ..record import Record
|
|
|
|
from .base import BaseProcessor
|
|
|
|
|
|
# Mark anything octoDNS is managing that way it can know it's safe to modify or
|
|
# delete. We'll take ownership of existing records that we're told to manage
|
|
# and thus "own" them going forward.
|
|
class OwnershipProcessor(BaseProcessor):
|
|
|
|
def __init__(self, name, txt_name='_owner', txt_value='*octodns*'):
|
|
super(OwnershipProcessor, self).__init__(name)
|
|
self.txt_name = txt_name
|
|
self.txt_value = txt_value
|
|
self._txt_values = [txt_value]
|
|
|
|
def process_source_zone(self, desired, *args, **kwargs):
|
|
for record in desired.records:
|
|
# Then create and add an ownership TXT for each of them
|
|
record_name = record.name.replace('*', '_wildcard')
|
|
if record.name:
|
|
name = f'{self.txt_name}.{record._type}.{record_name}'
|
|
else:
|
|
name = f'{self.txt_name}.{record._type}'
|
|
txt = Record.new(desired, name, {
|
|
'type': 'TXT',
|
|
'ttl': 60,
|
|
'value': self.txt_value,
|
|
})
|
|
desired.add_record(txt)
|
|
|
|
return desired
|
|
|
|
def _is_ownership(self, record):
|
|
return record._type == 'TXT' and \
|
|
record.name.startswith(self.txt_name) \
|
|
and record.values == self._txt_values
|
|
|
|
def process_plan(self, plan, *args, **kwargs):
|
|
if not plan:
|
|
# If we don't have any change there's nothing to do
|
|
return plan
|
|
|
|
# First find all the ownership info
|
|
owned = defaultdict(dict)
|
|
# We need to look for ownership in both the desired and existing
|
|
# states, many things will show up in both, but that's fine.
|
|
for record in list(plan.existing.records) + list(plan.desired.records):
|
|
if self._is_ownership(record):
|
|
pieces = record.name.split('.', 2)
|
|
if len(pieces) > 2:
|
|
_, _type, name = pieces
|
|
name = name.replace('_wildcard', '*')
|
|
else:
|
|
_type = pieces[1]
|
|
name = ''
|
|
owned[name][_type.upper()] = True
|
|
|
|
# Cases:
|
|
# - Configured in source
|
|
# - We'll fully CRU/manage it adding ownership TXT,
|
|
# thanks to process_source_zone, if needed
|
|
# - Not in source
|
|
# - Has an ownership TXT - delete it & the ownership TXT
|
|
# - Does not have an ownership TXT - don't delete it
|
|
# - Special records like octodns-meta
|
|
# - Should be left alone and should not have ownerthis TXTs
|
|
|
|
filtered_changes = []
|
|
for change in plan.changes:
|
|
record = change.record
|
|
|
|
if not self._is_ownership(record) and \
|
|
record._type not in owned[record.name] and \
|
|
record.name != 'octodns-meta':
|
|
# It's not an ownership TXT, it's not owned, and it's not
|
|
# special we're going to ignore it
|
|
continue
|
|
|
|
# We own this record or owned it up until now so whatever the
|
|
# change is we should do
|
|
filtered_changes.append(change)
|
|
|
|
if plan.changes != filtered_changes:
|
|
return Plan(plan.existing, plan.desired, filtered_changes,
|
|
plan.exists, plan.update_pcent_threshold,
|
|
plan.delete_pcent_threshold)
|
|
|
|
return plan
|