[PR #6334/64e797d0 backport][stable-6] dig: Support multiple domains in a single lookup (#6346)

dig: Support multiple domains in a single lookup (#6334)

* dig: Support multiple domains in a single lookup (#6334)

The docs for this plugin indicated that multiple domains could be
specified at once, but the code did not support multiple domains.

* Address review feedback.

(cherry picked from commit 64e797d077)

Co-authored-by: Kevin P. Fleming <kpfleming@users.noreply.github.com>

changelogs/fragments/6334-dig-support-multiple-domains.yml

@@ -0,0 +1,3 @@
+minor_changes:
+  - dig lookup plugin - Support multiple domains to be queried as indicated in docs
+    (https://github.com/ansible-collections/community.general/pull/6334).

plugins/lookup/dig.py

@@ -61,6 +61,7 @@ DOCUMENTATION = '''
         description:
           - Return empty result without empty strings, and return empty list instead of C(NXDOMAIN).
           - The default for this option will likely change to C(true) in the future.
+          - This option will be forced to C(true) if multiple domains to be queried are specified.
         default: false
         type: bool
         version_added: 6.0.0
@@ -95,6 +96,21 @@ EXAMPLES = """
     msg: "MX record for gmail.com {{ item }}"
   with_items: "{{ lookup('community.general.dig', 'gmail.com./MX', wantlist=true) }}"
 
+- name: Lookup multiple names at once
+  ansible.builtin.debug:
+    msg: "A record found {{ item }}"
+  loop: "{{ query('community.general.dig', 'example.org.', 'example.com.', 'gmail.com.') }}"
+
+- name: Lookup multiple names at once (from list variable)
+  ansible.builtin.debug:
+    msg: "A record found {{ item }}"
+  loop: "{{ query('community.general.dig', *hosts) }}"
+  vars:
+    hosts:
+      - example.org.
+      - example.com.
+      - gmail.com.
+
 - ansible.builtin.debug:
     msg: "Reverse DNS for 192.0.2.5 is {{ lookup('community.general.dig', '192.0.2.5/PTR') }}"
 - ansible.builtin.debug:
@@ -308,7 +324,7 @@ class LookupModule(LookupBase):
         edns_size = 4096
         myres.use_edns(0, ednsflags=dns.flags.DO, payload=edns_size)
 
-        domain = None
+        domains = []
         qtype = self.get_option('qtype')
         flat = self.get_option('flat')
         fail_on_error = self.get_option('fail_on_error')
@@ -365,24 +381,32 @@ class LookupModule(LookupBase):
             if '/' in t:
                 try:
                     domain, qtype = t.split('/')
+                    domains.append(domain)
                 except Exception:
-                    domain = t
+                    domains.append(t)
             else:
-                domain = t
+                domains.append(t)
 
         # print "--- domain = {0} qtype={1} rdclass={2}".format(domain, qtype, rdclass)
 
-        ret = []
-
         if qtype.upper() == 'PTR':
+            reversed_domains = []
+            for domain in domains:
                 try:
                     n = dns.reversename.from_address(domain)
-                domain = n.to_text()
+                    reversed_domains.append(n.to_text())
                 except dns.exception.SyntaxError:
                     pass
                 except Exception as e:
                     raise AnsibleError("dns.reversename unhandled exception %s" % to_native(e))
+            domains = reversed_domains
 
+        if len(domains) > 1:
+            real_empty = True
+
+        ret = []
+
+        for domain in domains:
             try:
                 answers = myres.query(domain, qtype, rdclass=rdclass)
                 for rdata in answers: