EDIT

/**

* Copyright (C) 2009 EDIT

* European Distributed Institute of Taxonomy 

* http://www.e-taxonomy.eu

* 

* The contents of this file are subject to the Mozilla Public License Version 1.1

* See LICENSE.TXT at the top of this package for the full license terms.

*/

package eu.etaxonomy.cdm.remote.controller.checklist;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.InputStreamReader;

import java.io.PrintWriter;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Collections;

import java.util.List;

import java.util.Set;

import java.util.UUID;

import javax.servlet.http.Cookie;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import org.apache.log4j.Logger;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.beans.propertyeditors.UUIDEditor;

import org.springframework.context.ApplicationContext;

import org.springframework.stereotype.Controller;

import org.springframework.web.bind.WebDataBinder;

import org.springframework.web.bind.annotation.InitBinder;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RequestMethod;

import org.springframework.web.bind.annotation.RequestParam;

import eu.etaxonomy.cdm.api.service.IClassificationService;

import eu.etaxonomy.cdm.api.service.IService;

import eu.etaxonomy.cdm.api.service.ITaxonNodeService;

import eu.etaxonomy.cdm.api.service.ITaxonService;

import eu.etaxonomy.cdm.api.service.ITermService;

import eu.etaxonomy.cdm.common.CdmUtils;

import eu.etaxonomy.cdm.hibernate.HibernateProxyHelper;

import eu.etaxonomy.cdm.io.common.CdmApplicationAwareDefaultExport;

import eu.etaxonomy.cdm.io.csv.redlist.out.CsvTaxExportConfiguratorRedlist;

import eu.etaxonomy.cdm.model.description.Feature;

import eu.etaxonomy.cdm.model.location.NamedArea;

import eu.etaxonomy.cdm.model.taxon.Classification;

import eu.etaxonomy.cdm.model.taxon.Taxon;

import eu.etaxonomy.cdm.remote.controller.AbstractController;

import eu.etaxonomy.cdm.remote.controller.ProgressMonitorController;

import eu.etaxonomy.cdm.remote.editor.UUIDListPropertyEditor;

import eu.etaxonomy.cdm.remote.editor.UuidList;

/**

 * @author a.oppermann

 * @created 20.09.2012

 * 

 */

@Controller

@RequestMapping(value = { "/csv" })

public class CsvExportController extends AbstractController{

	/**

	 * 

	 */

	@Autowired

	private ApplicationContext appContext;

	@Autowired 

	private ITermService termService;

	@Autowired 

	private IClassificationService classificationService;

	@Autowired 

	private ITaxonNodeService taxonNodeService;

	@Autowired 

	private ITaxonService taxonService;

	@Autowired

	public ProgressMonitorController progressMonitorController;

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

	/**

	 * Helper method, which allows to convert strings directly into uuids.

	 * 

	 * @param binder Special DataBinder for data binding from web request parameters to JavaBean objects.

	 */

    @InitBinder

    public void initBinder(WebDataBinder binder) {

        binder.registerCustomEditor(UuidList.class, new UUIDListPropertyEditor());

//        binder.registerCustomEditor(NamedArea.class, new NamedAreaPropertyEditor());

        binder.registerCustomEditor(UUID.class, new UUIDEditor());

    }

    /**

     * Fetches data from the application context and forwards the stream to the HttpServletResponse, which offers a file download.

     *

     * @param featureUuids List of uuids to download/select {@link Feature feature}features

     * @param taxonUuid Selected {@link Classification classification} to iterate the {@link Taxon}

     * @param response HttpServletResponse which returns the ByteArrayOutputStream

     */

	@RequestMapping(value = { "exportRedlist" }, method = { RequestMethod.POST })

