package com.qianwen.core.notify.provider.email.embedded;

import com.alibaba.fastjson.JSONObject;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import javax.mail.internet.MimeMessage;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.qianwen.core.http.HttpRequest;
import com.qianwen.core.log.exception.ServiceException;
import com.qianwen.core.notify.DefaultNotifyType;
import com.qianwen.core.notify.NotifyType;
import com.qianwen.core.notify.Provider;
import com.qianwen.core.notify.notifier.AbstractNotifier;
import com.qianwen.core.notify.notifier.NotifierProperties;
import com.qianwen.core.notify.provider.email.EmailProvider;
import com.qianwen.core.notify.provider.email.embedded.EmailTemplate;
import com.qianwen.core.notify.template.Template;
import com.qianwen.core.notify.template.TemplateManager;
import com.qianwen.core.tool.api.ResultCode;
import com.qianwen.core.tool.metadata.Values;
import com.qianwen.core.tool.utils.ExpressionUtils;
import com.qianwen.core.tool.utils.Func;
import com.qianwen.core.tool.utils.StringUtil;
import org.springframework.core.io.ByteArrayResource;
import org.springframework.core.io.InputStreamResource;
import org.springframework.core.io.InputStreamSource;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.mail.javamail.JavaMailSenderImpl;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.util.StringUtils;

/* loaded from: blade-starter-notify-9.3.0.0-SNAPSHOT.jar:org/springblade/core/notify/provider/email/embedded/DefaultEmailNotifier.class */
public class DefaultEmailNotifier extends AbstractNotifier<EmailTemplate> {
    private static final Logger log = LoggerFactory.getLogger(DefaultEmailNotifier.class);
    private String sender;
    private JavaMailSender javaMailSender;
    private NotifierProperties notifierProperties;
    private String notifierId;


    public String getSender() {
        return this.sender;
    }

    public void setSender(final String sender) {
        this.sender = sender;
    }

    public JavaMailSender getJavaMailSender() {
        return this.javaMailSender;
    }

    public void setJavaMailSender(final JavaMailSender javaMailSender) {
        this.javaMailSender = javaMailSender;
    }

    public void setNotifierProperties(final NotifierProperties notifierProperties) {
        this.notifierProperties = notifierProperties;
    }

    public DefaultEmailNotifier(NotifierProperties properties, TemplateManager templateManager) {
        super(templateManager);
        this.notifierId = properties.getId();
        DefaultEmailProperties emailProperties = (DefaultEmailProperties) new JSONObject(properties.getConfiguration()).toJavaObject(DefaultEmailProperties.class);
        JavaMailSenderImpl mailSender = new JavaMailSenderImpl();
        mailSender.setHost(emailProperties.getHost());
        mailSender.setPort(emailProperties.getPort());
        mailSender.setUsername(emailProperties.getUsername());
        mailSender.setPassword(emailProperties.getPassword());
        mailSender.setJavaMailProperties(emailProperties.createJavaMailProperties());
        this.sender = emailProperties.getSender();
        this.javaMailSender = mailSender;
        this.notifierProperties = properties;
    }

    @Override // com.qianwen.core.notify.notifier.Notifier
    public String getNotifierId() {
        return this.notifierId;
    }

    @Override // com.qianwen.core.notify.notifier.Notifier
    public NotifyType getType() {
        return DefaultNotifyType.email;
    }

    @Override // com.qianwen.core.notify.notifier.Notifier
    public Provider getProvider() {
        return EmailProvider.embedded;
    }

    @Override // com.qianwen.core.notify.notifier.Notifier
    public NotifierProperties getNotifierProperties() {
        return this.notifierProperties;
    }

    public void send(EmailTemplate template, String traceId, Values context, List<String> notifiedParty) {
        ParsedEmailTemplate parsedEmailTemplate = convert(template, context.getAllValues());
        doSend(parsedEmailTemplate, notifiedParty);
    }

