package com.iplatform.file.support;

import com.iplatform.base.util.UserUtils;
import com.iplatform.file.FileStoreCallback;
import com.iplatform.file.service.FileServiceImpl;
import com.iplatform.file.util.ImageUtils;
import com.iplatform.model.po.S_user_core;
import com.iplatform.model.po.SfAttachment;
import com.iplatform.model.po.SfAttachment_mapper;
import com.walker.file.DefaultFileInfo;
import com.walker.file.FileInfo;
import com.walker.file.FileStoreType;
import com.walker.infrastructure.utils.DateUtils;
import com.walker.infrastructure.utils.StringUtils;
import com.walker.web.UserPrincipal;
import com.walker.web.WebRuntimeException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;

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

public class AttachmentJdbcCallback implements FileStoreCallback {

    protected final transient Logger logger = LoggerFactory.getLogger(getClass());

    @Override
    public void onCallback(FileInfo fileInfo, FileStoreType fileStoreType) {
        if(logger.isDebugEnabled()){
            logger.debug("保存文件:{}", fileInfo);
        }
        String fileExt = fileInfo.getFileExt();

        SfAttachment attachment = new SfAttachment();
        attachment.setAttId(Long.parseLong(fileInfo.getId()));
        attachment.setCreateTime(DateUtils.getDateTimeNumber());
        attachment.setUpdateTime(attachment.getCreateTime());

        // 2023-08-01
        try{
            UserPrincipal<S_user_core> userPrincipal = UserUtils.getCurrentUserPrincipal();
            if(userPrincipal != null){
                attachment.setUserName(userPrincipal.getUserName());
            } else {
                attachment.setUserName("前端用户");
            }
        } catch (WebRuntimeException ex){
            logger.debug("未找到当前用户，文件记录人为空");
            attachment.setUserName("未知用户");
        }

        attachment.setName(fileInfo.getFileName());
        attachment.setSattDir(fileInfo.getUrl());
        attachment.setAttType(fileExt);   // 这里目前是后缀名，后续可参考电商系统存放:mimeType
        attachment.setAttSize(fileInfo.getFileSize());
        attachment.setFileStoreType(fileStoreType.getIndex());
        if(StringUtils.isNotEmpty(fileInfo.getGroupId())){
            attachment.setGroupId(fileInfo.getGroupId());
        }
        if(ImageUtils.isImage(fileExt) || ImageUtils.isVideo(fileExt)){
            // 暂时存放为1，表示媒体格式
            attachment.setImageType(1);
        } else {
            // 非媒体格式
            attachment.setImageType(0);
        }
        // pid 和 owner 需要传过来
        attachment.setPid(fileInfo.getBusinessType());
        String owner = fileInfo.getOwner();
        if(StringUtils.isEmpty(owner)){
            attachment.setOwner(-1);
        } else {
            attachment.setOwner(Integer.parseInt(owner));
        }
        this.fileService.insert(attachment);
    }

    @Override
    public FileInfo onAcquireFileInfo(String id) {
        SfAttachment attachment = this.fileService.get(new SfAttachment(Long.parseLong(id)));
        if(attachment == null){
            return null;
        }
        return this.toFileInfo(attachment);
    }

    @Override
    public List<FileInfo> onAcquireFileInfoList(List<String> ids) {
        if(StringUtils.isEmptyList(ids)){
            return null;
        }
        MapSqlParameterSource sqlParameterSource = new MapSqlParameterSource();
        sqlParameterSource.addValue("ids", ids);
        List<SfAttachment> list = this.fileService.sqlListObjectWhereIn("select * from sf_attachment where att_id in(:ids)"
                , SfAttachment_mapper.ROW_MAPPER, sqlParameterSource);
        if(StringUtils.isEmptyList(list)){
            return null;
        }
        List<FileInfo> fileInfoList = new ArrayList<>(list.size());
        for(SfAttachment s_file : list){
            fileInfoList.add(this.toFileInfo(s_file));
        }
        return fileInfoList;
    }

    private FileInfo toFileInfo(SfAttachment attachment){
        if(attachment == null){
            return null;
        }
        DefaultFileInfo fileInfo = new DefaultFileInfo();
        fileInfo.setId(String.valueOf(attachment.getAttId()));
        fileInfo.setFileSize(attachment.getAttSize());
        fileInfo.setUrl(attachment.getSattDir());
        fileInfo.setFileExt(attachment.getAttType());
        fileInfo.setFileName(attachment.getName());
        fileInfo.setFileStoreType(FileStoreType.getType(attachment.getFileStoreType()));
        fileInfo.setGroupId(attachment.getGroupId());
        fileInfo.setBusinessType(attachment.getPid());
        fileInfo.setOwner(String.valueOf(attachment.getOwner()));
        return fileInfo;
    }

    public void setFileService(FileServiceImpl fileService) {
        this.fileService = fileService;
    }

    private FileServiceImpl fileService;

}