csns.web.controller.UserControllerS.java Source code

Introduction

Here is the source code for csns.web.controller.UserControllerS.java

Source

/*
 * This file is part of the CSNetwork Services (CSNS) project.
 * 
 * Copyright 2013-2014, Chengyu Sun (csun@calstatela.edu).
 * 
 * CSNS is free software: you can redistribute it and/or modify it under the
 * terms of the GNU Affero General Public License as published by the Free
 * Software Foundation, either version 3 of the License, or (at your option)
 * any later version.
 * 
 * CSNS is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for
 * more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with CSNS. If not, see http://www.gnu.org/licenses/agpl.html.
 */
package csns.web.controller;

import java.text.SimpleDateFormat;
import java.util.Date;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.propertyeditors.CustomDateEditor;
import org.springframework.security.authentication.encoding.PasswordEncoder;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.util.StringUtils;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.WebDataBinder;
import org.springframework.web.bind.annotation.InitBinder;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.SessionAttributes;
import org.springframework.web.bind.support.SessionStatus;
import org.springframework.web.multipart.MultipartFile;

import csns.model.core.File;
import csns.model.core.User;
import csns.model.core.dao.UserDao;
import csns.security.SecurityUtils;
import csns.util.FileIO;
import csns.util.ImageUtils;
import csns.web.validator.AddUserValidator;
import csns.web.validator.EditUserValidator;
import csns.web.validator.RegistrationValidator;

@Controller
@SessionAttributes("user")
@SuppressWarnings("deprecation")
public class UserControllerS {

    @Autowired
    private UserDao userDao;

    @Autowired
    private AddUserValidator addUserValidator;

    @Autowired
    private EditUserValidator editUserValidator;

    @Autowired
    private RegistrationValidator registrationValidator;

    @Autowired
    private PasswordEncoder passwordEncoder;

    @Autowired
    private FileIO fileIO;

    @Autowired
    private ImageUtils imageUtils;

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

    /*
     * The default Spring date property editor does not accept null or empty
     * string so we have to explicitly register one that does.
     */
    @InitBinder
    public void initBinder(WebDataBinder binder) {
        binder.registerCustomEditor(Date.class, new CustomDateEditor(new SimpleDateFormat("MM/dd/yyyy"), true));
    }

    @RequestMapping(value = "/user/add", method = RequestMethod.GET)
    public String add(ModelMap models) {
        models.put("user", new User());
        return "user/add";
    }

    @RequestMapping(value = "/user/add", method = RequestMethod.POST)
    public String add(@ModelAttribute User user,
            @RequestParam(value = "file", required = false) MultipartFile uploadedFile, BindingResult bindingResult,
            SessionStatus sessionStatus) {
        addUserValidator.validate(user, bindingResult);
        if (bindingResult.hasErrors())
            return "user/add";

        user.setUsername(user.getCin());
        user.setPassword(passwordEncoder.encodePassword(user.getCin(), null));
        if (!StringUtils.hasText(user.getPrimaryEmail()))
            user.setPrimaryEmail(user.getCin() + "@localhost");
        user.setTemporary(true);
        user = userDao.saveUser(user);

        logger.info(SecurityUtils.getUser().getUsername() + " created user " + user.getId());

        handleProfilePicture(user, uploadedFile);

        sessionStatus.setComplete();
        return "redirect:/user/view?id=" + user.getId();
    }

    @RequestMapping(value = "/user/edit", method = RequestMethod.GET)
    public String edit(@RequestParam Long id, ModelMap models) {
        models.put("user", userDao.getUser(id));
        return "user/edit";
    }

    @RequestMapping(value = "/user/edit", method = RequestMethod.POST)
    public String edit(@ModelAttribute User user,
            @RequestParam(value = "file", required = false) MultipartFile uploadedFile, BindingResult bindingResult,
            SessionStatus sessionStatus) {
        editUserValidator.validate(user, bindingResult);
        if (bindingResult.hasErrors())
            return "user/edit";

        String password = user.getPassword1();
        if (StringUtils.hasText(password))
            user.setPassword(passwordEncoder.encodePassword(password, null));
        user = userDao.saveUser(user);

        logger.info(SecurityUtils.getUser().getUsername() + " edit user " + user.getId());

        handleProfilePicture(user, uploadedFile);

        sessionStatus.setComplete();
        return "redirect:/user/view?id=" + user.getId();
    }

    private void handleProfilePicture(User user, MultipartFile uploadedFile) {
        if (uploadedFile == null || uploadedFile.isEmpty())
            return;

        File picture0 = fileIO.save(uploadedFile, user, true);
        File picture1 = imageUtils.resizeToProfilePicture(picture0);
        File picture2 = imageUtils.resizeToProfileThumbnail(picture0);
        user.setOriginalPicture(picture0);
        user.setProfilePicture(picture1);
        user.setProfileThumbnail(picture2);
        user = userDao.saveUser(user);

        logger.info("Saved profile picture for " + user.getUsername());
    }

    @RequestMapping(value = "/register", method = RequestMethod.GET)
    public String register(ModelMap models) {
        User user = SecurityUtils.getUser().clone();
        user.setUsername(null);
        user.setPrimaryEmail(null);
        models.put("user", user);
        return "register";
    }

    @RequestMapping(value = "/register", method = RequestMethod.POST)
    public String register(@ModelAttribute("user") User cmd,
            @RequestParam(value = "file", required = false) MultipartFile uploadedFile, BindingResult bindingResult,
            SessionStatus sessionStatus) {
        registrationValidator.validate(cmd, bindingResult);
        if (bindingResult.hasErrors())
            return "register";

        User user = userDao.getUser(SecurityUtils.getUser().getId());
        user.copySelfEditableFieldsFrom(cmd);
        user.setUsername(cmd.getUsername());
        user.setPassword(passwordEncoder.encodePassword(cmd.getPassword1(), null));
        user.setTemporary(false);
        userDao.saveUser(user);

        logger.info(user.getUsername() + " completed registration");

        handleProfilePicture(user, uploadedFile);

        sessionStatus.setComplete();
        return "redirect:/j_spring_security_logout";
    }

}