	public void doExportRedlist(

			@RequestParam(value = "features", required = false) UuidList featureUuids,

			@RequestParam(value = "classificationUuid", required = false) String classificationUuid,

			@RequestParam(value = "taxonUuid", required = false) String taxonUuid,

            @RequestParam(value = "area", required = false) UuidList areas,

			@RequestParam(value = "downloadTokenValueId", required = false) String downloadTokenValueId,

			HttpServletResponse response,

			HttpServletRequest request) {

		ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();

		UUID taxonNodeUuid = null;

		Classification classification = classificationService.load(UUID.fromString(classificationUuid));

		if(CdmUtils.isNotBlank(taxonUuid)){

			Taxon taxon = HibernateProxyHelper.deproxy(taxonService.load(UUID.fromString(taxonUuid), Arrays.asList(new String[]{"taxonNodes.classification"})), Taxon.class);

			taxonNodeUuid = classification.getNode(taxon).getUuid();

		}

		else{

			taxonNodeUuid = classification.getRootNode().getUuid();

		}

		CsvTaxExportConfiguratorRedlist config = setTaxExportConfigurator(taxonNodeUuid, featureUuids, areas, byteArrayOutputStream);

		CdmApplicationAwareDefaultExport<?> defaultExport = (CdmApplicationAwareDefaultExport<?>) appContext.getBean("defaultExport");

		logger.info("Start export...");

		logger.info("doExportRedlist()" + requestPathAndQuery(request));

		defaultExport.invoke(config);

		try {

			/*

			 *  Fetch data from the appContext and forward stream to HttpServleResponse

			 *  

			 *  FIXME: Large Data could be out of memory

			 *  

			 *  HTPP Error Break

			 */

			ByteArrayInputStream bais = new ByteArrayInputStream(byteArrayOutputStream.toByteArray());//byteArrayOutputStream.toByteArray()

			InputStreamReader isr = new InputStreamReader(bais);

			Cookie progressCookie = new Cookie("fileDownloadToken", downloadTokenValueId);

			progressCookie.setPath("/");

			progressCookie.setMaxAge(60);

			response.addCookie(progressCookie);

			response.setContentType("text/csv; charset=utf-8");

			response.setHeader("Content-Disposition", "attachment; filename=\""+config.getClassificationTitleCache()+".txt\"");

			PrintWriter printWriter = response.getWriter();

			int i;

			while((i = isr.read())!= -1){

				printWriter.write(i);

			}

			byteArrayOutputStream.flush();

			isr.close();

			byteArrayOutputStream.close();

			printWriter.flush();

			printWriter.close();

		} catch (Exception e) {

			logger.error("error generating feed", e);

		}

	}

	/**

	 * Cofiguration method to set the configuration details for the defaultExport in the application context.

	 * 

	 * @param taxonNodeUuid pass-through the selected {@link Classification classification}

	 * @param featureUuids pass-through the selected {@link Feature feature} of a {@link Taxon}, in order to fetch it.

	 * @param areas 

	 * @param byteArrayOutputStream pass-through the stream to write out the data later.

	 * @return the CsvTaxExportConfiguratorRedlist config

	 */

	private CsvTaxExportConfiguratorRedlist setTaxExportConfigurator(UUID taxonNodeUuid, UuidList featureUuids, UuidList areas, ByteArrayOutputStream byteArrayOutputStream) {

		@SuppressWarnings({ "unchecked", "rawtypes" })

		Set<UUID> taxonNodeUuids = Collections.singleton(taxonNodeUuid); 

		String destination = System.getProperty("java.io.tmpdir");

		List<Feature> features = new ArrayList<Feature>();

		if(featureUuids != null){

			for(UUID uuid : featureUuids) {

				features.add((Feature) termService.find(uuid));

			}

		}

		List<NamedArea> selectedAreas = new ArrayList<NamedArea>();

		if(areas != null){

			for(UUID area:areas){

				logger.info(area);

				selectedAreas.add((NamedArea)termService.find(area));

			}

		}

		CsvTaxExportConfiguratorRedlist config = CsvTaxExportConfiguratorRedlist.NewInstance(null, new File(destination));

		config.setHasHeaderLines(true);

		config.setFieldsTerminatedBy("\t");

		config.setTaxonNodeUuids(taxonNodeUuids);

		config.setByteArrayOutputStream(byteArrayOutputStream);

		if(features != null)config.setFeatures(features);

        config.setNamedAreas(selectedAreas);

		return config;

	}

	/* (non-Javadoc)

	 * @see eu.etaxonomy.cdm.remote.controller.AbstractController#setService(eu.etaxonomy.cdm.api.service.IService)

	 */

	@Override

	public void setService(IService service) {

		// TODO Auto-generated method stub

	}