ieplus/ie-admin/src/main/java/com/ruoyi/web/controller/dz/DzCardsController.java

package com.ruoyi.web.controller.dz;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
import javax.servlet.http.HttpServletResponse;
import javax.validation.ValidationException;

import com.alibaba.fastjson2.JSONArray;
import com.alibaba.fastjson2.JSONObject;
import com.google.common.collect.Lists;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.enums.ExamType;
import com.ruoyi.common.enums.UserRegStatus;
import com.ruoyi.common.utils.NumberUtils;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.common.utils.bean.BeanUtils;
import com.ruoyi.dz.domain.DzAgent;
import com.ruoyi.dz.service.IDzAgentService;
import com.ruoyi.enums.CardAction;
import com.ruoyi.enums.CardType;
import com.ruoyi.enums.UserTypeEnum;
import com.ruoyi.system.service.ISysUserService;
import com.ruoyi.web.domain.CardUserBody;
import com.ruoyi.web.domain.DzCardExport;
import com.ruoyi.web.service.SysLoginService;
import com.ruoyi.web.service.SysRegisterService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import org.apache.commons.lang3.ArrayUtils;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.CollectionUtils;
import org.springframework.web.bind.annotation.*;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.dz.domain.DzCards;
import com.ruoyi.dz.service.IDzCardsService;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.common.core.page.TableDataInfo;

/**
 * 学习卡Controller
 *
 * @author ruoyi
 * @date 2025-09-12
 */
@RestController
@RequestMapping("/dz/cards")
@Api(tags = "后台-学习卡管理")
public class DzCardsController extends BaseController
{
    @Autowired
    private IDzCardsService dzCardsService;
    @Autowired
    private IDzAgentService agentService;
    @Autowired
    private ISysUserService sysUserService;
    @Autowired
    private SysLoginService sysLoginService;
    @Autowired
    private SysRegisterService sysRegisterService;

    /**
     * 查询学习卡列表
     */
    @PreAuthorize("@ss.hasPermi('dz:cards:list')")
    @GetMapping("/list")
    @ApiOperation("列表")
    public TableDataInfo list(DzCards dzCards)
    {
        startPage();
        List<DzCards> list = dzCardsService.selectDzCardsList(prepare(dzCards));
        return getDataTable(list);
    }

    private DzCards prepare(DzCards dzCards) {
        SysUser sysUser = SecurityUtils.getLoginUser().getUser();
        String userType = sysUser.getUserType();
        if(UserTypeEnum.isInstitution(userType)){
            dzCards.setDeptId(SecurityUtils.getDeptId());
        } else if(UserTypeEnum.isSchool(userType)){
            dzCards.setCampusId(sysUser.getUserTypeId());
        }
        if((UserTypeEnum.isAgent(userType)||UserTypeEnum.isTeacher(userType))&&null==dzCards.getAgentId()){
            //代理商及老师只能查看分配给自己的卡
            DzAgent agent = agentService.selectDzAgentByUserId(SecurityUtils.getUserId());
            if (null!=agent){
                if(NumberUtils.isPositive(agent.getParentId())) {
                    dzCards.setLeafAgentId(agent.getAgentId());
                } else {
                    dzCards.setAgentId(agent.getAgentId());
                }
            }
        }
        return dzCards;
    }

