com.haulmont.cuba.core.sys.CubaMailSender.java Source code

Introduction

Here is the source code for com.haulmont.cuba.core.sys.CubaMailSender.java

Source

/*
 * Copyright (c) 2008-2016 Haulmont.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

package com.haulmont.cuba.core.sys;

import com.haulmont.cuba.core.app.EmailerConfig;
import com.haulmont.cuba.core.global.Configuration;
import org.apache.commons.lang.StringUtils;
import org.springframework.mail.javamail.JavaMailSenderImpl;

import org.springframework.stereotype.Component;
import javax.inject.Inject;
import javax.mail.Session;
import java.util.HashSet;
import java.util.Properties;
import java.util.Set;

@Component(CubaMailSender.NAME)
public class CubaMailSender extends JavaMailSenderImpl {

    public static final String NAME = "cuba_MailSender";

    protected EmailerConfig config;

    private boolean propertiesInitialized;

    @Inject
    public void setConfiguration(Configuration configuration) {
        config = configuration.getConfig(EmailerConfig.class);
    }

    @Override
    public String getHost() {
        return config.getSmtpHost();
    }

    @Override
    public void setHost(String host) {
        throw new UnsupportedOperationException("Use cuba.email.* properties");
    }

    @Override
    public int getPort() {
        return config.getSmtpPort();
    }

    @Override
    public void setPort(int port) {
        throw new UnsupportedOperationException("Use cuba.email.* properties");
    }

    @Override
    public String getUsername() {
        return config.getSmtpAuthRequired() && !StringUtils.isBlank(config.getSmtpUser()) ? config.getSmtpUser()
                : null;
    }

    @Override
    public void setUsername(String username) {
        throw new UnsupportedOperationException("Use cuba.email.* properties");
    }

    @Override
    public String getPassword() {
        return config.getSmtpAuthRequired() && !StringUtils.isBlank(config.getSmtpPassword())
                ? config.getSmtpPassword()
                : null;
    }

    @Override
    public void setPassword(String password) {
        throw new UnsupportedOperationException("Use cuba.email.* properties");
    }

    @Override
    public synchronized Session getSession() {
        if (!propertiesInitialized) {
            Properties properties = createJavaMailProperties();
            setJavaMailProperties(properties);
            propertiesInitialized = true;
        }
        return super.getSession();
    }

    protected Properties createJavaMailProperties() {
        long connectionTimeoutMillis = config.getSmtpConnectionTimeoutSec() * 1000;
        long timeoutMillis = config.getSmtpTimeoutSec() * 1000;

        Properties properties = new Properties();
        properties.setProperty("mail.smtp.auth", String.valueOf(config.getSmtpAuthRequired()));
        properties.setProperty("mail.smtp.starttls.enable", String.valueOf(config.getSmtpStarttlsEnable()));
        properties.setProperty("mail.smtp.connectiontimeout", String.valueOf(connectionTimeoutMillis));
        properties.setProperty("mail.smtp.timeout", String.valueOf(timeoutMillis));
        properties.setProperty("mail.smtp.ssl.enable", String.valueOf(config.getSmtpSslEnabled()));

        Set excludedProperties = new HashSet<>(properties.keySet());
        for (String name : AppContext.getPropertyNames()) {
            if (includeJavaMailProperty(name, excludedProperties)) {
                String value = AppContext.getProperty(name);
                if (value != null) {
                    properties.put(name, value);
                }
            }
        }
        return properties;
    }

    protected boolean includeJavaMailProperty(String name, Set excludedProperties) {
        return name.startsWith("mail.") && !excludedProperties.contains(name);
    }
}