package cn.iocoder.yudao.module.wxcp.service.member;

import cn.iocoder.yudao.module.wxcp.controller.admin.member.vo.WxCpDepartmentRespVO;
import cn.iocoder.yudao.module.wxcp.controller.admin.member.vo.WxCpMemberRespVO;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import me.chanjar.weixin.cp.api.WxCpService;
import me.chanjar.weixin.cp.bean.WxCpDepart;
import me.chanjar.weixin.cp.bean.WxCpUser;
import org.springframework.stereotype.Service;
import org.springframework.validation.annotation.Validated;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.wxcp.enums.ErrorCodeConstants.WXCP_API_ERROR;

/**
 * 企业微信通讯录 Service 实现类
 *
 * @author 超级管理员
 */
@Service
@Validated
@Slf4j
public class WxCpMemberServiceImpl implements WxCpMemberService {

    /**
     * 企业微信根部门ID固定为 1
     */
    private static final Long ROOT_DEPARTMENT_ID = 1L;

    @Resource
    private WxCpService wxCpService;

    @Override
    public List<WxCpDepartmentRespVO> getDepartmentList() {
        List<WxCpDepart> departs = listDeparts();
        return departs.stream().map(this::toDeptVO).toList();
    }

    @Override
    public List<WxCpDepartmentRespVO> getDepartmentTree() {
        List<WxCpDepart> departs = listDeparts();
        Map<Long, WxCpDepartmentRespVO> voMap = new LinkedHashMap<>();
        for (WxCpDepart d : departs) {
            voMap.put(d.getId(), toDeptVO(d));
        }
        List<WxCpDepartmentRespVO> roots = new ArrayList<>();
        for (WxCpDepartmentRespVO vo : voMap.values()) {
            Long parentId = vo.getParentId();
            WxCpDepartmentRespVO parent = parentId == null ? null : voMap.get(parentId);
            // 找不到父节点的（含顶级部门）都作为根返回，避免因数据缺失导致部门"消失"
            if (parent == null || parent == vo) {
                roots.add(vo);
            } else {
                parent.getChildren().add(vo);
            }
        }
        return roots;
    }

    @Override
    public List<WxCpMemberRespVO> getMemberList(Long departmentId, Boolean fetchChild) {
        Long deptId = departmentId != null ? departmentId : ROOT_DEPARTMENT_ID;
        boolean child = fetchChild == null || fetchChild;
        List<WxCpUser> users;
        try {
            users = wxCpService.getUserService().listSimpleByDepartment(deptId, child, null);
        } catch (Exception e) {
            log.error("[getMemberList][获取企业微信成员失败 deptId={} fetchChild={}]", deptId, child, e);
            throw exception(WXCP_API_ERROR);
        }

        // 部门名映射，用于把成员的部门ID翻译成可读名称
        Map<Long, String> deptNameMap = new LinkedHashMap<>();
        for (WxCpDepart d : listDeparts()) {
            deptNameMap.put(d.getId(), d.getName());
        }

        return users.stream().map(u -> toMemberVO(u, deptNameMap)).toList();
    }

    /**
     * 获取全部部门（平铺）。企业微信 department/list 不传 id 时返回全量。
     */
    private List<WxCpDepart> listDeparts() {
        try {
            List<WxCpDepart> departs = wxCpService.getDepartmentService().list(null);
            return departs != null ? departs : List.of();
        } catch (Exception e) {
            log.error("[listDeparts][获取企业微信部门失败]", e);
            throw exception(WXCP_API_ERROR);
        }
    }

    private WxCpDepartmentRespVO toDeptVO(WxCpDepart depart) {
        WxCpDepartmentRespVO vo = new WxCpDepartmentRespVO();
        vo.setId(depart.getId());
        vo.setName(depart.getName());
        vo.setParentId(depart.getParentId());
        vo.setOrder(depart.getOrder());
        return vo;
    }

    private WxCpMemberRespVO toMemberVO(WxCpUser user, Map<Long, String> deptNameMap) {
        WxCpMemberRespVO vo = new WxCpMemberRespVO();
        vo.setUserId(user.getUserId());
        vo.setName(user.getName());
        vo.setMobile(user.getMobile());
        vo.setEmail(user.getEmail());
        vo.setAvatar(user.getAvatar());
        // 优先取 position，老接口可能只返回 positions 数组
        String position = user.getPosition();
        if (position == null && user.getPositions() != null && user.getPositions().length > 0) {
            position = String.join("、", user.getPositions());
        }
        vo.setPosition(position);
        vo.setStatus(user.getStatus());
        // Gender.getCode() 返回基本类型 int，不能与 null 组成三元表达式，这里单独判断
        if (user.getGender() != null) {
            vo.setGender(user.getGender().getCode());
        }

        Long[] deptIds = user.getDepartIds();
        if (deptIds != null && deptIds.length > 0) {
            List<Long> ids = Arrays.asList(deptIds);
            vo.setDepartmentIds(ids);
            vo.setDepartmentNames(String.join("、", ids.stream()
                    .map(id -> deptNameMap.getOrDefault(id, "部门" + id))
                    .toList()));
        }
        return vo;
    }

}