package org.cateproject.controller.captcha;
   
   import java.awt.image.BufferedImage;
   import java.io.ByteArrayOutputStream;
   import java.io.OutputStream;
   import java.util.Locale;
   
   import javax.imageio.ImageIO;
   import javax.servlet.http.HttpServletResponse;
  
  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 org.springframework.web.bind.annotation.RequestMethod;
  import org.springframework.web.servlet.ModelAndView;
  
  import com.octo.captcha.service.CaptchaServiceException;
  import com.octo.captcha.service.image.ImageCaptchaService;
  
  @Controller
  public class CaptchaController {
  	
  	private ImageCaptchaService imageCaptchaService;
  	
  	@Autowired
  	public void setImageCaptchaService(ImageCaptchaService imageCaptchaService) {
  		this.imageCaptchaService = imageCaptchaService;
  	}
  
  	@RequestMapping(value = "/captcha/{captchaId}",method = RequestMethod.GET)
  	public void handleRequest(@PathVariable(value = "captchaId") String captchaId, 
  			                          Locale locale, HttpServletResponse response) throws Exception {
  		byte[] captchaChallengeAsJpeg = null;
  	    // the output stream to render the captcha image as jpeg into
  	    ByteArrayOutputStream jpegOutputStream = new ByteArrayOutputStream();
  	    try {
  	        // call the ImageCaptchaService getChallenge method
  	        BufferedImage challenge = imageCaptchaService.getImageChallengeForID(captchaId,locale);
  	            
  	        // a jpeg encoder
  	        ImageIO.write(challenge, "jpeg", jpegOutputStream);
  	    } catch (IllegalArgumentException e) {
  	            response.sendError(HttpServletResponse.SC_BAD_REQUEST);
  	    } catch (CaptchaServiceException e) {
  	        response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
  	    }
  
  	    captchaChallengeAsJpeg = jpegOutputStream.toByteArray();
  
  	    // flush it in the response
  	    response.setHeader("Cache-Control", "no-store");
  	    response.setHeader("Pragma", "no-cache");
  	    response.setDateHeader("Expires", 0);
  	    response.setContentType("image/jpeg");
  	    OutputStream outputStream = response.getOutputStream();
  	    outputStream.write(captchaChallengeAsJpeg);
  	    outputStream.flush();
  	    outputStream.close();
  	}
  
  }