package com.ms.ssm.controller;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;

import javax.imageio.ImageIO;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;

@Controller
public class ValidCodeController {

	@RequestMapping("validcode")
	public void ValidCode(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
			
			BufferedImage image = new BufferedImage(200, 100, BufferedImage.TYPE_INT_RGB);
			
			
			Graphics2D gra = image.createGraphics();
			
			gra.setColor(Color.WHITE);
			
			gra.fillRect(0, 0, 200, 100);
			
			List<Integer> randList = new ArrayList<Integer>();
			Random random =new Random();
			for (int i = 0 ;i<4;i++) {
				randList.add(random.nextInt(10));
			}
		
			gra.setFont(new Font("宋体",Font.ITALIC|Font.BOLD,40));
			Color[] colors = new Color[]{Color.RED,Color.YELLOW,Color.BLUE,Color.GREEN,Color.PINK,Color.GRAY};
			for (int i = 0; i < randList.size(); i++) {
				gra.setColor(colors[random.nextInt(colors.length)]);
				gra.drawString(randList.get(i)+"", i*40, 70+(random.nextInt(21)-10));
			}
			
			for (int i = 0; i < 2; i++) {
				gra.setColor(colors[random.nextInt(colors.length)]);
			
				gra.drawLine(0, random.nextInt(101), 200, random.nextInt(101));
			}
			
			ServletOutputStream outputStream = resp.getOutputStream();
			
			ImageIO.write(image, "jpg", outputStream);
			
		
			HttpSession session = req.getSession();
			session.setAttribute("code", ""+randList.get(0)+randList.get(1)+randList.get(2)+randList.get(3));
		}
	}