mirror of
https://github.com/Hello-hao/Tbed.git
synced 2024-04-21 12:32:10 +00:00
415 lines
16 KiB
Java
415 lines
16 KiB
Java
package cn.hellohao.controller;
|
|
|
|
import cn.hellohao.auth.token.JWTUtil;
|
|
import cn.hellohao.config.GlobalConstant;
|
|
import cn.hellohao.pojo.*;
|
|
import cn.hellohao.service.*;
|
|
import cn.hellohao.service.impl.UploadServicel;
|
|
import cn.hellohao.service.impl.deleImages;
|
|
import cn.hellohao.utils.GetIPS;
|
|
import cn.hellohao.utils.MyVersion;
|
|
import cn.hellohao.utils.SetFiles;
|
|
import cn.hutool.captcha.CaptchaUtil;
|
|
import cn.hutool.captcha.ShearCaptcha;
|
|
import cn.hutool.captcha.generator.MathGenerator;
|
|
import com.alibaba.fastjson.JSONArray;
|
|
import com.alibaba.fastjson.JSONObject;
|
|
import org.apache.commons.lang3.StringUtils;
|
|
import org.apache.shiro.SecurityUtils;
|
|
import org.apache.shiro.authc.UsernamePasswordToken;
|
|
import org.apache.shiro.subject.Subject;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.stereotype.Controller;
|
|
import org.springframework.ui.Model;
|
|
import org.springframework.web.bind.annotation.*;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.nio.file.Path;
|
|
import java.nio.file.Paths;
|
|
import java.util.List;
|
|
import java.util.UUID;
|
|
import java.util.concurrent.CompletableFuture;
|
|
|
|
@Controller
|
|
public class IndexController {
|
|
@Autowired private ImgService imgService;
|
|
@Autowired private SysConfigService sysConfigService;
|
|
@Autowired private ConfdataService confdataService;
|
|
@Autowired private UploadConfigService uploadConfigService;
|
|
@Autowired private UploadServicel uploadServicel;
|
|
@Autowired private deleImages deleimages;
|
|
@Autowired private IRedisService iRedisService;
|
|
@Autowired private AppClientService appClientService;
|
|
|
|
@RequestMapping(value = "/")
|
|
public String Welcome(Model model, HttpServletRequest httpServletRequest) {
|
|
model.addAttribute("name", "服务端程序(开源版)");
|
|
model.addAttribute("version", "20230622");
|
|
model.addAttribute("ip", GetIPS.getIpAddr(httpServletRequest));
|
|
model.addAttribute("links", "https://github.com/Hello-hao/tbed");
|
|
return "welcome";
|
|
}
|
|
|
|
@RequestMapping(value = "/webInfo")
|
|
@ResponseBody
|
|
public Msg webInfo() {
|
|
Msg msg = new Msg();
|
|
final Confdata confdata = confdataService.selectConfdata("config");
|
|
JSONObject cd = JSONObject.parseObject(confdata.getJsondata());
|
|
UploadConfig updateConfig = uploadConfigService.getUpdateConfig();
|
|
SysConfig sysConfig = sysConfigService.getstate();
|
|
JSONObject jsonObject = new JSONObject();
|
|
AppClient appClient = appClientService.getAppClientData("app");
|
|
jsonObject.put("webname", cd.getString("webname"));
|
|
jsonObject.put("websubtitle", cd.getString("websubtitle"));
|
|
jsonObject.put("keywords", cd.getString("keywords"));
|
|
jsonObject.put("webms", cd.getString("webms"));
|
|
jsonObject.put("explain", cd.getString("explain"));
|
|
jsonObject.put("baidu", cd.getString("baidu"));
|
|
jsonObject.put("links", cd.getString("links"));
|
|
jsonObject.put("aboutinfo", cd.getString("aboutinfo"));
|
|
jsonObject.put("logo", cd.getString("logo"));
|
|
jsonObject.put("webfavicons", cd.getString("webfavicons"));
|
|
jsonObject.put("api", updateConfig.getApi());
|
|
jsonObject.put("register", sysConfig.getRegister());
|
|
jsonObject.put("isuse", appClient.getIsuse());
|
|
jsonObject.put("clientname", appClient.getAppname());
|
|
jsonObject.put("clientlogo", appClient.getApplogo());
|
|
jsonObject.put("appupdate", appClient.getAppupdate());
|
|
msg.setData(jsonObject);
|
|
return msg;
|
|
}
|
|
@PostMapping("/uploadChunkFile")
|
|
@ResponseBody
|
|
@CrossOrigin
|
|
public Msg uploadChunk(HttpServletRequest request,Chunk chunk) {
|
|
Msg msg = new Msg();
|
|
MultipartFile file = chunk.getFile();
|
|
try {
|
|
byte[] bytes = file.getBytes();
|
|
Path path = Paths.get(SetFiles.generatePath(GlobalConstant.HELLOHAOTEMPIMG_PATH+ File.separator+chunk.getUuid(), chunk));
|
|
java.nio.file.Files.write(path, bytes);
|
|
JSONObject jsonObject = new JSONObject();
|
|
jsonObject.put("chunkNumber",chunk.getChunkNumber());
|
|
jsonObject.put("totalChunks",chunk.getTotalChunks());
|
|
msg.setData(jsonObject);
|
|
msg.setCode("200");
|
|
return msg;
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
msg.setCode("500");
|
|
return msg;
|
|
}
|
|
}
|
|
|
|
@PostMapping("/processFile")
|
|
@ResponseBody
|
|
@CrossOrigin
|
|
public Msg processFile(HttpServletRequest request,@RequestParam(value = "data", defaultValue = "") String data){
|
|
Msg msg = new Msg();
|
|
try{
|
|
Subject subject = SecurityUtils.getSubject();
|
|
User user = (User) subject.getPrincipal();
|
|
CompletableFuture<Msg> ret = uploadServicel.mergeFile(request,data,user);
|
|
msg = ret.get();
|
|
}catch (Exception e){
|
|
msg.setCode("500");
|
|
msg.setInfo("处理文件错误，上传失败");
|
|
}
|
|
return msg;
|
|
}
|
|
|
|
@PostMapping(value = {"/upload", "/client/upload"})
|
|
@ResponseBody
|
|
public Msg upimg(
|
|
HttpServletRequest request,
|
|
@RequestParam(value = "file", required = false) MultipartFile multipartFile,
|
|
Integer day,@RequestParam(value = "md5",required = false) String md5) {
|
|
JSONArray jsonArray = new JSONArray();
|
|
Subject subject = SecurityUtils.getSubject();
|
|
User user = (User) subject.getPrincipal();
|
|
String originalFilename = multipartFile.getOriginalFilename();
|
|
if(StringUtils.isBlank(originalFilename)){
|
|
originalFilename = "未命名图像";
|
|
}
|
|
File file = SetFiles.changeFile_new(multipartFile);
|
|
return uploadServicel.uploadForLoc(request, file,originalFilename, day, null,md5);
|
|
}
|
|
|
|
@PostMapping(value = {"/uploadForUrl", "/client/uploadForUrl"})
|
|
@ResponseBody
|
|
public Msg upurlimg(
|
|
HttpServletRequest request, @RequestParam(value = "data", defaultValue = "") String data) {
|
|
JSONObject jsonObj = JSONObject.parseObject(data);
|
|
Integer setday = jsonObj.getInteger("day");
|
|
String imgUrl = jsonObj.getString("imgUrl");
|
|
String referer = jsonObj.getString("referer");
|
|
String[] URLArr = imgUrl.split("\n");
|
|
|
|
Subject subject = SecurityUtils.getSubject();
|
|
User user = (User) subject.getPrincipal();
|
|
UploadConfig updateConfig = uploadConfigService.getUpdateConfig();
|
|
int syscounts = 0;
|
|
if (null == user) {
|
|
syscounts = updateConfig.getImgcounttourists();
|
|
} else {
|
|
syscounts = updateConfig.getImgcountuser();
|
|
}
|
|
Msg retMsg = new Msg();
|
|
JSONObject jsonObject = new JSONObject();
|
|
JSONArray retArray = new JSONArray();
|
|
int errcounts = 0;
|
|
int excess = 0;
|
|
for (int i = 0; i < URLArr.length; i++) {
|
|
int temp = i + 1;
|
|
if (syscounts >= temp) {
|
|
JSONObject imgJson = new JSONObject();
|
|
imgJson.put("imgUrl",URLArr[i]);
|
|
imgJson.put("referer",referer);
|
|
Msg msg = uploadServicel.uploadForLoc(request, null,"URL转存图像", setday, imgJson,null);
|
|
if (!msg.getCode().equals("200")) {
|
|
errcounts++;
|
|
} else {
|
|
retArray.add(msg);
|
|
}
|
|
} else {
|
|
excess++;
|
|
}
|
|
}
|
|
jsonObject.put("counts", URLArr.length);
|
|
jsonObject.put("errcounts", errcounts);
|
|
jsonObject.put("excess", excess);
|
|
jsonObject.put("urls", retArray);
|
|
retMsg.setData(jsonObject);
|
|
return retMsg;
|
|
}
|
|
|
|
@RequestMapping(value = "/getUploadInfo")
|
|
@ResponseBody
|
|
public Msg getUploadInfo() {
|
|
Msg msg = new Msg();
|
|
JSONObject jsonObject = new JSONObject();
|
|
Subject subject = SecurityUtils.getSubject();
|
|
User user = (User) subject.getPrincipal();
|
|
try {
|
|
UploadConfig updateConfig = uploadConfigService.getUpdateConfig();
|
|
jsonObject.put("suffix", updateConfig.getSuffix().split(","));
|
|
if (null == user) {
|
|
jsonObject.put("filesize", Integer.valueOf(updateConfig.getFilesizetourists()) / 1024);
|
|
jsonObject.put("imgcount", updateConfig.getImgcounttourists());
|
|
jsonObject.put("uploadSwitch", updateConfig.getIsupdate());
|
|
jsonObject.put("uploadInfo", "您登陆后才能使用此功能哦");
|
|
} else {
|
|
jsonObject.put("filesize", Integer.valueOf(updateConfig.getFilesizeuser()) / 1024);
|
|
jsonObject.put("imgcount", updateConfig.getImgcountuser());
|
|
jsonObject.put("uploadSwitch", updateConfig.getUserclose());
|
|
jsonObject.put("uploadInfo", "系统暂时关闭了用户上传功能");
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
msg.setData(jsonObject);
|
|
return msg;
|
|
}
|
|
|
|
@RequestMapping("/checkStatus")
|
|
@ResponseBody
|
|
public Msg checkStatus(HttpServletRequest request) {
|
|
Msg msg = new Msg();
|
|
String token = request.getHeader("Authorization");
|
|
if (token != null) {
|
|
JSONObject tokenJson = JWTUtil.checkToken(token);
|
|
if (tokenJson.getBoolean("check")) {
|
|
Subject subject = SecurityUtils.getSubject();
|
|
UsernamePasswordToken tokenOBJ =
|
|
new UsernamePasswordToken(
|
|
tokenJson.getString("email"), tokenJson.getString("password"));
|
|
tokenOBJ.setRememberMe(true);
|
|
try {
|
|
subject.login(tokenOBJ);
|
|
SecurityUtils.getSubject().getSession().setTimeout(3600000);
|
|
User u = (User) subject.getPrincipal();
|
|
JSONObject jsonObject = new JSONObject();
|
|
jsonObject.put("RoleLevel", u.getLevel() == 2 ? "admin" : "user");
|
|
jsonObject.put("userName", u.getUsername());
|
|
msg.setCode("200");
|
|
msg.setData(jsonObject);
|
|
} catch (Exception e) {
|
|
msg.setCode("40041");
|
|
msg.setInfo("登录失效，请重新登录");
|
|
System.err.println("登录失效，请重新登录");
|
|
e.printStackTrace();
|
|
}
|
|
} else {
|
|
msg.setCode("40041");
|
|
msg.setInfo("登录失效，请重新登录");
|
|
}
|
|
} else {
|
|
msg.setCode("40040");
|
|
msg.setInfo("当前未登录，请先登录");
|
|
}
|
|
return msg;
|
|
}
|
|
|
|
@PostMapping("/verifyCode")
|
|
@ResponseBody
|
|
public Msg verifyCode() {
|
|
Msg msg = new Msg();
|
|
try {
|
|
ShearCaptcha captcha = CaptchaUtil.createShearCaptcha(90, 35, 2, 2);
|
|
captcha.setGenerator(new MathGenerator(1));
|
|
String code = getVerifyCodeOperator(captcha.getCode());
|
|
String uid = UUID.randomUUID().toString().replace("-", "").toLowerCase();
|
|
iRedisService.setValue("verifyCode_" + uid, code);
|
|
JSONObject jsonObject = new JSONObject();
|
|
jsonObject.put("codeKey", uid);
|
|
jsonObject.put("codeImg", captcha.getImageBase64());
|
|
msg.setData(jsonObject);
|
|
return msg;
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
msg.setCode("500");
|
|
return msg;
|
|
}
|
|
}
|
|
|
|
@PostMapping("/verifyCodeForRegister")
|
|
@ResponseBody
|
|
public Msg verifyCodeForRegister() {
|
|
Msg msg = new Msg();
|
|
try {
|
|
ShearCaptcha captcha = CaptchaUtil.createShearCaptcha(90, 35, 2, 2);
|
|
captcha.setGenerator(new MathGenerator(1));
|
|
String code = getVerifyCodeOperator(captcha.getCode());
|
|
String uid = UUID.randomUUID().toString().replace("-", "").toLowerCase();
|
|
iRedisService.setValue("verifyCodeForRegister_" + uid, code);
|
|
JSONObject jsonObject = new JSONObject();
|
|
jsonObject.put("codeKey", uid);
|
|
jsonObject.put("codeImg", captcha.getImageBase64());
|
|
msg.setData(jsonObject);
|
|
return msg;
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
msg.setCode("500");
|
|
return msg;
|
|
}
|
|
}
|
|
|
|
@PostMapping("/verifyCodeForRetrieve")
|
|
@ResponseBody
|
|
public Msg verifyCodeForRetrieve() {
|
|
Msg msg = new Msg();
|
|
try {
|
|
ShearCaptcha captcha = CaptchaUtil.createShearCaptcha(90, 35, 2, 2);
|
|
captcha.setGenerator(new MathGenerator(1));
|
|
String code = getVerifyCodeOperator(captcha.getCode());
|
|
String uid = UUID.randomUUID().toString().replace("-", "").toLowerCase();
|
|
iRedisService.setValue("verifyCodeForRetrieve_" + uid, code);
|
|
JSONObject jsonObject = new JSONObject();
|
|
jsonObject.put("codeKey", uid);
|
|
jsonObject.put("codeImg", captcha.getImageBase64());
|
|
msg.setData(jsonObject);
|
|
return msg;
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
msg.setCode("500");
|
|
return msg;
|
|
}
|
|
}
|
|
|
|
public static String getVerifyCodeOperator(String str) {
|
|
int a = Integer.valueOf(str.substring(0, 1));
|
|
String yxf = str.substring(1, 2);
|
|
int b = Integer.valueOf(str.substring(2, 3));
|
|
switch (yxf) {
|
|
case "*":
|
|
return Integer.toString(a * b);
|
|
case "+":
|
|
return Integer.toString(a + b);
|
|
case ",":
|
|
default:
|
|
return Integer.toString(82);
|
|
case "-":
|
|
return Integer.toString(a - b);
|
|
}
|
|
}
|
|
|
|
@PostMapping({"/deleImagesByUid", "/client/deleImagesByUid"})
|
|
@ResponseBody
|
|
public Msg deleImagesByUid(@RequestParam(value = "data", defaultValue = "") String data) {
|
|
Msg msg = new Msg();
|
|
JSONObject jsonObj = JSONObject.parseObject(data);
|
|
String imguid = jsonObj.getString("imguid");
|
|
Images image = imgService.selectImgUrlByImgUID(imguid);
|
|
if (null == image) {
|
|
msg.setInfo("图像已不存在，删除成功");
|
|
return msg;
|
|
}
|
|
Subject subject = SecurityUtils.getSubject();
|
|
User user = (User) subject.getPrincipal();
|
|
if (null != user) {
|
|
if (!user.getId().equals(image.getUserid())) {
|
|
msg.setInfo("删除失败，该图片不允许你执行操作");
|
|
msg.setCode("100403");
|
|
return msg;
|
|
}
|
|
}
|
|
msg = deleimages.dele(null, image.getId());
|
|
List<Long> Delids = (List<Long>) msg.getData();
|
|
if (!Delids.contains(image.getId())) {
|
|
msg.setCode("500");
|
|
msg.setInfo("图像删除失败");
|
|
} else {
|
|
msg.setCode("200");
|
|
msg.setInfo("图像已成功删除");
|
|
}
|
|
return msg;
|
|
}
|
|
|
|
@PostMapping("/getClientVersion")
|
|
@ResponseBody
|
|
public Msg getClientVersion(@RequestParam(value = "data", defaultValue = "") String data) {
|
|
Msg msg = new Msg();
|
|
msg.setCode("000");
|
|
try {
|
|
JSONObject jsonObj = JSONObject.parseObject(data);
|
|
String appv = jsonObj.getString("version");
|
|
AppClient app = appClientService.getAppClientData("app");
|
|
if (!StringUtils.isBlank(app.getAppupdate())) {
|
|
String version = app.getAppupdate().replaceAll("\\.", "");
|
|
if (MyVersion.compareVersion(app.getAppupdate(), appv) == 1) {
|
|
JSONObject jsonObject = new JSONObject();
|
|
jsonObject.put("winpackurl", app.getWinpackurl());
|
|
jsonObject.put("macpackurl", app.getMacpackurl());
|
|
msg.setCode("200");
|
|
msg.setData(jsonObject);
|
|
}
|
|
}
|
|
} catch (Exception e) {
|
|
msg.setCode("000");
|
|
}
|
|
return msg;
|
|
}
|
|
|
|
@RequestMapping("/authError")
|
|
@ResponseBody
|
|
public Msg authError(HttpServletRequest request) {
|
|
Msg msg = new Msg();
|
|
msg.setCode("4031");
|
|
msg.setInfo("You don't have authority");
|
|
return msg;
|
|
}
|
|
|
|
@RequestMapping("/jurisError")
|
|
@ResponseBody
|
|
public Msg jurisError(HttpServletRequest request) {
|
|
Msg msg = new Msg();
|
|
msg.setCode("4031");
|
|
msg.setInfo("Authentication request failed");
|
|
return msg;
|
|
}
|
|
}
|