package com.tievd.cube.modules.system.controller;

import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.tievd.cube.commons.constant.CacheConst;
import com.tievd.cube.commons.constant.CommonConst;
import com.tievd.cube.commons.base.CubeController;
import com.tievd.cube.commons.mybatisplus.QueryGenerator;
import com.tievd.cube.commons.annotations.AutoLog;
import com.tievd.cube.commons.annotations.DictMethod;
import com.tievd.cube.commons.base.Result;
import com.tievd.cube.commons.easyexcel.model.ImportExcel;
import com.tievd.cube.commons.utils.SystemContextUtil;
import com.tievd.cube.commons.utils.YouBianCodeUtil;
import com.tievd.cube.commons.utils.web.HttpServletUtil;
import com.tievd.cube.modules.system.entity.SysDepart;
import com.tievd.cube.modules.system.entity.SysUser;
import com.tievd.cube.modules.system.model.DepartIdModel;
import com.tievd.cube.modules.system.model.LoginUser;
import com.tievd.cube.modules.system.model.SysDepartTreeModel;
import com.tievd.cube.modules.system.service.ISysDepartService;
import com.tievd.cube.modules.system.service.ISysUserDepartService;
import com.tievd.cube.modules.system.service.ISysUserService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cache.annotation.CacheEvict;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.util.*;

/**
 * 部门
 *
 * @author xinwuy
 * @since 2021-05-25
 */
@Slf4j
@Tag(name = "部门管理相关接口")
@RestController
@RequestMapping("/sys/depart")
public class SysDepartController extends CubeController<SysDepart, ISysDepartService> {

    @Autowired
    private ISysUserService sysUserService;
    @Autowired
    private ISysUserDepartService sysUserDepartService;

    /**
     * 查询我的部门
     *
     * @apiNote 以树结构方式返回，上级才返回数据
     */
    @GetMapping("/queryMyDeptTreeList")
    @Operation(summary = "查询我的部门")
    public Result<List<SysDepartTreeModel>> queryMyDeptTreeList() {
        LoginUser user = SystemContextUtil.currentLoginUser();
        if (user.getUserIdentity() != null && user.getUserIdentity().equals(CommonConst.USER_IDENTITY_2)) {
            List<SysDepartTreeModel> list = service.queryMyDeptTreeList(user.getDepartIds());
            return Result.ok(user.getUserIdentity().toString(), list);
        }
        return Result.ok(user.getUserIdentity().toString(), null);
    }

    /**
     * 查询所有部门
     *
     * @return 部门树
     * @apiNote 以树结构方式返回
     */
    @GetMapping("/queryTreeList")
    @Operation(summary = "查询所有部门")
    public Result<List<SysDepartTreeModel>> queryTreeList(boolean myself) {
        List<SysDepartTreeModel> list;
        if (myself) {
            LoginUser user = SystemContextUtil.currentLoginUser();
            list = service.queryMyDeptTreeList(user.getDepartIds());
        } else {
            list = service.queryTreeList();
        }
        return Result.ok(list);
    }

    /**
     * 新增部门
     *
     * @param sysDepart 部门信息
     */
    @AutoLog("新增部门")
    @PostMapping("/add")
    @Operation(summary = "新增部门")
    public Result<?> add(@RequestBody SysDepart sysDepart) {
        return service.saveDepartData(sysDepart);
    }

    /**
     * 更新部门信息
     *
     * @param sysDepart 部门信息
     */
    @AutoLog("更新部门信息")
    @PutMapping("/edit")
    @Operation(summary = "更新部门信息")
    public Result<?> edit(@RequestBody SysDepart sysDepart) {
        String username = StpUtil.getLoginIdAsString();
        sysDepart.setUpdateBy(username);
        SysDepart sysDepartEntity = service.getById(sysDepart.getId());
        if (sysDepartEntity == null) {
            return Result.error("未找到对应数据！");
        }
        service.updateDepartDataById(sysDepart, username);
        return Result.ok();
    }

    /**
     * 删除部门
     *
     * @param id 主键id
     */
    @AutoLog("删除部门")
    @DeleteMapping("/delete")
    @Operation(summary = "删除部门")
    public Result<?> delete(@RequestParam String id) {
        SysDepart sysDepart = service.getById(id);
        if (sysDepart == null) {
            return Result.error("未找到对应数据！");
        }
        service.delete(id);
        return Result.ok();
    }

