package util;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.Image;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Random;

import javax.imageio.ImageIO;

public class Verification {
  private static final String ALPHABET = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz1234567890";
  
  /**
   * 生成一個(gè)寬為width， 高為height， 驗(yàn)證碼為code的圖片
   * @param width 圖片的寬
   * @param height 圖片的高
   * @param code 驗(yàn)證碼字符串
   * @return 返回圖片驗(yàn)證碼
   */
  public static BufferedImage getImage(int width, int height, String code){
    return getImage(width, height, code, 20);
  }
  /**
   * 生成一個(gè)寬為width， 高為height， 驗(yàn)證碼為code的圖片，圖片中干擾線的條數(shù)為lineCnt
   * @param width 圖片的寬
   * @param height 圖片的高
   * @param code 驗(yàn)證碼字符串
   * @param lineCnt 干擾線的條數(shù)，建議為10條左右，可根據(jù)結(jié)果適當(dāng)調(diào)整
   * @return 返回圖片驗(yàn)證碼
   */
  public static BufferedImage getImage(int width, int height, String code, int lineCnt){
    return createImage(width, height, code, lineCnt, 0.01);
  }
  /**
   * 生成一個(gè)寬為width， 高為height， 驗(yàn)證碼為code的圖片，圖片中干擾線的條數(shù)為lineCnt
   * 噪聲比為noiseRate，即圖片中噪音像素點(diǎn)的百分比
   * @param width 圖片的寬
   * @param height 圖片的高
   * @param code 驗(yàn)證碼字符串
   * @param lineCnt 干擾線的條數(shù)，建議為10條左右，可根據(jù)結(jié)果適當(dāng)調(diào)整
   * @param noiseRate 圖片中噪音像素點(diǎn)占總像素的百分比
   * @return 返回圖片驗(yàn)證碼
   */
  public static BufferedImage getImage(int width, int height, String code, int lineCnt, double noiseRate){
    return createImage(width, height, code, lineCnt, noiseRate);
  }
  
  /**
   * 
   * 生成一個(gè)寬為width， 高為height， 驗(yàn)證碼為code的圖片，圖片中干擾線的條數(shù)為lineCnt
   * 噪聲比為noiseRate，即圖片中噪音像素點(diǎn)的百分比
   * @param width 圖片的寬
   * @param height 圖片的高
   * @param code 驗(yàn)證碼字符串
   * @param lineCnt 干擾線的條數(shù)，建議為10條左右，可根據(jù)結(jié)果適當(dāng)調(diào)整
   * @param noiseRate 圖片中噪音像素點(diǎn)占總像素的百分比
   * @return 返回圖片驗(yàn)證碼
   */
  private static BufferedImage createImage(int width, int height, String code, int lineCnt, double noiseRate){
    int fontWidth = ((int)(width * 0.8)) / code.length();
    int fontHeight = (int)(height * 0.7);
    //為了在任意的width和height下都能生成良好的驗(yàn)證碼，
    //字體的大小為fontWdith何fontHeight中的小者，
    int fontSize = Math.min(fontWidth, fontHeight);
    //drawString時(shí)要用到
    int paddingX = (int) (width * 0.1);
    int paddingY = height - (height - fontSize) / 2;
    
    //創(chuàng)建圖像
    BufferedImage buffimg = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
    //獲得畫筆
    Graphics g = buffimg.getGraphics();
    //設(shè)置畫筆的顏色
    g.setColor(getRandColor(200, 255));
    //然后填充一個(gè)矩形，即設(shè)置背景色
    g.fillRect(0, 0, width, height);
    
    // 設(shè)置干擾線
    for (int i = 0; i < lineCnt; i++) {
        //隨機(jī)獲取干擾線的起點(diǎn)和終點(diǎn)
      int xs = (int)(Math.random() * width);
      int ys = (int)(Math.random() * height);
      int xe = (int)(Math.random() * width);
      int ye = (int)(Math.random() * height);
      g.setColor(getRandColor(1, 255));
      g.drawLine(xs, ys, xe, ye);
    }
    // 添加噪點(diǎn)
    int area = (int) (noiseRate * width * height);
    for(int i=0; i<area; ++i){
        int x = (int)(Math.random() * width);
        int y = (int)(Math.random() * height);
        buffimg.setRGB(x, y, (int)(Math.random() * 255));
    }
    //設(shè)置字體
    Font font = new Font("Ravie", Font.PLAIN, fontSize);
    g.setFont(font);
    
    for(int i=0; i<code.length(); ++i){
        String ch = code.substring(i, i+1);
        g.setColor(getRandColor(1, 199));
        g.drawString(ch, paddingX + fontWidth * i, paddingY);
    }
    return buffimg;
    
  }
  /**
   * 獲取隨機(jī)的顏色，r,g,b的取值在L到R之間
   * @param L 左區(qū)間
   * @param R 右區(qū)間
   * @return 返回隨機(jī)顏色值
   */
  private static Color getRandColor(int L, int R){
    if(L > 255)
      L = 255;
    if(R > 255)
      R = 255;
    if(L < 0)
      L = 0;
    if(R < 0)
      R = 0;
    int interval = R - L; 
    int r = L + (int)(Math.random() * interval);
    int g = L + (int)(Math.random() * interval);
    int b = L + (int)(Math.random() * interval);
    return new Color(r, g, b);
  }

