package com.njcn.rdms.module.project.dal.mysql.status;

import com.njcn.rdms.framework.common.pojo.PageResult;
import com.njcn.rdms.framework.mybatis.core.mapper.BaseMapperX;
import com.njcn.rdms.framework.mybatis.core.query.LambdaQueryWrapperX;
import com.njcn.rdms.module.project.controller.admin.status.vo.model.ObjectStatusModelPageReqVO;
import com.njcn.rdms.module.project.dal.dataobject.status.ObjectStatusModelDO;
import org.apache.ibatis.annotations.Delete;
import org.apache.ibatis.annotations.Mapper;
import org.springframework.util.StringUtils;

import java.util.List;
import java.util.stream.Collectors;

@Mapper
public interface ObjectStatusModelMapper extends BaseMapperX<ObjectStatusModelDO> {

    default PageResult<ObjectStatusModelDO> selectPage(ObjectStatusModelPageReqVO reqVO) {
        LambdaQueryWrapperX<ObjectStatusModelDO> queryWrapper = new LambdaQueryWrapperX<>();
        if (StringUtils.hasText(reqVO.getKeyword())) {
            queryWrapper.and(wrapper -> wrapper.like(ObjectStatusModelDO::getStatusCode, reqVO.getKeyword())
                    .or()
                    .like(ObjectStatusModelDO::getStatusName, reqVO.getKeyword()));
        }
        queryWrapper.eqIfPresent(ObjectStatusModelDO::getObjectType, reqVO.getObjectType())
                .eqIfPresent(ObjectStatusModelDO::getStatus, reqVO.getStatus())
                .eqIfPresent(ObjectStatusModelDO::getInitialFlag, reqVO.getInitialFlag())
                .eqIfPresent(ObjectStatusModelDO::getTerminalFlag, reqVO.getTerminalFlag())
//                .betweenIfPresent(BaseDO::getUpdateTime, reqVO.getUpdateTime())
//                .orderByAsc(ObjectStatusModelDO::getObjectType)
                .orderByAsc(ObjectStatusModelDO::getSort)
                .orderByAsc(ObjectStatusModelDO::getUpdateTime);
        return selectPage(reqVO, queryWrapper);
    }

    default ObjectStatusModelDO selectByObjectTypeAndStatusCode(String objectType, String statusCode) {
        return selectOne(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getStatusCode, statusCode));
    }

    default ObjectStatusModelDO selectByObjectTypeAndStatusCodeEnabled(String objectType, String statusCode) {
        return selectOne(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getStatusCode, statusCode)
                .eq(ObjectStatusModelDO::getStatus, 0));
    }

    default ObjectStatusModelDO selectInitialByObjectTypeEnabled(String objectType) {
        return selectOne(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getInitialFlag, true)
                .eq(ObjectStatusModelDO::getStatus, 0));
    }

    default ObjectStatusModelDO selectInitialByObjectType(String objectType) {
        return selectOne(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getInitialFlag, true));
    }


    default List<ObjectStatusModelDO> selectListByObjectType(String objectType) {
        return selectList(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .orderByAsc(ObjectStatusModelDO::getSort));
    }

    default List<ObjectStatusModelDO> selectListByObjectTypeEnabled(String objectType) {
        return selectList(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getStatus, 0)
                .orderByAsc(ObjectStatusModelDO::getSort));
    }

    /**
     * 查询某对象类型下所有已启用的终态状态码。
     */
    default List<String> selectTerminalStatusCodesByObjectTypeEnabled(String objectType) {
        return selectList(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getStatus, 0)
                .eq(ObjectStatusModelDO::getTerminalFlag, true))
                .stream()
                .map(ObjectStatusModelDO::getStatusCode)
                .collect(Collectors.toList());
    }

    /**
     * 查询某对象类型下所有已启用、且不参与上层进度统计的状态码。
     */
    default List<String> selectProgressExcludedStatusCodesByObjectTypeEnabled(String objectType) {
        return selectList(new LambdaQueryWrapperX<ObjectStatusModelDO>()
                .eq(ObjectStatusModelDO::getObjectType, objectType)
                .eq(ObjectStatusModelDO::getStatus, 0)
                .eq(ObjectStatusModelDO::getProgressExcludedFlag, true))
                .stream()
                .map(ObjectStatusModelDO::getStatusCode)
                .collect(Collectors.toList());
    }

    /**
     * 物理删除
     */
    @Delete("DELETE FROM rdms_object_status_model WHERE id = #{id}")
    int physicalDeleteById(Long id);
}