    /**
     * 批量删除部门
     *
     * @param ids id数组（逗号分割）
     */
    @AutoLog("批量删除部门")
    @DeleteMapping("/deleteBatch")
    @Operation(summary = "批量删除部门")
    public Result<?> deleteBatch(@RequestParam String ids) {
        if (ids == null || "".equals(ids.trim())) {
            return Result.error("参数不识别！");
        }
        service.deleteBatchWithChildren(Arrays.asList(ids.split(",")));
        return Result.ok();
    }

    /**
     * 查询部门信息
     *
     * @apiNote 添加或编辑页面对该方法发起请求, 以树结构形式加载所有部门的名称, 方便用户的操作
     */
    @GetMapping("/queryIdTree")
    @Operation(summary = "查询部门信息")
    public Result<List<DepartIdModel>> queryIdTree() {
        List<DepartIdModel> list = service.queryDepartIdTreeList();
        return Result.ok(list);
    }

    /**
     * 部门搜索
     *
     * @param keyWord      关键词
     * @param myDeptSearch 我的部门下搜索
     * @return 部门列表树
     */
    @GetMapping("/searchBy")
    @Operation(summary = "部门搜索")
    public Result<List<SysDepartTreeModel>> searchBy(@Parameter(name = "关键词") String keyWord, @Parameter(name = "我的部门下搜索") @RequestParam(required = false) String myDeptSearch) {
        //部门查询，myDeptSearch为1时为我的部门查询，登录用户为上级时查只查负责部门下数据
        LoginUser user = SystemContextUtil.currentLoginUser();
        String departIds = null;
        if (user.getUserIdentity() != null && user.getUserIdentity().equals(CommonConst.USER_IDENTITY_2)) {
            departIds = user.getDepartIds();
        }
        List<SysDepartTreeModel> treeList = service.searchBy(keyWord, myDeptSearch, departIds);
        if (treeList == null || treeList.size() == 0) {
            return Result.error("未查询匹配数据！");
        }
        return Result.ok(treeList);
    }

    /**
     * 导出部门数据为Excel
     *
     * @param sysDepart 查询参数
     */
    @GetMapping("/exportXls")
    @Operation(summary = "导出Excel")
    public void exportXls(SysDepart sysDepart, HttpServletRequest request, HttpServletResponse response) throws IOException {
        QueryWrapper<SysDepart> queryWrapper = QueryGenerator.initQueryWrapper(sysDepart, request.getParameterMap());
        List<SysDepart> pageList = service.list(queryWrapper);
        //按字典排序
        pageList.sort(Comparator.comparing(SysDepart::getOrgCode));
        //导出文件名称
        String date = DateUtil.format(new Date(), "yyyyMMddHHmmss");
        HttpServletUtil.addDownloadHeader(response, "部门数据-" + date + easyExcel.getExtension());
        easyExcel.export(pageList, response.getOutputStream(), SystemContextUtil.dictTranslator());
    }

