From d126b193932df406fd77e621e20be78c5b30e325 Mon Sep 17 00:00:00 2001
From: sin <2943460818@qq.com>
Date: Tue, 12 Mar 2019 02:04:15 +0800
Subject: [PATCH] =?UTF-8?q?-=20=E4=BC=98=E5=8C=96=E8=8A=82=E7=82=B9?=
 =?UTF-8?q?=E6=9F=A5=E6=89=BE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 admin-web/src/models/admin/roleList.js | 86 ++++++++++++++++++--------
 admin-web/src/pages/Admin/RoleList.js  |  6 +-
 2 files changed, 63 insertions(+), 29 deletions(-)

diff --git a/admin-web/src/models/admin/roleList.js b/admin-web/src/models/admin/roleList.js
index dfd2b26f..32071b53 100644
--- a/admin-web/src/models/admin/roleList.js
+++ b/admin-web/src/models/admin/roleList.js
@@ -7,7 +7,6 @@ import {
   queryRole,
   queryRoleResourceTree,
   roleAssignResource,
-  resourceTree,
 } from '../../services/admin';
 
 function buildTreeNode(nodes, titleKey, nodeKey) {
@@ -22,10 +21,57 @@ function buildTreeNode(nodes, titleKey, nodeKey) {
   });
 }
 
-function getKeys(treeData) {
-  return treeData.map(item => {
-    return item.key;
-  });
+function findNodes(id, nodes) {
+  const res = [];
+  for (let i = 0; i < nodes.length; i += 1) {
+    const node = nodes[i];
+    if (node.key === id) {
+      res.push(node.key);
+      break;
+    } else {
+      const childNodes = findNodes(id, node.children);
+      if (childNodes.length) {
+        res.push(node.key);
+        for (let j = 0; j < childNodes.length; j += 1) {
+          res.push(childNodes[j]);
+        }
+        break;
+      }
+    }
+  }
+  return res;
+}
+
+function findAllNodes(resourceIds, nodes) {
+  const findNodesArray = [];
+  for (let i = 0; i < resourceIds.length; i += 1) {
+    const findNodesData = findNodes(resourceIds[i], nodes);
+    if (findNodesData) {
+      for (let j = 0; j < findNodesData.length; j += 1) {
+        const jD = findNodesData[j];
+        if (findNodesArray.indexOf(jD) === -1) {
+          findNodesArray.push(jD);
+        }
+      }
+    }
+  }
+  return findNodesArray;
+}
+
+function findCheckedKeys(nodes) {
+  let res = [];
+  for (let i = 0; i < nodes.length; i += 1) {
+    const node = nodes[i];
+    if (node.children) {
+      const findChildrenNodes = findCheckedKeys(node.children);
+      if (findChildrenNodes) {
+        res = res.concat(findChildrenNodes);
+      }
+    } else if (node.assigned === true) {
+      res.push(node.id);
+    }
+  }
+  return res;
 }
 
 export default {
@@ -38,7 +84,6 @@ export default {
     pageSize: 10,
 
     roleTreeData: [],
-    resourceTreeData: [],
     checkedKeys: [],
     assignModalLoading: true,
   },
@@ -99,31 +144,16 @@ export default {
         payload: true,
       });
 
-      const roleResourceResponse = yield call(queryRoleResourceTree, payload);
-      const resourceTreeResponse = yield call(resourceTree);
-
-      const roleTreeData = buildTreeNode(roleResourceResponse.data, 'displayName', 'id');
-      const resourceTreeData = buildTreeNode(resourceTreeResponse.data, 'displayName', 'id');
-      const roleTreeIdKeys = getKeys(roleTreeData);
-      const resourceTreeIdKeys = getKeys(resourceTreeData);
-
-      const checkedKeys = roleTreeIdKeys.filter(roleKey => {
-        let res = false;
-        resourceTreeIdKeys.map(key => {
-          if (key === roleKey) {
-            res = true;
-          }
-          return key;
-        });
-        return res;
-      });
+      const response = yield call(queryRoleResourceTree, payload);
+      const roleResourceTree = response.data;
+      const roleTreeData = buildTreeNode(roleResourceTree, 'displayName', 'id');
+      const checkedKeys = findCheckedKeys(roleResourceTree);
 
       yield put({
         type: 'querySuccess',
         payload: {
           checkedKeys,
           roleTreeData,
-          resourceTreeData,
         },
       });
 
@@ -133,10 +163,12 @@ export default {
       });
     },
     *roleAssignResource({ payload }, { call }) {
-      const { id, resourceIds } = payload;
+      const { id, resourceIds, roleTreeData } = payload;
+      const assignNodes = findAllNodes(resourceIds, roleTreeData);
+
       const params = {
         id,
-        resourceIds: arrayToStringParams(resourceIds),
+        resourceIds: arrayToStringParams(assignNodes),
       };
       const response = yield call(roleAssignResource, params);
       if (response.code === 0) {
diff --git a/admin-web/src/pages/Admin/RoleList.js b/admin-web/src/pages/Admin/RoleList.js
index 3870d539..99abfcf3 100644
--- a/admin-web/src/pages/Admin/RoleList.js
+++ b/admin-web/src/pages/Admin/RoleList.js
@@ -100,6 +100,7 @@ const AssignModal = Form.create()(props => {
             <Tree
               defaultExpandAll={true}
               checkable={true}
+              multiple={true}
               checkedKeys={checkedKeys}
               onCheck={handleCheckBoxClick}
             >
@@ -212,6 +213,7 @@ class RoleList extends PureComponent {
       payload: {
         id: roleAssignRecord.id,
         resourceIds: data.checkedKeys,
+        roleTreeData: data.roleTreeData,
       },
     });
     this.handleAssignModalVisibleClose(false);
@@ -282,7 +284,7 @@ class RoleList extends PureComponent {
   render() {
     const { list, data } = this.props;
 
-    const { pageNo, pageSize, count, resourceTreeData, checkedKeys, assignModalLoading } = data;
+    const { pageNo, pageSize, count, roleTreeData, checkedKeys, assignModalLoading } = data;
     const { modalVisible, modalType, initValues, roleAssignVisible } = this.state;
 
     const parentMethods = {
@@ -350,7 +352,7 @@ class RoleList extends PureComponent {
         <CreateForm {...parentMethods} modalVisible={modalVisible} />
         <AssignModal
           loading={assignModalLoading}
-          treeData={resourceTreeData}
+          treeData={roleTreeData}
           checkedKeys={checkedKeys}
           handleOk={this.handleAssignOK}
           modalVisible={roleAssignVisible}