/*
 * 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.reports.servlet;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.OutputStream;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.lang.StringUtils;
import org.mifos.application.admin.servicefacade.ViewOrganizationSettingsServiceFacade;
import org.mifos.application.servicefacade.ApplicationContextProvider;
import org.mifos.reports.admindocuments.business.AdminDocumentBO;
import org.mifos.reports.business.ReportsBO;

public class ReportTemplateDownloadServlet extends HttpServlet {

	@Override
	protected void doGet(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		String realPath = request.getParameter("realPath");
		boolean isReportAnAdminDocument = false;
		if (StringUtils.isBlank(realPath)) {
			realPath = "report";
		} else {
			isReportAnAdminDocument = true;
		}

		File dir = new File(
				ApplicationContextProvider.getBean(ViewOrganizationSettingsServiceFacade.class).getUploadStorageDirectory(),
				realPath);
		String reportFileName = "";
		if (isReportAnAdminDocument) {
			reportFileName = ((AdminDocumentBO) request.getSession()
					.getAttribute("reportsBO")).getAdminDocumentIdentifier();
		} else {
			reportFileName = ((ReportsBO) request.getSession().getAttribute(
					"reportsBO")).getReportsJasperMap().getReportJasper();
		}

		File file = new File(dir, reportFileName);

		if (!file.exists()) {
			file = new File(new File(getServletContext().getRealPath("/")
					+ realPath), reportFileName);
		}
		try {
			BufferedInputStream is = new BufferedInputStream(
					new FileInputStream(file));
			response.setContentType("application/x-msdownload;charset=UTF-8");
			response.setHeader("Content-Disposition", "attachment; filename="
					+ reportFileName);
			OutputStream os = response.getOutputStream();
			byte[] buffer = new byte[4096];
			int bytesRead = 0;
			while ((bytesRead = is.read(buffer, 0, 4096)) != -1) {
				os.write(buffer, 0, bytesRead);
			}
			os.close();
			is.close();
		} catch (FileNotFoundException e) {
			response.sendRedirect("birtAdminDocumentUploadAction.do?method=getFileNotFoundPage");
		}
		request.getSession().setAttribute("reportsBO", null);
	}
}