EncryptController.java

/**
 * Copyright 2012, Board of Regents of the University of
 * Wisconsin System. See the NOTICE file distributed with
 * this work for additional information regarding copyright
 * ownership. Board of Regents of the University of Wisconsin
 * System licenses this file to you 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 edu.wisc.doit.tcrypt.controller;

import java.io.IOException;
import java.io.InputStream;
import java.util.Set;
import java.util.TreeSet;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.FilenameUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.ExceptionHandler;
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.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.servlet.ModelAndView;

import edu.wisc.doit.tcrypt.FileEncrypter;
import edu.wisc.doit.tcrypt.dao.IKeysKeeper;
import edu.wisc.doit.tcrypt.exception.ServiceErrorException;
import edu.wisc.doit.tcrypt.exception.ValidationException;
import edu.wisc.doit.tcrypt.vo.ServiceKey;

@Controller
public class EncryptController extends BaseController {

	private final IKeysKeeper keysKeeper;
	
	@Autowired
	public EncryptController(IKeysKeeper keysKeeper) {
		this.keysKeeper = keysKeeper;
	}
	
	//Request actions
	
	@RequestMapping(value = "/encrypt", method = RequestMethod.GET)
	public ModelAndView encryptTextInit() throws Exception {
		ModelAndView modelAndView = new ModelAndView("encryptToken");
		modelAndView.addObject("serviceNames", keysKeeper.getListOfServiceNames());
		return modelAndView;
	}
	
	@RequestMapping(value = "/encrypt/{selectedServiceName:.*}", method = RequestMethod.GET)
	public ModelAndView encryptTextInit(@PathVariable String selectedServiceName) throws Exception {
		ModelAndView modelAndView = encryptTextInit();
		modelAndView.addObject("selectedServiceName",selectedServiceName);
		return modelAndView;
	}
	
	@RequestMapping(value = "/encryptionServices", method = RequestMethod.GET)
	public @ResponseBody Set<String> getServiceNamesInJSON()
	{
		final TreeSet<String> serviceNames = new TreeSet<>(String.CASE_INSENSITIVE_ORDER);
		serviceNames.addAll(keysKeeper.getListOfServiceNames());
        return serviceNames;
	}
	
	@RequestMapping(value = "/encryptFile", method = RequestMethod.POST) 
	public ModelAndView encryptFile(@RequestParam("fileToEncrypt") MultipartFile file, @RequestParam("selectedServiceName") String serviceName, HttpServletResponse response) throws Exception {
	    if (file.isEmpty()) {
    		ModelAndView modelAndView = encryptTextInit();
    		modelAndView.addObject("selectedServiceName",serviceName);
    		return modelAndView;
	    }
		
	    final FileEncrypter fileEncrypter = this.getFileEncrypter(serviceName);
	    final String filename = FilenameUtils.getName(file.getOriginalFilename());
	    
	    response.setHeader("Content-Type", "application/x-tar");
	    response.setHeader("Content-Disposition", "attachment; filename=\"" + filename + ".tar" + "\"");

	    final long size = file.getSize();
	    try (   final InputStream inputStream = file.getInputStream();
	            final ServletOutputStream outputStream = response.getOutputStream()) {

	        fileEncrypter.encrypt(filename, (int)size, inputStream, outputStream);
	    }
		
	    return null;
	}
	
	//Exception Handlers
	
	@ExceptionHandler(ValidationException.class)
	public ModelAndView handleException(ValidationException e) throws Exception {
		ModelAndView mav;
		try {
			mav = encryptTextInit();
		} catch (Exception e1) {
			logger.error("Error resetting view after error",e);
			throw new Exception(e);
		}
		
		mav.addObject(e.getMessage());
		
		return mav;
	}
	
	@ExceptionHandler(ServiceErrorException.class)
	public ModelAndView handleException(ServiceErrorException e) throws Exception {
		ModelAndView mav;
		try {
			mav = encryptTextInit();
		} catch (Exception e1) {
			logger.error("Error resetting view after error",e);
			throw new Exception(e);
		}
		
		mav.addObject("errorMessage",e.getMessage());
		mav.addObject("zero",e.getServiceName());
		
		return mav;
	}
	


    private FileEncrypter getFileEncrypter (String serviceName) throws ServiceErrorException, IOException {
        ServiceKey sk = keysKeeper.getServiceKey(serviceName);
        if (sk == null) {
            throw new ServiceErrorException(serviceName, "error.serviceKeyNotFound");
        }
        
        return sk.getFileEncrypter();
    }
}