mirror of
https://gitee.com/kekingcn/file-online-preview.git
synced 2026-03-14 05:03:49 +08:00
206 lines
8.4 KiB
Java
206 lines
8.4 KiB
Java
package com.yudianbank.web.controller;
|
||
|
||
import com.yudianbank.param.ReturnResponse;
|
||
import com.yudianbank.utils.*;
|
||
import org.apache.commons.io.IOUtils;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.beans.factory.annotation.Value;
|
||
import org.springframework.stereotype.Controller;
|
||
import org.springframework.ui.Model;
|
||
import org.springframework.util.StringUtils;
|
||
import org.springframework.web.bind.annotation.RequestMapping;
|
||
import org.springframework.web.bind.annotation.RequestMethod;
|
||
|
||
import javax.servlet.http.HttpServletRequest;
|
||
import javax.servlet.http.HttpServletResponse;
|
||
import java.io.File;
|
||
import java.io.IOException;
|
||
import java.io.InputStream;
|
||
import java.io.UnsupportedEncodingException;
|
||
import java.net.HttpURLConnection;
|
||
import java.net.URL;
|
||
import java.net.URLConnection;
|
||
import java.net.URLDecoder;
|
||
import java.util.Arrays;
|
||
|
||
/**
|
||
* @author yudian-it
|
||
*/
|
||
@Controller
|
||
public class OnlinePreviewController {
|
||
@Autowired
|
||
private OfficeToPdf officeToPdf;
|
||
@Autowired
|
||
FileUtils fileUtils;
|
||
@Autowired
|
||
DownloadUtils downloadUtils;
|
||
@Autowired
|
||
ZipReader zipReader;
|
||
@Autowired
|
||
SimTextUtil simTextUtil;
|
||
@Value("${simText}")
|
||
String[] simText;
|
||
|
||
@Value("${file.dir}")
|
||
String fileDir;
|
||
|
||
/**
|
||
* @param url
|
||
* @param model
|
||
* @return
|
||
*/
|
||
@RequestMapping(value = "onlinePreview",method = RequestMethod.GET)
|
||
public String onlinePreview(String url, Model model, HttpServletRequest req) throws UnsupportedEncodingException {
|
||
// 路径转码
|
||
String decodedUrl = URLDecoder.decode(url, "utf-8");
|
||
String type = typeFromUrl(url);
|
||
String suffix = suffixFromUrl(url);
|
||
// 抽取文件并返回文件列表
|
||
String fileName = fileUtils.getFileNameFromURL(decodedUrl);
|
||
model.addAttribute("fileType", suffix);
|
||
if (type.equalsIgnoreCase("picture")) {
|
||
model.addAttribute("imgurl", url);
|
||
return "picture";
|
||
} else if (type.equalsIgnoreCase("simText")){
|
||
ReturnResponse<String> response = simTextUtil.readSimText(decodedUrl, fileName);
|
||
if (0 != response.getCode()) {
|
||
model.addAttribute("msg", response.getMsg());
|
||
return "fileNotSupported";
|
||
}
|
||
model.addAttribute("ordinaryUrl", response.getMsg());
|
||
return "txt";
|
||
} else if(type.equalsIgnoreCase("pdf")){
|
||
model.addAttribute("pdfUrl",url);
|
||
return "pdf";
|
||
} else if(type.equalsIgnoreCase("compress")){
|
||
String fileTree = null;
|
||
// 判断文件名是否存在(redis缓存读取)
|
||
if (!StringUtils.hasText(fileUtils.getConvertedFile(fileName))) {
|
||
ReturnResponse<String> response = downloadUtils.downLoad(decodedUrl, suffix, fileName);
|
||
if (0 != response.getCode()) {
|
||
model.addAttribute("msg", response.getMsg());
|
||
return "fileNotSupported";
|
||
}
|
||
String filePath = response.getContent();
|
||
if ("zip".equalsIgnoreCase(suffix)
|
||
|| "jar".equalsIgnoreCase(suffix)
|
||
|| "gzip".equalsIgnoreCase(suffix)) {
|
||
fileTree = zipReader.readZipFile(filePath);
|
||
} else if ("rar".equalsIgnoreCase(suffix)) {
|
||
fileTree = zipReader.unRar(filePath);
|
||
}
|
||
fileUtils.addConvertedFile(fileName, fileTree);
|
||
}else {
|
||
fileTree = fileUtils.getConvertedFile(fileName);
|
||
}
|
||
System.out.println("返回文件tree》》》》》》》》》》》》》》》》》》》");
|
||
if (null != fileTree) {
|
||
model.addAttribute("fileTree",fileTree);
|
||
return "compress";
|
||
}else {
|
||
model.addAttribute("msg", "压缩文件类型不受支持，尝试在压缩的时候选择RAR4格式");
|
||
return "fileNotSupported";
|
||
}
|
||
} else if ("office".equalsIgnoreCase(type)) {
|
||
boolean isHtml = suffix.equalsIgnoreCase("xls")
|
||
|| suffix.equalsIgnoreCase("xlsx");
|
||
String pdfName = fileName.substring(0, fileName.lastIndexOf(".") + 1) + (isHtml ? "html" : "pdf");
|
||
// 判断之前是否已转换过，如果转换过，直接返回，否则执行转换
|
||
if (!fileUtils.listConvertedFiles().containsKey(pdfName)) {
|
||
String filePath = fileDir + fileName;
|
||
if (!new File(filePath).exists()) {
|
||
ReturnResponse<String> response = downloadUtils.downLoad(decodedUrl, suffix, null);
|
||
if (0 != response.getCode()) {
|
||
model.addAttribute("msg", response.getMsg());
|
||
return "fileNotSupported";
|
||
}
|
||
filePath = response.getContent();
|
||
}
|
||
String outFilePath = fileDir + pdfName;
|
||
if (StringUtils.hasText(outFilePath)) {
|
||
officeToPdf.openOfficeToPDF(filePath, outFilePath);
|
||
File f = new File(filePath);
|
||
if (f.exists()) {
|
||
f.delete();
|
||
}
|
||
if (isHtml) {
|
||
// 对转换后的文件进行操作(改变编码方式)
|
||
fileUtils.doActionConvertedFile(outFilePath);
|
||
}
|
||
// 加入缓存
|
||
fileUtils.addConvertedFile(pdfName, fileUtils.getRelativePath(outFilePath));
|
||
}
|
||
}
|
||
model.addAttribute("pdfUrl", pdfName);
|
||
return isHtml ? "html" : "pdf";
|
||
}else {
|
||
model.addAttribute("msg", "系统还不支持该格式文件的在线预览，" +
|
||
"如有需要请按下方显示的邮箱地址联系系统维护人员");
|
||
return "fileNotSupported";
|
||
}
|
||
}
|
||
|
||
private String suffixFromUrl(String url) {
|
||
String nonPramStr = url.substring(0, url.indexOf("?") != -1 ? url.indexOf("?"): url.length());
|
||
String fileName = nonPramStr.substring(nonPramStr.lastIndexOf("/") + 1);
|
||
String fileType = fileName.substring(fileName.lastIndexOf(".") + 1);
|
||
return fileType;
|
||
}
|
||
|
||
/**
|
||
* 查看文件类型(防止参数中存在.点号或者其他特殊字符，所以先抽取文件名，然后再获取文件类型)
|
||
* @param url
|
||
* @return
|
||
*/
|
||
private String typeFromUrl(String url) {
|
||
String nonPramStr = url.substring(0, url.indexOf("?") != -1 ? url.indexOf("?"): url.length());
|
||
String fileName = nonPramStr.substring(nonPramStr.lastIndexOf("/") + 1);
|
||
String fileType = fileName.substring(fileName.lastIndexOf(".") + 1);
|
||
if (fileUtils.listPictureTypes().contains(fileType.toLowerCase())) {
|
||
fileType = "picture";
|
||
}
|
||
if (fileUtils.listArchiveTypes().contains(fileType.toLowerCase())) {
|
||
fileType = "compress";
|
||
}
|
||
if (fileUtils.listOfficeTypes().contains(fileType.toLowerCase())) {
|
||
fileType = "office";
|
||
}
|
||
if (Arrays.asList(simText).contains(fileType.toLowerCase())) {
|
||
fileType = "simText";
|
||
}
|
||
return fileType;
|
||
}
|
||
|
||
/**
|
||
* 根据url获取文件内容
|
||
* 当pdfjs读取存在跨域问题的文件时将通过此接口读取
|
||
* @param urlPath
|
||
* @param resp
|
||
*/
|
||
@RequestMapping(value = "/getCorsFile", method = RequestMethod.GET)
|
||
public void getCorsFile(String urlPath, HttpServletResponse resp) {
|
||
InputStream inputStream = null;
|
||
try {
|
||
String strUrl = urlPath.trim();
|
||
URL url=new URL(strUrl);
|
||
//打开请求连接
|
||
URLConnection connection = url.openConnection();
|
||
HttpURLConnection httpURLConnection=(HttpURLConnection) connection;
|
||
httpURLConnection.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
|
||
inputStream = httpURLConnection.getInputStream();
|
||
byte[] bs = new byte[1024];
|
||
int len;
|
||
while(-1 != (len = inputStream.read(bs))) {
|
||
resp.getOutputStream().write(bs, 0, len);
|
||
}
|
||
} catch (IOException e) {
|
||
e.printStackTrace();
|
||
} finally {
|
||
if(inputStream != null) {
|
||
IOUtils.closeQuietly(inputStream);
|
||
}
|
||
}
|
||
}
|
||
|
||
}
|