com.gnizr.web.action.user.RequestPasswordReset.java Source code

Introduction

Here is the source code for com.gnizr.web.action.user.RequestPasswordReset.java

Source

/*
 * gnizr is a trademark of Image Matters LLC in the United States.
 * 
 * The contents of this file are subject to the Mozilla Public License Version
 * 1.1 (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.mozilla.org/MPL/
 * 
 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either expressed or implied. See the License
 * for the specific language governing rights and limitations under the License.
 * 
 * The Initial Contributor of the Original Code is Image Matters LLC.
 * Portions created by the Initial Contributor are Copyright (C) 2007
 * Image Matters LLC. All Rights Reserved.
 */
package com.gnizr.web.action.user;

import java.util.HashMap;
import java.util.Map;

import org.apache.log4j.Logger;
import org.springframework.mail.MailSender;
import org.springframework.mail.SimpleMailMessage;
import org.springframework.ui.freemarker.FreeMarkerTemplateUtils;

import com.gnizr.core.exceptions.NoSuchUserException;
import com.gnizr.core.user.UserManager;
import com.gnizr.core.util.TokenManager;
import com.gnizr.db.dao.User;
import com.gnizr.web.action.AbstractAction;
import com.gnizr.web.action.error.ActionErrorCode;

import freemarker.template.Configuration;
import freemarker.template.Template;

public class RequestPasswordReset extends AbstractAction {

    /**
     * 
     */
    private static final long serialVersionUID = 2177445666023620630L;

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

    private String username;

    private UserManager userManager;
    private TokenManager tokenManager;
    private MailSender mailSender;
    private SimpleMailMessage verifyResetTemplate;
    private Configuration freemarkerEngine;

    public Configuration getFreemarkerEngine() {
        return freemarkerEngine;
    }

    public void setFreemarkerEngine(Configuration freemarkerEngine) {
        this.freemarkerEngine = freemarkerEngine;
    }

    public SimpleMailMessage getVerifyResetTemplate() {
        return verifyResetTemplate;
    }

    public void setVerifyResetTemplate(SimpleMailMessage templateMessage) {
        this.verifyResetTemplate = templateMessage;
    }

    public MailSender getMailSender() {
        return mailSender;
    }

    public void setMailSender(MailSender mailSender) {
        this.mailSender = mailSender;
    }

    public String getUsername() {
        return username;
    }

    public void setUsername(String username) {
        this.username = username;
    }

    public UserManager getUserManager() {
        return userManager;
    }

    public void setUserManager(UserManager userManager) {
        this.userManager = userManager;
    }

    public TokenManager getTokenManager() {
        return tokenManager;
    }

    public void setTokenManager(TokenManager passwordManager) {
        this.tokenManager = passwordManager;
    }

    @Override
    protected String go() throws Exception {
        User user = null;
        try {
            user = userManager.getUser(username);
        } catch (NoSuchUserException e) {
            logger.debug("RequestPasswordReset: No such user in the system. Username = " + username);
            addActionError(String.valueOf(ActionErrorCode.ERROR_NO_SUCH_USER));
            return INPUT;
        }

        String token = tokenManager.createResetToken(user);
        if (token != null) {
            if (sendPasswordResetEmail(token, user) == false) {
                return ERROR;
            }
        } else {
            addActionError(String.valueOf(ActionErrorCode.ERROR_NO_RESET_TOKEN));
        }
        return SUCCESS;
    }

    private boolean sendPasswordResetEmail(String token, User user) {
        Map<String, Object> model = new HashMap<String, Object>();
        model.put("token", token);
        model.put("username", user.getUsername());
        model.put("gnizrConfiguration", getGnizrConfiguration());

        if (getVerifyResetTemplate() == null) {
            logger.error("RequestPasswordReset: templateMessge bean is not defined");
            addActionError(String.valueOf(ActionErrorCode.ERROR_CONFIG));
            return false;
        }
        String toEmail = user.getEmail();
        if (toEmail == null) {
            logger.error("RequestPasswordReset: the email of user " + user.getUsername() + " is not defined");
            addActionError(String.valueOf(ActionErrorCode.ERROR_EMAIL_UNDEF));
            return false;
        }
        SimpleMailMessage msg = new SimpleMailMessage(getVerifyResetTemplate());
        msg.setTo(toEmail);

        if (msg.getFrom() == null) {
            String contactEmail = getGnizrConfiguration().getSiteContactEmail();
            if (contactEmail != null) {
                msg.setFrom(contactEmail);
            } else {
                msg.setFrom("help@localhost");
            }
        }

        Template fmTemplate = null;
        String text = null;
        try {
            fmTemplate = freemarkerEngine.getTemplate("login/notifyreset-template.ftl");
            text = FreeMarkerTemplateUtils.processTemplateIntoString(fmTemplate, model);
        } catch (Exception e) {
            logger.error("RequestPasswordReset: error creating message template from Freemarker engine");
        }

        msg.setText(text);

        if (getMailSender() == null) {
            logger.error("RequestPasswordReset: mailSender bean is not defined");
            addActionError(String.valueOf(ActionErrorCode.ERROR_CONFIG));
            return false;
        }
        try {
            getMailSender().send(msg);
            return true;
        } catch (Exception e) {
            logger.error("RequestPasswordReset: send mail error. " + e);
            addActionError(String.valueOf(ActionErrorCode.ERROR_INTERNAL));
        }
        return false;
    }
}