package com.doumee.api.common;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.doumee.api.BaseController;
|
import com.doumee.biz.system.SystemDictDataBiz;
|
import com.doumee.core.annotation.trace.Trace;
|
import com.doumee.core.utils.Constants;
|
import com.doumee.core.utils.DateUtil;
|
import com.doumee.core.utils.aliyun.ALiYunUtil;
|
import com.doumee.core.wx.WxMiniConfig;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiImplicitParam;
|
import io.swagger.annotations.ApiImplicitParams;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import me.chanjar.weixin.common.error.WxErrorException;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.multipart.MultipartHttpServletRequest;
|
import org.springframework.web.multipart.commons.CommonsMultipartResolver;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.File;
|
import java.io.IOException;
|
import java.io.PrintWriter;
|
import java.util.HashMap;
|
import java.util.Iterator;
|
import java.util.Map;
|
import java.util.UUID;
|
|
/**
|
* @author Eva.Caesar Liu
|
* @date 2023/02/14 11:14
|
*/
|
@Api(tags = "公共接口")
|
@Trace(exclude = true)
|
@RestController
|
@RequestMapping("/public")
|
@Slf4j
|
public class PublicController extends BaseController {
|
|
|
@Autowired
|
private SystemDictDataBiz systemDictDataBiz;
|
|
|
@ApiOperation(value = "上传", notes = "上传", httpMethod = "POST", position = 6)
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "folder", value = "文件夹", required = true, paramType = "query", dataType = "String", dataTypeClass = String.class),
|
})
|
@PostMapping(value = "/upload", headers = "content-type=multipart/form-data")
|
public void uploadMobile(String folder, HttpServletRequest request, HttpServletResponse response) throws Exception {
|
MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
|
upload(multipartRequest, response, folder + "/",
|
systemDictDataBiz.queryByCode(Constants.OSS,Constants.BUCKETNAME).getCode(),
|
systemDictDataBiz.queryByCode(Constants.OSS,Constants.ACCESS_ID).getCode(),
|
systemDictDataBiz.queryByCode(Constants.OSS,Constants.ACCESS_KEY).getCode(),
|
systemDictDataBiz.queryByCode(Constants.OSS,Constants.RESOURCE_PATH).getCode(),
|
systemDictDataBiz.queryByCode(Constants.OSS,Constants.ENDPOINT).getCode());
|
}
|
|
|
@ApiOperation(value = "上传图片", notes = "上传图片", httpMethod = "POST", position = 6)
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "folder", value = "文件夹", required = true, paramType = "query", dataType = "String", dataTypeClass = String.class),
|
})
|
@PostMapping(value = "/uploadPicture", headers = "content-type=multipart/form-data")
|
public void uploadPicture(String folder, HttpServletRequest request, HttpServletResponse response) throws Exception {
|
MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
|
Iterator<String> fileNames = multipartRequest.getFileNames();
|
Map<String, Object> context = new HashMap<>();
|
while (fileNames.hasNext()){
|
MultipartFile file = multipartRequest.getFile(fileNames.next());
|
String fileName = file.getOriginalFilename();
|
String suffix = fileName.substring(fileName.lastIndexOf("."));
|
if ( !StringUtils.equalsIgnoreCase(suffix, ".jpg") || !StringUtils.equalsIgnoreCase(suffix, ".png")) {
|
context.put("code", 0);
|
context.put("message", "对不起，文件格式上传有误！");
|
}
|
}
|
upload(multipartRequest, response, folder + "/",
|
systemDictDataBiz.queryByCode(Constants.OSS,Constants.BUCKETNAME).getCode(),
|
systemDictDataBiz.queryByCode(Constants.OSS,Constants.ACCESS_ID).getCode(),
|
systemDictDataBiz.queryByCode(Constants.OSS,Constants.ACCESS_KEY).getCode(),
|
systemDictDataBiz.queryByCode(Constants.OSS,Constants.RESOURCE_PATH).getCode(),
|
systemDictDataBiz.queryByCode(Constants.OSS,Constants.ENDPOINT).getCode());
|
}
|
|
public void upload(HttpServletRequest request, HttpServletResponse response, String folder, String bucketName,
|
String access_id, String access_key, String resourcePath, String endpoint) throws Exception {
|
response.setCharacterEncoding("UTF-8");
|
response.setContentType("text/html;charset=UTF-8");
|
Map<String, Object> context = new HashMap<>();
|
CommonsMultipartResolver multipartResovler = new CommonsMultipartResolver();
|
if (multipartResovler.isMultipart(request)) {
|
MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
|
Iterator<String> it = multipartRequest.getFileNames();
|
while (it.hasNext()) {
|
MultipartFile file = multipartRequest.getFile((String) it.next());
|
if (file != null) {
|
String uploadFileName = file.getOriginalFilename();
|
String originname = uploadFileName;
|
if (originname.lastIndexOf("/") >= 0) {
|
originname = originname.substring(originname.lastIndexOf("/") + 1);
|
}
|
String nfix = "";
|
if (StringUtils.isNotBlank(uploadFileName)) {
|
nfix = uploadFileName.substring(uploadFileName.lastIndexOf("."));
|
}
|
if (StringUtils.equalsIgnoreCase(nfix, ".exe")) {
|
context.put("code", 4000);
|
context.put("message", "对不起，文件格式\".exe\"上传有误！");
|
return;
|
}
|
if (StringUtils.equalsIgnoreCase(nfix, ".dll")) {
|
context.put("code", 4000);
|
context.put("message", "对不起，文件格式\".dll\"上传有误！");
|
return;
|
}
|
|
if (isImageFile(nfix)) {
|
File tempFile = null;
|
try {
|
tempFile = File.createTempFile("wx_check_", nfix);
|
java.io.FileOutputStream fos = new java.io.FileOutputStream(tempFile);
|
fos.write(file.getBytes());
|
fos.close();
|
boolean isSafe = WxMiniConfig.wxMaService.getSecurityService().checkImage(tempFile);
|
if (!isSafe) {
|
context.put("code", 4001);
|
context.put("message", "图片内容不合规，请更换图片后重试");
|
writerJson(response, context);
|
return;
|
}
|
} catch (WxErrorException e) {
|
log.warn("微信图片安全检测失败: {}", e.getMessage());
|
} finally {
|
if (tempFile != null && tempFile.exists()) {
|
tempFile.delete();
|
}
|
}
|
}
|
|
String nowDate = DateUtil.getNowShortDate();
|
String fileName = UUID.randomUUID().toString() + nfix;
|
String tempFileName = nowDate + "/" + fileName;
|
String key = folder + tempFileName;
|
ALiYunUtil obs = new ALiYunUtil(endpoint,access_id, access_key);
|
if (obs.uploadOnlineObject(file.getInputStream(),bucketName, key,null)) {
|
context.put("success", true);
|
context.put("code", 200);
|
context.put("errno",0);
|
JSONObject fileJSON = new JSONObject();
|
fileJSON.put("url", resourcePath + key);
|
fileJSON.put("imgaddr", tempFileName);
|
fileJSON.put("imgname", fileName);
|
fileJSON.put("originname", originname);
|
context.put("data",fileJSON);
|
context.put("message","请求成功");
|
writerJson(response, context);
|
return;
|
} else {
|
context.put("code", 0);
|
context.put("message", "上传失败");
|
writerJson(response, context);
|
return;
|
}
|
}
|
|
}
|
}
|
context.put("code", 0);
|
context.put("message", "上传失败");
|
context.put("errno",0);
|
writerJson(response, context);
|
return;
|
}
|
|
private boolean isImageFile(String suffix) {
|
if (StringUtils.isBlank(suffix)) {
|
return false;
|
}
|
String lowerSuffix = suffix.toLowerCase();
|
return lowerSuffix.equals(".jpg") || lowerSuffix.equals(".jpeg")
|
|| lowerSuffix.equals(".png") || lowerSuffix.equals(".gif")
|
|| lowerSuffix.equals(".bmp") || lowerSuffix.equals(".webp");
|
}
|
|
public static void writerJson(HttpServletResponse response, Object object) {
|
response.setContentType("application/json");
|
writer(response, JSONObject.toJSONString(object));
|
}
|
|
private static void writer(HttpServletResponse response, String str) {
|
try {
|
StringBuffer result = new StringBuffer();
|
//设置页面不缓存
|
response.setHeader("Pragma", "No-cache");
|
response.setHeader("Cache-Control", "no-cache");
|
response.setCharacterEncoding("UTF-8");
|
PrintWriter out = null;
|
out = response.getWriter();
|
out.print(str);
|
out.flush();
|
out.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
