添加Database类型资产链接支持 (#138)

* [update]添加Database支持

* [update]添加Database支持

* [update]修改bug

* [update]修改bug

* [update]修改函数名

* [update]删除多余判断，database登录没有走这个逻辑。
Co-authored-by: BaiJiangJie <32935519+BaiJiangJie@users.noreply.github.com>
Co-authored-by: 老广 <ibuler@qq.com>

src/app/elements/asset-tree/asset-tree.component.html

@@ -5,6 +5,9 @@
 
   <ul id="remoteAppsTree" class="ztree">
   </ul>
+
+  <ul id="DBAppsTree" class="ztree">
+  </ul>
 </div>
 
 

src/app/elements/asset-tree/asset-tree.component.ts

@@ -42,6 +42,7 @@ export class ElementAssetTreeComponent implements OnInit, OnDestroy {
   expandNodes: any;
   assetsTree: any;
   remoteAppsTree: any;
+  DBAppsTree: any;
   isShowRMenu = false;
   rightClickSelectNode: any;
   hasLoginTo = false;
@@ -127,7 +128,20 @@ export class ElementAssetTreeComponent implements OnInit, OnDestroy {
     this.remoteAppsTree.destroy();
     this.initRemoteAppsTree();
   }
-  
+
+  refreshDBAppsTree() {
+    this.DBAppsTree.destroy();
+    this.initDBAppsTree();
+  }
+
+  onDBAppsTreeNodeClick(event, treeId, treeNode, clickFlag) {
+    if (treeNode.isParent) {
+      this.DBAppsTree.expandNode(treeNode);
+    } else {
+      this._http.getUserProfile().subscribe();
+      this.connectAsset(treeNode);
+    }
+  }
   onRemoteAppsNodeClick(event, treeId, treeNode, clickFlag) {
     if (treeNode.isParent) {
       this.remoteAppsTree.expandNode(treeNode);
@@ -156,17 +170,37 @@ export class ElementAssetTreeComponent implements OnInit, OnDestroy {
       }
     );
   }
+  initDBAppsTree() {
+    const setting = Object.assign({}, this.setting);
+    setting['callback'] = {
+      onClick: this.onDBAppsTreeNodeClick.bind(this),
+      onRightClick: this.onRightClick.bind(this)
+    };
+    this._http.getMyGrantedDBApps().subscribe(
+      resp => {
+        if (resp.length === 1) {
+          return;
+        }
+        const tree = $.fn.zTree.init($('#DBAppsTree'), setting, resp);
+        this.DBAppsTree = tree;
+        this.rootNodeAddDom(tree, () => {
+          this.refreshDBAppsTree();
+        });
+      }
+    );
+  }
 
   initTree() {
     this.initAssetsTree();
     this.initRemoteAppsTree();
+    this.initDBAppsTree();
   }
 
   connectAsset(node: TreeNode) {
     const evt = new ConnectEvt(node, 'asset');
     connectEvt.next(evt);
   }
-  
+
   rootNodeAddDom(ztree, callback) {
     const tId = ztree.setting.treeId + '_tree_refresh';
     const refreshIcon = '<a id=' + tId + ' class="tree-refresh">' +

src/app/elements/connect/connect.component.ts

@@ -83,6 +83,9 @@ export class ElementConnectComponent implements OnInit, OnDestroy {
       case 'remote_app':
         this.connectRemoteApp(node);
         break;
+      case 'database_app':
+        this.connectDatabaseApp(node);
+        break;
       default:
         alert('Unknown type: ' + node.meta.type);
     }
@@ -100,6 +103,18 @@ export class ElementConnectComponent implements OnInit, OnDestroy {
     }
   }
 
+  async connectDatabaseApp(node: TreeNode) {
+    this._logger.debug('Connect remote app: ', node.id);
+    const systemUsers = await this._http.getMyDatabaseAppSystemUsers(node.id).toPromise();
+    let sysUser = await this.selectLoginSystemUsers(systemUsers);
+    sysUser = await this.manualSetUserAuthLoginIfNeed(sysUser);
+    if (sysUser && sysUser.id) {
+      this.loginDatabaseApp(node, sysUser);
+    } else {
+      alert('该主机没有授权系统用户');
+    }
+  }
+
   selectLoginSystemUsers(systemUsers: Array<SystemUser>): Promise<SystemUser> {
     const systemUserMaxPriority = this.filterMaxPrioritySystemUsers(systemUsers);
     let user: SystemUser;
@@ -184,6 +199,20 @@ export class ElementConnectComponent implements OnInit, OnDestroy {
       this.onNewView.emit(view);
     }
   }
+  loginDatabaseApp(node: TreeNode, user: SystemUser) {
+    if (node) {
+      const view = new View();
+      view.host = node;
+      view.nick = node.name;
+      view.connected = true;
+      view.editable = false;
+      view.closed = false;
+      view.DatabaseApp = node.id;
+      view.user = user;
+      view.type = 'database';
+      this.onNewView.emit(view);
+    }
+  }
 
   connectFileManager(node: TreeNode) {
     const host = node.meta.asset as Asset;

src/app/elements/content-window/content-window.component.html

@@ -3,7 +3,7 @@
     [view]="view"
     [host]="view.host"
     [sysUser]="view.user"
-    *ngIf="view.type=='ssh'"
+    *ngIf="view.type=='ssh' || view.type=='database'"
   >
   </elements-ssh-term>
   <elements-guacamole

src/app/elements/ssh-term/ssh-term.component.ts

@@ -34,7 +34,7 @@ export class ElementSshTermComponent implements OnInit, OnDestroy {
       this.ws = sock;
       this.connectHost();
     });
-    this.view.type = 'ssh';
+    // this.view.type = 'ssh';
   }
 
   newTerm() {
@@ -74,7 +74,8 @@ export class ElementSshTermComponent implements OnInit, OnDestroy {
         uuid: this.host.id,
         userid: this.sysUser.id,
         secret: this.secret,
-        size: [this.term.cols, this.term.rows]
+        size: [this.term.cols, this.term.rows],
+        type: this.view.type
       };
       this.ws.emit('host', data);
     }

src/app/model.ts

@@ -116,6 +116,7 @@ export class View {
   room: string;
   Rdp: any;
   Term: any;
+  DatabaseApp: string;
 }
 
 export class ViewAction {

src/app/services/http.ts

@@ -99,11 +99,24 @@ export class HttpService {
     return this.http.get<Array<TreeNode>>(url);
   }
 
+  getMyGrantedDBApps(id?: string) {
+    let url = '/api/v1/perms/user/database-apps/tree/';
+    if (id) {
+      url += `?id=${id}&only=1`;
+    }
+    return this.http.get<Array<TreeNode>>(url);
+  }
+
   getMyRemoteAppSystemUsers(remoteAppId: string) {
     const url = `/api/v1/perms/users/remote-apps/${remoteAppId}/system-users/`;
     return this.http.get<Array<SystemUser>>(url);
   }
 
+  getMyDatabaseAppSystemUsers(DatabaseAppId: string) {
+    const url = `/api/v1/perms/users/database-apps/${DatabaseAppId}/system-users/`;
+    return this.http.get<Array<SystemUser>>(url);
+  }
+
   getMyAssetSystemUsers(assetId: string) {
     const url = `/api/v1/perms/users/assets/${assetId}/system-users/`;
     return this.http.get<Array<SystemUser>>(url);
@@ -122,12 +135,10 @@ export class HttpService {
       return this.delete(url);
     }
   }
-
   getFavoriteAssets() {
     const url = '/api/v1/assets/favorite-assets/';
     return this.http.get<Array<any>>(url);
   }
-
   getGuacamoleToken(user_id: string, authToken: string) {
     const body = new HttpParams()
       .set('username', user_id)