package com.ibeetl.admin.core.web;
|
|
import com.ibeetl.admin.core.entity.CoreOrg;
|
import com.ibeetl.admin.core.entity.CoreUser;
|
import com.ibeetl.admin.core.rbac.UserLoginInfo;
|
import com.ibeetl.admin.core.rbac.tree.MenuItem;
|
import com.ibeetl.admin.core.service.CorePlatformService;
|
import com.ibeetl.admin.core.service.CoreUserService;
|
import com.ibeetl.admin.core.util.HttpRequestLocal;
|
import com.ibeetl.admin.core.util.PlatformException;
|
import com.ibeetl.admin.core.util.TokenUtils;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.servlet.ModelAndView;
|
|
import javax.imageio.ImageIO;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import javax.servlet.http.HttpSession;
|
import java.awt.*;
|
import java.awt.image.BufferedImage;
|
import java.io.IOException;
|
import java.util.Enumeration;
|
import java.util.Random;
|
|
//@Controller
|
@RestController
|
@Slf4j
|
public class IndexController {
|
|
@Autowired
|
CorePlatformService platformService;
|
|
@Autowired
|
CoreUserService userService;
|
|
@Autowired
|
HttpRequestLocal httpRequestLocal;
|
|
|
|
|
@RequestMapping("/")
|
public ModelAndView login() {
|
ModelAndView view = new ModelAndView("/login.html");
|
return view;
|
}
|
|
/**
|
* 登陆验证码
|
*/
|
@RequestMapping("/verify")
|
void verifyCode(HttpServletRequest req, HttpServletResponse resp)throws IOException {
|
// 创建图片
|
int width = 80;
|
int height = 40;
|
BufferedImage image = new BufferedImage(width,height,BufferedImage.TYPE_INT_RGB);
|
|
// 创建图层获得画板
|
Graphics g = image.getGraphics();
|
// 确认画笔颜色
|
g.setColor(Color.BLACK);
|
//填充矩形
|
g.fillRect(0,0,width-2,height-2);
|
// String dataString="ABCDEFGHIJHLMNOPQRSTUVWXYZabcdefghijklmnopqlstuvwxyz1234567890";
|
String dataString="1234567890";
|
//设置字体
|
g.setFont(new Font("宋体",Font.BOLD,30));
|
//缓存随机生成的字符
|
StringBuffer buf = new StringBuffer();
|
Random random = new Random();
|
|
//截取字符
|
for(int i=0;i<4;i++){
|
//设置字体颜色 随机
|
g.setColor(new Color(random.nextInt(255),random.nextInt(255),random.nextInt(255)));
|
//获得一个随机字符
|
int index = random.nextInt(10);
|
String str = dataString.substring(index,index+1);
|
//加入画板
|
g.drawString(str,20*i,30);
|
buf.append(str);
|
}
|
|
|
//干扰线
|
for(int i=0;i<10;i++){
|
g.setColor(new Color(random.nextInt(255),random.nextInt(255),random.nextInt(255)));
|
g.setColor(new Color(16,16,16));
|
g.drawLine(random.nextInt(width),random.nextInt(height),random.nextInt(width),random.nextInt(height)) ;
|
}
|
HttpSession session = req.getSession();
|
log.info("获取的验证码是: {}" , buf.toString());
|
session.setAttribute("cap",buf.toString());
|
|
//设置响应类型
|
resp.setContentType("image/jpeg");
|
//将图片发送给浏览器
|
ImageIO.write(image,"jpg",resp.getOutputStream());
|
}
|
|
@PostMapping("/login.do")
|
public ModelAndView login(String code, String password, String verify, HttpServletRequest request) {
|
|
String ipAddr = request.getRequestURL().toString();
|
if(!request.getSession().getAttribute("cap").toString().equalsIgnoreCase(verify)){
|
if(ipAddr.indexOf("192.168.0.100") != -1){
|
|
}else{
|
throw new PlatformException("验证码错误");
|
}
|
}
|
UserLoginInfo info = userService.login(code, password);
|
if (info == null) {
|
throw new PlatformException("用户名密码错");
|
}
|
CoreUser user = info.getUser();
|
CoreOrg currentOrg = info.getOrgs().get(0);
|
for (CoreOrg org : info.getOrgs()) {
|
if (org.getId() == user.getOrgId()) {
|
currentOrg = org;
|
break;
|
}
|
}
|
|
info.setCurrentOrg(currentOrg);
|
// 记录登录信息到session
|
this.platformService.setLoginUser(info.getUser(), info.getCurrentOrg(), info.getOrgs());
|
ModelAndView view = new ModelAndView("redirect:/index.do");
|
return view;
|
}
|
|
@RequestMapping("/index.do")
|
public ModelAndView index() {
|
ModelAndView view = new ModelAndView("/index.html");
|
CoreUser currentUser = platformService.getCurrentUser();
|
Long orgId = platformService.getCurrentOrgId();
|
MenuItem menuItem = platformService.getMenuItem(currentUser.getId(), orgId);
|
view.addObject("menus", menuItem);
|
return view;
|
}
|
|
@RequestMapping("/logout.do")
|
public ModelAndView logout(HttpServletRequest request) {
|
HttpSession session = request.getSession();
|
Enumeration eum = session.getAttributeNames();
|
while(eum.hasMoreElements()) {
|
String key = (String)eum.nextElement();
|
session.removeAttribute(key);
|
}
|
ModelAndView view = new ModelAndView("redirect:/");
|
return view;
|
}
|
@RequestMapping("/changeOrg.do")
|
public ModelAndView changeOrg(HttpServletRequest request,Long orgId) {
|
platformService.changeOrg(orgId);
|
ModelAndView view = new ModelAndView("redirect:/index.do");
|
return view;
|
}
|
|
}
|
