mirror of
https://github.com/Hello-hao/Tbed.git
synced 2024-04-21 12:32:10 +00:00
497 lines
26 KiB
Java
497 lines
26 KiB
Java
package cn.hellohao.controller;
|
|
|
|
import java.io.FileInputStream;
|
|
import java.text.SimpleDateFormat;
|
|
import java.util.*;
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import javax.servlet.http.HttpSession;
|
|
|
|
import cn.hellohao.pojo.*;
|
|
import cn.hellohao.service.*;
|
|
import cn.hellohao.service.impl.*;
|
|
import cn.hellohao.utils.IPPortUtil;
|
|
import cn.hellohao.utils.*;
|
|
import com.alibaba.fastjson.JSONObject;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.stereotype.Controller;
|
|
import org.springframework.ui.Model;
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import org.springframework.web.bind.annotation.RequestParam;
|
|
import org.springframework.web.bind.annotation.ResponseBody;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
import com.alibaba.fastjson.JSONArray;
|
|
|
|
@Controller
|
|
public class UpdateImgController {
|
|
@Autowired
|
|
private NOSImageupload nOSImageupload;
|
|
@Autowired
|
|
private UserService userService;
|
|
@Autowired
|
|
private KeysService keysService;
|
|
@Autowired
|
|
private OSSImageupload ossImageupload;
|
|
@Autowired
|
|
private ConfigService configService;
|
|
@Autowired
|
|
private UploadConfigService uploadConfigService;
|
|
@Autowired
|
|
private USSImageupload ussImageupload;
|
|
@Autowired
|
|
private KODOImageupload kodoImageupload;
|
|
@Autowired
|
|
private COSImageupload cosImageupload;
|
|
@Autowired
|
|
private FTPImageupload ftpImageupload;
|
|
|
|
@RequestMapping({"/", "/index"})
|
|
public String indexImg(Model model, HttpSession httpSession) {
|
|
Print.Normal("欢迎使用Hellohao图床源码。者也许是最好用的java图床项目。");
|
|
Print.Normal("当前项目路径："+System.getProperty("user.dir"));
|
|
Config config = configService.getSourceype();//查询当前系统使用的存储源类型。
|
|
UploadConfig uploadConfig = uploadConfigService.getUpdateConfig();
|
|
Integer filesizetourists = 0;
|
|
Integer filesizeuser = 0;
|
|
Integer imgcounttourists = 0;
|
|
Integer imgcountuser = 0;
|
|
if(uploadConfig.getFilesizetourists()!=null){filesizetourists = uploadConfig.getFilesizetourists();}
|
|
if(uploadConfig.getFilesizeuser()!=null){filesizeuser = uploadConfig.getFilesizeuser();}
|
|
if(uploadConfig.getImgcounttourists()!=null){imgcounttourists = uploadConfig.getImgcounttourists();}
|
|
if(uploadConfig.getImgcountuser()!=null){imgcountuser = uploadConfig.getImgcountuser();}
|
|
//Boolean b =false;
|
|
Keys key = null;
|
|
if(config.getSourcekey()!=0 && config.getSourcekey()!=5){
|
|
key= keysService.selectKeys(config.getSourcekey());//然后根据类型再查询key
|
|
//b = StringUtils.doNull(key);//判断对象是否有空值
|
|
if(key.getStorageType()!=0 && key.getStorageType()!=null){
|
|
if(key.getStorageType()==1){
|
|
nOSImageupload.Initialize(key);//实例化网易
|
|
}else if (key.getStorageType()==2){
|
|
OSSImageupload.Initialize(key);
|
|
}else if(key.getStorageType()==3){
|
|
USSImageupload.Initialize(key);
|
|
}else if(key.getStorageType()==4){
|
|
KODOImageupload.Initialize(key);
|
|
}else if(key.getStorageType()==6){
|
|
COSImageupload.Initialize(key);
|
|
}else if(key.getStorageType()==7){
|
|
FTPImageupload.Initialize(key);
|
|
}
|
|
else{
|
|
Print.Normal("为获取到存储参数，或者使用存储源是本地的。");
|
|
}
|
|
}
|
|
}
|
|
//if(b){
|
|
|
|
//}
|
|
User u = (User) httpSession.getAttribute("user");
|
|
String email = (String) httpSession.getAttribute("email");
|
|
if (email != null) {
|
|
//登陆成功
|
|
Integer ret = userService.login(u.getEmail(), u.getPassword());
|
|
if (ret > 0) {
|
|
User user = userService.getUsers(u.getEmail());
|
|
model.addAttribute("username", user.getUsername());
|
|
model.addAttribute("level", user.getLevel());
|
|
model.addAttribute("loginid", 100);
|
|
model.addAttribute("imgcount", imgcountuser);
|
|
model.addAttribute("filesize", filesizeuser*1024*1024);
|
|
|
|
} else {
|
|
model.addAttribute("loginid", -1);
|
|
model.addAttribute("imgcount", imgcounttourists);
|
|
}
|
|
} else {
|
|
model.addAttribute("loginid", -2);
|
|
model.addAttribute("imgcount", imgcounttourists);
|
|
model.addAttribute("filesize", filesizetourists*1024*1024);
|
|
}
|
|
model.addAttribute("suffix", uploadConfig.getSuffix());
|
|
model.addAttribute("config", config);
|
|
model.addAttribute("uploadConfig", uploadConfig);
|
|
Integer isupdate = 1;
|
|
if(uploadConfig.getIsupdate()!=1){
|
|
isupdate = (u == null) ? 0: 1;//如果u等于null那么isupdate就等于0，否则等于1
|
|
//isupdate=0;
|
|
}
|
|
model.addAttribute("ykxz", isupdate);
|
|
return "index";
|
|
|
|
}
|
|
|
|
@RequestMapping(value = "/upimg")
|
|
@ResponseBody
|
|
public String upimg( HttpSession session
|
|
, @RequestParam(value = "file", required = false) MultipartFile[] file) throws Exception {
|
|
|
|
Config config = configService.getSourceype();//查询当前系统使用的存储源类型。
|
|
UploadConfig uploadConfig = uploadConfigService.getUpdateConfig();
|
|
User u = (User) session.getAttribute("user");
|
|
JSONArray jsonArray = new JSONArray();
|
|
JSONObject jsonObject = new JSONObject();
|
|
Keys key = keysService.selectKeys(config.getSourcekey());
|
|
Boolean b =false;
|
|
if(config.getSourcekey()==5){
|
|
b =true;
|
|
}else{
|
|
b = StringUtils.doNull(config.getSourcekey(),key);//判断对象是否有空值
|
|
}
|
|
if(b){
|
|
long stime = System.currentTimeMillis();
|
|
String userpath = "tourist";
|
|
if(uploadConfig.getUrltype()==2){
|
|
java.text.DateFormat dateFormat = new java.text.SimpleDateFormat("yyyy/MM/dd");
|
|
userpath = dateFormat.format(new Date());
|
|
}else{
|
|
if (u != null) {
|
|
userpath = u.getUsername();
|
|
}
|
|
}
|
|
Map<String, MultipartFile> map = new HashMap<>();
|
|
for (MultipartFile multipartFile : file) {
|
|
// 获取ImageReader对象的迭代器
|
|
//获取文件名
|
|
String fileName = multipartFile.getOriginalFilename();
|
|
String lastname = fileName.substring(fileName.lastIndexOf(".") + 1);//获取文件后缀
|
|
if (!multipartFile.isEmpty()) { //判断文件是否为空
|
|
map.put(lastname, multipartFile);
|
|
System.out.println("===name==="+multipartFile.getOriginalFilename()); //文件名
|
|
//multipartFile.getSize(); //文件大小
|
|
}
|
|
}
|
|
Map<ReturnImage, Integer> m = null;
|
|
if(key.getStorageType()==1){
|
|
m = nOSImageupload.Imageupload(map, userpath,null);
|
|
}else if (key.getStorageType()==2){
|
|
m = ossImageupload.ImageuploadOSS(map, userpath,null);
|
|
}else if(key.getStorageType()==3){
|
|
m = ussImageupload.ImageuploadUSS(map, userpath,null);
|
|
}else if(key.getStorageType()==4){
|
|
m = kodoImageupload.ImageuploadKODO(map, userpath,null);
|
|
}else if(key.getStorageType()==5){
|
|
m = LocUpdateImg.ImageuploadLOC(map, userpath,null);
|
|
}else if(key.getStorageType()==6){
|
|
m = cosImageupload.ImageuploadCOS(map, userpath,null);
|
|
}else if(key.getStorageType()==7){
|
|
m = ftpImageupload.ImageuploadFTP(map, userpath,null);
|
|
}
|
|
else{
|
|
System.err.println("未获取到对象存储参数，上传失败。");
|
|
}
|
|
Images img = new Images();
|
|
SimpleDateFormat df=new SimpleDateFormat("yyyy-MM-dd");
|
|
String times = df.format(new Date());
|
|
System.out.println("上传图片的时间是："+times);
|
|
for (Map.Entry<ReturnImage, Integer> entry : m.entrySet()) {
|
|
if(key.getStorageType()==5){
|
|
if(config.getDomain()!=null){
|
|
jsonObject.put("imgurls",config.getDomain()+"/links/"+entry.getKey().getImgurl());
|
|
jsonObject.put("imgnames",entry.getKey().getImgname());
|
|
img.setImgurl(config.getDomain()+"/links/"+entry.getKey().getImgurl());//图片链接
|
|
}else{
|
|
jsonObject.put("imgurls",config.getDomain()+"/links/"+entry.getKey().getImgurl());
|
|
jsonObject.put("imgnames",entry.getKey().getImgname());
|
|
img.setImgurl("http://"+IPPortUtil.getLocalIP()+":"+IPPortUtil.getLocalPort()+"/links/"+entry.getKey().getImgurl());//图片链接
|
|
}
|
|
}else{
|
|
jsonObject.put("imgurls",entry.getKey().getImgurl());
|
|
jsonObject.put("imgnames",entry.getKey().getImgname());
|
|
img.setImgurl(entry.getKey().getImgurl());//图片链接
|
|
}
|
|
jsonArray.add(jsonObject);
|
|
img.setUpdatetime(times);
|
|
img.setSource(key.getStorageType());
|
|
if (u == null) {
|
|
img.setUserid(0);
|
|
} else {
|
|
img.setUserid(u.getId());//用户id
|
|
}
|
|
img.setSizes((entry.getValue()) / 1024);
|
|
img.setImgname(SetText.getSubString(entry.getKey().getImgurl(), key.getRequestAddress() + "/", ""));
|
|
img.setAbnormal(0);
|
|
userService.insertimg(img);
|
|
long etime = System.currentTimeMillis();
|
|
System.out.println("上传图片所用时长：" + String.valueOf(etime - stime) + "ms");
|
|
}
|
|
|
|
}else{
|
|
jsonObject.put("imgurls",-1);
|
|
}
|
|
|
|
//开辟新进程鉴黄。现在已经改成定时器
|
|
//查询鉴黄功能是否启动，1为启用
|
|
// Imgreview imgreview = imgreviewService.selectByPrimaryKey(1);
|
|
// if (imgreview.getUsing() == 1) {
|
|
// //上传完成后开辟新的线程进行图片鉴黄。
|
|
// JianHuangThread thread = new JianHuangThread(imgreviewService, key, u, m);
|
|
// thread.start();
|
|
// }
|
|
System.out.println(jsonArray.toString());
|
|
return jsonObject.toString();
|
|
}
|
|
|
|
|
|
|
|
//根据网络图片url上传
|
|
@PostMapping(value = "/upurlimg")
|
|
@ResponseBody
|
|
public String upurlimg(HttpSession session, String imgurl, HttpServletRequest request) throws Exception {
|
|
Config config = configService.getSourceype();//查询当前系统使用的存储源类型。
|
|
UploadConfig uploadConfig = uploadConfigService.getUpdateConfig();
|
|
User u = (User) session.getAttribute("user");
|
|
String userpath = "tourist";
|
|
if(uploadConfig.getUrltype()==2){
|
|
java.text.DateFormat dateFormat = new java.text.SimpleDateFormat("yyyy/MM/dd");
|
|
userpath = dateFormat.format(new Date());
|
|
}else{
|
|
if (u != null) {
|
|
userpath = u.getUsername();
|
|
}
|
|
}
|
|
JSONArray jsonArray = new JSONArray();
|
|
Keys key = keysService.selectKeys(config.getSourcekey());
|
|
long imgsize = ImgUrlUtil.getFileLength(imgurl);
|
|
Integer youke = uploadConfig.getFilesizetourists();
|
|
Integer yonghu = uploadConfig.getFilesizeuser();
|
|
String uuid= UUID.randomUUID().toString().replace("-", "");
|
|
Boolean bo =false;
|
|
if(config.getSourcekey()==5){
|
|
bo =true;
|
|
}else{
|
|
bo = StringUtils.doNull(config.getSourcekey(),key);//判断对象是否有空值
|
|
}
|
|
//先判断对象存储key是不是null
|
|
Print.warning("上传地址是："+request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/");
|
|
if(bo){
|
|
long stime = System.currentTimeMillis();
|
|
//判断是会员还是游客
|
|
if(u!=null){
|
|
//判断文件大小
|
|
if(imgsize>0 && imgsize<=(yonghu*1024*1024)){
|
|
try{
|
|
boolean bl =ImgUrlUtil.downLoadFromUrl(imgurl,
|
|
uuid, request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/");
|
|
if(bl==true){
|
|
FileInputStream is = new FileInputStream(request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/"+uuid);
|
|
byte[] b = new byte[3];
|
|
is.read(b, 0, b.length);
|
|
String xxx = ImgUrlUtil.bytesToHexString(b);
|
|
xxx = xxx.toUpperCase();
|
|
System.out.println("头文件是：" + xxx);
|
|
String ooo = TypeDict.checkType(xxx);
|
|
System.out.println("后缀名是：" + ooo);
|
|
if(is!=null){
|
|
is.close();
|
|
}
|
|
//判断文件头是否是图片
|
|
if(!ooo.equals("0000")){
|
|
Map<String, String> map = new HashMap<>();
|
|
map.put(ooo, request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/"+uuid);
|
|
Map<ReturnImage, Integer> m = null;
|
|
if(key.getStorageType()==1){
|
|
m = nOSImageupload.Imageupload(null, userpath,map);
|
|
}else if (key.getStorageType()==2){
|
|
m = ossImageupload.ImageuploadOSS(null, userpath,map);
|
|
}else if(key.getStorageType()==3){
|
|
m = ussImageupload.ImageuploadUSS(null, userpath,map);
|
|
}else if(key.getStorageType()==4){
|
|
m = kodoImageupload.ImageuploadKODO(null, userpath,map);
|
|
}else if(key.getStorageType()==5){
|
|
m = LocUpdateImg.ImageuploadLOC(null,userpath,map);
|
|
}else if(key.getStorageType()==6){
|
|
m = cosImageupload.ImageuploadCOS(null,userpath,map);
|
|
}else if(key.getStorageType()==7){
|
|
m = ftpImageupload.ImageuploadFTP(null,userpath,map);
|
|
}
|
|
else{
|
|
System.err.println("未获取到对象存储参数，上传失败。");
|
|
}
|
|
Images img = new Images();
|
|
SimpleDateFormat df=new SimpleDateFormat("yyyy-MM-dd");
|
|
String times = df.format(new Date());
|
|
System.out.println("上传图片的时间是："+times);
|
|
for (Map.Entry<ReturnImage, Integer> entry : m.entrySet()) {
|
|
if(key.getStorageType()==5){
|
|
if(config.getDomain()!=null){
|
|
jsonArray.add(config.getDomain()+"/links/"+entry.getKey().getImgurl());
|
|
img.setImgurl(config.getDomain()+"/links/"+entry.getKey().getImgurl());//图片链接
|
|
}else{
|
|
jsonArray.add(config.getDomain()+"/links/"+entry.getKey().getImgurl());
|
|
img.setImgurl("http://"+IPPortUtil.getLocalIP()+":"+IPPortUtil.getLocalPort()+"/links/"+entry.getKey().getImgurl());//图片链接
|
|
}
|
|
}else{
|
|
jsonArray.add(entry.getKey().getImgurl());
|
|
img.setImgurl(entry.getKey().getImgurl());
|
|
}
|
|
img.setUpdatetime(times);
|
|
img.setSource(key.getStorageType());
|
|
if (u == null) {
|
|
img.setUserid(0);//用户id
|
|
} else {
|
|
img.setUserid(u.getId());//用户id
|
|
}
|
|
img.setSizes((entry.getValue()));
|
|
img.setImgname(SetText.getSubString(entry.getKey().getImgurl(), key.getRequestAddress() + "/", ""));
|
|
img.setAbnormal(0);
|
|
userService.insertimg(img);
|
|
long etime = System.currentTimeMillis();
|
|
System.out.println("上传图片所用时长：" + String.valueOf(etime - stime) + "ms");
|
|
}
|
|
}else{
|
|
jsonArray.add(-3);
|
|
}
|
|
}
|
|
}catch (Exception e) {
|
|
// TODO: handle exception
|
|
Print.warning(e.toString());
|
|
jsonArray.add(-4);
|
|
}
|
|
}else{
|
|
//文件过大
|
|
jsonArray.add(-2);
|
|
}
|
|
}else{
|
|
if(imgsize>0 && imgsize<=(youke*1024*1024)){
|
|
try{
|
|
boolean bl = ImgUrlUtil.downLoadFromUrl(imgurl,
|
|
uuid, request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/");
|
|
if(bl==true){
|
|
FileInputStream is = new FileInputStream(request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/"+uuid);
|
|
byte[] b = new byte[3];
|
|
is.read(b, 0, b.length);
|
|
String xxx = ImgUrlUtil.bytesToHexString(b);
|
|
xxx = xxx.toUpperCase();
|
|
//System.out.println("头文件是：" + xxx);
|
|
String ooo = TypeDict.checkType(xxx);
|
|
//System.out.println("后缀名是：" + ooo);
|
|
if(is!=null){
|
|
is.close();
|
|
}
|
|
//判断文件头是否是图片
|
|
if(!xxx.equals("0000")){
|
|
Map<String, String> map = new HashMap<>();
|
|
map.put(ooo, request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/"+uuid);
|
|
Map<ReturnImage, Integer> m = null;
|
|
if(key.getStorageType()==1){
|
|
m = nOSImageupload.Imageupload(null, userpath,map);
|
|
}else if (key.getStorageType()==2){
|
|
m = ossImageupload.ImageuploadOSS(null, userpath,map);
|
|
}else if(key.getStorageType()==3){
|
|
m = ussImageupload.ImageuploadUSS(null, userpath,map);
|
|
}else if(key.getStorageType()==4){
|
|
m = kodoImageupload.ImageuploadKODO(null, userpath,map);
|
|
}else if(key.getStorageType()==5){
|
|
m =LocUpdateImg.ImageuploadLOC(null,userpath, map);
|
|
}else if(key.getStorageType()==6){
|
|
m =cosImageupload.ImageuploadCOS(null,userpath, map);
|
|
}else if(key.getStorageType()==7){
|
|
m = ftpImageupload.ImageuploadFTP(null,userpath,map);
|
|
}
|
|
else{
|
|
System.err.println("未获取到对象存储参数，上传失败。");
|
|
}
|
|
Images img = new Images();
|
|
SimpleDateFormat df=new SimpleDateFormat("yyyy-MM-dd");
|
|
String times = df.format(new Date());
|
|
System.out.println("上传图片的时间是："+times);
|
|
for (Map.Entry<ReturnImage, Integer> entry : m.entrySet()) {
|
|
if(key.getStorageType()==5){
|
|
if(config.getDomain()!=null){
|
|
jsonArray.add(config.getDomain()+"/links/"+entry.getKey().getImgurl());
|
|
img.setImgurl(config.getDomain()+"/links/"+entry.getKey().getImgurl());//图片链接
|
|
}else{
|
|
jsonArray.add(config.getDomain()+"/links/"+entry.getKey().getImgurl());
|
|
img.setImgurl("http://"+IPPortUtil.getLocalIP()+":"+IPPortUtil.getLocalPort()+"/links/"+entry.getKey().getImgurl());//图片链接
|
|
}
|
|
}else{
|
|
jsonArray.add(entry.getKey().getImgurl());
|
|
}
|
|
//img.setImgurl(entry.getKey());//图片链接
|
|
img.setUpdatetime(times);
|
|
img.setSource(key.getStorageType());
|
|
if (u == null) {
|
|
img.setUserid(0);//用户id
|
|
} else {
|
|
img.setUserid(u.getId());//用户id
|
|
}
|
|
img.setSizes((entry.getValue()));
|
|
img.setImgname(SetText.getSubString(entry.getKey().getImgurl(), key.getRequestAddress() + "/", ""));
|
|
img.setAbnormal(0);
|
|
userService.insertimg(img);
|
|
long etime = System.currentTimeMillis();
|
|
System.out.println("上传图片所用时长：" + String.valueOf(etime - stime) + "ms");
|
|
}
|
|
}else{
|
|
jsonArray.add(-3);
|
|
}
|
|
}
|
|
}catch (Exception e) {
|
|
// TODO: handle exception
|
|
Print.warning(e.toString());
|
|
jsonArray.add(-4);
|
|
}
|
|
}else{
|
|
//文件过大
|
|
jsonArray.add(-2);
|
|
}
|
|
}
|
|
}else{
|
|
jsonArray.add(-1);
|
|
}
|
|
|
|
return jsonArray.toString();
|
|
/**
|
|
* 错误返回值含义：
|
|
* -1 存储源key未配置
|
|
* -2 目标图片太大或者不存在
|
|
* -3 文件类型不符合要求
|
|
* */
|
|
}
|
|
|
|
//刪除用戶
|
|
@RequestMapping("/sentence")
|
|
@ResponseBody
|
|
public String sentence(HttpSession session, Integer id) {
|
|
JSONArray jsonArray = new JSONArray();
|
|
String text = Sentence.getURLContent();
|
|
jsonArray.add(text);
|
|
return jsonArray.toString();
|
|
}
|
|
|
|
//ajax查询用户是否已经登录
|
|
@RequestMapping(value = "/islogin")
|
|
@ResponseBody
|
|
public String islogin(HttpSession session) {
|
|
JSONObject jsonObject = new JSONObject();
|
|
User user = (User) session.getAttribute("user");
|
|
if(user!=null){
|
|
if (user.getEmail() != null && user.getPassword() != null) {
|
|
jsonObject.put("username",user.getUsername());
|
|
jsonObject.put("level",user.getLevel());
|
|
jsonObject.put("lgoinret",1);
|
|
}else{
|
|
jsonObject.put("lgoinret",0);
|
|
}
|
|
}
|
|
return jsonObject.toString();
|
|
}
|
|
|
|
@RequestMapping(value = "/getsentence")
|
|
@ResponseBody
|
|
public String getsentence(HttpSession session) {
|
|
JSONObject jsonObject = new JSONObject();
|
|
String yiyan = Sentence.getURLContent();
|
|
|
|
return yiyan;
|
|
}
|
|
|
|
|
|
@RequestMapping("/err")
|
|
public String err() {
|
|
return "err";
|
|
}
|
|
|
|
}
|