cn.dreampie.captcha.CaptchaRender Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jfinal-captcha Show documentation
jfinal captcha plugins
The newest version!
package cn.dreampie.captcha;

import cn.dreampie.captcha.background.BackgroundFactory;
import cn.dreampie.captcha.color.ColorFactory;
import cn.dreampie.captcha.color.RandomColorFactory;
import cn.dreampie.captcha.filter.ConfigurableFilterFactory;
import cn.dreampie.captcha.filter.FilterFactory;
import cn.dreampie.captcha.filter.library.WobbleImageOp;
import cn.dreampie.captcha.filter.predefined.*;
import cn.dreampie.captcha.font.RandomFontFactory;
import cn.dreampie.captcha.service.Captcha;
import cn.dreampie.captcha.service.ConfigurableCaptchaService;
import cn.dreampie.captcha.text.renderer.BestFitTextRenderer;
import cn.dreampie.captcha.text.renderer.TextRenderer;
import cn.dreampie.captcha.word.RandomWordFactory;
import cn.dreampie.encription.EncriptionKit;
import com.jfinal.render.Render;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.imageio.ImageIO;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpSession;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.awt.image.BufferedImageOp;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Date;
import java.util.Random;

/**
 * Created by wangrenhui on 13-12-31.
 */
public class CaptchaRender extends Render {

  private Logger logger = LoggerFactory.getLogger(getClass());
  private String code = "ABCDEFGHIJKLMNPQRSTUVWXYZ123456789";
  private int font_min_num = 4;
  private int font_max_num = 4;
  private int font_min_size = 20;
  private int font_max_size = 20;
  private double x_amplitude = 1.6;
  private double y_amplitude = 0.8;
  private int top_margin = 1;
  private int bottom_margin = 1;
  private int width = 118;
  private int height = 41;

  private String captchaName = "captcha";
  private ConfigurableCaptchaService configurableCaptchaService = new ConfigurableCaptchaService();
  private ColorFactory colorFactory = null;
  private RandomFontFactory fontFactory = null;
  private RandomWordFactory wordFactory = null;
  private TextRenderer textRenderer = null;
  private BackgroundFactory backgroundFactory = null;
  //滤镜特效
  private FilterFactory filter = null;
  private static Random random = new Random();
  /**
   * 背景色
   */
  private Color bgColor = null;

  /**
   * 验证码字符颜色
   */
  private Color drawColor = new Color(0, 0, 0);
  /**
   * 背景元素的颜色
   */
  private Color drawBgColor = new Color(102, 102, 102);

  private boolean randomColor = false;
  /**
   * 噪点数量
   */
  private int artifactNum = 50;

  private int lineNum = 0;

  private void initCaptchService() {

    // 颜色创建工厂,使用一定范围内的随机色

    if (randomColor)
      colorFactory = new RandomColorFactory();
    else
      colorFactory = new ColorFactory() {

        public Color getColor(int index) {
          return drawColor;//new Color(118,102,102);
        }
      };

    configurableCaptchaService.setColorFactory(colorFactory);

    // 随机字体生成器
    fontFactory = new RandomFontFactory();
    fontFactory.setMaxSize(font_max_size);
    fontFactory.setMinSize(font_min_size);
    configurableCaptchaService.setFontFactory(fontFactory);

    // 随机字符生成器,去除掉容易混淆的字母和数字,如o和0等
    wordFactory = new RandomWordFactory();
    wordFactory.setCharacters(code);
    wordFactory.setMaxLength(font_max_num);
    wordFactory.setMinLength(font_min_num);
    configurableCaptchaService.setWordFactory(wordFactory);

    // 自定义验证码图片背景
    if (backgroundFactory == null) {
      backgroundFactory = new SimpleBackgroundFactory(bgColor, randomColor ? null : drawBgColor, artifactNum, lineNum);
    }
    configurableCaptchaService.setBackgroundFactory(backgroundFactory);

    // 图片滤镜设置
    int filterNum;
    if (filter == null) {
      filterNum = random.nextInt(4);
    } else
      filterNum = filter.value();

    switch (filterNum) {
      case 0:
        configurableCaptchaService.setFilterFactory(new CurvesRippleFilterFactory(configurableCaptchaService.getColorFactory()));
        break;
      case 1:
        configurableCaptchaService.setFilterFactory(new MarbleRippleFilterFactory());
        break;
      case 2:
        configurableCaptchaService.setFilterFactory(new DoubleRippleFilterFactory());
        break;
      case 3:
        configurableCaptchaService.setFilterFactory(new WobbleRippleFilterFactory());
        break;
      case 4:
        configurableCaptchaService.setFilterFactory(new DiffuseRippleFilterFactory());
        break;
      default:
        //默认效果
        configurableCaptchaService.setFilterFactory(new CurvesRippleFilterFactory(configurableCaptchaService.getColorFactory()));
//        ConfigurableFilterFactory filterFactory = new ConfigurableFilterFactory();
//
//        java.util.List filters = new ArrayList();
//        WobbleImageOp wobbleImageOp = new WobbleImageOp();
//        wobbleImageOp.setEdgeMode(BufferedImage.TYPE_INT_ARGB);
//        wobbleImageOp.setxAmplitude(x_amplitude);
//        wobbleImageOp.setyAmplitude(y_amplitude);
//        filters.add(wobbleImageOp);
//        filterFactory.setFilters(filters);
//
//        configurableCaptchaService.setFilterFactory(filterFactory);
        break;
    }

    // 文字渲染器设置
    textRenderer = new BestFitTextRenderer();
    textRenderer.setBottomMargin(bottom_margin);
    textRenderer.setTopMargin(top_margin);
    configurableCaptchaService.setTextRenderer(textRenderer);

    // 验证码图片的大小
    configurableCaptchaService.setWidth(width);
    configurableCaptchaService.setHeight(height);
  }

