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 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;

/**
 * Created by wangrenhui on 13-12-31.
 */
public class PatchcaRender extends Render {
  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 static boolean use_code = true;

  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);
  }

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

  public PatchcaRender(int minnum, int maxnum, int width, int height) {
    if (minnum <= 0) {
      minnum = MIN_NUM;
    }
    if (maxnum <= 0) {
      maxnum = MAX_NUM;
    }
    if (width <= 0) {
      width = WIDTH;
    }
    if (height <= 0) {
      height = HEIGHT;
    }
    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(FONT_MAX_SIZE);
    fontFactory.setMinSize(FONT_MIN_SIZE);
    configurableCaptchaService.setFontFactory(fontFactory);

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

    // 自定义验证码图片背景
    SimpleBackgroundFactory backgroundFactory = new SimpleBackgroundFactory();
    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();

    //System.out.println(validationCode);
    Session session = SecurityUtils.getSubject().getSession();
    session.setAttribute(AppConstants.CAPTCHA_NAME, EncriptionUtils.encrypt(captchaCode));
//    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();
        }
    }

  }

  /**
   * 验证码验证
   *
   * @param controller controller
   * @param inputCode  valid code
   * @return boolean
   */
  public static boolean validate(Controller controller, String inputCode) {
    if (StrKit.isBlank(inputCode))
      return false;
    try {
      if (use_code) {
        inputCode = inputCode.toUpperCase();
        inputCode = EncriptionUtils.encrypt(inputCode);
        return inputCode.equals(controller.getCookie(AppConstants.CAPTCHA_NAME));
      }
    } catch (Exception e) {
      e.printStackTrace();
    }
    return false;
  }
}