  /**
   * 隨機(jī)生成若干個(gè)由大小寫字母和數(shù)字組成的字符串
   * @param len 隨機(jī)生成len個(gè)字符
   * @return 返回隨機(jī)生成的若干個(gè)由大小寫字母和數(shù)字組成的字符串
   */
  public static String getRandCode(int len){
    String code = "";
    for(int i=0; i<len; ++i){
      int index = (int)(Math.random() * ALPHABET.length());
      code = code + ALPHABET.charAt(index);
    }
    return code;
  }
  /**
   * 將圖片轉(zhuǎn)為byte數(shù)組
   * @param image 圖片
   * @return 返回byte數(shù)組
   * @throws IOException
   */
  public static byte[] getByteArray(BufferedImage image) throws IOException{
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    ImageIO.write(image, "png", baos);
    return baos.toByteArray();
    //ByteArrayOutputStream 不需要close
    
  }
}

protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
    HttpSession session = request.getSession();
    //隨機(jī)生成字符串，并寫入session
    String code = Verification.getRandCode(4);
    session.setAttribute("verification", code);
    BufferedImage image = util.Verification.getImage(100,30, code, 5);
    response.setContentType("image/png");
    
    OutputStream out = response.getOutputStream();
    out.write(util.Verification.getByteArray(image));
    out.flush();
    out.close();
    
  }
 

<%@page import="util.Verification"%>
<%@ page language="java" contentType="text/html; charset=UTF-8"
  pageEncoding="UTF-8"%>
<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">
<html>
<head>
<meta http-equiv="Content-Type" content="text/html; charset=UTF-8">

<title>Insert title here</title>

<script type="text/javascript">
    function refreshcode(){
      document.getElementById("verification").src= "/verificationCode/verificationCode?hehe="+Math.random();
    }
  </script>
</head>
<body>
  
  <form action="<%=request.getContextPath()+"/checkVerification" %>" method="post">
    驗(yàn)證碼:<input type="text" name="submitVerification">
    <img id="verification" alt="" title="看不清點(diǎn)擊刷新驗(yàn)證碼" src="<%=request.getContextPath()+"/verificationCode" %>"
    onclick="refreshcode()"><br>
    <input type="submit" name="submit" value="提交">
  </form>
  
</body>
</html>

protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
  HttpSession session = request.getSession();
  String verification = (String)session.getAttribute("verification");
  String submitVerification = request.getParameter("submitVerification");
  PrintWriter out = response.getWriter();
  if(verification!=null && submitVerification!=null){
   if(verification.toLowerCase().equals(submitVerification.toLowerCase())){
    out.println("yes!!!");
   }
   else{
    out.println("no!!!");
   }
   
  }
  else{
   out.println("no!!!");
  }
  session.removeAttribute("verification");//防止用戶重復(fù)提交表單

 }

 /**
  * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse response)
  */
 protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
  // TODO Auto-generated method stub
  doGet(request, response);
 }