    /**
     * 导出学习卡列表
     */
    @PreAuthorize("@ss.hasPermi('dz:cards:export')")
    @Log(title = "学习卡", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    public void export(HttpServletResponse response, DzCards dzCards)
    {
        List<DzCards> list = dzCardsService.selectDzCardsList(prepare(dzCards));
        for(DzCards dc : list) {
            dc.getDeptId();

        }
        ExcelUtil<DzCardExport> util = new ExcelUtil<DzCardExport>(DzCardExport.class);
        util.exportExcel(response, list.stream().map(t -> {
            DzCardExport exp = new DzCardExport();
            BeanUtils.copyProperties(t, exp);
            return exp;
        }).collect(Collectors.toList()), "学习卡数据");
    }

    /**
     * 获取学习卡详细信息
     */
    @PreAuthorize("@ss.hasPermi('dz:cards:query')")
    @GetMapping(value = "/{cardId}")
    @ApiOperation("详情")
    public AjaxResult getInfo(@PathVariable("cardId") Long cardId)
    {
        return success(dzCardsService.selectDzCardsByCardId(cardId));
    }

    /**
     * 新增学习卡
     */
    @PreAuthorize("@ss.hasPermi('dz:cards:add')")
    @Log(title = "学习卡", businessType = BusinessType.INSERT)
    @PostMapping
    public AjaxResult add(@RequestBody DzCards dzCards)
    {
        return toAjax(dzCardsService.insertDzCards(dzCards));
    }

    /**
     * 修改学习卡
     */
    @PreAuthorize("@ss.hasPermi('dz:cards:edit')")
    @Log(title = "学习卡", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult edit(@RequestBody DzCards dzCards)
    {
        return toAjax(dzCardsService.updateDzCards(dzCards));
    }

    /**
     * 删除学习卡
     */
    @PreAuthorize("@ss.hasPermi('dz:cards:remove')")
    @Log(title = "学习卡", businessType = BusinessType.DELETE)
	@DeleteMapping("/{cardIds}")
    public AjaxResult remove(@PathVariable Long[] cardIds)
    {
        return toAjax(dzCardsService.deleteDzCardsByCardIds(cardIds));
    }

    @Log(title = "制卡", businessType = BusinessType.INSERT)
    @PostMapping("/issueCard")
    @ApiOperation("制卡")
    public AjaxResult issueCard(@ApiParam("机构") Long institutionId, @ApiParam("卡类型") String type, @ApiParam("数量") Integer count)
    {
        dzCardsService.issueCard(institutionId, getCardType(institutionId, type), count);
        return AjaxResult.success();
    }

    @Log(title = "分配开卡", businessType = BusinessType.INSERT)
    @PostMapping("/assignCard")
    @ApiOperation("分配即开卡")
    public AjaxResult assignCard(@ApiParam("代理商") @RequestParam Long agentId, @ApiParam("卡类型") String type, @ApiParam(value = "卡类型") Integer cardType,
                                 @ApiParam("开始号") @RequestParam String begin, @ApiParam("结束号") @RequestParam String end,
                                 @ApiParam("省份") @RequestParam(required = false) String location,
                                 @ApiParam("考生类型") @RequestParam(required = false) ExamType examType,
                                 @ApiParam("学校") @RequestParam(required = false) Long schoolId,
                                 @ApiParam("有效天数") @RequestParam(required = false) Integer days)
    {
        SysUser sysUser = SecurityUtils.getLoginUser().getUser();
        DzAgent agent = agentService.selectDzAgentByAgentId(agentId);
        CardType ct = getCardType(agent.getDeptId(), null != cardType ? cardType.toString() : type);
        if(UserTypeEnum.Agent.getVal().equals(sysUser.getUserType())) { // 代理商分配
            if (!sysUser.getUserTypeId().equals(agent.getParentId())) {
                throw new ValidationException("只能分配给下级代理");
            }
            dzCardsService.assignCard(sysUser.getUserTypeId(), agentId, ct, begin, end, location, examType, schoolId, days);
        } else if(UserTypeEnum.Institution.getVal().equals(sysUser.getUserType())) { // 机构分配卡给自己的下级
            if (!sysUser.getDeptId().equals(agent.getDeptId())) {
                throw new ValidationException("只能分配给下级代理");
            }
            dzCardsService.assignCard(NumberUtils.isPositive(agent.getParentId()) ? agent.getParentId() : agentId, agentId, ct, begin, end, location, examType, schoolId, days);
        } else { // 平台指定, TODO 暂只支持二级代理
            dzCardsService.assignCard(NumberUtils.isPositive(agent.getParentId()) ? agent.getParentId() : agentId, agentId, ct, begin, end, location, examType, schoolId, days);
        }
        return AjaxResult.success();
    }

    @Log(title = "直接开卡", businessType = BusinessType.INSERT)
    @PostMapping("/openCard")
    @ApiOperation("直接开卡")
    public AjaxResult openCard(@ApiParam("代理商") Long agentId, @ApiParam("省份") String location, @ApiParam("学校") Long schoolId, @ApiParam("开始号") String begin, @ApiParam("结束号") String end,
                               @ApiParam("有效天数") @RequestParam(required = false) Integer days)
    {
        return AjaxResult.error("暂不支持，现在分配即开卡"); // success(dzCardsService.openCard(agentId, location, schoolId, begin, end, days));
    }

    @Log(title = "分配校区", businessType = BusinessType.INSERT)
    @PostMapping("/changeCampus")
    @ApiOperation("分配校区")
    public AjaxResult changeCampus(@ApiParam("校区") Long campusId, @ApiParam("开始号") String begin, @ApiParam("结束号") String end)
    {
        return AjaxResult.success(dzCardsService.changeCampus(campusId, begin, end));
    }

    @Log(title = "修改卡", businessType = BusinessType.UPDATE)
    @PostMapping("/changeCard")
    @ApiOperation("修改卡 重卡/关卡/支付/续费/结算")
    public AjaxResult changeCard(@ApiParam("操作") @RequestParam CardAction action, @ApiParam("卡ID") @RequestParam Long[] cardIds)
    {
        dzCardsService.changeCard(action, cardIds);
        if (CardAction.Close.equals(action) && ArrayUtils.isNotEmpty(cardIds)) {
            sysLoginService.resetTokens(sysUserService.selectUserByCardIds(Arrays.asList(cardIds)));
        } else if (CardAction.ReOpen.equals(action) && ArrayUtils.isNotEmpty(cardIds)) {
            List<SysUser> sysUsers = sysUserService.selectUserByCardIds(Arrays.asList(cardIds));
            for (SysUser user : sysUsers) {
                if(UserRegStatus.User.equals(user.getRegStatus())) {
                    SysUser upUser = new SysUser();
                    upUser.setUserId(user.getUserId());
                    upUser.setRegStatus(UserRegStatus.Student);
                    sysUserService.updateUserProfile(upUser);
                }
            }
            sysLoginService.resetTokens(sysUsers);
        }
        return AjaxResult.success();
    }

    @Log(title = "修改卡用户信息", businessType = BusinessType.UPDATE)
    @PostMapping("/updateCardUser")
    @ApiOperation("修改卡用户信息")
    public AjaxResult updateCardUser(@RequestBody CardUserBody cardUserBody) {
        sysRegisterService.improve(cardUserBody);
        return AjaxResult.success();
    }

    @GetMapping(value = "/cardUser/{cardId}")
    @ApiOperation("卡用户详情")
    public AjaxResult getCardUser(@PathVariable("cardId") Long cardId)
    {
        CardUserBody cardUserBody = new CardUserBody();
        DzCards dzCards = dzCardsService.selectDzCardsByCardId(cardId);
        if(null == dzCards) {
            throw new ValidationException("卡号无效");
        }
        List<SysUser> sysUserList = sysUserService.selectUserByCardIds(Lists.newArrayList(cardId));
        if(CollectionUtils.isEmpty(sysUserList)) {
            throw new ValidationException("卡用户无效");
        }
        SysUser sysUser = sysUserList.get(0);
        BeanUtils.copyProperties(sysUser, cardUserBody, "password", "schoolId", "classId");
        cardUserBody.setMobile(sysUser.getPhonenumber());
        if (null != sysUser.getDirectedStudy()) {
            cardUserBody.setDirectionStudy(JSONArray.parse(sysUser.getDirectedStudy()));
        }
        BeanUtils.copyProperties(dzCards, cardUserBody, "password", "nickName", "location", "examType", "endYear");
        return success(cardUserBody);
    }

    private CardType getCardType(Long institutionId, String type) {
        if(!NumberUtils.isNumeric(type)) {
            return CardType.valueOf(type);
        }
        if(Integer.parseInt(type) == 9) {
            return CardType.Experience;
        }
        return (null == institutionId || institutionId.equals(101L)) ? CardType.Platform : CardType.Dept;
    }

}