From e604283141d046a3f4b432889e1e94f727bc8648 Mon Sep 17 00:00:00 2001
From: vapao <live1989@foxmail.com>
Date: Wed, 27 Oct 2021 17:01:05 +0800
Subject: [PATCH] =?UTF-8?q?#=20=E6=B7=BB=E5=8A=A0web=E7=BB=88=E7=AB=AF?=
 =?UTF-8?q?=E6=90=9C=E7=B4=A2=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 spug_web/src/pages/ssh/index.js          | 31 +++++++++++++++++++++---
 spug_web/src/pages/ssh/index.module.less |  6 ++++-
 2 files changed, 32 insertions(+), 5 deletions(-)

diff --git a/spug_web/src/pages/ssh/index.js b/spug_web/src/pages/ssh/index.js
index 8b7e805..5f1e085 100644
--- a/spug_web/src/pages/ssh/index.js
+++ b/spug_web/src/pages/ssh/index.js
@@ -5,12 +5,12 @@
  */
 import React, { useEffect, useState } from 'react';
 import { observer } from 'mobx-react';
-import { Tabs, Tree, Spin } from 'antd';
-import { FolderOutlined, FolderOpenOutlined, CloudServerOutlined } from '@ant-design/icons';
+import { Tabs, Tree, Input, Spin } from 'antd';
+import { FolderOutlined, FolderOpenOutlined, CloudServerOutlined, SearchOutlined } from '@ant-design/icons';
 import { NotFound, AuthButton } from 'components';
 import Terminal from './Terminal';
 import FileManager from './FileManager';
-import { http, hasPermission } from 'libs';
+import { http, hasPermission, includes } from 'libs';
 import styles from './index.module.less';
 import LogoSpugText from 'layout/logo-spug-txt.png';
 import lds from 'lodash';
@@ -19,7 +19,9 @@ import lds from 'lodash';
 function WebSSH(props) {
   const [visible, setVisible] = useState(false);
   const [fetching, setFetching] = useState(true);
+  const [rawTreeData, setRawTreeData] = useState([]);
   const [treeData, setTreeData] = useState([]);
+  const [searchValue, setSearchValue] = useState();
   const [hosts, setHosts] = useState([]);
   const [activeId, setActiveId] = useState();
   const [hostId, setHostId] = useState();
@@ -28,11 +30,30 @@ function WebSSH(props) {
     window.document.title = 'Spug web terminal'
     window.addEventListener('beforeunload', leaveTips)
     http.get('/api/host/group/?with_hosts=1')
-      .then(res => setTreeData(res.treeData))
+      .then(res => setRawTreeData(res.treeData))
       .finally(() => setFetching(false))
     return () => window.removeEventListener('beforeunload', leaveTips)
   }, [])
 
+  useEffect(() => {
+    if (searchValue) {
+      const newTreeData = []
+      const loop = (data) => {
+        for (let item of data) {
+          if (item.children) {
+            loop(item.children)
+          } else if (item.isLeaf && includes(item.title, searchValue)) {
+            newTreeData.push(item)
+          }
+        }
+      }
+      loop(rawTreeData)
+      setTreeData(newTreeData)
+    } else {
+      setTreeData(rawTreeData)
+    }
+  }, [rawTreeData, searchValue])
+
   function leaveTips(e) {
     e.returnValue = '确定要离开页面？'
   }
@@ -97,6 +118,8 @@ function WebSSH(props) {
         </div>
         <div className={styles.hosts}>
           <Spin spinning={fetching}>
+            <Input allowClear className={styles.search} prefix={<SearchOutlined style={{color: '#999'}}/>}
+                   placeholder="输入检索" onChange={e => setSearchValue(e.target.value)}/>
             <Tree.DirectoryTree
               defaultExpandAll
               expandAction="doubleClick"
diff --git a/spug_web/src/pages/ssh/index.module.less b/spug_web/src/pages/ssh/index.module.less
index d7ca9d9..722469d 100644
--- a/spug_web/src/pages/ssh/index.module.less
+++ b/spug_web/src/pages/ssh/index.module.less
@@ -19,7 +19,6 @@
             }
         }
         .hosts {
-            margin-top: 12px;
             :global(.ant-tree-node-content-wrapper) {
                 overflow: hidden;
                 text-overflow: ellipsis;
@@ -28,6 +27,11 @@
             :global(.ant-tree) {
                 background-color: #fafafa;
             }
+
+            .search {
+                margin: 12px 10px;
+                width: 260px;
+            }
         }
     }
     .content {