/**
|
* Copyright 2018-2020 stylefeng & fengshuonan (https://gitee.com/stylefeng)
|
* <p>
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
* <p>
|
* http://www.apache.org/licenses/LICENSE-2.0
|
* <p>
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
*/
|
package cn.stylefeng.guns.sys.modular.system.controller;
|
|
import cn.stylefeng.guns.base.consts.ConstantsContext;
|
import cn.stylefeng.roses.core.util.FileUtil;
|
import com.google.code.kaptcha.Constants;
|
import com.google.code.kaptcha.Producer;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.web.bind.annotation.PathVariable;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import javax.imageio.ImageIO;
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import javax.servlet.http.HttpSession;
|
import java.awt.image.BufferedImage;
|
import java.io.IOException;
|
|
/**
|
* 验证码生成
|
*
|
* @author fengshuonan
|
* @date 2017-05-05 23:10
|
*/
|
@Controller
|
@RequestMapping("/kaptcha")
|
public class KaptchaController {
|
|
@Autowired
|
private Producer producer;
|
|
/**
|
* 生成验证码
|
*/
|
@RequestMapping("")
|
public void index(HttpServletRequest request, HttpServletResponse response) {
|
HttpSession session = request.getSession();
|
|
response.setDateHeader("Expires", 0);
|
|
// Set standard HTTP/1.1 no-cache headers.
|
response.setHeader("Cache-Control", "no-store, no-cache, must-revalidate");
|
|
// Set IE extended HTTP/1.1 no-cache headers (use addHeader).
|
response.addHeader("Cache-Control", "post-check=0, pre-check=0");
|
|
// Set standard HTTP/1.0 no-cache header.
|
response.setHeader("Pragma", "no-cache");
|
|
// return a jpeg
|
response.setContentType("image/jpeg");
|
|
// create the text for the image
|
String capText = producer.createText();
|
|
// store the text in the session
|
session.setAttribute(Constants.KAPTCHA_SESSION_KEY, capText);
|
|
// create the image with the text
|
BufferedImage bi = producer.createImage(capText);
|
ServletOutputStream out = null;
|
try {
|
out = response.getOutputStream();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
|
// write the data out
|
try {
|
ImageIO.write(bi, "jpg", out);
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
try {
|
try {
|
out.flush();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
} finally {
|
try {
|
out.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
|
/**
|
* 返回图片
|
*
|
* @author stylefeng
|
* @Date 2017/5/24 23:00
|
*/
|
@RequestMapping("/{pictureId}")
|
public void renderPicture(@PathVariable("pictureId") String pictureId, HttpServletResponse response) {
|
String path = ConstantsContext.getFileUploadPath() + pictureId;
|
try {
|
byte[] bytes = FileUtil.toByteArray(path);
|
response.getOutputStream().write(bytes);
|
} catch (Exception e) {
|
//如果找不到图片就返回一个默认图片
|
try {
|
response.sendRedirect("/static/img/girl.gif");
|
} catch (IOException e1) {
|
e1.printStackTrace();
|
}
|
}
|
}
|
}
|
