package org.jeecg.modules.weekly.controller;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.apache.shiro.SecurityUtils;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.aspect.annotation.AutoLog;
import org.jeecg.common.system.base.controller.JeecgController;
import org.jeecg.common.system.query.QueryGenerator;
import org.jeecg.common.system.vo.LoginUser;
import org.jeecg.modules.weekly.entity.WekUserRelation;
import org.jeecg.modules.weekly.feign.SystemClient;
import org.jeecg.modules.weekly.feign.model.SysUser;
import org.jeecg.modules.weekly.service.IWekUserRelationService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.*;
import java.util.stream.Collectors;

/**
 * @Description: wek_user_relation
 * @Author: jeecg-boot
 * @Date: 2024-01-09
 * @Version: V1.0
 */
@Api(tags = "wek_user_relation")
@RestController
@RequestMapping("/wek/relation")
@Slf4j
public class WekUserRelationController extends JeecgController<WekUserRelation, IWekUserRelationService> {
    @Autowired
    private IWekUserRelationService wekUserRelationService;
    @Autowired
    private SystemClient systemClient;

    /**
     * 分页列表查询
     *
     * @param wekUserRelation
     * @param pageNo
     * @param pageSize
     * @param req
     * @return
     */
    @AutoLog(value = "wek_user_relation-分页列表查询")
    @ApiOperation(value = "wek_user_relation-分页列表查询", notes = "wek_user_relation-分页列表查询")
    @GetMapping(value = "/list")
    public Result<?> queryPageList(WekUserRelation wekUserRelation,
                                   @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo,
                                   @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize,
                                   HttpServletRequest req) {
        if (StringUtils.isEmpty(wekUserRelation.getUser())) {
            return Result.error("用户不能为空!");
        }
        QueryWrapper<WekUserRelation> queryWrapper = QueryGenerator.initQueryWrapper(wekUserRelation, req.getParameterMap());
        Page<WekUserRelation> page = new Page<WekUserRelation>(pageNo, pageSize);
        IPage<WekUserRelation> pageList = wekUserRelationService.page(page, queryWrapper);
        return Result.OK(pageList);
    }

    /**
     * 添加
     *
     * @param wekUserRelation
     * @return
     */
    @AutoLog(value = "wek_user_relation-添加")
    @ApiOperation(value = "wek_user_relation-添加", notes = "wek_user_relation-添加")
    @PostMapping(value = "/add")
    public Result<?> add(@RequestBody WekUserRelation wekUserRelation) {
        if (StringUtils.isEmpty(wekUserRelation.getUser()) || StringUtils.isEmpty(wekUserRelation.getSuperiors())) {
            return Result.error("用户或上级不能为空!");
        }
        QueryWrapper<WekUserRelation> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(WekUserRelation::getUser, wekUserRelation.getUser());
        queryWrapper.lambda().eq(WekUserRelation::getSuperiors, wekUserRelation.getSuperiors());
        List<WekUserRelation> list = wekUserRelationService.list(queryWrapper);
        if (list.size() < 1) {
            wekUserRelationService.save(wekUserRelation);
        } else {
            return Result.error("上级已存在！");
        }

        return Result.OK("添加成功！");
    }

    /**
     * 编辑
     *
     * @param wekUserRelation
     * @return
     */
    @AutoLog(value = "wek_user_relation-编辑")
    @ApiOperation(value = "wek_user_relation-编辑", notes = "wek_user_relation-编辑")
    @PutMapping(value = "/edit")
    public Result<?> edit(@RequestBody WekUserRelation wekUserRelation) {
        wekUserRelationService.updateById(wekUserRelation);
        return Result.OK("编辑成功!");
    }

    /**
     * 通过id删除
     *
     * @param id
     * @return
     */
    @AutoLog(value = "wek_user_relation-通过id删除")
    @ApiOperation(value = "wek_user_relation-通过id删除", notes = "wek_user_relation-通过id删除")
    @DeleteMapping(value = "/delete")
    public Result<?> delete(@RequestParam(name = "id", required = true) String id) {
        wekUserRelationService.removeById(id);
        return Result.OK("删除成功!");
    }

    /**
     * 批量删除
     *
     * @param ids
     * @return
     */
    @AutoLog(value = "wek_user_relation-批量删除")
    @ApiOperation(value = "wek_user_relation-批量删除", notes = "wek_user_relation-批量删除")
    @DeleteMapping(value = "/deleteBatch")
    public Result<?> deleteBatch(@RequestParam(name = "ids", required = true) String ids) {
        this.wekUserRelationService.removeByIds(Arrays.asList(ids.split(",")));
        return Result.OK("批量删除成功!");
    }

