OnlyOffice/src/main/java/com/zdd/controller/FileController.java

package com.zdd.controller;

import com.zdd.config.IpConfig;
import com.zdd.domain.FileItem;
import com.zdd.exceptions.APIException;
import com.zdd.exceptions.APIExceptionType;
import com.zdd.utils.OfficeManager;
import com.zdd.utils.OfficeUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.List;
import java.util.Map;

@Controller
@RequestMapping("/api/file/**")
public class FileController {

    private static Logger logger = LoggerFactory.getLogger(FileController.class);

    @Autowired
    private OfficeManager officeManager;

    @RequestMapping("/list")
    @ResponseBody
    public List<FileItem> getFiles() {
        return officeManager.getFiles();
    }

    @RequestMapping("/upload")
    @ResponseBody
    public void upload(@RequestParam("file") MultipartFile file) throws APIException, IOException {
        if (!file.isEmpty()) {
            if (!officeManager.checkExtname(file.getOriginalFilename())) {
                throw new APIException(APIExceptionType.FileExtError);
            }
            if (!officeManager.checkFileSize(file.getSize())) {
                throw new APIException(APIExceptionType.FileTooLarge);
            }
            officeManager.save(file.getInputStream(), file.getOriginalFilename());
        }
    }

    @RequestMapping("/download")
    public void download(String id, HttpServletResponse response) throws IOException {
        FileItem fileItem = officeManager.findById(id);
        if (fileItem != null) {
            response.setHeader("Content-Disposition", "attachment; filename=" + fileItem.getFileName());
            officeManager.read(response.getOutputStream(), fileItem.getFileName());
        }
    }


    @RequestMapping("/edit")
    public String onlineEdit(@RequestParam(name = "id", required = true) String id,
                             @RequestParam(name = "edit", required = false, defaultValue = "false") boolean edit,
                             String userId,
                             String userName,
                             String type, Map<String, Object> map) {
        FileItem file = officeManager.findById(id);
        String ip = IpConfig.getInetAddress().getHostAddress();
        String port = IpConfig.getPort();
        String localhost = "http://" + ip + ":" + port;
        if (file != null) {
            String fileName = file.getFileName();
            String fileType = file.getFileType();
            String[] split = fileName.split("\\.");
            map.put("url", localhost + "/localfile/" + fileName);
            map.put("apiurl", officeManager.getOfficeSettings().getApiUrl());
            map.put("fileName", fileName);
            map.put("fileType", fileType);
            map.put("callback", localhost + "/api/online/recallback/" + split[0] + "/" + split[1]);
            map.put("docType", OfficeUtils.GetFileType(file.getFileName()));
            map.put("key", id);
            map.put("userId", userId);
            map.put("userName", userName);
            map.put("type", type);
            map.put("edit", edit);
            return "/onlyoffice";
        }
        return "error";
    }

//    @RequestMapping("/edit")
//    public String onlineEdit(Map<String, Object> map,
//                             @RequestParam(name = "id", required = true) String id,
//                             @RequestParam(name = "edit", required = false, defaultValue = "false") boolean edit,
//                             String userId,
//                             String userName,
//                             String type,
//                             HttpServletRequest request) throws UnsupportedEncodingException {
//        FileItem file = officeManager.findById(id);
//        if (file != null) {
//            map.put("api", officeManager.getOfficeSettings().getApiUrl());
//            map.put("fileName", file.getFileName());
//            map.put("fileType", file.getFileType());
//            map.put("docType", OfficeUtils.GetFileType(file.getFileName()));
//            map.put("downloadUrl", ip + "/api/file/download?id=" + file.getId());
//            String callbackQuery = "?type=callback&fileName=" +
//                    URLEncoder.encode(file.getFileName(), "UTF-8") +
//                    "&userId" + URLEncoder.encode(userId, "UTF-8") +
//                    "&userName=" + URLEncoder.encode(userName, "UTF-8");
//            map.put("configCallback", ip + "/api/online/callback" + callbackQuery);
//            map.put("callback", ip + "/api/online/callback");
//            map.put("id", file.getId());
//            map.put("userId", userId);
//            map.put("userName", userName);
//            map.put("mode", edit ? "edit" : "view");
//            map.put("type", type);
//            return "/online";
//        }
//        return "";
//    }
}