    protected void doSend(ParsedEmailTemplate template, List<String> sendTo) {
    	try {
	        MimeMessage mimeMessage = this.javaMailSender.createMimeMessage();
	        MimeMessageHelper helper = new MimeMessageHelper(mimeMessage, true, "utf-8");
	        helper.setFrom(this.sender);
	        helper.setTo((String[]) sendTo.toArray(new String[0]));
	        helper.setSubject(template.getSubject());
	        helper.setText(new String(template.getContent().getBytes(), StandardCharsets.UTF_8), true);
	        for (Map.Entry<String, String> entry : template.getAttachments().entrySet()) {
	            String attachmentFileName = entry.getKey();
	            InputStreamSource fileResource = convertResource(entry.getValue());
	            if (Func.isNotEmpty(fileResource)) {
	                helper.addAttachment(attachmentFileName, fileResource);
	            }
	        }
	        for (Map.Entry<String, String> entry2 : template.getImages().entrySet()) {
	            String attachmentFileName2 = entry2.getKey();
	            InputStreamSource fileResource2 = convertResource(entry2.getValue());
	            if (Func.isNotEmpty(fileResource2)) {
	                helper.addInline(attachmentFileName2, fileResource2, "application/octet-stream");
	            }
	        }
	        this.javaMailSender.send(mimeMessage);
    	}catch(Exception ex) {
    		throw new RuntimeException(ex);
    	}
    }

    protected InputStreamSource convertResource(String resource) {
    	if (resource.startsWith("http")) {
    	      HashMap<Object, Object> headers = new HashMap<>();
    	      headers.put("Accept", "application/octet-stream");
    	      byte[] fileBytes = HttpRequest.get(resource).addHeader("Accept", "application/octet-stream").execute().onFailed((k, v) -> log.info(StringUtil.format("获取附件资源{}失败,原因{}", new Object[] { k.url(), v.getMessage() }))).asBytes();
    	      return (InputStreamSource)new ByteArrayResource(fileBytes);
    	    } 
    	    try {
    	      return (InputStreamSource)new InputStreamResource(new FileInputStream(resource));
    	    } catch (FileNotFoundException e) {
    	      log.error("处理邮件中资源文件异常:{},堆栈信息如下{}", e.getMessage(), e.getStackTrace());
    	      return null;
    	    } 
    }

    protected ParsedEmailTemplate convert(EmailTemplate template, Map<String, Object> context) {
        String subject = template.getSubject();
        String content = template.getContent();
        if (StringUtils.isEmpty(subject) || StringUtils.isEmpty(content)) {
            throw new ServiceException(ResultCode.PARAM_MISS, new Throwable("模板内容错误，content 或者 subject 不能为空."));
        }
        String sendContent = render(content, context);
        List<EmailTemplate.Attachment> tempAttachments = template.getAttachments();
        Map<String, String> attachments = new HashMap<>();
        if (tempAttachments != null) {
            for (EmailTemplate.Attachment tempAttachment : tempAttachments) {
                attachments.put(tempAttachment.getName(), render(tempAttachment.getLocation(), context));
            }
        }
        return ParsedEmailTemplate.builder().attachments(attachments).images(extractSendContentImage(sendContent)).content(sendContent).subject(render(subject, context)).build();
    }

    private Map<String, String> extractSendContentImage(String sendContent) {
        Map<String, String> images = new HashMap<>();
        Document doc = Jsoup.parse(sendContent);
        Iterator it = doc.getElementsByTag("img").iterator();
        while (it.hasNext()) {
            Element src = (Element) it.next();
            String s = src.attr("src");
            if (!s.startsWith("http")) {
                String tempKey = Func.randomUUID();
                src.attr("src", "cid:".concat(tempKey));
                images.put(tempKey, s);
            }
        }
        return images;
    }

    private String render(String str, Map<String, Object> context) {
        return ExpressionUtils.analytical(str, context, "spel");
    }

    @Override // com.qianwen.core.notify.notifier.Notifier
    public void close() {
    }
}