package org.sunxin.lesson.jsp.ch24;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.util.Random;
import javax.imageio.ImageIO;

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

public class RandomCodeServlet extends HttpServlet
{
    //驗證碼圖片的寬度。
    private int width=60;
    //驗證碼圖片的高度。
    private int height=20;
    
    protected void service(HttpServletRequest req, HttpServletResponse resp)
                    throws ServletException, java.io.IOException
    {   
        BufferedImage buffImg=new BufferedImage(width,height,
                                                BufferedImage.TYPE_INT_RGB);
        Graphics2D g=buffImg.createGraphics();
        
        //創建一個隨機數生成器類。
        Random random=new Random();
        
        g.setColor(Color.WHITE);
        g.fillRect(0,0,width,height);
        
        //創建字體，字體的大小應該根據圖片的高度來定。
        Font font=new Font("Times New Roman",Font.PLAIN,18);
        //設置字體。
        g.setFont(font);
        
        //畫邊框。
        g.setColor(Color.BLACK);
        g.drawRect(0,0,width-1,height-1);
        
        //隨機產生160條干擾線，使圖象中的認證碼不易被其它程序探測到。
        g.setColor(Color.GRAY);
        for (int i=0;i<160;i++)
        {
            int x = random.nextInt(width);
            int y = random.nextInt(height);
            int xl = random.nextInt(12);
            int yl = random.nextInt(12);
            g.drawLine(x,y,x+xl,y+yl);
        }
        
        
        //randomCode用于保存隨機產生的驗證碼，以便用戶登錄后進行驗證。
        StringBuffer randomCode=new StringBuffer();
        int red=0,green=0,blue=0;
        
        //隨機產生4位數字的驗證碼。
        for (int i=0;i<4;i++)
        {
            String strRand=String.valueOf(random.nextInt(10));
            
            //產生隨機的顏色分量來構造顏色值，這樣輸出的每位數字的顏色值都將不同。
            red=random.nextInt(110);
            green=random.nextInt(50);
            blue=random.nextInt(50);
            
            //將驗證碼繪制到圖像中。
            g.setColor(new Color(red,green,blue));
            g.drawString(strRand,13*i+6,16);
            
            //將產生的四個隨機數組合在一起。
            randomCode.append(strRand);
        }
        //將四位數字的驗證碼保存到Session中。
        HttpSession session=req.getSession();
        session.setAttribute("randomCode",randomCode.toString());
        
        //禁止圖像緩存。
        resp.setHeader("Pragma","no-cache");
        resp.setHeader("Cache-Control","no-cache");
        resp.setDateHeader("Expires", 0);
        
        resp.setContentType("image/jpeg");
        
        //將圖像輸出到Servlet輸出流中。
        ServletOutputStream sos=resp.getOutputStream();
        ImageIO.write(buffImg, "jpeg",sos);
        sos.close();
    }
}