/*
 *      Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *
 *  Redistributions of source code must retain the above copyright notice,
 *  this list of conditions and the following disclaimer.
 *  Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in the
 *  documentation and/or other materials provided with the distribution.
 *  Neither the name of the dreamlu.net developer nor the names of its
 *  contributors may be used to endorse or promote products derived from
 *  this software without specific prior written permission.
 *  Author: Chill 庄骞 (smallchill@163.com)
 */
package org.sxkj.system.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import io.swagger.annotations.*;
import lombok.AllArgsConstructor;
import org.apache.commons.lang3.StringUtils;
import org.springblade.core.boot.ctrl.BladeController;
import org.springblade.core.cache.utils.CacheUtil;
import org.springblade.core.excel.util.ExcelUtil;
import org.springblade.core.log.annotation.ApiLog;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.mp.support.Query;
import org.springblade.core.secure.BladeUser;
import org.springblade.core.secure.annotation.PreAuth;
import org.springblade.core.tenant.annotation.NonDS;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.constant.BladeConstant;
import org.springblade.core.tool.constant.RoleConstant;
import org.springblade.core.tool.support.Kv;
import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.core.tool.utils.DateUtil;
import org.springblade.core.tool.utils.Func;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.multipart.MultipartFile;
import org.sxkj.common.constant.WordOrderConstant;
import org.sxkj.common.node.TreeStringNode;
import org.sxkj.common.utils.HeaderUtils;
import org.sxkj.common.utils.OrderNumUtils;
import org.sxkj.system.cache.DictCache;
import org.sxkj.system.cache.UserCache;
import org.sxkj.system.entity.Dept;
import org.sxkj.system.entity.Region;
import org.sxkj.system.entity.User;
import org.sxkj.system.enums.DictEnum;
import org.sxkj.system.excel.DeptExcel;
import org.sxkj.system.excel.DeptImporter;
import org.sxkj.system.excel.RegionExcel;
import org.sxkj.system.excel.RegionImporter;
import org.sxkj.system.mapper.DeptMapper;
import org.sxkj.system.mapper.UserMapper;
import org.sxkj.system.param.DeptAddParam;
import org.sxkj.system.param.DeptExportParam;
import org.sxkj.system.param.DeptPageParam;
import org.sxkj.system.service.IDeptService;
import org.sxkj.system.vo.DeptVO;
import org.sxkj.system.wrapper.DeptWrapper;
import org.springframework.web.bind.annotation.*;
import springfox.documentation.annotations.ApiIgnore;

import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;

import static org.springblade.core.cache.constant.CacheConstant.SYS_CACHE;

/**
 * 控制器
 *
 * @author Chill
 */
@NonDS
@RestController
@AllArgsConstructor
@RequestMapping("/dept")
@Api(value = "部门", tags = "部门")
public class DeptController extends BladeController {

	private final IDeptService deptService;
	@Autowired
	private DeptMapper dictMapper;
	@Autowired
	private UserMapper userMapper;

	/**
	 * 详情
	 */
	@GetMapping("/detail")
	@ApiOperationSupport(order = 1)
	@ApiOperation(value = "详情", notes = "传入dept")
	public R<DeptVO> detail(Dept dept) {
		Dept detail = deptService.getOne(Condition.getQueryWrapper(dept));
		return R.data(DeptWrapper.build().entityVO(detail));
	}

	/**
	 * 机构 自定义分页
	 */
	@GetMapping("/page")
	@ApiOperationSupport(order = 3)
	@ApiOperation(value = "分页", notes = " ")
	public R<IPage<DeptVO>> page(DeptPageParam deptPageParam, Query query) {
		IPage<DeptVO> pages = deptService.selectDeptPage(Condition.getPage(query), deptPageParam);
		return R.data(pages);
	}
	/**
	 * 列表
	 */
	@GetMapping("/list")
	@ApiImplicitParams({
		@ApiImplicitParam(name = "deptName", value = "部门名称", paramType = "query", dataType = "string"),
		@ApiImplicitParam(name = "fullName", value = "部门全称", paramType = "query", dataType = "string")
	})
	@ApiOperationSupport(order = 2)
	@ApiOperation(value = "列表", notes = "传入dept")
	public R<List<DeptVO>> list(@ApiIgnore @RequestParam Map<String, Object> dept, BladeUser bladeUser) {
		QueryWrapper<Dept> queryWrapper = Condition.getQueryWrapper(dept, Dept.class);
		List<Dept> list = deptService.list((!bladeUser.getTenantId().equals(BladeConstant.ADMIN_TENANT_ID)) ? queryWrapper.lambda().eq(Dept::getTenantId, bladeUser.getTenantId()) : queryWrapper);
		return R.data(DeptWrapper.build().listNodeVO(list));
	}