    /**
     * 通过Excel导入部门数据
     */
    @PostMapping("/importExcel")
    @AutoLog("部门管理-通过Excel导入部门数据")
    @Operation(summary = "Excel导入")
    @CacheEvict(cacheNames = CacheConst.SYS_DEPARTS_CACHE, allEntries = true)
    public Result<?> importExcel(HttpServletRequest request) throws Exception {
        MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
        List<String> errorMessages = new ArrayList<>();
        int successLines = 0, errorLines = 0;
        List<SysDepart> listSysDeparts;
        Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
        for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
            MultipartFile file = entity.getValue();// 获取上传文件对象
            try (InputStream inputStream = file.getInputStream()) {
                ImportExcel excel = new ImportExcel();
                excel.setInputStream(inputStream);
                // orgCode编码长度
                int codeLength = YouBianCodeUtil.zhanweiLength;
                listSysDeparts = easyExcel.read(SysDepart.class, excel, SystemContextUtil.dictTranslator());
                //按长度排序
                listSysDeparts.sort(Comparator.comparingInt(arg0 -> arg0.getOrgCode().length()));
                for (SysDepart sysDepart : listSysDeparts) {
                    String orgCode = sysDepart.getOrgCode();
                    if (orgCode.length() > codeLength) {
                        String parentCode = orgCode.substring(0, orgCode.length() - codeLength);
                        QueryWrapper<SysDepart> queryWrapper = new QueryWrapper<>();
                        queryWrapper.eq("org_code", parentCode);
                        SysDepart parentDept = service.getOne(queryWrapper);
                        if (parentDept != null) {
                            sysDepart.setParentId(parentDept.getId());
                        } else {
                            sysDepart.setParentId("");
                        }
                    } else {
                        sysDepart.setParentId("");
                    }
                    sysDepart.setOrgType(sysDepart.getOrgCode().length() / codeLength + "");
                    sysDepart.setDelFlag(CommonConst.NOT_DELETED);
                    try {
                        service.save(sysDepart);
                        successLines++;
                    } catch (Exception e) {
                        errorLines++;
                        int lineNumber = 1;
                        errorMessages.add("第 " + lineNumber + " 行：存在错误，忽略导入。【" + e.getMessage() + "】");
                    }
                }
            }
        }
        JSONObject returnData = JSONUtil.createObj();
        returnData.set("errorLines", errorLines);
        returnData.set("successLines", successLines);
        returnData.set("errorMessages", errorMessages);
        return Result.ok(returnData);
    }

    /**
     * 查询所有部门信息
     *
     * @param id 查询指定多个id的部门信息
     * @return 部门信息
     */
    @DictMethod
    @Deprecated
    @GetMapping("/listAll")
    @Operation(summary = "查询所有部门信息")
    public Result<List<SysDepart>> listAll(@Parameter(name = "部门id", description = "指定多个逗号分隔") @RequestParam(required = false) String id) {
        LambdaQueryWrapper<SysDepart> query = new LambdaQueryWrapper<>();
        query.orderByAsc(SysDepart::getOrgCode);
        if (StrUtil.isNotEmpty(id)) {
            Object[] arr = id.split(",");
            query.in(SysDepart::getId, arr);
        }
        List<SysDepart> sysDeparts = service.list(query);
        return Result.ok(sysDeparts);
    }

    /**
     * 根据关键词查询部门信息
     *
     * @param keyWord 关键词
     * @apiNote 以树结构方式返回
     */
    @Deprecated
    @GetMapping("/queryTreeByKeyWord")
    @Operation(summary = "根据关键词查询部门信息")
    public Result<?> queryTreeByKeyWord(@Parameter(name = "关键词") @RequestParam(required = false) String keyWord) {
        List<SysDepartTreeModel> list = service.queryTreeByKeyWord(keyWord);
        //根据keyWord获取用户信息
        LambdaQueryWrapper<SysUser> queryUser = new LambdaQueryWrapper<>();
        queryUser.eq(SysUser::getDelFlag, CommonConst.NOT_DELETED);
        queryUser.and(i -> i.like(SysUser::getUsername, keyWord).or().like(SysUser::getRealname, keyWord));
        List<SysUser> sysUsers = sysUserService.list(queryUser);
        Map<String, Object> returnData = new HashMap<>();
        returnData.put("userList", sysUsers);
        returnData.put("departList", list);
        return Result.ok(returnData);
    }

    /**
     * 根据部门编码获取部门信息
     *
     * @param orgCode 部门编码
     * @return 部门信息
     */
    @Deprecated
    @DictMethod
    @GetMapping("/getDepartName")
    @Operation(summary = "根据部门编码获取部门信息")
    public Result<SysDepart> getDepartName(@Parameter(name = "部门编码") @RequestParam String orgCode) {
        LambdaQueryWrapper<SysDepart> query = new LambdaQueryWrapper<>();
        query.eq(SysDepart::getOrgCode, orgCode);
        SysDepart sysDepart = service.getOne(query);
        return Result.ok(sysDepart);
    }

    /**
     * 根据部门id获取用户信息
     *
     * @param id 部门id
     * @return 用户列表
     */
    @Deprecated
    @GetMapping("/getUsersByDepartId")
    @Operation(summary = "根据部门id获取用户信息")
    public Result<List<SysUser>> getUsersByDepartId(@RequestParam String id) {
        List<SysUser> sysUsers = sysUserDepartService.queryUserByDepId(id);
        return Result.ok(sysUsers);
    }
}