embulk-output-send_email 0.1.3 → 0.1.5

data/src/main/java/org/embulk/output/send_email/SendEmailOutputPlugin.java

@@ -1,29 +1,30 @@
 package org.embulk.output.send_email;
 
 
+import java.io.File;
 import java.util.*;
 import java.util.stream.Collectors;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
+import org.apache.commons.lang3.StringUtils;
 import org.embulk.config.*;
 import org.embulk.spi.*;
+import org.jsoup.Jsoup;
+
 import javax.mail.*;
 import javax.mail.internet.InternetAddress;
 import javax.mail.internet.MimeBodyPart;
 import javax.mail.internet.MimeMessage;
 import javax.mail.internet.MimeMultipart;
-import javax.validation.constraints.Null;
-
-import static ch.qos.logback.core.joran.action.ActionConst.NULL;
 
 public class SendEmailOutputPlugin
         implements OutputPlugin {
     public interface PluginTask
             extends Task {
 
-        @Config("file_type")
+        @Config("format_type")
         @ConfigDefault("\"json\"")
-        public String getFileType();
+        public String getFormatType();
 
         @Config("to")
         public List<String> getTO();
@@ -66,9 +67,17 @@ public class SendEmailOutputPlugin
         @ConfigDefault("\"\"")
         public String getUserName();
 
-        @Config("smtp_enable")
+        @Config("template")
+        @ConfigDefault("\"\"")
+        public String getEmailTemplate();
+
+        @Config("is_html")
+        @ConfigDefault("false")
+        public boolean getIsHTML();
+
+        @Config("enable_starttls")
         @ConfigDefault("\"true\"")
-        public String getSmtpEnable();
+        public String getEnableStarttls();
     }
 
     @Override
@@ -76,6 +85,7 @@ public class SendEmailOutputPlugin
                                   Schema schema, int taskCount,
                                   OutputPlugin.Control control) {
         PluginTask task = config.loadConfig(PluginTask.class);
+        System.out.println(task);
         control.run(task.dump());
         return Exec.newConfigDiff();
     }
@@ -204,7 +214,7 @@ public class SendEmailOutputPlugin
             try {
                 Properties properties = new Properties();
                 properties.put("mail.smtp.auth", task.getAuth());
-                properties.put("mail.smtp.starttls.enable", task.getSmtpEnable());
+                properties.put("mail.smtp.starttls.enable", task.getEnableStarttls());
 
                 properties.put("mail.smtp.ssl.protocols", task.getProtocol());
                 properties.put("mail.smtp.host", task.getHost());
@@ -243,11 +253,18 @@ public class SendEmailOutputPlugin
                 }
                 message.addRecipients(Message.RecipientType.TO, InternetAddress.parse(listStringTo));
                 message.setSubject(task.getSubject());
-
-                if (task.getFileType().equalsIgnoreCase("html")) {
-                    setHtmlContent(message, mapList, schema);
-                } else if (task.getFileType().equalsIgnoreCase("json")) {
-                    setJsonContent(message, mapList, schema);
+                if(task.getEmailTemplate().length()!=0){
+                    if(task.getIsHTML()) {
+                        setEmailTemplate(message, mapList, schema, task);
+                    }else{
+                        setTextContent(message,mapList,schema,task);
+                    }
+                }else{
+                    if (task.getFormatType().equalsIgnoreCase("html")) {
+                        setHtmlContent(message,mapList,schema);
+                    } else if (task.getFormatType().equalsIgnoreCase("json")) {
+                        setJsonContent(message, mapList, schema);
+                    }
                 }
                 return message;
             } catch (Exception e) {
@@ -256,16 +273,15 @@ public class SendEmailOutputPlugin
             }
         }
 