	/**
	 * 懒加载列表
	 */
	@GetMapping("/lazy-list")
	@ApiImplicitParams({
		@ApiImplicitParam(name = "deptName", value = "部门名称", paramType = "query", dataType = "string"),
		@ApiImplicitParam(name = "fullName", value = "部门全称", paramType = "query", dataType = "string")
	})
	@ApiOperationSupport(order = 3)
	@ApiOperation(value = "懒加载列表", notes = "传入dept")
	public R<List<DeptVO>> lazyList(@ApiIgnore @RequestParam Map<String, Object> dept, Long parentId, BladeUser bladeUser) {
		List<DeptVO> list = deptService.lazyList(bladeUser.getTenantId(), parentId, dept);
		for (DeptVO lists : list) {
			if (StringUtils.isNotEmpty(lists.getAreaCode())) {
				lists.setAreaName(dictMapper.getSingleRegionName(lists.getAreaCode()).replace(" ", ""));

			}
		}
		return R.data(DeptWrapper.build().listNodeLazyVO(list));
	}


	/**
	 * 获取部门树形结构
	 *
	 * @return
	 */
	@GetMapping("/tree")
	@ApiOperationSupport(order = 4)
	@ApiOperation(value = "树形结构", notes = "树形结构")
	public R<List<DeptVO>> tree(String tenantId, BladeUser bladeUser) {
		List<DeptVO> tree = deptService.tree(Func.toStrWithEmpty(tenantId, bladeUser.getTenantId()));
		return R.data(tree);
	}

	/**
	 * 懒加载获取部门树形结构
	 */
	@GetMapping("/lazy-tree")
	@ApiOperationSupport(order = 5)
	@ApiOperation(value = "懒加载树形结构", notes = "树形结构")
	public R<List<DeptVO>> lazyTree(String tenantId, Long parentId, Integer level, BladeUser bladeUser) {
		List<DeptVO> tree = deptService.lazyTree(Func.toStrWithEmpty(tenantId, bladeUser.getTenantId()), parentId, level);
		return R.data(tree);
	}

	/**
	 * 懒加载获取部门树形结构(子集)
	 */
	@GetMapping("/getChildLazyTree")
	@ApiOperationSupport(order = 14)
	@ApiOperation(value = "懒加载获取部门树形结构(子集)", notes = "树形结构")
	public R<List<DeptVO>> getChildLazyTree(@ApiIgnore @RequestParam Map<String, Object> dept, Long parentId) {
		List<DeptVO> tree = deptService.getChildLazyTree(dept, parentId);
		for (DeptVO lists : tree) {
			if (StringUtils.isNotEmpty(lists.getAreaCode())) {
				lists.setAreaName(dictMapper.getSingleRegionName(lists.getAreaCode()).replace(" ", ""));
			}
		}
		return R.data(DeptWrapper.build().listNodeLazyVO(tree));
	}

	/**
	 * 新增或修改
	 */
	@PostMapping("/submit")
	@ApiOperationSupport(order = 6)
	@ApiOperation(value = "新增或修改", notes = "传入dept")
	@ApiLog("组织机构信息新增或修改")
	public R submit(@Valid @RequestBody DeptAddParam dept) {
		Dept deptEntity = Objects.requireNonNull(BeanUtil.copy(dept, Dept.class));
		String times = OrderNumUtils.initOrderNum2(WordOrderConstant.ORG_CODE);
		String deptCode = WordOrderConstant.ORG_PREFIX + times;
		deptEntity.setDeptCode(deptCode);
		if (deptService.submit(deptEntity)) {
			CacheUtil.clear(SYS_CACHE);
			CacheUtil.clear(SYS_CACHE, Boolean.FALSE);
			List<User> userList = userMapper.selectList(new LambdaQueryWrapper<User>().eq(User::getDeptId, dept.getId()));
			if (!userList.isEmpty()) {
				List<Long> userIds = userList.stream().map(User::getId).collect(Collectors.toList());
				userMapper.updateUserAreaCode(dept.getAreaCode(), userIds);
			}
			// 返回懒加载树更新节点所需字段
			Kv kv = Kv.create().set("id", String.valueOf(dept.getId())).set("tenantId", dept.getTenantId())
				.set("deptCategoryName", DictCache.getValue(DictEnum.ORG_CATEGORY, dept.getDeptCategory()));
			return R.data(kv);
		}

		return R.fail("操作失败");
	}

	/**
	 * 删除
	 */
//	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	@PostMapping("/remove")
	@ApiOperationSupport(order = 7)
	@ApiOperation(value = "删除", notes = "传入ids")
	@ApiLog("组织机构信息删除")
	public R remove(@ApiParam(value = "主键集合", required = true) @RequestParam String ids) {
		CacheUtil.clear(SYS_CACHE);
		CacheUtil.clear(SYS_CACHE, Boolean.FALSE);
		return R.status(deptService.removeDept(ids));
	}

