package org.jeecg.modules.doc.controller;

import com.alibaba.fastjson.JSON;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.apache.shiro.SecurityUtils;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.system.vo.LoginUser;
import org.jeecg.modules.doc.component.AsyncTaskComp;
import org.jeecg.modules.doc.component.DocFileDealComp;
import org.jeecg.modules.doc.component.PathPermissionComp;
import org.jeecg.modules.doc.constant.Constant;
import org.jeecg.modules.doc.dto.BatchDeleteRecoveryFileDTO;
import org.jeecg.modules.doc.dto.DeleteRecoveryFileDTO;
import org.jeecg.modules.doc.entity.DocFilePath;
import org.jeecg.modules.doc.entity.RecoveryFile;
import org.jeecg.modules.doc.dto.RestoreFileDTO;
import org.jeecg.modules.doc.service.IRecoveryFileService;
import org.jeecg.modules.doc.vo.RecoveryFileListVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

@Slf4j
@RestController
@RequestMapping("/recoveryfile")
public class RecoveryFileController {

    @Autowired
    private IRecoveryFileService recoveryFileService;

    @Resource
    AsyncTaskComp asyncTaskComp;

    @Autowired
    PathPermissionComp pathPermissionComp;

    @Autowired
    DocFileDealComp fileDealComp;


    @ApiOperation(value = "删除回收文件", notes = "删除回收文件", tags = {"recoveryfile"})
    @RequestMapping(value = "/deleterecoveryfile", method = RequestMethod.POST)
    @ResponseBody
    public Result<String> deleteRecoveryFile(@RequestBody DeleteRecoveryFileDTO deleteRecoveryFileDTO) {

        RecoveryFile recoveryFile = recoveryFileService.getById(deleteRecoveryFileDTO.getRecoveryFileId());

        asyncTaskComp.deleteUserFile(recoveryFile.getPathId());

        recoveryFileService.removeById(deleteRecoveryFileDTO.getRecoveryFileId());
        return Result.OK("删除成功");
    }



    @ApiOperation(value = "批量删除回收文件", notes = "批量删除回收文件", tags = {"recoveryfile"})
    @RequestMapping(value = "/batchdelete", method = RequestMethod.POST)
    @ResponseBody
    public Result<String> batchDeleteRecoveryFile(@RequestBody BatchDeleteRecoveryFileDTO batchDeleteRecoveryFileDTO) {
        List<RecoveryFile> recoveryFileList = JSON.parseArray(batchDeleteRecoveryFileDTO.getRecoveryFileIds(), RecoveryFile.class);
        for (RecoveryFile recoveryFile : recoveryFileList) {
            RecoveryFile recoveryFile1 = recoveryFileService.getById(recoveryFile.getRecoveryFileId());

            if (recoveryFile1 != null) {
                asyncTaskComp.deleteUserFile(recoveryFile1.getPathId());

                recoveryFileService.removeById(recoveryFile1.getRecoveryFileId());
            }

        }
        return Result.OK("批量删除成功");
    }

    @ApiOperation(value = "回收文件列表", notes = "回收文件列表", tags = {"recoveryfile"})
    @RequestMapping(value = "/list", method = RequestMethod.POST)
    @ResponseBody
    public Map<String, Object> getRecoveryFileList() {
        Map<String, Object> map = new HashMap<>();
        LoginUser loginUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
        // 访问、下载、管理权限包
        Map<String, Map<String, DocFilePath>> pathPermission = pathPermissionComp.getPathPermission();
        // 拥有访问权限的文件id集合
        Map<String, DocFilePath> visits =  pathPermission.get(Constant.VISIT);
        // 拥有下载权限的文件id集合
        //Map<String, DocFilePath> downloads =  pathPermission.get(Constant.DOWNLOAD);
        // 拥有管理权限的文件id集合
        Map<String, DocFilePath> manages =  pathPermission.get(Constant.MANAGE);


        Result<List<RecoveryFileListVo>> restResult = new Result<>();
        List<RecoveryFileListVo> filePaths = recoveryFileService.selectRecoveryFileList(loginUser.getId());
        // 当前查询结果中可访问的文件集合
        filePaths = filePaths.stream().filter(
                i -> visits.get(i.getPathId())!=null || fileDealComp.isParentsOrChild(visits,i, true)
        ).collect(Collectors.toList());

//        Set<String> downloadSet = filePaths.stream().filter(
//                i -> downloads.get(i.getPathId()) != null || fileDealComp.isParentsOrChild(downloads,i, false)
//        ).map(DocFilePath::getPathId).collect(Collectors.toSet());
        Set<String> manageSet = filePaths.stream().filter(
                i -> manages.get(i.getPathId()) != null || fileDealComp.isParentsOrChild(manages,i, false)
        ).map(RecoveryFileListVo::getPathId).collect(Collectors.toSet());
        restResult.setResult(filePaths);
        restResult.setSuccess(true);
        map.put("result", filePaths);
        map.put("success", true);
        map.put("managePrem", manageSet);
        return map;
    }

    @ApiOperation(value = "还原文件", notes = "还原文件", tags = {"recoveryfile"})
    @RequestMapping(value = "/restorefile", method = RequestMethod.POST)
    @ResponseBody
    public Result restoreFile(@RequestBody RestoreFileDTO restoreFileDto) {
        LoginUser loginUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
        recoveryFileService.restorefile(restoreFileDto.getDeleteBatchNum(), restoreFileDto.getFilePath(), restoreFileDto.getPathId(), loginUser.getId());
        return Result.OK("还原成功！");
    }

}