VelocityMessageComposer.java

/* Copyright 2014 Danish Maritime Authority.
 *
 *  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 net.maritimecloud.portal.infrastructure.mail;

import java.util.HashMap;
import java.util.Map;
import org.apache.velocity.app.VelocityEngine;
import org.springframework.ui.velocity.VelocityEngineUtils;

/**
 *
 * @author Christoffer Børrild
 */
public class VelocityMessageComposer implements MessageComposer {

    private static final String TEMPLATE_SIGN_UP_ACTIVATION_MESSAGE = "templates/signUpActivationMessage.vm.html";
    private static final String TEMPLATE_RESET_PASSWORD_MESSAGE = "templates/resetPasswordMessage.vm.html";
    private static final String TEMPLATE_CONFIRM_CHANGED_EMAIL_ADDRESS_MESSAGE = "templates/confirmChangedEmailAddressMessage.vm.html";
    private final VelocityEngine velocityEngine;
    private final String spaUrl;

    public VelocityMessageComposer(VelocityEngine velocityEngine, String spaUrl) {
        assert (velocityEngine != null);
        assert (spaUrl != null);
        this.velocityEngine = velocityEngine;
        this.spaUrl = spaUrl;
    }

    @Override
    public String composeSignUpActivationMessage(String username, String verificationCode) {
        assertNotNull(verificationCode);
        return compose(createModel(username, verificationCode), TEMPLATE_SIGN_UP_ACTIVATION_MESSAGE);
    }

    @Override
    public String composeResetPasswordMessage(String username, String resetPasswordCode) {
        assertNotNull(resetPasswordCode);
        return compose(createModel(username, resetPasswordCode), TEMPLATE_RESET_PASSWORD_MESSAGE);
    }

    @Override
    public String composeConfirmChangedEmailAddressMessage(String username, String verificationCode) {
        assertNotNull(verificationCode);
        return compose(createModel(username, verificationCode), TEMPLATE_CONFIRM_CHANGED_EMAIL_ADDRESS_MESSAGE);
    }

    private void assertNotNull(Object value) throws IllegalStateException {
        if (value == null) {
            throw new IllegalStateException("User object is missing activationid or reset key!");
        }
    }

    private String compose(Map model, String templateFilename) {
        return VelocityEngineUtils.mergeTemplateIntoString(velocityEngine, templateFilename, "UTF-8", model);
    }

    private Map createModel(String username, String key) {
        Map model = new HashMap();
        model.put("username", username);
        model.put("baseUrl", spaUrl);
        model.put("activationid", key);
        return model;
    }

}