org.mifos.ui.core.controller.UploadLogoController.java Source code

Introduction

Here is the source code for org.mifos.ui.core.controller.UploadLogoController.java

Source

/*
 * Copyright (c) 2005-2011 Grameen Foundation USA
 *  All rights reserved.
 *
 *  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.
 *
 *  See also http://www.apache.org/licenses/LICENSE-2.0.html for an
 *  explanation of the license and how it is applied.
 */

package org.mifos.ui.core.controller;

import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;
import java.util.List;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.io.IOUtils;
import org.mifos.config.servicefacade.LogoServiceFacade;
import org.mifos.framework.util.ConfigurationLocator;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.MessageSource;
import org.springframework.core.io.Resource;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.validation.BindingResult;
import org.springframework.validation.ObjectError;
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.servlet.support.RequestContextUtils;

@Controller
@edu.umd.cs.findbugs.annotations.SuppressWarnings(value = { "RV_RETURN_VALUE_IGNORED_BAD_PRACTICE",
        "SE_BAD_FIELD" })
public class UploadLogoController {

    public static final String LOGO_DIRECTORY = "logo";

    @Autowired
    private LogoServiceFacade logoServiceFacade;

    @Autowired
    private MessageSource messageSource;

    @ModelAttribute("breadcrumbs")
    public List<BreadCrumbsLinks> showBreadCrumbs() {
        return new AdminBreadcrumbBuilder().withLink("admin.uploadNewLogo", "uploadNewLogo.ftl").build();
    }

    @RequestMapping(value = "/uploadNewLogo.ftl", method = RequestMethod.GET)
    public String showUploadNewLogoPage(Model model) {
        model.addAttribute(new LogoUpload());
        return "uploadNewLogo";
    }

    @RequestMapping(value = "/uploadNewLogo.ftl", method = RequestMethod.POST)
    public String uploadNewLogo(@ModelAttribute LogoUpload logoUpload, BindingResult result, Model model,
            HttpServletRequest request) {
        String[] availableContentTypes = { "image/png", "image/gif", "image/jpeg", "image/pjpeg" };
        if (Arrays.asList(availableContentTypes).contains(logoUpload.getFile().getContentType())) {
            try {
                logoServiceFacade.uploadNewLogo(logoUpload.getFile());
                model.addAttribute("success", true);
            } catch (IOException e) {
                result.addError(new ObjectError("logoUpload", messageSource
                        .getMessage("admin.uploadLogo.ioexception", null, RequestContextUtils.getLocale(request))));
            }
        } else {
            result.addError(new ObjectError("logoUpload", messageSource.getMessage("admin.uploadLogo.badType", null,
                    RequestContextUtils.getLocale(request))));
        }
        return "uploadNewLogo";
    }

    @RequestMapping("/getMifosLogo")
    public ResponseEntity<byte[]> getMifosLogo() throws IOException {
        ConfigurationLocator configurationLocator = new ConfigurationLocator();
        Resource logo = configurationLocator.getUploadedMifosLogo();
        InputStream in = logo.getInputStream();
        byte[] logoContent = IOUtils.toByteArray(in);
        in.close();

        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.IMAGE_PNG);

        return new ResponseEntity<byte[]>(logoContent, headers, HttpStatus.OK);
    }

}