cn.dreampie.common.plugin.patchca.PatchcaRender Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jfinal-dreampie Show documentation
jfinal shiro-freemarker plugins
There is a newer version: 1.2.2
package cn.dreampie.common.plugin.patchca;

import cn.dreampie.common.config.AppConstants;
import cn.dreampie.common.plugin.patchca.color.ColorFactory;
import cn.dreampie.common.plugin.patchca.filter.ConfigurableFilterFactory;
import cn.dreampie.common.plugin.patchca.filter.library.WobbleImageOp;
import cn.dreampie.common.plugin.patchca.font.RandomFontFactory;
import cn.dreampie.common.plugin.patchca.service.Captcha;
import cn.dreampie.common.plugin.patchca.service.ConfigurableCaptchaService;
import cn.dreampie.common.plugin.patchca.text.renderer.BestFitTextRenderer;
import cn.dreampie.common.plugin.patchca.text.renderer.TextRenderer;
import cn.dreampie.common.plugin.patchca.word.RandomWordFactory;
import cn.dreampie.common.util.security.EncriptionUtils;
import com.jfinal.core.Controller;
import com.jfinal.kit.StrKit;
import com.jfinal.render.Render;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.session.Session;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

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

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

  private Logger logger = LoggerFactory.getLogger(getClass());
  private static final String CODE_CHAR = "0123456789";
  private static final int MIN_NUM = 4;
  private static final int MAX_NUM = 4;
  private static final int FONT_MIN_SIZE = 20;
  private static final int FONT_MAX_SIZE = 20;
  private static final double X_AMPLITUDE = 1.6;
  private static final double Y_AMPLITUDE = 0.8;
  private static final int TOP_MARGIN = 1;
  private static final int BOTTOM_MARGIN = 1;
  private static final int WIDTH = 118;
  private static final int HEIGHT = 41;

  private String captchaName = AppConstants.CAPTCHA_NAME;
  private float alpha = 1.0f;
  private ConfigurableCaptchaService configurableCaptchaService = null;
  private ColorFactory colorFactory = null;
  private RandomFontFactory fontFactory = null;
  private RandomWordFactory wordFactory = null;
  private TextRenderer textRenderer = null;

  public PatchcaRender() {
    this(MIN_NUM, MAX_NUM, WIDTH, HEIGHT, FONT_MIN_SIZE, FONT_MAX_SIZE, null);
  }

  public PatchcaRender(int num) {
    this(num, num, WIDTH, HEIGHT, FONT_MIN_SIZE, FONT_MAX_SIZE, null);
  }

  public PatchcaRender(int minnum, int maxnum, int width, int height, int fontsize) {
    this(minnum, maxnum, width, height, fontsize, fontsize, null);
  }

  public PatchcaRender(int minnum, int maxnum, int width, int height, int fontsize, String code) {
    this(minnum, maxnum, width, height, fontsize, fontsize, code);
  }

  public PatchcaRender(int minnum, int maxnum, int width, int height, int fontmin, int fontmax, String code) {
    if (minnum <= 0) {
      minnum = MIN_NUM;
    }
    if (maxnum <= 0) {
      maxnum = MAX_NUM;
    }
    if (width <= 0) {
      width = WIDTH;
    }
    if (height <= 0) {
      height = HEIGHT;
    }

    if (fontmin <= 0) {
      fontmin = FONT_MIN_SIZE;
    }

    if (fontmax <= 0) {
      fontmax = FONT_MAX_SIZE;
    }

    if (!(code != null && !code.isEmpty())) {
      code = CODE_CHAR;
    }

    configurableCaptchaService = new ConfigurableCaptchaService();

    // 颜色创建工厂,使用一定范围内的随机色
    //colorFactory = new RandomColorFactory();
    colorFactory = new ColorFactory() {

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

    configurableCaptchaService.setColorFactory(colorFactory);

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

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

    // 自定义验证码图片背景
    SimpleBackgroundFactory backgroundFactory = new SimpleBackgroundFactory(alpha);
    configurableCaptchaService.setBackgroundFactory(backgroundFactory);

    // 图片滤镜设置
    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);

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

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

  /**
   * 输出
   */
  public void render() {
    ServletOutputStream outputStream = null;

    // 得到验证码对象,有验证码图片和验证码字符串
    Captcha captcha = configurableCaptchaService.getCaptcha();
    // 取得验证码字符串放入Session
    String captchaCode = captcha.getChallenge();
    if (logger.isDebugEnabled()) {
      logger.debug("captcha:" + captchaCode);
    }
    //System.out.println(validationCode);
    Session session = SecurityUtils.getSubject().getSession();
    session.setAttribute(captchaName, EncriptionUtils.encrypt(captchaCode));
    session.setAttribute(captchaName + "_time", new Date().getTime());
//    CookieUtils.addCookie(request, response, AppConstants.CAPTCHA_NAME, EncriptionUtils.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 float getAlpha() {
    return alpha;
  }

  public void setAlpha(float alpha) {
    this.alpha = alpha;
  }
}