    /**
     * 通过id查询
     *
     * @param id
     * @return
     */
    @AutoLog(value = "wek_user_relation-通过id查询")
    @ApiOperation(value = "wek_user_relation-通过id查询", notes = "wek_user_relation-通过id查询")
    @GetMapping(value = "/queryById")
    public Result<?> queryById(@RequestParam(name = "id", required = true) String id) {
        WekUserRelation wekUserRelation = wekUserRelationService.getById(id);
        if (wekUserRelation == null) {
            return Result.error("未找到对应数据");
        }
        return Result.OK(wekUserRelation);
    }


    @AutoLog(value = "wek_user_relation-查询下级")
    @ApiOperation(value = "wek_user_relation-查询下级", notes = "wek_user_relation-查询下级")
    @GetMapping(value = "/queryJunior")
    public Result<?> queryJunior(@RequestParam(name = "username", required = false) String username) {
        //查询登录用户的下级
        LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
        QueryWrapper<WekUserRelation> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(WekUserRelation::getSuperiors, sysUser.getUsername());
        List<WekUserRelation> list = wekUserRelationService.list(queryWrapper);


        //如果登录用户是部门上级  1-普通用户  2-上级
        Integer userIdentity = sysUser.getUserIdentity();

        List<Map<String, Object>> root = new ArrayList<>();
        List<Map<String, Object>> gList1 = new ArrayList<>();
        List<Map<String, Object>> gList2 = new ArrayList<>();
        Map<String, Object> group1 = new HashMap<>();
        group1.put("title", "直属下级");
        group1.put("children", gList1);
        group1.put("type", 1);

        Map<String, Object> group2 = new HashMap<>();
        group2.put("title", "其他下级");
        group2.put("children", gList2);
        group2.put("type", 1);

        Map<String, Object> gl1 = new HashMap<>();
        gl1.put("username", "gl1");
        gl1.put("realname", "概览");
        gl1.put("type", 1);
        gList1.add(gl1);
        Map<String, Object> gl2 = new HashMap<>();
        gl2.put("username", "gl2");
        gl2.put("realname", "概览");
        gl2.put("type", 1);
        gList2.add(gl2);

        list.forEach(i -> {
            Map<String, Object> item = new HashMap<>();
            item.put("username", i.getUser());
            item.put("realname", i.getUserName());
            item.put("type", 2);
            gList1.add(item);

            QueryWrapper<WekUserRelation> wrapper = new QueryWrapper<>();
            wrapper.lambda().eq(WekUserRelation::getSuperiors, i.getUser());
            List<WekUserRelation> subList = wekUserRelationService.list(wrapper);
            subList.forEach(sub -> {
                Map<String, Object> s = new HashMap<>();
                s.put("username", sub.getUser());
                s.put("realname", sub.getUserName());
                s.put("type", 2);
                gList2.add(s);
            });
        });

        //如果是上级需要把部门所有人员添加为下级
        if (userIdentity != null && userIdentity == 2) {
            Set<String> namesToRemove = gList1.stream()
                    .map(map -> (String) map.get("username"))
                    .collect(Collectors.toSet());
            List<SysUser> depUserList = systemClient.queryUserByDepCode(sysUser.getOrgCode()).getResult();
            if (depUserList != null) {
                depUserList = depUserList.stream().filter(i -> i.getUserIdentity() == null || (i.getUserIdentity()!=null && i.getUserIdentity() == 1)).collect(Collectors.toList());
                List<Map<String, Object>> oList = new ArrayList<>();
                depUserList.forEach(user -> {
                    Map<String, Object> o = new HashMap<>();
                    o.put("username", user.getUsername());
                    o.put("realname", user.getRealname());
                    o.put("type", 2);
                    oList.add(o);
                });
                // 移除 name 值相同的元素
                oList.removeIf(map -> namesToRemove.contains(map.get("username")));
                group2.put("children", oList);
            }

        }


        if (list.size() < 1) {
            return Result.error("未找到对应数据");
        }
        root.add(group1);
        root.add(group2);
        return Result.OK(root);
    }

    /**
     * 导出excel
     *
     * @param request
     * @param wekUserRelation
     */
    @RequestMapping(value = "/exportXls")
    public ModelAndView exportXls(HttpServletRequest request, WekUserRelation wekUserRelation) {
        return super.exportXls(request, wekUserRelation, WekUserRelation.class, "wek_user_relation");
    }

    /**
     * 通过excel导入数据
     *
     * @param request
     * @param response
     * @return
     */
    @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
    public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
        return super.importExcel(request, response, WekUserRelation.class);
    }

}