package com.ruoyi.web.controller.system;

import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.page.TableDataInfo;
import com.ruoyi.system.domain.InvitationBlacklist;
import com.ruoyi.system.domain.IpBlacklist;
import com.ruoyi.system.service.impl.InvitationBlacklistServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.Date;
import java.util.List;

@RestController
@RequestMapping("/system/Invitation")
public class InvitationBlacklistController extends BaseController {

    @Autowired
    InvitationBlacklistServiceImpl service;

    /**
     * 邀请码黑名单列表
     */
//    @PreAuthorize("@ss.hasPermi('im:group:list')")
    @GetMapping("/list")
    public TableDataInfo list(@RequestParam(value = "keywords", required = false) String keywords)
    {
        // 创建查询条件包装器
        LambdaQueryWrapper<InvitationBlacklist> queryWrapper = new LambdaQueryWrapper<>();

        // 只有当 keyword 不为空时才添加 OR 条件
        if (ObjectUtil.isNotEmpty(keywords)) {
            queryWrapper.and(wrapper -> wrapper.eq(InvitationBlacklist::getInvitationCode,keywords)
            );
        }

        // 默认按创建时间倒序
        queryWrapper.orderByDesc(InvitationBlacklist::getCreateTime);
        startPage();
        List<InvitationBlacklist> list = service.list(queryWrapper);
        return getDataTable(list);
    }


    /**
     * 新增邀请码黑名单
     */
    @PostMapping("/addCode")
    public AjaxResult addIp(@RequestParam("code") String code) {

        long count = service.count(new LambdaQueryWrapper<>(InvitationBlacklist.class).eq(InvitationBlacklist::getInvitationCode, code));
        if(count > 0){
            return AjaxResult.error("邀请码已存在！");
        }
        InvitationBlacklist invitationBlacklist = new InvitationBlacklist();
        invitationBlacklist.setInvitationCode(code);
        invitationBlacklist.setCreateTime(new Date());
        service.save(invitationBlacklist);
        return AjaxResult.success("添加成功");
    }

    /**
     * 删除邀请码黑名单
     */
    @PostMapping("/deleteCode")
    public AjaxResult deleteCode(@RequestParam("id") Integer id) {

        InvitationBlacklist invitationBlacklist  = service.getById(id);
        if(ObjectUtil.isEmpty(invitationBlacklist)){
            return AjaxResult.error("邀请码不存在！");
        }
        service.removeById(id);
        return AjaxResult.success("删除成功");
    }


}