	/**
	 * 下拉数据源
	 */
	@GetMapping("/select")
	@ApiOperationSupport(order = 8)
	@ApiOperation(value = "下拉数据源", notes = "传入id集合")
	public R<List<Dept>> select(Long userId, String deptId) {
		if (Func.isNotEmpty(userId)) {
			User user = UserCache.getUser(userId);
			deptId = user.getDeptId();
		}
		List<Dept> list = deptService.list(Wrappers.<Dept>lambdaQuery().in(Dept::getId, Func.toLongList(deptId)));
		return R.data(list);
	}


	/**
	 * 获取部门树形结构
	 *
	 * @param dept
	 * @return
	 */
	@GetMapping("/getTree")
	@ApiOperationSupport(order = 9)
	@ApiOperation(value = "树形结构", notes = "树形结构")
	public R getTree(DeptVO dept) {
		List<TreeStringNode> tree = deptService.getTree(dept);
		return R.data(tree);
	}

	/**
	 * 自定义详情查询-根据系统配置id
	 *
	 * @param dept
	 * @return
	 */
	@GetMapping("/getDetailBySysConfigId")
	@ApiOperationSupport(order = 10)
	@ApiOperation(value = "自定义详情查询", notes = "传入dept")
	public R<DeptVO> getDetailBySysConfigId(DeptVO dept) {
		DeptVO detail = deptService.getDetailBySysConfigId(dept);
		return R.data(detail);
	}

	/**
	 * 下拉数据源
	 *
	 * @param deptId
	 * @return
	 */
	@GetMapping("/selectDeptList")
	@ApiOperationSupport(order = 11)
	@ApiOperation(value = "下拉数据源", notes = "传入id集合")
	public R selectDeptList(String deptId) {
		List<DeptVO> list = deptService.selectDeptList(deptId);
		return R.data(list);
	}

	/**
	 * 查询父级机构信息（一级机构）
	 *
	 * @param deptId
	 * @return
	 */
	@GetMapping("/getParentDeptInfo")
	@ApiOperationSupport(order = 12)
	@ApiOperation(value = "通过id查询", notes = "传入id集合")
	public R getParentDeptInfo(String deptId) {
		DeptVO list = deptService.getParentDeptInfo(deptId);
		return R.data(list);
	}

	/**
	 * 查询机构树（除本机机构以外的机构）
	 *
	 * @param deptId
	 * @param deviceSn 设备编号
	 * @return
	 */
	@GetMapping("/getDeptTreeByNotItself")
	@ApiOperationSupport(order = 13)
	@ApiOperation(value = "查询机构树（除本机机构以外的机构）")
	public R getDeptTreeByNotItself(String deptId, String deviceSn) {
		List<TreeStringNode> list = deptService.getDeptTreeByNotItself(deptId, deviceSn);
		return R.data(list);
	}

	/**
	 * 查询机构树（除本机机构以外的机构）
	 *
	 * @return
	 */
	@GetMapping("/deptsByAreaCode")
	@ApiOperationSupport(order = 13)
	@ApiOperation(value = "查询机构树（根据区域）")
	public R deptsByAreaCode(@RequestParam(required = false) String areaCode) {
		String areaCodez = HeaderUtils.getAreaCode();
		if (StringUtils.isNotBlank(areaCode)) {
			areaCodez = areaCode;
		}
		List<DeptVO> list = deptService.deptsByAreaCode(areaCodez);
		return R.data(list);
	}

	/**
	 * 导入机构数据
	 */
	@PostMapping("import-region")
	@ApiOperationSupport(order = 14)
	@ApiOperation(value = "导入机构", notes = "传入excel")
	public R importRegion(MultipartFile file, Integer isCovered) {
		DeptImporter deptImporter = new DeptImporter(deptService, isCovered == 1);
		ExcelUtil.save(file, deptImporter, DeptExcel.class);
		return R.success("操作成功");
	}

	/**
	 * 导出机构数据
	 */
	@GetMapping("export-region")
	@ApiOperationSupport(order = 15)
	@ApiOperation(value = "导出机构", notes = "传入user")
	public void exportRegion(DeptExportParam dept, HttpServletResponse response) {
 		List<DeptExcel> list = deptService.exportDept(dept);
		ExcelUtil.export(response, "机构数据" + DateUtil.time(), "机构数据表", list, DeptExcel.class);
	}

	/**
	 * 导出模板
	 */
	@GetMapping("export-template")
	@ApiOperationSupport(order = 16)
	@ApiOperation(value = "导出模板")
	public void exportUser(HttpServletResponse response) {
		List<DeptExcel> list = new ArrayList<>();
		ExcelUtil.export(response, "机构模板", "机构表", list, DeptExcel.class);
	}

}