package org.sunxin.lesson.jsp.ch22;

import java.io.*;
import java.sql.*;
import javax.naming.*;
import javax.servlet.*;
import javax.servlet.http.*;
import javax.sql.*;

public class DownloadServlet extends HttpServlet
{
    private DataSource ds=null;

    public void init() throws ServletException
    {
        try
        {
            Context ctx = new InitialContext();
            ds = (DataSource) ctx.lookup("java:comp/env/jdbc/bookstore");
        }
        catch (NamingException ex)
        {
            System.err.println(ex);
        }
    }

    public void doGet(HttpServletRequest request, HttpServletResponse response) throws
        ServletException, IOException
    {
        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;

        String strId = request.getParameter("id");
        if (null == strId || strId.equals(""))
        {
            return;
        }
        File file=null;
        InputStream is=null;
        String fileName=null;
        int fileSize=0;

        //對從硬盤上下載文件進行處理。
        if(strId.equals("111"))
        {
            file=new File("D:\\tomcat5.exe");
            is=new FileInputStream(file);
            fileName=file.getName();
            fileSize=(int)file.length();
        }
        //對從數據庫中下載文件進行處理。
        else
        {
            int id=Integer.parseInt(strId);
            try
            {
                conn = ds.getConnection();
                pstmt = conn.prepareStatement(
                    "select * from uploadfile where id=?");
                pstmt.setInt(1,id);
                rs = pstmt.executeQuery();
                if(rs.next())
                {
                    fileName=rs.getString("filename");
                    fileSize=rs.getInt("filesize");
                    is=rs.getBinaryStream("data");
                }
                else
                {
                    response.setContentType("text/html;charset=gb2312");
                    PrintWriter out=response.getWriter();
                    out.print("沒有找到要下載的文件，請聯系");
                    out.println("<a href=\"mailto:admin@sunxin.org\">管理員</a>");
                    out.close();
                    return;
                }
            }
            catch(SQLException e)
            {
                System.err.println(e);
            }
        }
        fileName=toUTF8String(fileName);
        //設置下載文件使用的報頭域。
        response.setContentType("application/x-msdownload");
        String str = "attachment; filename="+fileName;
        response.setHeader("Content-Disposition", str);
        response.setContentLength(fileSize);

        //得到響應對象的輸出流，用于向客戶端輸出二進制數據。
        ServletOutputStream sos=response.getOutputStream();
        byte[] data = new byte[2048];
        int len = 0;
        while((len=is.read(data)) >0)
        {
            sos.write(data,0,len); //向瀏覽器輸出文件數據
        }
        is.close(); //關閉輸入流
        sos.close();//關閉輸出流
        if(rs!=null)
        {
            try
            {
                rs.close();
            }
            catch(SQLException e)
            {
                System.err.println(e);
            }
            rs=null;
        }
        if(pstmt!=null)
        {
            try
            {
                pstmt.close();
            }
            catch(SQLException e)
            {
                System.err.println(e);
            }
            pstmt=null;
        }
        if(conn!=null)
        {
            try
            {
                conn.close();
            }
            catch(SQLException e)
            {
                System.err.println(e);
            }
            conn=null;
        }
    }

    public void doPost(HttpServletRequest request, HttpServletResponse response) throws
        ServletException, IOException
    {
        doGet(request, response);
    }

    public static String toUTF8String(String str)
    {
        StringBuffer sb = new StringBuffer();
        int len=str.length();

        for (int i=0;i<len;i++)
        {
            char c=str.charAt(i);
            if (c >= 0 && c <= 255)
            {
                sb.append(c);
            }
            else
            {
                byte[] b;
                try
                {
                    b = Character.toString(c).getBytes("UTF-8");
                }
                catch (UnsupportedEncodingException ex)
                {
                    System.err.println(ex);
                    b = null;
                }
                for (int j = 0; j < b.length; j++)
                {
                    int k = b[j];
                    if(k<0)
                    {
                        k &= 255;
                    }
                    sb.append("%" + Integer.toHexString(k).toUpperCase());
                }
            }
        }
        return sb.toString();
    }
}