<template>
  <div style="padding-bottom:50px">
    <p v-if="data.hasEnt">
      请选择权限：
    </p>
    <!-- 树状结构 -->
    <a-tree
      v-if="data.hasEnt"
      v-model:checkedKeys="data.checkedKeys" 
      :tree-data="data.treeData"
      :field-names="data.replaceFields"
      :checkable="true"
    />
  </div>
</template>

<script lang="ts" setup>
import { $getEntTree, API_URL_ROLE_ENT_RELA_LIST, req } from '@/api/manage'
import { reactive,ref,getCurrentInstance } from 'vue'
const { $infoBox,$access, $hasAgentEnt } = getCurrentInstance()!.appContext.config.globalProperties

    const data = reactive ({
      hasEnt: $access('ENT_UR_ROLE_DIST'),
      recordId: null, // 更新对象ID
      treeData: [],
      replaceFields: { key: 'entId', title: 'entName' }, // 配置替换字段
      checkedKeys: [], // 已选中的节点
      allEntList: {} // 由于antd vue关联操作，无法直接获取到父ID, 需要内部自行维护一套数据结构 {entId: {pid, children}}
    })


defineExpose({initTree,getSelectedEntIdList})
 function initTree(recordId) { // 弹层打开事件
      // 判断是否有权限访问
      if (!data.hasEnt) {
        return false
      }

      // 重置数据
      data.checkedKeys = []
      data.treeData = []
      data.allEntList = {}

      data.recordId = recordId
      // 获取全部权限的树状结构
      $getEntTree('PLATFORM').then(res => {
        // 前置拦截代理商
        let preposition = []
        res.forEach(item => {
          if (!$hasAgentEnt()) {
            if (item.entName.includes('代理商')) return
          } 
          preposition.push(item as never)
        })
        
        data.treeData = preposition
        // 存储所有的菜单权限集合
       recursionTreeData(res, (item) => {
          data.allEntList[item.entId] = { pid: item.pid, children: item.children || [] }
        })

        // 查询所有的已分配的权限集合 (默认为 0 ， 无数据)
        req.list(API_URL_ROLE_ENT_RELA_LIST, { roleId: recordId || 'NONE', pageSize: -1 }).then(res2 => {
          const checkedEntIdList:any = [] // 所有已分配的权限集合（兼容antd vue 仅保留子节点）

          res2.records.map(item => {
            if (data.allEntList[item.entId] && data.allEntList[item.entId].children.length <= 0) { // 说明是子节点
              checkedEntIdList.push(item.entId)
            }
          })

          data.checkedKeys = checkedEntIdList
        })
      })
    }

   function getSelectedEntIdList() { // 获取已选择的列表集合
      // 判断是否有权限访问
      if (!data.hasEnt) {
        return false
      }
      
      const reqData:any = []

      data.checkedKeys.map(item => {
        const pidList:any = [] // 当前权限的所有的父节点IDList
        getAllPid(item, pidList)
        pidList.map(pid => {
          if (reqData.indexOf(pid) < 0) {
            reqData.push(pid)
          }
        })
      })
      return reqData
    }

    // 递归遍历树状结构数据
  function  recursionTreeData (entTreeData, func) {
      for (let i = 0; i < entTreeData.length; i++) {

        const thisEnt = entTreeData[i]
        
        if (thisEnt.children && thisEnt.children.length > 0) {
          recursionTreeData(thisEnt.children, func)
        }
        func(thisEnt)
      }
    }

   function getAllPid (entId, array) { // 获取所有的PID
      if (data.allEntList[entId] && entId !== 'ROOT') {
        array.push(entId)
       getAllPid(data.allEntList[entId].pid, array)
      }
    }
</script>
