<script setup lang="ts">
import { computed, ref, onMounted } from 'vue'

import { useVbenModal } from '@vben/common-ui'
import { $t } from '@vben/locales'
import { cloneDeep } from '@vben/utils'

import { useVbenForm } from '#/adapter/form'
import { authGroupAdd, authGroupInfo, authGroupUpdate } from '#/api/sis/authGroup'
import { defaultFormValueGetter, useBeforeCloseDiff } from '#/utils/popup'
import { queryTree } from '#/api/sis/authRecord'
import { message, Tree } from 'ant-design-vue'

import { modalSchema } from './data'

const emit = defineEmits<{ reload: [] }>()

const isUpdate = ref(false)
const title = computed(() => {
  return isUpdate.value ? $t('pages.common.edit') : $t('pages.common.add')
})

const [BasicForm, formApi] = useVbenForm({
  commonConfig: {
    // 默认占满两列
    formItemClass: 'col-span-2',
    // 默认label宽度 px
    labelWidth: 80,
    // 通用配置项 会影响到所有表单项
    componentProps: {
      class: 'w-full',
    }
  },
  schema: modalSchema(),
  showDefaultActions: false,
  wrapperClass: 'grid-cols-2',
})

const { onBeforeClose, markInitialized, resetInitialized } = useBeforeCloseDiff(
  {
    initializedGetter: defaultFormValueGetter(formApi),
    currentGetter: defaultFormValueGetter(formApi),
  },
)

const [BasicModal, modalApi] = useVbenModal({
  // 在这里更改宽度
  class: 'w-[550px]',
  fullscreenButton: false,
  onBeforeClose,
  onClosed: handleClosed,
  onConfirm: handleConfirm,
  onOpenChange: async (isOpen) => {
    if (!isOpen) {
      return null
    }
    modalApi.modalLoading(true)
    loadDeviceTree()
    const { id } = modalApi.getData() as { id?: number | string }
    isUpdate.value = !!id
    if (isUpdate.value && id) {
      const record = await authGroupInfo(id)

      checkedKeys.value = (record.acIds || []).concat(record.floorIds || [])

      await formApi.setValues(record)
    }
    await markInitialized()

    modalApi.modalLoading(false)
  },
})

async function handleConfirm() {
  try {
    modalApi.lock(true)
    const { valid } = await formApi.validate()
    if (!valid) {
      return
    }

    if (eleIds.value.length === 0 && acIds.value.length === 0) {
      message.error('请选择授权设备')
      return
    }

    const data = cloneDeep(await formApi.getValues())
    data.acIds = acIds.value
    data.eleIds = eleIds.value
    data.floorIds = floorIds.value
    await (isUpdate.value ? authGroupUpdate(data) : authGroupAdd(data))
    resetInitialized()
    emit('reload')
    modalApi.close()
  } catch (error) {
    console.error(error)
  } finally {
    modalApi.lock(false)
  }
}

async function handleClosed() {
  await formApi.resetForm()
  checkedKeys.value = []
  resetInitialized()
}



const treeData = ref([])
function loadDeviceTree() {
  queryTree().then((data: any) => {
    treeData.value = data
  })
}


const fieldNames = {
  title: 'title',
  label: 'label',
  key: 'code',
  children: 'children',
}

const checkedKeys = ref<any[]>([])
type Key = string | number
const eleIds = ref<any[]>([])
const acIds = ref<any[]>([])
const floorIds = ref<any[]>([])
function handleCheck(checked: Key[] | { checked: Key[]; halfChecked: Key[] }, info: any) {

  acIds.value = []
  eleIds.value = []
  floorIds.value = []

  info.checkedNodesPositions.forEach((item: any) => {
    switch (item.node.label) {
      case 'accessControl':
        acIds.value = acIds.value.concat(item.node.code)
        break
      case 'floor':
        floorIds.value = floorIds.value.concat(item.node.code)
        eleIds.value = eleIds.value.concat(item.node.parentCode)
        break
      case 'elevator':
        eleIds.value = eleIds.value.concat(item.node.code)
        break
    }
  })

  // 去重
  acIds.value = [...new Set(acIds.value)]
  eleIds.value = [...new Set(eleIds.value)]
  floorIds.value = [...new Set(floorIds.value)]
}
</script>

<template>
  <BasicModal :title="title">
    <BasicForm />
    <div class="p-4">
      <Tree checkable :tree-data="treeData" :fieldNames="fieldNames" v-model:checkedKeys="checkedKeys"
        :onCheck="handleCheck">
      </Tree>
    </div>
  </BasicModal>
</template>

