package cn.iocoder.yudao.module.research.controller.admin.application;

import cn.hutool.core.collection.CollUtil;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.number.NumberUtils;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.research.controller.admin.application.vo.ResearchApplicationPageReqVO;
import cn.iocoder.yudao.module.research.controller.admin.application.vo.ResearchApplicationRespVO;
import cn.iocoder.yudao.module.research.controller.admin.application.vo.ResearchApplicationSaveReqVO;
import cn.iocoder.yudao.module.research.dal.dataobject.application.ResearchApplicationDO;
import cn.iocoder.yudao.module.research.dal.dataobject.guideline.ResearchGuidelineDO;
import cn.iocoder.yudao.module.research.service.application.ResearchApplicationService;
import cn.iocoder.yudao.module.research.service.application.ResearchApplicationServiceImpl;
import cn.iocoder.yudao.module.research.service.guideline.ResearchGuidelineService;
import cn.iocoder.yudao.module.system.api.storage.StorageAttachmentApi;
import cn.iocoder.yudao.module.system.api.storage.dto.StorageBlobRespDTO;
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import jakarta.validation.Valid;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;

import java.util.*;
import java.util.stream.Collectors;

import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;
import static cn.iocoder.yudao.framework.common.util.collection.MapUtils.findAndThen;

@Tag(name = "管理后台 - 课题申报")
@RestController
@RequestMapping("/research/application")
@Validated
public class ResearchApplicationController {

    public static final String APPLICATION_RECORD_TYPE =
            ResearchApplicationServiceImpl.APPLICATION_RECORD_TYPE;

    @Resource
    private ResearchApplicationService applicationService;

    @Resource
    private ResearchGuidelineService guidelineService;

    @Resource
    private AdminUserApi adminUserApi;

    @Resource
    private StorageAttachmentApi storageAttachmentApi;

    @PostMapping("/create")
    @Operation(summary = "创建课题申报书")
    @PreAuthorize("@ss.hasPermission('research:application:create')")
    public CommonResult<Long> createApplication(@Valid @RequestBody ResearchApplicationSaveReqVO createReqVO) {
        return success(applicationService.createApplication(createReqVO));
    }

    @PutMapping("/update")
    @Operation(summary = "更新课题申报书")
    @PreAuthorize("@ss.hasPermission('research:application:update')")
    public CommonResult<Boolean> updateApplication(@Valid @RequestBody ResearchApplicationSaveReqVO updateReqVO) {
        applicationService.updateApplication(updateReqVO);
        return success(true);
    }

    @DeleteMapping("/delete")
    @Operation(summary = "删除课题申报书")
    @Parameter(name = "id", description = "编号", required = true)
    @PreAuthorize("@ss.hasPermission('research:application:delete')")
    public CommonResult<Boolean> deleteApplication(@RequestParam("id") Long id) {
        applicationService.deleteApplication(id);
        return success(true);
    }

    @GetMapping("/get")
    @Operation(summary = "获得课题申报书")
    @Parameter(name = "id", description = "编号", required = true, example = "1024")
    @PreAuthorize("@ss.hasPermission('research:application:query')")
    public CommonResult<ResearchApplicationRespVO> getApplication(@RequestParam("id") Long id) {
        return success(buildApplicationDetail(applicationService.getApplication(id)));
    }

    @GetMapping("/page")
    @Operation(summary = "获得课题申报书分页")
    @PreAuthorize("@ss.hasPermission('research:application:query')")
    public CommonResult<PageResult<ResearchApplicationRespVO>> getApplicationPage(
            @Valid ResearchApplicationPageReqVO pageReqVO) {
        PageResult<ResearchApplicationDO> pageResult = applicationService.getApplicationPage(pageReqVO);
        return success(new PageResult<>(buildApplicationDetailList(pageResult.getList()), pageResult.getTotal()));
    }

    @PostMapping("/submit")
    @Operation(summary = "提交课题申报书审批")
    @PreAuthorize("@ss.hasPermission('research:application:submit')")
    public CommonResult<Boolean> submitApplication(@RequestParam("id") Long id,
                                                   @RequestParam("processDefinitionKey") String processDefinitionKey) {
        applicationService.submitApplication(id, processDefinitionKey);
        return success(true);
    }

    @GetMapping("/approve-process-list")
    @Operation(summary = "获取课题立项审批流程列表")
    @PreAuthorize("@ss.hasPermission('research:application:query')")
    public CommonResult<List<Map<String, Object>>> getApproveProcessDefinitionList() {
        return success(applicationService.getApproveProcessDefinitionList());
    }

    @PostMapping("/publish-notice")
    @Operation(summary = "发布立项通知")
    @Parameter(name = "id", description = "编号", required = true)
    @PreAuthorize("@ss.hasPermission('research:application:notice')")
    public CommonResult<Boolean> publishNotice(@RequestParam("id") Long id) {
        applicationService.publishNotice(id);
        return success(true);
    }

    // ==================== 内部方法 ====================

    private ResearchApplicationRespVO buildApplicationDetail(ResearchApplicationDO application) {
        if (application == null) {
            return null;
        }
        return buildApplicationDetailList(Collections.singletonList(application)).get(0);
    }

    private List<ResearchApplicationRespVO> buildApplicationDetailList(List<ResearchApplicationDO> list) {
        if (CollUtil.isEmpty(list)) {
            return Collections.emptyList();
        }

        // 1.1 获取创建人
        Set<Long> creatorIds = convertSet(list, application -> NumberUtils.parseLong(application.getCreator()));
        Map<Long, AdminUserRespDTO> creatorMap = adminUserApi.getUserMap(creatorIds);
        // 1.2 获取附件
        Map<Long, List<StorageBlobRespDTO>> attachmentMap = list.stream()
                .collect(Collectors.toMap(ResearchApplicationDO::getId,
                        r -> storageAttachmentApi.listAttachments(APPLICATION_RECORD_TYPE, r.getId())));
        // 1.3 获取指南名称
        Set<Long> guidelineIds = convertSet(list, ResearchApplicationDO::getGuidelineId);
        Map<Long, ResearchGuidelineDO> guidelineMap = guidelineService.getGuidelineList(guidelineIds).stream()
                .collect(Collectors.toMap(ResearchGuidelineDO::getId, g -> g));

        // 2. 拼接结果
        return BeanUtils.toBean(list, ResearchApplicationRespVO.class, (vo) -> {
            findAndThen(creatorMap, NumberUtils.parseLong(vo.getCreator()),
                    user -> vo.setCreatorName(user.getNickname()));
            List<StorageBlobRespDTO> attachments = attachmentMap.getOrDefault(vo.getId(), Collections.emptyList());
            vo.setAttachmentList(attachments);
            vo.setHasAttachment(CollUtil.isNotEmpty(attachments));
            if (vo.getGuidelineId() != null) {
                findAndThen(guidelineMap, vo.getGuidelineId(), g -> vo.setGuidelineTitle(g.getTitle()));
            }
        });
    }

}