Add 'content' flag for including template content. (#13)

Bug fix to describe template, not cluster.

Signed-off-by: Webster Mudge <wmudge@cloudera.com>

Author: wmudge

plugins/modules/datahub_template_info.py

@@ -42,6 +42,14 @@
     required: False
     aliases:
       - template
+  return_content:
+    description: Flag dictating if cluster template content is returned
+    type: bool
+    required: False
+    default: False
+    aliases:
+     - template_content
+     - content
 extends_documentation_fragment:
   - cloudera.cloud.cdp_sdk_options
   - cloudera.cloud.cdp_auth_options
@@ -56,6 +64,11 @@
 # Gather detailed information about a named Datahub
 - cloudera.cloud.datahub_template_info:
     name: example-template
+    
+# Gather detailed information about a named Datahub, including the template contents in JSON
+- cloudera.cloud.datahub_template_info:
+    name: example-template
+    return_content: yes
 '''
 
 RETURN = r'''
@@ -90,6 +103,13 @@
       description: The status of the cluster template.
       returned: always
       type: str
+      sample:
+        - DEFAULT
+        - USER_MANAGED
+    clusterTemplateContent:
+      description: The cluster template contents, in JSON.
+      returned: when specified
+      type: str
     tags:
       description: Tags added to the cluster template
       type: dict
@@ -121,27 +141,53 @@ def __init__(self, module):
 
         # Set variables
         self.name = self._get_param('name')
+        self.content = self._get_param('return_content')
 
         # Initialize return values
         self.templates = []
 
+        # Initialize internal values
+        self.all_templates = []
+
         # Execute logic process
         self.process()
 
     @CdpModule._Decorators.process_debug
     def process(self):
-        if self.name:  # Note that both None and '' will trigger this
-            template_single = self.cdpy.datahub.describe_cluster(self.name)
-            if template_single is not None:
-                self.templates.append(template_single)
+        self.all_templates = self.cdpy.datahub.list_cluster_templates()
+        
+        if self.name:
+            short_desc = next((t for t in self.all_templates if t['crn'] == self.name 
+                               or t['clusterTemplateName'] == self.name), None)
+            if short_desc is not None:
+              if self.content:
+                self.templates.append(self._describe_template(short_desc))
+              else:
+                self.templates.append(short_desc)
+            else:
+              self.module.warn("Template not found, '%s'" % self.name)
         else:
-            self.templates = self.cdpy.datahub.list_cluster_templates()
-
+            if self.content:
+              for short_desc in self.all_templates:
+                self.templates.append(self._describe_template(short_desc))
+            else:
+              self.templates = self.all_templates
+
+    def _describe_template(self, short_desc):
+      full_desc = self.cdpy.datahub.describe_cluster_template(short_desc['crn'])
+      if full_desc is not None:
+          full_desc.update(productVersion=short_desc['productVersion'])
+          return full_desc
+      else:
+        self.module.fail_json(msg="Failed to retrieve Cluster Template content, '%s'" %
+                              short_desc['clusterTemplateName'])
+        
 
 def main():
     module = AnsibleModule(
         argument_spec=CdpModule.argument_spec(
-            name=dict(required=False, type='str', aliases=['template'])
+            name=dict(required=False, type='str', aliases=['template', 'crn']),
+            return_content=dict(required=False, type='bool', default=False, aliases=['template_content', 'content'])
         ),
         supports_check_mode=True
     )