com.excilys.ebi.bank.web.controller.account.transfer.operations.TransferPerformController.java Source code

Introduction

Here is the source code for com.excilys.ebi.bank.web.controller.account.transfer.operations.TransferPerformController.java

Source

/**
 * Copyright 2011-2012 eBusiness Information, Groupe Excilys (www.excilys.com)
 *
 * 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.excilys.ebi.bank.web.controller.account.transfer.operations;

import static com.google.common.collect.Collections2.filter;

import java.math.BigDecimal;
import java.util.List;

import javax.validation.Valid;

import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.propertyeditors.CustomNumberEditor;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
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.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;

import com.excilys.ebi.bank.model.entity.Account;
import com.excilys.ebi.bank.service.BankService;
import com.excilys.ebi.bank.service.UnsufficientBalanceException;
import com.excilys.ebi.bank.web.interceptor.account.AccountModelAttribute;
import com.excilys.ebi.bank.web.interceptor.page.WebPage;
import com.excilys.ebi.bank.web.interceptor.page.WebPageModelAttribute;
import com.excilys.ebi.bank.web.messages.Message;
import com.excilys.ebi.bank.web.messages.MessageHelper;
import com.excilys.ebi.bank.web.security.SecurityUtils;
import com.excilys.ebi.utils.spring.log.slf4j.InjectLogger;
import com.google.common.base.Predicate;

@Controller
@RequestMapping("/private/bank/account/{accountNumber}/transfers/perform.html")
@AccountModelAttribute
@WebPageModelAttribute(WebPage.TRANSFER_PERFORM)
public class TransferPerformController {

    @InjectLogger
    private Logger logger;

    @Autowired
    private BankService bankService;

    @InitBinder
    public void initBinder(WebDataBinder binder) {
        // accept "," as a decimal separator
        binder.registerCustomEditor(BigDecimal.class, new CustomNumberEditor(BigDecimal.class, true) {
            @Override
            public void setAsText(String text) throws IllegalArgumentException {
                super.setAsText(StringUtils.replaceChars(text, ',', '.'));
            }
        });
    }

    @RequestMapping(method = RequestMethod.GET)
    public String displayForm(@PathVariable String accountNumber, final @ModelAttribute TransferCommand command,
            ModelMap model) {

        List<Account> accounts = bankService.findAccountsByUser(SecurityUtils.getCurrentUser());
        model.put("debitableAccounts", accounts);

        if (command.getDebitedAccountNumber() == null) {
            command.setDebitedAccountNumber(accountNumber);
        }

        model.put("creditableAccounts", filter(accounts, new Predicate<Account>() {
            @Override
            public boolean apply(Account input) {
                return !input.getNumber().equals(command.getDebitedAccountNumber());
            }
        }));

        return "private/bank/account/transfers/perform";
    }

    @RequestMapping(method = RequestMethod.POST)
    public String performTransfer(@PathVariable String accountNumber,
            @ModelAttribute @Valid TransferCommand command, BindingResult result, ModelMap model,
            RedirectAttributes redirectAttributes) {

        if (!result.hasErrors()) {
            try {
                Integer debitedAccountId = bankService.findAccountIdByNumber(command.getDebitedAccountNumber());
                Integer creditedAccountId = bankService.findAccountIdByNumber(command.getCreditedAccountNumber());
                bankService.performTransfer(debitedAccountId, creditedAccountId, command.getAmount());

                MessageHelper.addFlashMessage(redirectAttributes, new Message("message.info.transfer.success"));

                return "redirect:/private/bank/accounts.html";

            } catch (UnsufficientBalanceException e) {
                logger.info("insufficient amount");
                result.rejectValue("amount", "message.error.transfer.amount.unsufficientBalance");
            }
        }

        return displayForm(accountNumber, command, model);
    }
}