Merge branch 'ww' into 'main'

fix: BUG in teambition && perf code

See merge request huang/yun-teng-iot-front!456

@@ -14,6 +14,7 @@
     "PROTOBUF",
     "PROTOBUF",
     "SNMP",
     "SNMP",
     "unref",
     "unref",
+    "vben",
     "VITE"
     "VITE"
   ]
   ]
 }
 }

@@ -34,7 +34,7 @@ enum DeviceManagerApi {
 }
 }
 
 
 export const devicePage = (params: DeviceQueryParam) => {
 export const devicePage = (params: DeviceQueryParam) => {
-  return defHttp.get<DeviceModel>({
+  return defHttp.get<PaginationResult<DeviceModel>>({
     url: DeviceManagerApi.DEVICE_URL,
     url: DeviceManagerApi.DEVICE_URL,
     params,
     params,
   });
   });
@@ -80,10 +80,9 @@ export const deleteDevice = (ids: string[]) => {
  * 查询设备配置
  * 查询设备配置
  * @param params pageSize page name
  * @param params pageSize page name
  */
  */
-export const deviceProfile = (params) => {
-  return defHttp.get({
+export const deviceProfile = () => {
+  return defHttp.get<DeviceRecord[]>({
     url: DeviceManagerApi.DEVICE_PROFILE_URL_ME,
     url: DeviceManagerApi.DEVICE_PROFILE_URL_ME,
-    params,
   });
   });
 };
 };
 
 

@@ -18,3 +18,32 @@ export interface RouteItem {
  * @description: Get menu return value
  * @description: Get menu return value
  */
  */
 export type getMenuListResultModel = RouteItem[];
 export type getMenuListResultModel = RouteItem[];
+
+export interface Meta {
+  title: string;
+  menuType: string;
+  status: string;
+  icon: string;
+  isLink: boolean;
+  ignoreKeepAlive: boolean;
+  hideMenu: boolean;
+}
+
+export interface MenuRecord {
+  id: string;
+  createTime: string;
+  updateTime: string;
+  name: string;
+  children: MenuRecord[];
+  path: string;
+  type: string;
+  permission: string;
+  sort: number;
+  component: string;
+  meta: Meta;
+  disabled?: boolean;
+  isDictCompareDisabled?: boolean;
+  icon?: string;
+  title?: string;
+  key?: string;
+}

@@ -181,9 +181,9 @@
         const newItem = JSON.parse(getItem);
         const newItem = JSON.parse(getItem);
         updataPersonlData.value = newItem;
         updataPersonlData.value = newItem;
       };
       };
-
-      const [registerModal, { closeModal }] = useModalInner(async (data) => {
+      const [registerModal, { closeModal }] = useModalInner(async () => {
         refreshCacheGetData();
         refreshCacheGetData();
+        const userInfo = userStore.getUserInfo;
         try {
         try {
           if (updataPersonlData.value != null) {
           if (updataPersonlData.value != null) {
             (peresonalPic.value = updataPersonlData.value.avatar),
             (peresonalPic.value = updataPersonlData.value.avatar),
@@ -194,13 +194,13 @@
               });
               });
             getPersonalDetailValue.value = updataPersonlData.value;
             getPersonalDetailValue.value = updataPersonlData.value;
           } else {
           } else {
-            if (data.userInfo) {
-              getPersonalDetailValue.value = data.userInfo;
-              peresonalPic.value = data.userInfo.avatar;
+            if (userInfo) {
+              getPersonalDetailValue.value = userInfo;
+              peresonalPic.value = userInfo.avatar;
               setFieldsValue({
               setFieldsValue({
-                realName: data.userInfo.realName,
-                phoneNumber: data.userInfo.phoneNumber,
-                email: data.userInfo.email,
+                realName: userInfo.realName,
+                phoneNumber: userInfo.phoneNumber,
+                email: userInfo.email,
               });
               });
             }
             }
           }
           }

@@ -62,10 +62,18 @@ export const step1Schemas: FormSchema[] = [
     label: '所属产品',
     label: '所属产品',
     required: true,
     required: true,
     component: 'ApiSelect',
     component: 'ApiSelect',
-    componentProps: ({ formActionType }) => {
+    componentProps: ({ formActionType, formModel }) => {
       const { setFieldsValue } = formActionType;
       const { setFieldsValue } = formActionType;
       return {
       return {
-        api: deviceProfile,
+        api: async () => {
+          const options = await deviceProfile();
+          const { profileId } = formModel;
+          if (profileId) {
+            const selectRecord = options.find((item) => item.tbProfileId === profileId);
+            selectRecord && setFieldsValue({ transportType: selectRecord!.transportType });
+          }
+          return options;
+        },
         labelField: 'name',
         labelField: 'name',
         valueField: 'tbProfileId',
         valueField: 'tbProfileId',
         onChange(
         onChange(
@@ -148,6 +156,7 @@ export const step1Schemas: FormSchema[] = [
     ifShow: ({ values }) => values.deviceType === 'SENSOR' && values.organizationId,
     ifShow: ({ values }) => values.deviceType === 'SENSOR' && values.organizationId,
     componentProps: ({ formModel }) => {
     componentProps: ({ formModel }) => {
       const { organizationId, transportType } = formModel;
       const { organizationId, transportType } = formModel;
+      if (![organizationId, transportType].every(Boolean)) return {};
       return {
       return {
         api: getGatewayDevice,
         api: getGatewayDevice,
         showSearch: true,
         showSearch: true,

@@ -224,31 +224,33 @@
       const [registerTbDetailDrawer, { openDrawer: openTbDeviceDrawer }] = useDrawer();
       const [registerTbDetailDrawer, { openDrawer: openTbDeviceDrawer }] = useDrawer();
       const [registerGatewayDetailDrawer, { openDrawer: openGatewayDetailDrawer }] = useDrawer();
       const [registerGatewayDetailDrawer, { openDrawer: openGatewayDetailDrawer }] = useDrawer();
 
 
-      const [registerTable, { reload, setSelectedRowKeys, setProps, setTableData, getForm }] =
-        useTable({
-          title: '设备列表',
-          api: devicePage,
-          immediate: immediateStatus.value,
-          columns,
-          formConfig: {
-            labelWidth: 100,
-            schemas: searchFormSchema,
-            resetFunc: resetFn,
-          },
-          useSearchForm: true,
-          showTableSetting: true,
-          bordered: true,
-          showIndexColumn: false,
-          rowKey: 'id',
-          searchInfo: searchInfo,
-          clickToRowSelect: false,
-          actionColumn: {
-            width: 200,
-            title: '操作',
-            slots: { customRender: 'action' },
-            fixed: 'right',
-          },
-        });
+      const [
+        registerTable,
+        { reload, setSelectedRowKeys, setProps, setTableData, getForm, setPagination },
+      ] = useTable({
+        title: '设备列表',
+        api: devicePage,
+        immediate: immediateStatus.value,
+        columns,
+        formConfig: {
+          labelWidth: 100,
+          schemas: searchFormSchema,
+          resetFunc: resetFn,
+        },
+        useSearchForm: true,
+        showTableSetting: true,
+        bordered: true,
+        showIndexColumn: false,
+        rowKey: 'id',
+        searchInfo: searchInfo,
+        clickToRowSelect: false,
+        actionColumn: {
+          width: 200,
+          title: '操作',
+          slots: { customRender: 'action' },
+          fixed: 'right',
+        },
+      });
       const { hasBatchDelete, handleDeleteOrBatchDelete, selectionOptions, resetSelectedRowKeys } =
       const { hasBatchDelete, handleDeleteOrBatchDelete, selectionOptions, resetSelectedRowKeys } =
         useBatchDelete(deleteDevice, handleSuccess, setProps);
         useBatchDelete(deleteDevice, handleSuccess, setProps);
       selectionOptions.rowSelection.getCheckboxProps = (record: Recordable) => {
       selectionOptions.rowSelection.getCheckboxProps = (record: Recordable) => {
@@ -275,14 +277,15 @@
       const deviceProfileId = ref('');
       const deviceProfileId = ref('');
       count.value = Number(getParams('count'));
       count.value = Number(getParams('count'));
       deviceProfileId.value = getParams('deviceProfileId') || '';
       deviceProfileId.value = getParams('deviceProfileId') || '';
-      console.log(deviceProfileId.value);
+
       const setRowClassName = async () => {
       const setRowClassName = async () => {
         if (deviceProfileId.value !== undefined) {
         if (deviceProfileId.value !== undefined) {
-          const { items } = await devicePage({
+          const { items, total } = await devicePage({
             page: 1,
             page: 1,
             pageSize: count.value === 0 ? 10 : count.value,
             pageSize: count.value === 0 ? 10 : count.value,
             deviceProfileId: deviceProfileId.value,
             deviceProfileId: deviceProfileId.value,
           });
           });
+          setPagination({ total });
           nextTick(() => {
           nextTick(() => {
             setTableData(items);
             setTableData(items);
             const { setFieldsValue, resetFields } = getForm();
             const { setFieldsValue, resetFields } = getForm();

@@ -9,18 +9,20 @@
   >
   >
     <BasicForm @register="registerForm">
     <BasicForm @register="registerForm">
       <template #menu>
       <template #menu>
-        <BasicTree
-          v-if="treeData.length"
-          checkable
-          toolbar
-          :expandedKeys="treeExpandData"
-          ref="treeRef"
-          :treeData="treeData"
-          :replaceFields="{ title: 'menuName' }"
-          :checkedKeys="roleMenus"
-          @check="handleCheckClick"
-          title="菜单分配"
-        />
+        <Spin :spinning="spinning">
+          <BasicTree
+            v-if="treeData.length"
+            checkable
+            toolbar
+            :expandedKeys="treeExpandData"
+            ref="treeRef"
+            :treeData="treeData"
+            :replace-fields="{ title: 'name', key: 'id' }"
+            :checkedKeys="roleMenus"
+            @check="handleCheckClick"
+            title="菜单分配"
+          />
+        </Spin>
       </template>
       </template>
     </BasicForm>
     </BasicForm>
   </BasicDrawer>
   </BasicDrawer>
@@ -28,7 +30,7 @@
 <script lang="ts">
 <script lang="ts">
   import { defineComponent, ref, computed, unref, nextTick } from 'vue';
   import { defineComponent, ref, computed, unref, nextTick } from 'vue';
   import { BasicForm, useForm } from '/@/components/Form/index';
   import { BasicForm, useForm } from '/@/components/Form/index';
-  import { formSchema } from './role.data';
+  import { formSchema, KeysTypeEnum, RoleMenuDictEnum } from './role.data';
   import { BasicDrawer, useDrawerInner } from '/@/components/Drawer';
   import { BasicDrawer, useDrawerInner } from '/@/components/Drawer';
   import { BasicTree, TreeItem } from '/@/components/Tree';
   import { BasicTree, TreeItem } from '/@/components/Tree';
   import { useMessage } from '/@/hooks/web/useMessage';
   import { useMessage } from '/@/hooks/web/useMessage';
@@ -36,22 +38,29 @@
   // 加载菜单数据
   // 加载菜单数据
   import { getMenuList, getMenusIdsByRoleId } from '/@/api/sys/menu';
   import { getMenuList, getMenusIdsByRoleId } from '/@/api/sys/menu';
   import { useI18n } from '/@/hooks/web/useI18n';
   import { useI18n } from '/@/hooks/web/useI18n';
-  import { RouteItem } from '/@/api/sys/model/menuModel';
+  import { MenuRecord } from '/@/api/sys/model/menuModel';
   import { saveOrUpdateRoleInfoWithMenu } from '/@/api/system/system';
   import { saveOrUpdateRoleInfoWithMenu } from '/@/api/system/system';
+  import { findDictItemByCode } from '/@/api/system/dict';
+  import { RoleEnum } from '/@/enums/roleEnum';
+  import { Spin } from 'ant-design-vue';
+  import { useUserStore } from '/@/store/modules/user';
+
+  type TreeData = MenuRecord & TreeItem;
+
   export default defineComponent({
   export default defineComponent({
     name: 'RoleDrawer',
     name: 'RoleDrawer',
-    components: { BasicDrawer, BasicForm, BasicTree },
+    components: { BasicDrawer, BasicForm, BasicTree, Spin },
     emits: ['success', 'register'],
     emits: ['success', 'register'],
     setup(_, { emit }) {
     setup(_, { emit }) {
-      const treeExpandData = ref([]);
+      const treeExpandData = ref<string[]>([]);
       const isUpdate = ref<boolean>(true);
       const isUpdate = ref<boolean>(true);
-      const treeData = ref<TreeItem[]>([]);
+      const treeData = ref<TreeData[]>([]);
       const roleMenus = ref<string[]>([]);
       const roleMenus = ref<string[]>([]);
       const allCheckedKeys = ref<string[]>([]);
       const allCheckedKeys = ref<string[]>([]);
       const roleId = ref<string>('');
       const roleId = ref<string>('');
       const treeRef = ref();
       const treeRef = ref();
-      const checked: any = ref([]); //需要选中的节点
-      const pidArr: any = ref([]); //获取父节点
+      const checked = ref<string[]>([]); //需要选中的节点
+      const spinning = ref(false);
 
 
       const [registerForm, { resetFields, setFieldsValue, validate }] = useForm({
       const [registerForm, { resetFields, setFieldsValue, validate }] = useForm({
         labelWidth: 100,
         labelWidth: 100,
@@ -59,24 +68,19 @@
         showActionButtonGroup: false,
         showActionButtonGroup: false,
       });
       });
 
 
-      // 递归函数，将RouteItem里面的字段换名称
-      function processChildren(items: RouteItem[]) {
-        items.forEach((item) => {
-          item.menuName = t(item.meta.title);
-          item.icon = item.meta.icon;
-          item.key = item.id;
-          if (item.children) {
-            processChildren(item.children);
+      const originMenus = ref();
+
+      const transformName = (data: TreeData[]) => {
+        return data.map((item) => {
+          item.name = t(item.name);
+          if (item.children && item.children.length) {
+            item.children = transformName(item.children as unknown as TreeData[]);
           }
           }
+          return item;
         });
         });
-      }
-      function lookForAllId(data = [], arr = []) {
-        for (const item of data) {
-          arr.push(item.id);
-        }
-        return arr;
-      }
-      const originMenus = ref();
+      };
+
+      const userStore = useUserStore();
       const [registerDrawer, { setDrawerProps, closeDrawer }] = useDrawerInner(async (data) => {
       const [registerDrawer, { setDrawerProps, closeDrawer }] = useDrawerInner(async (data) => {
         allCheckedKeys.value = [];
         allCheckedKeys.value = [];
         originMenus.value = [];
         originMenus.value = [];
@@ -84,53 +88,53 @@
         resetFields();
         resetFields();
         roleId.value = '';
         roleId.value = '';
         // 在打开弹窗时清除所有选择的菜单
         // 在打开弹窗时清除所有选择的菜单
-        treeRef.value && treeRef.value.checkAll(false);
+        treeRef.value && treeRef.value?.setCheckedKeys([]);
         isUpdate.value = data.isUpdate;
         isUpdate.value = data.isUpdate;
-        // 需要在setFieldsValue之前先填充treeData，否则Tree组件可能会报key not exist警告
-        if (!unref(treeData).length) {
-          // 获取全部的菜单
-          const menuListModel = await getMenuList();
-          processChildren(menuListModel);
-          treeData.value = menuListModel;
-          //修复角色菜单新增-全部展开问题-只展开第一级即可
-          nextTick(() => {
-            const getAllIds = lookForAllId(treeData.value, []);
-            treeExpandData.value = getAllIds;
-          });
-        }
-        // 更新
-        if (unref(isUpdate)) {
-          allCheckedKeys.value = [];
-          checked.value = [];
-          //通过角色id去获取角色对应的菜单的ids
-          roleMenus.value = await getMenusIdsByRoleId(data.record.id);
-          originMenus.value = [...roleMenus.value];
-
-          const getAllIds = lookForAllId(treeData.value, []);
-          treeExpandData.value = getAllIds;
-          for (let item of treeData.value) {
-            if (item?.children != 0) {
-              pidArr.value.push(item.key);
-            }
-            for (let item1 of item?.children) {
-              if (item1?.children != 0) {
-                pidArr.value.push(item1.key);
-              }
-            }
+        const roleType = data?.record?.roleType || userStore.getRoleList.at(0);
+
+        try {
+          spinning.value = true;
+          // 需要在setFieldsValue之前先填充treeData，否则Tree组件可能会报key not exist警告
+
+          if (!unref(treeData).length) {
+            // 获取全部的菜单
+            const menuListModel = await getMenuList();
+            treeData.value = transformName(menuListModel as unknown as TreeData[]);
+            //修复角色菜单新增-全部展开问题-只展开第一级即可
+            await nextTick();
+            const getExpandKeys = unref(treeData).map((item) => item.id);
+            treeExpandData.value = getExpandKeys;
           }
           }
-          for (let item of roleMenus.value) {
-            let isP = pidArr.value.includes(item);
-            if (!isP) {
-              checked.value.push(item);
-            }
+
+          const keys = await getPermissionByRole(roleType);
+          const { keyType } = RoleMenuDictEnum[roleType];
+          treeData.value = getPermissionTreeData(
+            unref(treeData) as unknown as TreeData[],
+            keys,
+            keyType
+          );
+
+          // 更新
+          if (unref(isUpdate)) {
+            allCheckedKeys.value = [];
+            checked.value = [];
+            roleId.value = data.record.id;
+
+            //通过角色id去获取角色对应的菜单的ids
+            originMenus.value = roleMenus.value = await getMenusIdsByRoleId(data.record.id);
+
+            const getExpandKeys = unref(treeData).map((item) => item.id);
+            treeExpandData.value = getExpandKeys;
+
+            await nextTick();
+            treeRef.value.setCheckedKeys(roleMenus.value);
+            setFieldsValue(data.record);
+          } else {
           }
           }
-          nextTick(() => {
-            treeRef.value.setCheckedKeys(checked.value);
-            // const expandTreeIds = lookForAllId(treeData.value);
-            // treeRef.value.setExpandedKeys(expandTreeIds);
-          });
-          roleId.value = data.record.id;
-          setFieldsValue(data.record);
+        } catch (error) {
+          throw error;
+        } finally {
+          spinning.value = false;
         }
         }
       });
       });
 
 
@@ -146,7 +150,7 @@
             name: values.name,
             name: values.name,
             remark: values.remark,
             remark: values.remark,
             status: values.status,
             status: values.status,
-            menu: allCheckedKeys.value.length ? allCheckedKeys.value : originMenus.value,
+            menu: unref(treeRef).getCheckedKeys() || [],
           };
           };
           if (req.menu == undefined) return createMessage.error('请勾选权限菜单');
           if (req.menu == undefined) return createMessage.error('请勾选权限菜单');
           saveOrUpdateRoleInfoWithMenu(req).then(() => {
           saveOrUpdateRoleInfoWithMenu(req).then(() => {
@@ -165,7 +169,74 @@
         allCheckedKeys.value = [...checkedNodes, ...halfCheckedKeys];
         allCheckedKeys.value = [...checkedNodes, ...halfCheckedKeys];
       };
       };
 
 
+      const getPermissionByRole = async (roleType: RoleEnum) => {
+        try {
+          const { key } = RoleMenuDictEnum[roleType];
+          const res = await findDictItemByCode({ dictCode: key });
+          return res.map((item) => item.itemValue);
+        } catch (error) {}
+        return [];
+      };
+
+      const getPermissionTreeData = (
+        data: MenuRecord[],
+        permissionKeys: string[],
+        keysType: KeysTypeEnum
+      ) => {
+        const setDisabled = (data: MenuRecord[], flag: boolean) => {
+          return data.map((item) => {
+            item.name = t(item.name);
+            if (item.children && item.children.length) {
+              item.children = setDisabled(item.children, flag);
+            }
+            return {
+              ...item,
+              disabled: flag,
+              icon: item.meta.icon,
+            } as TreeData;
+          });
+        };
+
+        const permissionCompare = (
+          data: MenuRecord[],
+          permissionKeys: string[],
+          keysType: KeysTypeEnum
+        ) => {
+          return data.map((item) => {
+            item.name = t(item.name);
+            const findFlag = permissionKeys.includes(item.permission);
+            if (findFlag) item.isDictCompareDisabled = true;
+            const disabledFlag = keysType === KeysTypeEnum.DISABLED ? findFlag : !findFlag;
+            item.disabled = disabledFlag;
+
+            if (item.isDictCompareDisabled && item.children && item.children.length) {
+              setDisabled(item.children, disabledFlag);
+            } else {
+              if (item.children && item.children.length) {
+                item.children = permissionCompare(item.children, permissionKeys, keysType);
+                item.disabled = item.children.every((temp) => temp.disabled);
+              }
+            }
+            return {
+              ...item,
+              icon: item.meta.icon,
+            } as TreeData;
+          });
+        };
+
+        const result = permissionCompare(data, permissionKeys, keysType).map((item) => {
+          if (item.children && item.children.length) {
+            const rootDisabledFlag = item.children.every((temp) => temp.disabled);
+            item.disabled = rootDisabledFlag;
+          }
+          return item;
+        });
+
+        return result;
+      };
+
       return {
       return {
+        spinning,
         registerDrawer,
         registerDrawer,
         registerForm,
         registerForm,
         getTitle,
         getTitle,
@@ -185,9 +256,11 @@
   :deep(.vben-basic-tree) {
   :deep(.vben-basic-tree) {
     width: 100% !important;
     width: 100% !important;
   }
   }
+
   :deep(.is-unflod) {
   :deep(.is-unflod) {
     display: none !important;
     display: none !important;
   }
   }
+
   :deep(.is-flod) {
   :deep(.is-flod) {
     display: none !important;
     display: none !important;
   }
   }

 import { BasicColumn } from '/@/components/Table';
 import { BasicColumn } from '/@/components/Table';
 import { FormSchema } from '/@/components/Table';
 import { FormSchema } from '/@/components/Table';
+import { RoleEnum } from '/@/enums/roleEnum';
+
+export enum KeysTypeEnum {
+  DISABLED = 'disabled',
+  ENABLED = 'enabled',
+}
+
+export const RoleMenuDictEnum: Recordable<{ key: string; keyType: KeysTypeEnum }> = {
+  [RoleEnum.PLATFORM_ADMIN]: { key: 'enabled_platform_admin_auth', keyType: KeysTypeEnum.ENABLED },
+  [RoleEnum.SYS_ADMIN]: { key: 'enabled_sysadmin_auth', keyType: KeysTypeEnum.ENABLED },
+  [RoleEnum.TENANT_ADMIN]: { key: 'disabled_tenant_auth', keyType: KeysTypeEnum.DISABLED },
+  [RoleEnum.CUSTOMER_USER]: { key: 'disabled_tenant_auth', keyType: KeysTypeEnum.DISABLED },
+};
 
 
 export const columns: BasicColumn[] = [
 export const columns: BasicColumn[] = [
   {
   {

@@ -244,6 +244,7 @@
           permissionStore.setPermCodeList(permissionList);
           permissionStore.setPermCodeList(permissionList);
           userStore.setUserInfo(userInfo);
           userStore.setUserInfo(userInfo);
           userStore.setRoleList(userInfo.roles as RoleEnum[]);
           userStore.setRoleList(userInfo.roles as RoleEnum[]);
+          permissionStore.buildRoutesAction();
           go(PageEnum.BASE_HOME);
           go(PageEnum.BASE_HOME);
         } catch (error) {
         } catch (error) {
         } finally {
         } finally {

@@ -9,19 +9,21 @@
   >
   >
     <BasicForm @register="registerForm">
     <BasicForm @register="registerForm">
       <template #menu>
       <template #menu>
-        <BasicTree
-          v-if="treeData.length > 0"
-          :treeData="treeData"
-          :expandedKeys="treeExpandData"
-          :replaceFields="{ title: 'menuName' }"
-          :checkedKeys="roleMenus"
-          @check="handleCheckClick"
-          checkable
-          toolbar
-          ref="treeRef"
-          title="权限分配"
-          :defaultExpandAll="true"
-        />
+        <Spin :spinning="spinning">
+          <BasicTree
+            v-if="treeData.length"
+            :treeData="treeData"
+            :expandedKeys="treeExpandData"
+            :replaceFields="{ title: 'name', key: 'id' }"
+            :checkedKeys="roleMenus"
+            @check="handleCheckClick"
+            checkable
+            toolbar
+            ref="treeRef"
+            title="权限分配"
+            :defaultExpandAll="true"
+          />
+        </Spin>
       </template>
       </template>
     </BasicForm>
     </BasicForm>
   </BasicDrawer>
   </BasicDrawer>
@@ -36,17 +38,22 @@
   // 加载菜单数据
   // 加载菜单数据
   import { getMenuList, getMenusIdsByRoleId } from '/@/api/sys/menu';
   import { getMenuList, getMenusIdsByRoleId } from '/@/api/sys/menu';
   import { useI18n } from '/@/hooks/web/useI18n';
   import { useI18n } from '/@/hooks/web/useI18n';
-  import { RouteItem } from '/@/api/sys/model/menuModel';
+  import { MenuRecord } from '/@/api/sys/model/menuModel';
   import { saveOrUpdateRoleInfoWithMenu } from '/@/api/system/system';
   import { saveOrUpdateRoleInfoWithMenu } from '/@/api/system/system';
   import { RoleEnum } from '/@/enums/roleEnum';
   import { RoleEnum } from '/@/enums/roleEnum';
   import { useMessage } from '/@/hooks/web/useMessage';
   import { useMessage } from '/@/hooks/web/useMessage';
+  import { KeysTypeEnum, RoleMenuDictEnum } from '../../system/role/role.data';
+  import { findDictItemByCode } from '/@/api/system/dict';
+  import { Spin } from 'ant-design-vue';
+
+  type TreeData = MenuRecord & TreeItem;
 
 
   export default defineComponent({
   export default defineComponent({
     name: 'RoleDrawer',
     name: 'RoleDrawer',
-    components: { BasicDrawer, BasicForm, BasicTree },
+    components: { BasicDrawer, BasicForm, BasicTree, Spin },
     emits: ['success', 'register'],
     emits: ['success', 'register'],
     setup(_, { emit }) {
     setup(_, { emit }) {
-      const treeExpandData = ref([]);
+      const treeExpandData = ref<string[]>([]);
 
 
       const isUpdate = ref(true);
       const isUpdate = ref(true);
       const treeData = ref<TreeItem[]>([]);
       const treeData = ref<TreeItem[]>([]);
@@ -54,8 +61,8 @@
       const allCheckedKeys = ref<string[]>([]);
       const allCheckedKeys = ref<string[]>([]);
       const roleId = ref('');
       const roleId = ref('');
       const treeRef = ref();
       const treeRef = ref();
-      const checked: any = ref([]); //需要选中的节点
-      const pidArr: any = ref([]); //获取父节点
+      const checked = ref<string[]>([]); //需要选中的节点
+      const spinning = ref(false);
 
 
       const [registerForm, { resetFields, setFieldsValue, validate }] = useForm({
       const [registerForm, { resetFields, setFieldsValue, validate }] = useForm({
         labelWidth: 90,
         labelWidth: 90,
@@ -63,77 +70,71 @@
         showActionButtonGroup: false,
         showActionButtonGroup: false,
       });
       });
 
 
-      function processChildren(items: RouteItem[]) {
-        items.map((item) => {
-          item.menuName = t(item.meta.title);
-          item.icon = item.meta.icon;
-          item.key = item.id;
-          if (item.children) {
-            processChildren(item.children);
+      const originMenus = ref();
+
+      const transformName = (data: TreeData[]) => {
+        return data.map((item) => {
+          item.name = t(item.name);
+          if (item.children && item.children.length) {
+            item.children = transformName(item.children as unknown as TreeData[]);
           }
           }
+          return item;
         });
         });
-      }
-      function lookForAllId(data = [], arr = []) {
-        for (const item of data) {
-          arr.push(item.id);
-        }
-        return arr;
-      }
+      };
 
 
-      const originMenus = ref();
       const [registerDrawer, { setDrawerProps, closeDrawer }] = useDrawerInner(async (data) => {
       const [registerDrawer, { setDrawerProps, closeDrawer }] = useDrawerInner(async (data) => {
-        setDrawerProps({ confirmLoading: false });
         allCheckedKeys.value = [];
         allCheckedKeys.value = [];
         originMenus.value = [];
         originMenus.value = [];
-        treeExpandData.value;
         allCheckedKeys.value.length = 0;
         allCheckedKeys.value.length = 0;
         resetFields();
         resetFields();
         roleId.value = '';
         roleId.value = '';
         // 在打开弹窗时清除所有选择的菜单
         // 在打开弹窗时清除所有选择的菜单
-        treeRef.value && treeRef.value.checkAll(false);
+        treeRef.value && treeRef.value?.setCheckedKeys([]);
         isUpdate.value = data.isUpdate;
         isUpdate.value = data.isUpdate;
-        // 需要在setFieldsValue之前先填充treeData，否则Tree组件可能会报key not exist警告
-        if (!unref(treeData).length) {
-          // 获取全部的菜单
-          const menuListModel = await getMenuList();
-          processChildren(menuListModel);
-          treeData.value = menuListModel;
-          //修复角色菜单新增-全部展开问题-只展开第一级即可
-          nextTick(() => {
-            const getAllIds = lookForAllId(treeData.value, []);
-            treeExpandData.value = getAllIds;
-          });
-        }
-        // 更新
-        if (unref(isUpdate)) {
-          allCheckedKeys.value = [];
-          checked.value = [];
-          //通过角色id去获取角色对应的菜单的ids
-          roleMenus.value = await getMenusIdsByRoleId(data.record.id);
-          originMenus.value = [...roleMenus.value];
-          const getAllIds = lookForAllId(treeData.value, []);
-          treeExpandData.value = getAllIds;
-          for (let item of treeData.value) {
-            if (item?.children != 0) {
-              pidArr.value.push(item.key);
-            }
-            for (let item1 of item?.children) {
-              if (item1?.children != 0) {
-                pidArr.value.push(item1.key);
-              }
-            }
+        const roleType = data?.record?.roleType || RoleEnum.TENANT_ADMIN;
+        try {
+          spinning.value = true;
+          // 需要在setFieldsValue之前先填充treeData，否则Tree组件可能会报key not exist警告
+
+          if (!unref(treeData).length) {
+            // 获取全部的菜单
+            const menuListModel = await getMenuList();
+            treeData.value = transformName(menuListModel as unknown as TreeData[]);
+            //修复角色菜单新增-全部展开问题-只展开第一级即可
+            await nextTick();
+            const getExpandKeys = unref(treeData).map((item) => item.id);
+            treeExpandData.value = getExpandKeys;
           }
           }
-          for (let item of roleMenus.value) {
-            let isP = pidArr.value.includes(item);
-            if (!isP) {
-              checked.value.push(item);
-            }
+
+          const keys = await getPermissionByRole(roleType);
+          const { keyType } = RoleMenuDictEnum[roleType];
+          treeData.value = getPermissionTreeData(
+            unref(treeData) as unknown as TreeData[],
+            keys,
+            keyType
+          );
+
+          // 更新
+          if (unref(isUpdate)) {
+            allCheckedKeys.value = [];
+            checked.value = [];
+            roleId.value = data.record.id;
+
+            //通过角色id去获取角色对应的菜单的ids
+            originMenus.value = roleMenus.value = await getMenusIdsByRoleId(data.record.id);
+
+            const getExpandKeys = unref(treeData).map((item) => item.id);
+            treeExpandData.value = getExpandKeys;
+
+            await nextTick();
+            treeRef.value.setCheckedKeys(roleMenus.value);
+            setFieldsValue(data.record);
+          } else {
           }
           }
-          nextTick(() => {
-            treeRef.value.setCheckedKeys(checked.value);
-          });
-          roleId.value = data.record.id;
-          setFieldsValue(data.record);
+        } catch (error) {
+          throw error;
+        } finally {
+          spinning.value = false;
         }
         }
       });
       });
       const getTitle = computed(() => (!unref(isUpdate) ? '新增角色' : '编辑角色'));
       const getTitle = computed(() => (!unref(isUpdate) ? '新增角色' : '编辑角色'));
@@ -150,7 +151,7 @@
             remark: values.remark,
             remark: values.remark,
             status: values.status,
             status: values.status,
             roleType: RoleEnum.TENANT_ADMIN,
             roleType: RoleEnum.TENANT_ADMIN,
-            menu: allCheckedKeys.value.length ? allCheckedKeys.value : originMenus.value,
+            menu: unref(treeRef).getCheckedKeys() || [],
           };
           };
           if (req.menu == undefined) return createMessage.error('请勾选权限菜单');
           if (req.menu == undefined) return createMessage.error('请勾选权限菜单');
           const res = await saveOrUpdateRoleInfoWithMenu(req);
           const res = await saveOrUpdateRoleInfoWithMenu(req);
@@ -173,7 +174,74 @@
         allCheckedKeys.value = [...checkedNodes, ...halfCheckedKeys];
         allCheckedKeys.value = [...checkedNodes, ...halfCheckedKeys];
       };
       };
 
 
+      const getPermissionByRole = async (roleType: RoleEnum) => {
+        try {
+          const { key } = RoleMenuDictEnum[roleType];
+          const res = await findDictItemByCode({ dictCode: key });
+          return res.map((item) => item.itemValue);
+        } catch (error) {}
+        return [];
+      };
+
+      const getPermissionTreeData = (
+        data: MenuRecord[],
+        permissionKeys: string[],
+        keysType: KeysTypeEnum
+      ) => {
+        const setDisabled = (data: MenuRecord[], flag: boolean) => {
+          return data.map((item) => {
+            item.name = t(item.name);
+            if (item.children && item.children.length) {
+              item.children = setDisabled(item.children, flag);
+            }
+            return {
+              ...item,
+              disabled: flag,
+              icon: item.meta.icon,
+            } as TreeData;
+          });
+        };
+
+        const permissionCompare = (
+          data: MenuRecord[],
+          permissionKeys: string[],
+          keysType: KeysTypeEnum
+        ) => {
+          return data.map((item) => {
+            item.name = t(item.name);
+            const findFlag = permissionKeys.includes(item.permission);
+            if (findFlag) item.isDictCompareDisabled = true;
+            const disabledFlag = keysType === KeysTypeEnum.DISABLED ? findFlag : !findFlag;
+            item.disabled = disabledFlag;
+
+            if (item.isDictCompareDisabled && item.children && item.children.length) {
+              setDisabled(item.children, disabledFlag);
+            } else {
+              if (item.children && item.children.length) {
+                item.children = permissionCompare(item.children, permissionKeys, keysType);
+                item.disabled = item.children.every((temp) => temp.disabled);
+              }
+            }
+            return {
+              ...item,
+              icon: item.meta.icon,
+            } as TreeData;
+          });
+        };
+
+        const result = permissionCompare(data, permissionKeys, keysType).map((item) => {
+          if (item.children && item.children.length) {
+            const rootDisabledFlag = item.children.every((temp) => temp.disabled);
+            item.disabled = rootDisabledFlag;
+          }
+          return item;
+        });
+
+        return result;
+      };
+
       return {
       return {
+        spinning,
         registerDrawer,
         registerDrawer,
         registerForm,
         registerForm,
         getTitle,
         getTitle,
@@ -192,9 +260,11 @@
   :deep(.vben-basic-tree) {
   :deep(.vben-basic-tree) {
     width: 100% !important;
     width: 100% !important;
   }
   }
+
   :deep(.is-unflod) {
   :deep(.is-unflod) {
     display: none !important;
     display: none !important;
   }
   }
+
   :deep(.is-flod) {
   :deep(.is-flod) {
     display: none !important;
     display: none !important;
   }
   }

@@ -40,6 +40,8 @@ export interface UserInfo {
   tenantName?: string;
   tenantName?: string;
   roles?: string[];
   roles?: string[];
   plainRoles?: PlainRoleInfo[];
   plainRoles?: PlainRoleInfo[];
+  phoneNumber?: string;
+  email?: string;
 }
 }
 
 
 export interface BeforeMiniState {
 export interface BeforeMiniState {