iHotel_student_houtai/src/main/java/com/template/controller/AdminController.java

package com.template.controller;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.template.api.AdminAPI;
import com.template.common.utils.RSAUtils;
import com.template.config.MySecurity;
import com.template.model.dto.SaveAdminDto;
import com.template.model.dto.UpdateAdminDto;
import com.template.model.pojo.Admin;
import com.template.model.pojo.AdminMenu;
import com.template.model.pojo.PermissionSetting;
import com.template.model.pojo.Users;
import com.template.model.result.CommonResult;
import com.template.model.vo.AdminVo;
import com.template.services.AdminMenuService;
import com.template.services.AdminService;
import com.template.services.PermissionSettingService;
import com.template.services.UsersService;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.List;

/**
 * <p>
 * 前端控制器
 * </p>
 *
 * @author ceshi
 * @since 2023-12-13
 */
@RestController
public class AdminController implements AdminAPI {

    @Autowired
    AdminService adminService;

    @Autowired
    AdminMenuService adminMenuService;

    @Autowired
    PermissionSettingService permissionSettingService;

    @Autowired
    UsersService usersService;

    @Autowired
    MySecurity mySecurity;

    @Override
    public CommonResult getListPage(int permissionSettingId, int page, int size, String keyWord, String startTime, String endTime, String status) {
        PermissionSetting permissionSetting = permissionSettingService.getById(permissionSettingId);
        if (ObjectUtils.isEmpty(permissionSetting)) {
            return CommonResult.fail("非法进入");
        }
//      判断该用户是否拥有权限
        //      判断该用户是否拥有权限
        String accountManagement = permissionSetting.getAccountManagement();
        if (!"0".equals(accountManagement) && !accountManagement.contains("4")) {
            return CommonResult.fail("此账号暂无该权限");
        }


        if (ObjectUtils.isEmpty(page) && page <= 0) {
            page = 1;
        }

        if (ObjectUtils.isEmpty(size) && size <= 0) {
            size = 10;
        }


        LambdaQueryWrapper<Admin> wrapper = new LambdaQueryWrapper<>();
        wrapper.between(ObjectUtils.isNotEmpty(startTime) && ObjectUtils.isNotEmpty(endTime), Admin::getCreateTime, startTime, endTime)
                .eq(ObjectUtils.isNotEmpty(status), Admin::getStatus, status)
                .like(ObjectUtils.isNotEmpty(keyWord), Admin::getName, keyWord)
                .orderByDesc(Admin::getCreateTime);

        IPage<Admin> page1 = adminService.page(new Page<>(page, size), wrapper);
        Page<AdminVo> adminVoPage = new Page<>();
        List<Admin> records = page1.getRecords();
        ArrayList<AdminVo> adminVos = new ArrayList<>();
        for (Admin record : records) {
            AdminVo adminVo = new AdminVo();
            BeanUtils.copyProperties(record,adminVo);
            String adminMenuId = record.getAdminMenuId();
            AdminMenu adminMenu = adminMenuService.getById(adminMenuId);
            adminVo.setAdminMenu(adminMenu);
            Users byId = usersService.getById(record.getUsersId());
            adminVo.setCardNumber(byId.getCardNumber());
//            解密
            try {
                String decrypt = RSAUtils.decrypt(record.getPassWord(), RSAUtils.getPrivateKey(mySecurity.privateKey));
                adminVo.setPassWord(decrypt);
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
            adminVos.add(adminVo);
        }
        BeanUtils.copyProperties(page1,adminVoPage);
        adminVoPage.setRecords(adminVos);


        return CommonResult.ok(adminVoPage);
    }

    @Override
    public CommonResult save(SaveAdminDto saveAdminDto) {
        String permissionSettingId = saveAdminDto.getPermissionSettingId();

        PermissionSetting permissionSetting = permissionSettingService.getById(permissionSettingId);
        if (ObjectUtils.isEmpty(permissionSetting)) {
            return CommonResult.fail("非法进入");
        }
//      判断该用户是否拥有权限
        //      判断该用户是否拥有权限
        String accountManagement = permissionSetting.getAccountManagement();
        if (!"0".equals(accountManagement) && !accountManagement.contains("1")) {
            return CommonResult.fail("此账号暂无该权限");
        }

//        通过cardNumber找到对应的用户
        String cardNumber = saveAdminDto.getCardNumber();
        LambdaQueryWrapper<Users> wrapper=new LambdaQueryWrapper<>();
        wrapper.eq(Users::getCardNumber,cardNumber);
        Users users = usersService.getOne(wrapper);
        if (ObjectUtils.isEmpty(users)) {
            return CommonResult.fail("暂无该用户");
        }

//  密码加密
        String passWord = saveAdminDto.getPassWord();
        try {
            Admin admin = new Admin();
            String encrypt = RSAUtils.encrypt(passWord, RSAUtils.getPublicKey(mySecurity.publicKey));
            BeanUtils.copyProperties(saveAdminDto, admin);
            admin.setPassWord(encrypt);
            admin.setStatus("1");
            admin.setUsersId(users.getId()+"");
            adminService.save(admin);
            return CommonResult.ok();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }


    }

    @Override
    public CommonResult update(UpdateAdminDto updateAdminDto) {
        String permissionSettingId = updateAdminDto.getPermissionSettingId();

        PermissionSetting permissionSetting = permissionSettingService.getById(permissionSettingId);
        if (ObjectUtils.isEmpty(permissionSetting)) {
            return CommonResult.fail("非法进入");
        }
//      判断该用户是否拥有权限
        //      判断该用户是否拥有权限
        String accountManagement = permissionSetting.getAccountManagement();
        if (!"0".equals(accountManagement) && !accountManagement.contains("2")) {
            return CommonResult.fail("此账号暂无该权限");
        }


//        通过cardNumber找到对应的用户
        String cardNumber = updateAdminDto.getCardNumber();
        LambdaQueryWrapper<Users> wrapper=new LambdaQueryWrapper<>();
        wrapper.eq(Users::getCardNumber,cardNumber);
        Users users = usersService.getOne(wrapper);
        if (ObjectUtils.isEmpty(users)) {
            return CommonResult.fail("暂无该用户");
        }

        Admin admin = updateAdminDto.getAdmin();
        try {
            String passWord = admin.getPassWord();
            String encrypt = RSAUtils.encrypt(passWord, RSAUtils.getPublicKey(mySecurity.publicKey));
            admin.setPassWord(encrypt);
            admin.setUsersId(users.getId()+"");
            boolean update = adminService.updateById(admin);
            if (update) {
                return CommonResult.ok();
            }
            return CommonResult.fail();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }


    }

    @Override
    public CommonResult delete(int permissionSettingId, int id) {
        PermissionSetting permissionSetting = permissionSettingService.getById(permissionSettingId);
        if (ObjectUtils.isEmpty(permissionSetting)) {
            return CommonResult.fail("非法进入");
        }
//      判断该用户是否拥有权限
        //      判断该用户是否拥有权限
        String accountManagement = permissionSetting.getAccountManagement();
        if (!"0".equals(accountManagement) && !accountManagement.contains("3")) {
            return CommonResult.fail("此账号暂无该权限");
        }

        boolean b = adminService.removeById(id);

        if (b) {
            return CommonResult.ok();
        }

        return CommonResult.fail();

    }

    @Override
    public CommonResult getUsersPage(int permissionSettingId, int page, int size, String keyWord) {

        if (ObjectUtils.isEmpty(page) && page <= 0) {
            page = 1;
        }

        if (ObjectUtils.isEmpty(size) && size <= 0) {
            size = 10;
        }

        PermissionSetting permissionSetting = permissionSettingService.getById(permissionSettingId);
        if (ObjectUtils.isEmpty(permissionSetting)) {
            return CommonResult.fail("非法进入");
        }
//      判断该用户是否拥有权限
        //      判断该用户是否拥有权限
        String accountManagement = permissionSetting.getAccountManagement();
        if (!"0".equals(accountManagement) && !accountManagement.contains("1")) {
            return CommonResult.fail("此账号暂无该权限");
        }

        LambdaQueryWrapper<Users> wrapperU = new LambdaQueryWrapper<>();
        wrapperU.like(ObjectUtils.isNotEmpty(keyWord), Users::getCardNumber, keyWord)
                .or()
                .like(ObjectUtils.isNotEmpty(keyWord), Users::getUserName, keyWord);

        IPage<Users> page1 = usersService.page(new Page<>(page, size), wrapperU);

        return CommonResult.ok(page1);
    }

    @Override
    public CommonResult userList(String adminMenuId) {
        LambdaQueryWrapper<Admin> wrapper=new LambdaQueryWrapper<>();
        wrapper.eq(Admin::getAdminMenuId,adminMenuId);
        List<Admin> list = adminService.list(wrapper);
        return CommonResult.ok(list);
    }


}