UserController.java

package vion.controller;

import cn.dev33.satoken.annotation.SaCheckPermission;
import cn.hutool.core.lang.Opt;
import cn.hutool.core.lang.tree.Tree;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.ModelAndView;
import vion.Global;
import vion.third.DingMod;
import vion.third.WechatMod;
import vion.dto.BaseDTO;
import vion.dto.DingDTO;
import vion.model.TaskTemp;
import vion.model.User;
import vion.service.ITaskTempService;
import vion.service.IUserService;
import vion.vo.UserVO;

import javax.servlet.http.HttpServletResponse;
import java.util.List;


@RestController
@RequestMapping(Global.BASE_URL)
@RequiredArgsConstructor
@Slf4j
public class UserController {

    private final IUserService userService;
    private final ITaskTempService taskTempService;
    private final DingMod dingMod;
    private final WechatMod wechatMod;

    @GetMapping("/users")
    @SaCheckPermission(value = "user:list", orRole="admin")
    public Page<User> getUserList(BaseDTO dto) {
        return userService.page(Page.of(dto.getPageNum(), dto.getPageSize()));
    }

    @GetMapping("/user")
    @SaCheckPermission(value = "user:query", orRole="admin")
    public User getUserByID(@RequestParam(name = "id") Integer id) {
        return userService.getById(id);
    }

    @PostMapping("/user")
    @SaCheckPermission(value = "user:edit", orRole="admin")
    public String update(@RequestBody List<User> data) {
        return Opt.ofEmptyAble(data).map(userService::updateBatchById).orElse(false) ? "更新成功" : "更新失败";
    }
    @GetMapping("/user/{token}")
    @SaCheckPermission("user:token:query")
    public UserVO getUserByToken(@PathVariable String token) {
        return Global.USERNAME_MAP.get(token);
    }

    @GetMapping("/orgTree")
    @SaCheckPermission(value = "user:orgTree", orRole="admin")
    public List<Tree<String>> getOrgTree(String deptId) {
        return userService.getOrgTree(deptId);
    }


    @GetMapping("/ding/callback/{target}")
    public Object dingCallback(@PathVariable String target, DingDTO dto, HttpServletResponse res) {
        return dingMod.dingCallback(target, dto, res);
    }

    @GetMapping("/wechat/callback")
    public Object wechatCallback(String code, Long taskTempId) {
        Object obj = wechatMod.wechatCallback(code);
        if (obj instanceof String) {
            return taskTempService.lambdaUpdate().set(TaskTemp::getOpenid, obj).eq(TaskTemp::getId, taskTempId).update() ? new ModelAndView("weChatNeedAttention") : new ModelAndView("weChatError");
        } else {
            return obj;
        }
    }

}