Get vault secret for smtp (#1129)

Co-authored-by: Jayasimha Raghavan <[email protected]>

unskript-ctl/config/unskript_ctl_config.yaml

@@ -193,18 +193,21 @@ notification:
     #    - Sendgrid - Sendgrid
     provider: ""
     SMTP:
+      vault-secret-path: ""
       smtp-host: ""
       smtp-user: ""
       smtp-password: ""
       to-email: ""
       from-email: ""
     SES:
+      vault-secret-path: ""
       access_key: ""
       secret_access: ""
       region: ""
       to-email: ""
       from-email: ""
     Sendgrid:
+      vault-secret-path: ""
       api_key: ""
       to-email: ""
       from-email: ""

unskript-ctl/unskript_ctl_notification.py

@@ -31,6 +31,7 @@
 from unskript_utils import *
 from unskript_ctl_version import *
 from unskript_ctl_factory import NotificationFactory
+from unskript_ctl_custom_notification import custom_email_notification_main
 
 # This class implements Notification function for Slack
 class SlackNotification(NotificationFactory):
@@ -65,8 +66,8 @@ def notify(self, **kwargs):
             self.logger.error("Result Empty: No results to notify")
             return False
 
-        if not self.validate_data(summary_results):
-            self.logger.debug("Given Summary Result does not validate against Slack Schema")
+        # if not self.validate_data(summary_results):
+        #     self.logger.debug("Given Summary Result does not validate against Slack Schema")
 
         message = self._generate_notification_message(summary_results)
         if not message:
@@ -199,8 +200,8 @@ def create_temp_files_of_failed_check_results(self,
         if not failed_result:
             self.logger.error("Failed Result is Empty")
             return list_of_failed_files
-        if not self.validate_data(failed_result, self.checks_schema_file):
-            self.logger.debug("Validation of Given Result failed against Notification Schema")
+        # if not self.validate_data(failed_result, self.checks_schema_file):
+        #     self.logger.debug("Validation of Given Result failed against Notification Schema")
 
         if failed_result and len(failed_result.get('result', [])):
             for result_item in failed_result['result']:
@@ -447,6 +448,7 @@ def create_email_header(self, title: str = None):
             '''
         return message
 
+
     def prepare_combined_email(self,
                                summary_results: list,
                                failed_result: dict,
@@ -849,6 +851,80 @@ def send_smtp_notification(self,
             self.logger.info(f"Notification sent successfully to {to_email}")
         return True
 
+
+class CustomSMTPNotification(EmailNotification):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
+    def notify(self, **kwargs):
+        super().notify(**kwargs)
+
+        summary_results = kwargs.get('summary_result', [])
+        failed_result = kwargs.get('failed_result', {})
+        output_metadata_file = kwargs.get('output_metadata_file')
+        to_email = os.environ.get("LB_NOTIFICATION_RECEIVER_EMAIL",
+                                  "[email protected]")
+
+        subject = kwargs.get('subject', self.email_config.get('email_subject_line', 'Run Result'))
+
+        parent_folder = self.execution_dir
+        target_file_name = None
+        tar_file_name = None
+        if output_metadata_file:
+            parent_folder = os.path.dirname(output_metadata_file)
+            target_name = os.path.basename(parent_folder)
+            tar_file_name = f"{target_name}" + '.tar.bz2'
+            target_file_name = os.path.join('/tmp', tar_file_name)
+            output_metadata_file = output_metadata_file.split('/')[-1]
+
+        message = self.create_email_header(title=None)
+        if summary_results and len(summary_results):
+            message += self.create_checks_summary_message(
+                summary_results=summary_results,
+                failed_result=failed_result
+            )
+        if len(failed_result) and self.send_failed_objects_as_attachment:
+            self.create_temp_files_of_failed_check_results(failed_result=failed_result)
+
+
+        if tar_file_name and len(os.listdir(self.execution_dir)):
+            self.create_tarball_archive(
+                tar_file_name=tar_file_name,
+                output_metadata_file=output_metadata_file,
+                parent_folder=parent_folder
+            )
+
+        email_attach_name = None
+        if target_file_name and os.path.exists(target_file_name):
+            email_attach_name = target_file_name
+
+        info_result = self.create_info_gathering_action_result()
+        if info_result:
+            message += info_result
+            self.create_info_legos_output_file()
+
+        if output_metadata_file:
+            message += self.create_script_summary_message(
+                output_metadata_file=output_metadata_file
+            )
+        message += "</body></html>"
+
+
+        retval = custom_email_notification_main(
+            _logger=self.logger,
+            email_subject=subject,
+            email_content = message,
+            email_recipient = to_email,
+            file_path = email_attach_name
+        )
+        if retval:
+            self.logger.info("Successfully Sent Email via SMTP Relay")
+        else:
+            self.logger.error("Failed to send email via SMTP Relay")
+
+        return retval
+
+
 # Usage:
 # n = Notification()
 # n.notify(
@@ -918,6 +994,13 @@ def _do_send_email():
                             from_email = kwargs.get('from_email', smtp.get('from-email')),
                             subject = kwargs.get('subject', self.email_config.get('email_subject_line', 'Run Result'))
                             )
+            elif self.email_config.get('provider').lower() == 'custom':
+                retval = CustomSMTPNotification().notify(
+                            summary_result = summary_results,
+                            failed_result = failed_objects,
+                            output_metadata_file = kwargs.get('output_metadata_file'),
+                            subject = kwargs.get('subject', self.email_config.get('email_subject_line', 'Run Result'))
+                            )
             elif self.email_config.get('provider').lower() == 'sendgrid':
                 sendgrid = self.email_config.get('Sendgrid')
                 retval = SendgridNotification().notify(