file-online-preview/server/src/main/java/cn/keking/service/impl/PictureFilePreviewImpl.java

package cn.keking.service.impl;

import cn.keking.model.FileAttribute;
import cn.keking.service.FileHandlerService;
import cn.keking.utils.KkFileUtils;
import org.springframework.stereotype.Service;
import org.springframework.ui.Model;
import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.List;

/**
 * Created by kl on 2018/1/17.
 * Content :图片文件处理
 */
@Service
public class PictureFilePreviewImpl extends CommonPreviewImpl {

    private final FileHandlerService fileHandlerService;

    public PictureFilePreviewImpl(FileHandlerService fileHandlerService, OtherFilePreviewImpl otherFilePreview) {
        super(fileHandlerService, otherFilePreview);
        this.fileHandlerService = fileHandlerService;
    }

    @Override
    public String filePreviewHandle(String url, Model model, FileAttribute fileAttribute) {
        url= KkFileUtils.htmlEscape(url);
        String suffix = fileAttribute.getSuffix();
        List<String> imgUrls = new ArrayList<>();
        imgUrls.add(url);
        String compressFileKey = fileAttribute.getCompressFileKey();
        List<String> zipImgUrls = fileHandlerService.getImgCache(compressFileKey);
        if (!CollectionUtils.isEmpty(zipImgUrls)) {
            imgUrls.addAll(zipImgUrls);
            model.addAttribute("imgUrls", imgUrls);
        }else {
            // 不是http开头，浏览器不能直接访问，需下载到本地
            super.filePreviewHandle(url, model, fileAttribute);
            if ( url.toLowerCase().startsWith("file") || url.toLowerCase().startsWith("ftp")) {
                model.addAttribute("imgUrls", fileAttribute.getName());
            }else {
                model.addAttribute("imgUrls", url);
            }

        }
        if(suffix.equalsIgnoreCase("heic")){
            return HEIC_FILE_PREVIEW_PAGE;
        }else {
            return PICTURE_FILE_PREVIEW_PAGE;
        }
    }
}