edit

src/app/routes/menu-manager/services/menu-manager.service.ts

@@ -1,23 +1,149 @@
 import { Injectable, Injector } from '@angular/core';
+import { Menu } from '@delon/theme';
 import { BaseService } from '@shared';
 
 @Injectable({
-  providedIn: 'root',
+  providedIn: 'root'
 })
 export class MenuManagerService extends BaseService {
   // 新增/更新菜单
   $api_add_one = `/api/mdc/cuc/functionInfo/saveFunctionInfo`;
-
   // 根据应用ID获取所有菜单
   $api_get_all = `/api/mdc/cuc/functionInfo/getAllFunctionInfoByAppId`;
-
+  // 获取菜单详情
+  $api_get_menu_by_id = `/api/mdc/cuc/functionInfo/getFunctionInfo`;
   // 根据应用ID获取菜单
   $api_get_one = `/api/mdc/cuc/functionInfo/getAllFunctionInfoByAppId?_allow_badcode=true`;
-
   // 删除多个菜单
   $api_del_many = `/api/mdc/cuc/functionInfo/deletebatchFunctionInfo`;
 
+  // 获取菜单下按钮权限列表
+  $api_get_functions_by_id = `/api/mdc/cuc/functionButton/getFunctionButtonByFunctionId`;
+  // 获取所有按钮信息
+  $api_get_functions= `/api/mdc/cuc/buttonInfo/getButtonInfoList`;
+  // 保存菜单按钮关联
+  $api_save_menu_function= `/api/mdc/cuc/functionButton/saveFunctionButton`;
+
   constructor(public injector: Injector) {
     super(injector);
   }
+
+  getMenuByAppID(appId: string) {
+    this.request(this.$api_get_one, { appId }, 'POST', false).subscribe(res => {
+      if (res) {
+        const menus = res.data;
+        if (res.data?.length > 0) {
+          this.deleteMenuByAppID(res.data);
+        } else {
+          this.msgSrv.success('菜单已清空');
+        }
+      }
+    });
+  }
+
+  deleteMenuByAppID(arr: Array<any>) {
+    let ids: any[] = arr?.map(item => item.id) || [];
+    arr.forEach(item => {
+      if (item.children?.length > 0) {
+        this.deleteMenuByAppID(item.children);
+      }
+    });
+    this.request(this.$api_del_many, ids).subscribe(res => {});
+  }
+
+  menuImport(enName: string, appId: string) {
+    this.http.request('GET', `assets/mocks/platform/${enName}.json`).subscribe((res: any) => {
+      this.addMenu(res.menu, appId);
+    });
+  }
+
+  addMenu(menus: Array<Menu>, appId: string, parentId: string = '') {
+    menus.forEach(r => {
+      if (parentId !== '') {
+        r.parentId = parentId;
+      }
+      this.request(this.$api_get_one, { appId }, 'POST', false).subscribe(res => {
+        // 如果res.data存在，则更新菜单
+        if (res.data) {
+          r.id = res.data.id;
+        }
+        this.addOne({ appId, ...r, isLeaf: !(r.children && r.children.length > 0) }).subscribe(result => {
+          if (result) {
+            if (r.children && r.children.length > 0) {
+              this.addMenu(r.children, appId, result.id);
+            }
+          }
+        });
+      });
+    });
+
+    // this.loadMenus(this.selectedPlatform.appId);
+  }
+
+  collapse(array: TreeNodeInterface[], data: TreeNodeInterface, $event: boolean): void {
+    if (!$event) {
+      if (data.children) {
+        data.children.forEach(d => {
+          const target = array.find(a => a.key === d.key)!;
+          target.expand = false;
+          this.collapse(array, target, false);
+        });
+      } else {
+        return;
+      }
+    }
+  }
+
+  convertTreeToList(root: TreeNodeInterface): TreeNodeInterface[] {
+    const stack: TreeNodeInterface[] = [];
+    const array: TreeNodeInterface[] = [];
+    const hashMap = {};
+    stack.push({ ...root, level: 0, expand: true });
+
+    while (stack.length !== 0) {
+      const node = stack.pop()!;
+      this.visitNode(node, hashMap, array);
+      if (node.children) {
+        for (let i = node.children.length - 1; i >= 0; i--) {
+          stack.push({
+            ...node.children[i],
+            level: node.level! + 1,
+            expand: false,
+            parent: node,
+            iconType: this.formatIcon(node.children[i].icon)
+          });
+        }
+      }
+    }
+
+    return array;
+  }
+
+  visitNode(node: TreeNodeInterface, hashMap: { [key: string]: boolean }, array: TreeNodeInterface[]): void {
+    if (!hashMap[node.key]) {
+      hashMap[node.key] = true;
+      array.push(node);
+    }
+  }
+
+  private formatIcon(icon: any) {
+    let value = icon;
+    // compatible `anticon anticon-user`
+    if (~icon.indexOf(`anticon-`)) {
+      value = value.split('-').slice(1).join('-');
+    }
+    return value;
+  }
+}
+
+export interface TreeNodeInterface {
+  key: string;
+  name: string;
+  age?: number;
+  level?: number;
+  expand?: boolean;
+  address?: string;
+  children?: TreeNodeInterface[];
+  parent?: TreeNodeInterface;
+  [key: string]: any;
 }