-        private static void setHtmlContent(Message message, ArrayList<LinkedHashMap<String, Object>> mapList, Schema schema) {
+        private static void setEmailTemplate(Message message, ArrayList<LinkedHashMap<String, Object>> mapList, Schema schema,PluginTask task) {
 
             try {
                 MimeMultipart multipart = new MimeMultipart();
                 BodyPart messageBodyPart = new MimeBodyPart();
                 StringBuilder email = new StringBuilder();
+                StringBuilder email1= new StringBuilder();
 
-                email.append("<!DOCTYPE html><html><body>"
-                        + "<table style='border:1px solid #96D4D4;border-collapse: collapse;width: 100%'>");
-
+                email.append("<table style='border:1px solid #96D4D4;border-collapse: collapse;width: 100%'>");
                 email.append("<tr>");
                 for (int i = 0; i < schema.size(); i++) {
                     email.append("<th style='border:2px solid black; border-collapse: collapse;border-color: #96D4D4'>");
@@ -284,9 +300,71 @@ public class SendEmailOutputPlugin
                     }
                     email.append("<tr>");
                 }
-                email.append("</table></body></html>");
+                email.append("</table>");
+                email.append("<br>");
+                String s=null;
+                try {
+                    File sourceFile = new File(task.getEmailTemplate());
+                    org.jsoup.nodes.Document doc = Jsoup.parse(sourceFile, "UTF-8");
+                    org.jsoup.nodes.Element elements = doc.body();
+                    s=StringUtils.replace(String.valueOf(elements),"{{data}}",email.toString());
+
+                } catch (Exception e) {
+                    // TODO Auto-generated catch block
+                    e.printStackTrace();
+                }
+                email1.append(s);
+                messageBodyPart.setContent(email1.toString(), "text/html");
+                multipart.addBodyPart(messageBodyPart);
+                message.setContent(multipart);
+            } catch (Exception e) {
+                System.out.println("Not able to send data");
+                throw new RuntimeException(e);
+            }
 
-                messageBodyPart.setContent(email.toString(), "text/html");
+        }
+        private static void setTextContent(Message message, ArrayList<LinkedHashMap<String, Object>> mapList, Schema schema,PluginTask task) {
+
+            try {
+                MimeMultipart multipart = new MimeMultipart();
+                BodyPart messageBodyPart = new MimeBodyPart();
+                StringBuilder email = new StringBuilder();
+
+                StringBuilder email1 = new StringBuilder();
+
+                email.append("\n");
+                email.append("\n");
+                for (int i = 0; i < schema.size(); i++) {
+                    email.append(schema.getColumn(i).getName().split("/")[0]);
+                    email.append(",");
+                }
+                email.append("\n");
+
+                for (int i = 0; i < mapList.size(); i++) {
+                    Map<String, Object> map = mapList.get(i);
+                    email.append("\n");
+                    for (Map.Entry<String, Object> entry : map.entrySet()) {
+                        email.append(entry.getValue());
+                        email.append(",");
+                    }
+                    email.append("\n");
+                }
+
+                String s=null;
+                try {
+                    File sourceFile = new File(task.getEmailTemplate());
+                    org.jsoup.nodes.Document doc = Jsoup.parse(sourceFile, "UTF-8");
+                    String elements = doc.text();
+                    s=StringUtils.replace(String.valueOf(elements),"{{data}}",email.toString());
+
+                } catch (Exception e) {
+                    // TODO Auto-generated catch block
+                    e.printStackTrace();
+                }
+               // email1.append("\n");
+                email1.append(s);
+                email1.append("\n");
+                messageBodyPart.setText(email1.toString());
                 multipart.addBodyPart(messageBodyPart);
                 message.setContent(multipart);
             } catch (Exception e) {
@@ -309,6 +387,55 @@ public class SendEmailOutputPlugin
 
         }
 
+        private static void setHtmlContent(Message message, ArrayList<LinkedHashMap<String, Object>> mapList, Schema schema) {
+
+            try {
+                MimeMultipart multipart = new MimeMultipart();
+                BodyPart messageBodyPart = new MimeBodyPart();
+                StringBuilder email = new StringBuilder();
+
+                email.append("<!DOCTYPE html><html><body>");
+                email.append("<p>Hello Team,</p>");
+                email.append("<header>\n" +
+                        "    <h3>This is the daily ETL Report</h3>\n" +
+                        "</header>");
+
+                email.append("<table style='border:1px solid #96D4D4;border-collapse: collapse;width: 100%'>");
+
+                email.append("<tr>");
+                for (int i = 0; i < schema.size(); i++) {
+                    email.append("<th style='border:2px solid black; border-collapse: collapse;border-color: #96D4D4'>");
+                    email.append(schema.getColumn(i).getName().split("/")[0]);
+                    email.append("</th>");
+                }
+                email.append("</tr>");
+
+                for (int i = 0; i < mapList.size(); i++) {
+                    Map<String, Object> map = mapList.get(i);
+                    email.append("<tr >");
+                    for (Map.Entry<String, Object> entry : map.entrySet()) {
+                        email.append("<td style='border:1px solid black;border-collapse: collapse;text-align:center;border-color: #96D4D4'>");
+                        email.append(entry.getValue());
+                        email.append("</td>");
+                    }
+                    email.append("<tr>");
+                }
+                email.append("</table>");
+                email.append("<footer>\n" +
+                        "  <p>Thanks</p>\n" +
+                        "  <p>By ETL process</p>\n" +
+                        "</footer>");
+                email.append("</body></html>");
+
+                messageBodyPart.setContent(email.toString(), "text/html");
+                multipart.addBodyPart(messageBodyPart);
+                message.setContent(multipart);
+            } catch (Exception e) {
+                System.out.println("Not able to send data");
+                throw new RuntimeException(e);
+            }
+
+        }
         @Override
         public void finish() {
 

data/src/test/java/org/embulk/output/send_email/TestSendEmailOutputPlugin.java

@@ -1,5 +1,5 @@
-package org.embulk.output.send_email;
-
-public class TestSendEmailOutputPlugin
-{
-}
+package org.embulk.output.send_email;
+
+public class TestSendEmailOutputPlugin
+{
+}

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: embulk-output-send_email
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - Infoobjects Inc.
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-09-26 00:00:00.000000000 Z
+date: 2022-09-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -53,7 +53,7 @@ files:
 - classpath/activation-1.1.jar
 - classpath/commons-codec-1.10.jar
 - classpath/commons-logging-1.2.jar
-- classpath/embulk-output-send_email-0.1.3.jar
+- classpath/embulk-output-send_email-0.1.5.jar
 - classpath/google-api-client-1.26.0.jar
 - classpath/google-api-client-java6-1.26.0.jar
 - classpath/google-api-services-gmail-v1-rev83-1.23.0.jar
@@ -66,11 +66,14 @@ files:
 - classpath/httpcore-4.4.9.jar
 - classpath/j2objc-annotations-1.1.jar
 - classpath/jackson-core-2.9.6.jar
+- classpath/jsoup-1.10.3.jar
 - classpath/jsr305-3.0.2.jar
 - classpath/mail-1.4.7.jar
 - config/checkstyle/checkstyle.xml
 - config/checkstyle/default.xml
 - example/config.yml
+- example/email.html
+- example/email.txt
 - gradle/wrapper/gradle-wrapper.jar
 - gradle/wrapper/gradle-wrapper.properties
 - gradlew