Subversion Repositories SmartDukaan

//@Value("${prod}")
package com.spice.profitmandi.common.services;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.InputStreamSource;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.stereotype.Service;

import javax.mail.Multipart;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeBodyPart;
import javax.mail.internet.MimeMessage;
import javax.mail.internet.MimeMultipart;
import java.io.File;
import java.io.Serializable;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

@Service
public class EmailServiceImpl implements EmailService {

    private static final Logger logger =
            LogManager.getLogger(EmailServiceImpl.class);

    @Value("${prod}")
    private boolean isProd;

    private final JavaMailSender mailSender;

    public EmailServiceImpl(JavaMailSender mailSender) {
        this.mailSender = mailSender;
    }


    @Override
    public void sendMailWithAttachments(
            JavaMailSender mailSender,
            String[] emailTo,
            String[] cc,
            String subject,
            String body
    ) throws Exception {

        MimeMessage message = mailSender.createMimeMessage();
        MimeMessageHelper helper = new MimeMessageHelper(message, true);

        helper.setTo(emailTo);
        helper.setSubject(subject);
        helper.setText(body, true);

        if (cc != null) {
            helper.setCc(cc);
        }

        helper.setFrom(new InternetAddress(
                "noreply@smartdukaan.com",
                "SmartDukaan Care"
        ));

        sendIfProd(message, emailTo, subject);
    }

    @Override
    public void sendMailWithAttachments(
            JavaMailSender mailSender,
            String[] emailTo,
            String[] cc,
            String subject,
            String body,
            List<File> attachments
    ) throws Exception {

        MimeMessage message = mailSender.createMimeMessage();
        MimeMessageHelper helper = new MimeMessageHelper(message, true);

        helper.setTo(emailTo);
        helper.setSubject(subject);
        helper.setText(body, true);

        if (cc != null) {
            helper.setCc(cc);
        }

        helper.setFrom(new InternetAddress(
                "noreply@smartdukaan.com",
                "SmartDukaan Care"
        ));

        if (attachments != null) {
            for (File file : attachments) {
                helper.addAttachment(file.getName(), file);
            }
        }

        sendIfProd(message, emailTo, subject);
    }

    @Override
    public void sendMailWithAttachments(
            JavaMailSender mailSender,
            String emailTo,
            String[] cc,
            String subject,
            String body,
            List<File> attachments
    ) throws Exception {
        // Delegate to array version for Utils.java compatibility
        sendMailWithAttachments(
                mailSender,
                new String[]{emailTo},
                cc,
                subject,
                body,
                attachments
        );
    }

    @Override
    public void sendMailWithAttachment(
            JavaMailSender mailSender,
            String[] emailTo,
            String[] cc,
            String subject,
            String body,
            String fileName,
            InputStreamSource inputStreamSource
    ) throws Exception {

        MimeMessage message = mailSender.createMimeMessage();
        MimeMessageHelper helper = new MimeMessageHelper(message, true);

        helper.setTo(emailTo);
        helper.setSubject(subject);
        helper.setText(body, true);

        if (cc != null) {
            helper.setCc(cc);
        }

        helper.addAttachment(fileName, inputStreamSource);

        helper.setFrom(new InternetAddress(
                "noreply@smartdukaan.com",
                "SmartDukaan Care"
        ));

        sendIfProd(message, emailTo, subject);
    }

    @Override
    public void sendMailWithAttachments(
            JavaMailSender mailSender,
            String[] emailTo,
            String[] cc,
            String[] bcc,
            String subject,
            String body,
            boolean html,
            Attachment... attachments
    ) throws Exception {

        MimeMessage message = mailSender.createMimeMessage();
        MimeMessageHelper helper = new MimeMessageHelper(message, true);

        helper.setTo(emailTo);
        helper.setSubject(subject);
        helper.setText(body, html);

        if (cc != null) helper.setCc(cc);
        if (bcc != null) helper.setBcc(bcc);

        helper.setFrom(new InternetAddress(
                "noreply@smartdukaan.com",
                "SmartDukaan Care"
        ));

        if (attachments != null) {
            for (Attachment attachment : attachments) {
                if (attachment != null) {
                    helper.addAttachment(
                            attachment.getFileName(),
                            attachment.getInputStreamSource()
                    );
                }
            }
        }

        sendIfProd(message, emailTo, subject);
    }


    @Override
    public void sendHtmlMailWithAttachments(
            JavaMailSender mailSender,
            String[] emailTo,
            String[] cc,
            String subject,
            String body,
            Attachment... attachments
    ) throws Exception {

        sendMailWithAttachments(
                mailSender,
                emailTo,
                cc,
                null,
                subject,
                body,
                true,
                attachments
        );
    }


    @Override
    public void sendEmbeddedHtmlMail(
            JavaMailSender mailSender,
            String[] emailTo,
            String[] cc,
            String subject,
            String body,
            Map<? extends Serializable, File> inlineImages
    ) throws Exception {

        MimeMessage message = mailSender.createMimeMessage();
        Multipart multipart = new MimeMultipart();

        MimeBodyPart bodyPart = new MimeBodyPart();
        bodyPart.setContent(body, "text/html");
        multipart.addBodyPart(bodyPart);

        if (inlineImages != null) {
            for (Map.Entry<? extends Serializable, File> entry : inlineImages.entrySet()) {
                MimeBodyPart imagePart = new MimeBodyPart();
                imagePart.setHeader("Content-ID", entry.getKey().toString());
                imagePart.setDisposition(MimeBodyPart.INLINE);
                imagePart.attachFile(entry.getValue());
                multipart.addBodyPart(imagePart);
            }
        }

        message.setContent(multipart);
        message.setSubject(subject);
        message.setRecipients(
                MimeMessage.RecipientType.TO,
                InternetAddress.parse(String.join(",", emailTo))
        );

        if (cc != null) {
            message.addRecipients(
                    MimeMessage.RecipientType.CC,
                    InternetAddress.parse(String.join(",", cc))
            );
        }

        message.setFrom(new InternetAddress(
                "noreply@smartdukaan.com",
                "SmartDukaan Care"
        ));

        sendIfProd(message, emailTo, subject);
    }


    private void sendIfProd(
            MimeMessage message,
            String[] emailTo,
            String subject
    ) {
        if (isProd) {
            mailSender.send(message);
        } else {
            logger.info(
                    "[MAIL BLOCKED - NON PROD] to={}, subject={}",
                    Arrays.toString(emailTo),
                    subject
            );
        }
    }
}