package com.gonvan.component.captcha;
 
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;
 
import javax.imageio.ImageIO;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
 
/**
 * Created by yuerzm on 2016/3/14.
 */
public class CaptchaFactory {
 
  private static final char[]   CODE_SEQUENCE    = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
    .toCharArray();
  private static final int    DEFAULT_WIDTH    = 60;
  private static final int    DEFAULT_HEIGHT   = 20;
  private static final int    DEFAULT_CODE_LEN  = 4;
  private static final int    DEFAULT_CODE_X   = 13;
  private static final int    DEFAULT_CODE_Y   = 16;
  private static final int    DEFAULT_FONT_SIZE  = 18;
  private static final String   DEFAULT_FONT_FAMILY = "Times New Roman";
  private static CaptchaFactory  instance      = new CaptchaFactory();
  private int           width        = DEFAULT_WIDTH;            // 定義圖片的width
  private int           height       = DEFAULT_HEIGHT;            // 定義圖片的height
  private int           length       = DEFAULT_CODE_LEN;           // 定義圖片上顯示驗(yàn)證碼的個(gè)數(shù)
  private int           xx         = DEFAULT_CODE_X;            // 定義圖片上顯示驗(yàn)證碼x坐標(biāo)
  private int           yy         = DEFAULT_CODE_Y;            // 定義圖片上顯示驗(yàn)證碼y坐標(biāo)
  private int           fontSize      = DEFAULT_FONT_SIZE;          // 定義圖片上顯示驗(yàn)證碼的字體大小
  private String         fontFamily     = DEFAULT_FONT_FAMILY;         // 定義圖片上顯示驗(yàn)證碼的個(gè)數(shù)
 
  private CaptchaFactory() {
  }
 
  public static CaptchaFactory getInstance() {
    return instance;
  }
 
  /**
   * 配置寬高
   *
   * @param w
   * @param h
   * @return
   */
  public CaptchaFactory configWidthAndHeight(int w, int h) {
    instance.width = w;
    instance.height = h;
    return instance;
  }
 
  /**
   * 配置坐標(biāo)
   *
   * @param x
   * @param y
   * @return
   */
  public CaptchaFactory configXY(int x, int y) {
    instance.xx = x;
    instance.yy = y;
    return instance;
  }
 
  /**
   * 配置字體大小
   *
   * @param fontSize
   * @return
   */
  public CaptchaFactory configFontSize(int fontSize) {
    instance.fontSize = fontSize;
    return instance;
  }
 
  /**
   * 配置字體
   *
   * @param fontFamily
   * @return
   */
  public CaptchaFactory configFontSize(String fontFamily) {
    instance.fontFamily = fontFamily;
    return instance;
  }
 
  public void write(HttpServletRequest request, HttpServletResponse response) throws IOException {
    // 將四位數(shù)字的驗(yàn)證碼保存到Session中。
    Map captcha = generate();
    String randomCode = (String) captcha.get("captchaCode");
    BufferedImage buffImg = (BufferedImage) captcha.get("captchaImg");
 
    HttpSession session = request.getSession();
    session.setAttribute("code", randomCode);
 
    // 禁止圖像緩存。
    response.setHeader("Pragma", "no-cache");
    response.setHeader("Cache-Control", "no-cache");
    response.setDateHeader("Expires", 0);
    response.setContentType("image/jpeg");
 
    // 將圖像輸出到Servlet輸出流中。
    ServletOutputStream outputStream = response.getOutputStream();
    ImageIO.write(buffImg, "jpeg", outputStream);
    outputStream.close();
  }
 
  public Map<String, Object> generate() throws IOException {
    // 定義圖像buffer
    BufferedImage buffImg = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
    Graphics gd = buffImg.getGraphics();
    // 設(shè)定背景色
    gd.setColor(getRandColor(200, 250));
    gd.fillRect(0, 0, width, height);
    // 設(shè)定字體，字體的大小應(yīng)該根據(jù)圖片的高度來(lái)定。
    gd.setFont(new Font(fontFamily, Font.PLAIN, fontSize));
 
    // 創(chuàng)建一個(gè)隨機(jī)數(shù)生成器類
    Random random = new Random();
 
    // 隨機(jī)產(chǎn)生40條干擾線，使圖象中的認(rèn)證碼不易被其它程序探測(cè)到。
    gd.setColor(getRandColor(160, 200));
    for (int i = 0; i < 155; i++) {
      int x = random.nextInt(width);
      int y = random.nextInt(height);
      int xl = random.nextInt(12);
      int yl = random.nextInt(12);
      gd.drawLine(x, y, x + xl, y + yl);
    }
 
    // randomCode用于保存隨機(jī)產(chǎn)生的驗(yàn)證碼，以便用戶登錄后進(jìn)行驗(yàn)證。
    StringBuffer randomCode = new StringBuffer();
    int red = 0, green = 0, blue = 0;
 
    // 隨機(jī)產(chǎn)生 length 個(gè)驗(yàn)證碼。
    for (int i = 0; i < length; i++) {
      // 得到隨機(jī)產(chǎn)生的驗(yàn)證碼數(shù)字。
      String code = String.valueOf(CODE_SEQUENCE[random.nextInt(36)]);
      // 產(chǎn)生隨機(jī)的顏色分量來(lái)構(gòu)造顏色值，這樣輸出的每位數(shù)字的顏色值都將不同。
      red = random.nextInt(110);
      green = random.nextInt(110);
      blue = random.nextInt(110);
 
      // 用隨機(jī)產(chǎn)生的顏色將驗(yàn)證碼繪制到圖像中。
      gd.setColor(new Color(red + 20, green + 20, blue + 20));
      gd.drawString(code, i * xx + 6, yy);
 
      // 將產(chǎn)生的隨機(jī)數(shù)組合在一起。
      randomCode.append(code);
    }
    Map<String, Object> retval = new HashMap<>();
    retval.put("captchaCode", randomCode.toString());
    retval.put("captchaImg", buffImg);
    return retval;
  }
 
  /**
   * 給定范圍獲得隨機(jī)顏色
   *
   * @param fc
   *      最小值
   * @param bc
   *      最大值
   * @return Color
   */
  private Color getRandColor(int fc, int bc) {
    Random random = new Random();
    if (fc > 255)
      fc = 255;
    if (bc > 255)
      bc = 255;
    int r = fc + random.nextInt(bc - fc);
    int g = fc + random.nextInt(bc - fc);
    int b = fc + random.nextInt(bc - fc);
    return new Color(r, g, b);
  }
}