package com;
import java.io.IOException;
import java.io.PrintWriter;
import java.net.URLEncoder;
import java.util.Date;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import com.sun.xml.internal.messaging.saaj.packaging.mime.internet.MimeUtility;
/**
 * 文件下载类。为了防止客户端浏览器直接打开目标文件（例如在装了MS Office套件的Windows中的IE浏览器可能就会直接在IE浏览器中打开你想下载的doc或者xls文件），在响应头里加入强制下载的MIME类型。
 */
public class DownloadFile extends HttpServlet {
  private static final Log log = LogFactory.getLog(DownloadFile.class);
  public void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException {
    long timeStart = 0;
    if(log.isDebugEnabled()){
      timeStart=System.currentTimeMillis();
    }
    response.setContentType("application/x-download charset=UTF-8");
    java.io.FileInputStream fis = null;
    String filepath = request.getRealPath("");
    javax.servlet.ServletOutputStream sos = null;
    // System.out.println("DownloadFile filename:" + filename);
    try {
      if(request.getParameter("filename")==null
          ||request.getParameter("showName")==null){
        return;
      }
      String filename = request.getParameter("filename");
      String showName = request.getParameter("showName");
      request.setCharacterEncoding("UTF-8");
      response.setCharacterEncoding("UTF-8");
      java.io.File file = new java.io.File(filepath + filename);
      if (!file.exists()) {
        log.error(file.getAbsolutePath() + " 文件不存在!");
        return;
      }
      // 读取文件流
      fis = new java.io.FileInputStream(file);
      // 设置下载保存的文件名
      sos = response.getOutputStream();
      showName += filename.substring(filename.lastIndexOf("."));
      String contentDisposition = "", browser = getBrowser(request);
      if ("IE".equals(browser)) {
        contentDisposition = "attachment; filename=" + URLEncoder.encode(showName, "UTF-8").replace("+", "%20");
      } else if ("CH".equals(browser)) {
        contentDisposition = "attachment; filename=" + MimeUtility.encodeText(showName, "UTF8", "B");
      } else if ("SF".equals(browser)) {
        contentDisposition = "attachment; filename=" + new String(showName.getBytes("UTF-8"), "ISO8859-1");
      } else {
        contentDisposition = "attachment; filename*=UTF-8''" + URLEncoder.encode(showName, "UTF-8").replace("+", "%20");
      }
      response.setHeader("Content-Disposition", contentDisposition);
      int byteCount = 0; 
      if (fis != null) {
        byte[] buff = new byte[1024];
        int bytesRead;
        while(-1 != (bytesRead = fis.read(buff, 0, buff.length))) {
          sos.write(buff, 0, bytesRead);
          sos.flush();
          byteCount += bytesRead;
        } 
      }
      sos.flush();
      if(log.isDebugEnabled()){
        log.debug("文件下载完成，文件大小："+ byteCount +"，总共用时:"+ (new Date().getTime() - timeStart) +"毫秒。");
      }
    } catch(IOException ioe) {
      ioe.printStackTrace();
    } finally {
      try {
        if(fis!=null){
          fis.close();
        }
      } catch (IOException e) {
      } finally {
        try {
          if(sos!=null){
            sos.close();
          }
        } catch (IOException e) {
        }
      }
    }
  }
  public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
    response.setContentType("text/html");
    PrintWriter out = response.getWriter();
    out.println("!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Transitional//EN\" \"http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd\">");
    out.println("html>");
    out.println(" head>");
    out.println("  title>文件下载/title>");
    out.println("  meta http-equiv=\"Content-Type\" content=\"text/html; charset=UTF-8\" />");
    out.println(" /head>");
    out.println(" body>");
    out.print(" This is ");
    out.print(this.getClass().getName());
    out.println(", using the POST method");
    out.println(" /body>");
    out.println("/html>");
    out.flush();
    out.close();
  }
  private String getBrowser(HttpServletRequest request) {
    String userAgent = request.getHeader("USER-AGENT").toLowerCase();
    if (userAgent != null) {
      if (userAgent.indexOf("msie") >= 0) {
        return "IE";
      } else if (userAgent.indexOf("mozilla") >= 0) {
        return "FF";
      } else if (userAgent.indexOf("applewebkit") >= 0) {
        return "CH";
      } else if (userAgent.indexOf("safari") >= 0) {
        return "SF";
      } else if (userAgent.indexOf("opera") >= 0) {
        return "OP";
      }
    }
    return null;
  }
}