  /**
   * you can  rewrite this  render
   * 输出
   */
  public void render() {
    //初始化
    initCaptchService();
    ServletOutputStream outputStream = null;

    // 得到验证码对象,有验证码图片和验证码字符串
    Captcha captcha = configurableCaptchaService.getCaptcha();
    // 取得验证码字符串放入Session
    String captchaCode = captcha.getChallenge();
    if (logger.isDebugEnabled()) {
      logger.debug("captcha:" + captchaCode);
    }
    //System.out.println(validationCode);
    HttpSession session = request.getSession();
    session.setAttribute(captchaName, EncriptionKit.encrypt(captchaCode.toLowerCase()));
    session.setAttribute(captchaName + "_time", new Date().getTime());
//    CookieUtils.addCookie(request, response, AppConstants.CAPTCHA_NAME, EncriptionKit.encrypt(captchaCode), -1);
    // 取得验证码图片并输出
    BufferedImage bufferedImage = captcha.getImage();

    try {
      outputStream = response.getOutputStream();
      ImageIO.write(bufferedImage, "png", outputStream);
    } catch (Exception e) {
      throw new RuntimeException(e);
    } finally {
      if (outputStream != null)
        try {
          outputStream.flush();
          outputStream.close();
        } catch (IOException e) {
          e.printStackTrace();
        }
    }

  }

  public String getCaptchaName() {
    return captchaName;
  }

  public void setCaptchaName(String captchaName) {
    this.captchaName = captchaName;
  }

  public BackgroundFactory getBackgroundFactory() {
    return backgroundFactory;
  }

  public void setBackgroundFactory(BackgroundFactory backgroundFactory) {
    this.backgroundFactory = backgroundFactory;
  }

  public Color getDrawColor() {
    return drawColor;
  }

  public void setDrawColor(Color drawColor) {
    this.drawColor = drawColor;
  }

  public Color getDrawBgColor() {
    return drawBgColor;
  }

  public void setDrawBgColor(Color drawBgColor) {
    this.drawBgColor = drawBgColor;
  }

  public Color getBgColor() {
    return bgColor;
  }

  public void setBgColor(Color bgColor) {
    this.bgColor = bgColor;
  }

  public void setFontNum(int font_min_num, int font_max_num) {
    this.font_min_num = font_min_num;
    this.font_max_num = font_max_num;
  }


  public void setFontSize(int font_min_size, int font_max_size) {
    this.font_min_size = font_min_size;
    this.font_max_size = font_max_size;
  }

  public void setFontMargin(int top_margin, int bottom_margin) {
    this.top_margin = top_margin;
    this.bottom_margin = bottom_margin;
  }

  public void setImgSize(int width, int height) {
    this.width = width;
    this.height = height;
  }

  public void setArtifactNum(int artifactNum) {
    this.artifactNum = artifactNum;
  }

  public void setLineNum(int lineNum) {
    this.lineNum = lineNum;
  }

  public void setCode(String code) {
    this.code = code;
  }

  public void setFilter(FilterFactory filter) {
    this.filter = filter;
  }

  public void setRandomColor(boolean randomColor) {
    this.randomColor = randomColor;
  }

  public enum FilterFactory {
    //曲面
    Curves(0),
    //大理石纹
    Marble(1),
    //对折
    Double(2),
    //颤动
    Wobble(3),
    //扩散
    Diffuse(4);
    private int value;

    private FilterFactory(int value) {
      this.value = value;
    }

    public int value() {
      return this.value;
    }
  }


  public static void main(String[] args) {
    int imgWidth = 400;
    int imgHeight = 300;
    File file = new File("/home/ice/图片/test.png");
    BufferedImage image = new BufferedImage(imgWidth, imgHeight,
        BufferedImage.TYPE_INT_ARGB);//RGB形式
    BackgroundFactory bf = new SimpleBackgroundFactory(new Color(255, 255, 0, 100));
    bf.fillBackground(image);
    try {
      ImageIO.write(image, "png", file);
    } catch (IOException e) {
      e.printStackTrace();
    }
  }
}