Added base_sld_domains filter

filter_plugins/domain_filters.py

@@ -0,0 +1,69 @@
+import re
+from ansible.errors import AnsibleFilterError
+
+class FilterModule(object):
+    """
+    Custom Ansible filter plugin:
+      - generate_all_domains: Flatten, dedupe, sort domains with optional www prefixes
+      - generate_base_sld_domains: Extract unique sld.tld domains from values and redirect sources
+    """
+
+    def filters(self):
+        return {
+            'generate_all_domains': self.generate_all_domains,
+            'generate_base_sld_domains': self.generate_base_sld_domains,
+        }
+
+    @staticmethod
+    def generate_all_domains(domains_dict, include_www=True):
+        """
+        Transform a dict of domains (values: str, list, dict) into a flat list,
+        optionally add 'www.' prefixes, dedupe and sort alphabetically.
+
+        Avoids infinite loops by snapshotting initial domain list for www prefixes.
+        """
+        try:
+            flat = FilterModule._flatten_domain_values(domains_dict)
+            if include_www:
+                # Snapshot original list to avoid extending while iterating
+                original = list(flat)
+                flat.extend([f"www.{d}" for d in original])
+            return sorted(set(flat))
+        except Exception as exc:
+            raise AnsibleFilterError(f"generate_all_domains failed: {exc}")
+
+    @staticmethod
+    def generate_base_sld_domains(domains_dict, redirect_mappings):
+        """
+        Flatten domains_dict and redirect_mappings, extract second-level + top-level domains.
+        redirect_mappings: list of dicts with key 'source'
+        """
+        try:
+            flat = FilterModule._flatten_domain_values(domains_dict)
+            for mapping in redirect_mappings or []:
+                src = mapping.get('source')
+                if isinstance(src, str):
+                    flat.append(src)
+                elif isinstance(src, list):
+                    flat.extend(src)
+
+            pattern = re.compile(r'^(?:.*\.)?([^.]+\.[^.]+)$')
+            slds = {m.group(1) for d in flat if (m := pattern.match(d))}
+            return sorted(slds)
+        except Exception as exc:
+            raise AnsibleFilterError(f"generate_base_sld_domains failed: {exc}")
+
+    @staticmethod
+    def _flatten_domain_values(domains_dict):
+        """
+        Helper to extract domain strings from dict values (str, list, dict).
+        """
+        flat = []
+        for val in (domains_dict or {}).values():
+            if isinstance(val, str):
+                flat.append(val)
+            elif isinstance(val, list):
+                flat.extend(val)
+            elif isinstance(val, dict):
+                flat.extend(val.values())
+        return flat

filter_plugins/generate_all_domains.py

@@ -1,47 +0,0 @@
-import os
-from ansible.errors import AnsibleFilterError
-
-class FilterModule(object):
-    """
-    Custom Ansible filter to generate a flattened, deduplicated,
-    and sorted list of domains, with optional 'www.' prefixes.
-    """
-
-    def filters(self):
-        return {
-            'generate_all_domains': self.generate_all_domains,
-        }
-
-    @staticmethod
-    def generate_all_domains(domains_dict, include_www=True):
-        """
-        Transform a dict of domains into a flat list of domain strings.
-        Values in domains_dict may be strings, lists, or dicts.
-        If include_www is True, also generate "www." variants.
-        The final list is deduplicated and sorted alphabetically.
-
-        :param domains_dict: dict where each value is str, list, or dict of domains
-        :param include_www:  bool indicating if 'www.' prefixes should be added
-        :return:              sorted list of unique domain names
-        """
-        try:
-            flat = []
-            for val in domains_dict.values():
-                if isinstance(val, str):
-                    flat.append(val)
-                elif isinstance(val, list):
-                    flat.extend(val)
-                elif isinstance(val, dict):
-                    flat.extend(val.values())
-                else:
-                    # skip unsupported types
-                    continue
-
-            if include_www:
-                flat.extend(['www.' + d for d in flat])
-
-            # dedupe and sort
-            return sorted(set(flat))
-
-        except Exception as exc:
-            raise AnsibleFilterError(f"generate_all_domains failed: {exc}")