From 0a1e92588c3292647f46609b92bde9d58bcc48c6 Mon Sep 17 00:00:00 2001
From: chaos-zhu <ggchaos1022@gmail.com>
Date: Tue, 30 Jul 2024 18:56:22 +0800
Subject: [PATCH] =?UTF-8?q?:sparkles:=20=E6=94=AF=E6=8C=81=E5=A4=9A?=
 =?UTF-8?q?=E4=BC=9A=E8=AF=9D=E5=90=8C=E6=AD=A5=E6=8C=87=E4=BB=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/app/controller/host.js                 |   7 +-
 server/app/db/README.md                       |   4 +-
 server/app/socket/terminal.js                 |   2 +-
 web/src/views/server/components/host-form.vue |   3 +-
 .../terminal/components/terminal-tab.vue      |  16 +-
 .../views/terminal/components/terminal.vue    | 176 ++++++++++++------
 web/src/views/terminal/index.vue              |  16 +-
 7 files changed, 155 insertions(+), 69 deletions(-)

diff --git a/server/app/controller/host.js b/server/app/controller/host.js
index 7971e1e..27389d5 100644
--- a/server/app/controller/host.js
+++ b/server/app/controller/host.js
@@ -6,8 +6,8 @@ async function getHostList({ res }) {
   data?.sort((a, b) => Number(b.index || 0) - Number(a.index || 0))
   for (const item of data) {
     let { username, port, authType, _id: id, credential } = item
+    console.log('解密凭证title: ', credential)
     if (credential) credential = await AESDecryptSync(credential)
-    // console.log(credential)
     const isConfig = Boolean(username && port && (item[authType]))
     Object.assign(item, { id, isConfig, password: '', privateKey: '', credential })
   }
@@ -98,7 +98,8 @@ async function importHost({
   // 过滤已存在的host
   let hostListSet = new Set(hostList.map(item => item.host))
   let newHostList = importHost.filter(item => !hostListSet.has(item.host))
-  if (newHostList.length === 0) return res.fail({ msg: '导入的实例已存在' })
+  let newHostListLen = newHostList.length
+  if (newHostListLen === 0) return res.fail({ msg: '导入的实例已存在' })
 
   let extraFiels = {
     expired: null, expiredNotify: false, group: 'default', consoleUrl: '', remark: '',
@@ -106,7 +107,7 @@ async function importHost({
   }
   newHostList = newHostList.map((item, index) => {
     item.port = Number(item.port) || 0
-    item.index = hostList.length + index + 1
+    item.index = newHostListLen - index
     return Object.assign(item, { ...extraFiels })
   })
   hostList.push(...newHostList)
diff --git a/server/app/db/README.md b/server/app/db/README.md
index 1272dc6..e844fa0 100644
--- a/server/app/db/README.md
+++ b/server/app/db/README.md
@@ -1,6 +1,6 @@
 db目录，初始化后自动生成
 
-**host-list.db**
+**host.db**
 
 > 存储服务器基本信息
 
@@ -8,7 +8,7 @@ db目录，初始化后自动生成
 
 > 用于加密的密钥相关
 
-**ssh-record.db**
+**credentials.db**
 
 > ssh密钥记录(加密存储)
 
diff --git a/server/app/socket/terminal.js b/server/app/socket/terminal.js
index adf6303..2621981 100644
--- a/server/app/socket/terminal.js
+++ b/server/app/socket/terminal.js
@@ -23,7 +23,7 @@ function createTerminal(socket, sshClient) {
 
     // 监听按键重置终端大小
     socket.on('resize', ({ rows, cols }) => {
-      consola.info('更改tty终端行&列: ', { rows, cols })
+      // consola.info('更改tty终端行&列: ', { rows, cols })
       stream.setWindow(rows, cols)
     })
   })
diff --git a/web/src/views/server/components/host-form.vue b/web/src/views/server/components/host-form.vue
index a03656d..982bb60 100644
--- a/web/src/views/server/components/host-form.vue
+++ b/web/src/views/server/components/host-form.vue
@@ -362,7 +362,8 @@ const handleSave = () => {
         $message({ type: 'success', center: true, message: msg })
       }
       visible.value = false
-      emit('update-list')
+      const { host, username, port, authType } = formData
+      emit('update-list', { isConfig: Boolean(username && port && (formData[authType])), host })
       Object.assign(hostForm, resetForm())
     })
 }
diff --git a/web/src/views/terminal/components/terminal-tab.vue b/web/src/views/terminal/components/terminal-tab.vue
index a64d503..39d4db5 100644
--- a/web/src/views/terminal/components/terminal-tab.vue
+++ b/web/src/views/terminal/components/terminal-tab.vue
@@ -3,7 +3,7 @@
 </template>
 
 <script setup>
-import { ref, onMounted, computed, onBeforeUnmount, getCurrentInstance } from 'vue'
+import { ref, onMounted, computed, onBeforeUnmount, getCurrentInstance, defineEmits } from 'vue'
 import { Terminal } from '@xterm/xterm'
 import '@xterm/xterm/css/xterm.css'
 import { FitAddon } from '@xterm/addon-fit'
@@ -19,9 +19,15 @@ const props = defineProps({
   host: {
     required: true,
     type: String
+  },
+  index: {
+    required: true,
+    type: Number
   }
 })
 
+const emit = defineEmits(['input',])
+
 const socket = ref(null)
 const term = ref(null)
 const command = ref('')
@@ -200,6 +206,8 @@ const onData = () => {
     let acsiiCode = key.codePointAt()
     if (acsiiCode === 22) return handlePaste()
     if (acsiiCode === 6) return searchBar.value.show()
+    emit('input', { idx: props.index, key })
+    // console.log('input:', key)
     socket.value.emit('input', key)
   })
 }
@@ -209,8 +217,10 @@ const handleClear = () => {
 }
 
 const handlePaste = async () => {
-  let str = await navigator.clipboard.readText()
-  socket.value.emit('input', str)
+  let key = await navigator.clipboard.readText()
+  emit('input', { idx: props.index, key })
+  // console.log('input:', key)
+  socket.value.emit('input', key)
   term.value.focus()
 }
 
diff --git a/web/src/views/terminal/components/terminal.vue b/web/src/views/terminal/components/terminal.vue
index 18c2af4..1534bac 100644
--- a/web/src/views/terminal/components/terminal.vue
+++ b/web/src/views/terminal/components/terminal.vue
@@ -1,23 +1,36 @@
 <template>
   <div class="terminal_wrap">
+    <div class="terminal_top">
+      <el-dropdown trigger="click">
+        <span class="link_text">新建连接<el-icon><arrow-down /></el-icon></span>
+        <template #dropdown>
+          <el-dropdown-menu>
+            <el-dropdown-item v-for="(item, index) in hostList" :key="index" @click="handleCommandHost(item)">
+              {{ item.name }} {{ item.host }}
+            </el-dropdown-item>
+          </el-dropdown-menu>
+        </template>
+      </el-dropdown>
+      <el-dropdown trigger="click">
+        <span class="link_text">会话同步<el-icon><arrow-down /></el-icon></span>
+        <template #dropdown>
+          <el-dropdown-menu>
+            <el-dropdown-item @click="handleSyncSession">
+              <el-icon v-show="isSyncAllSession"><Select class="action_icon" /></el-icon>
+              <span>同步键盘输入到所有会话</span>
+            </el-dropdown-item>
+            <!-- <el-dropdown-item @click="handleSyncSession">
+              同步键盘输入到部分会话
+            </el-dropdown-item> -->
+          </el-dropdown-menu>
+        </template>
+      </el-dropdown>
+      <!-- <div class="link_text fullscreen" @click="handleFullScreen">全屏</div> -->
+      <el-icon class="full_icon">
+        <FullScreen class="icon" @click="handleFullScreen" />
+      </el-icon>
+    </div>
     <div class="info_box">
-      <div class="top">
-        <el-icon>
-          <FullScreen class="full_icon" @click="handleFullScreen" />
-        </el-icon>
-        <el-dropdown trigger="click">
-          <div class="action_wrap">
-            <span class="link_host">连接<el-icon class="el-icon--right"><arrow-down /></el-icon></span>
-          </div>
-          <template #dropdown>
-            <el-dropdown-menu>
-              <el-dropdown-item v-for="(item, index) in hostList" :key="index" @click="handleCommandHost(item)">
-                {{ item.name }} {{ item.host }}
-              </el-dropdown-item>
-            </el-dropdown-menu>
-          </template>
-        </el-dropdown>
-      </div>
       <InfoSide
         ref="infoSideRef"
         v-model:show-input-command="showInputCommand"
@@ -42,23 +55,35 @@
           :closable="true"
         >
           <div class="tab_content_wrap" :style="{ height: mainHeight + 'px' }">
-            <TerminalTab ref="terminalTabRefs" :host="item.host" />
+            <TerminalTab
+              ref="terminalTabRefs"
+              :index="index"
+              :host="item.host"
+              @input="terminalInput"
+            />
             <Sftp :host="item.host" @resize="resizeTerminal" />
           </div>
         </el-tab-pane>
       </el-tabs>
     </div>
     <InputCommand v-model:show="showInputCommand" @input-command="handleInputCommand" />
+    <HostForm
+      v-model:show="hostFormVisible"
+      :default-data="updateHostData"
+      @update-list="handleUpdateList"
+      @closed="updateHostData = null"
+    />
   </div>
 </template>
 
 <script setup>
 import { ref, defineEmits, computed, defineProps, getCurrentInstance, watch, onMounted, onBeforeUnmount } from 'vue'
-import { ArrowDown, FullScreen } from '@element-plus/icons-vue'
+import { ArrowDown, FullScreen, Select } from '@element-plus/icons-vue'
 import TerminalTab from './terminal-tab.vue'
 import InfoSide from './info-side.vue'
 import Sftp from './sftp.vue'
 import InputCommand from '@/components/input-command/index.vue'
+import HostForm from '../../server/components/host-form.vue'
 
 const { proxy: { $nextTick, $store, $message } } = getCurrentInstance()
 
@@ -71,13 +96,15 @@ const props = defineProps({
 
 const emit = defineEmits(['closed', 'removeTab', 'add-host',])
 
-const activeTabIndex = ref(0)
-const isFullScreen = ref(false)
 const showInputCommand = ref(false)
-const visible = ref(true)
 const infoSideRef = ref(null)
 const terminalTabRefs = ref([])
+let activeTabIndex = ref(0)
+let visible = ref(true)
 let mainHeight = ref('')
+let isSyncAllSession = ref(false)
+let hostFormVisible = ref(false)
+let updateHostData = ref(null)
 
 const terminalTabs = computed(() => props.terminalTabs)
 const terminalTabsLen = computed(() => props.terminalTabs.length)
@@ -95,6 +122,19 @@ onBeforeUnmount(() => {
   window.removeEventListener('resize', handleResizeTerminalSftp)
 })
 
+const handleUpdateList = async ({ isConfig, host }) => {
+  try {
+    await $store.getHostList()
+    if (isConfig) {
+      let targetHost = hostList.value.find(item => item.host === host)
+      if (targetHost !== -1) emit('add-host', targetHost)
+    }
+  } catch (err) {
+    $message.error('获取实例列表失败')
+    console.error('获取实例列表失败: ', err)
+  }
+}
+
 function handleResizeTerminalSftp() {
   $nextTick(() => {
     mainHeight.value = document.querySelector('.terminals_sftp_wrap').offsetHeight - 45 // 45 is tab-header height+15
@@ -102,10 +142,31 @@ function handleResizeTerminalSftp() {
 }
 
 const handleCommandHost = (host) => {
-  if (!host.isConfig) return $message.warning('请先配置SSH连接信息')
+  if (!host.isConfig) {
+    $message.warning('请先配置SSH连接信息')
+    hostFormVisible.value = true
+    updateHostData.value = { ...host }
+    return
+  }
   emit('add-host', host)
 }
 
+const handleSyncSession = () => {
+  isSyncAllSession.value = !isSyncAllSession.value
+  if (isSyncAllSession.value) $message.success('已开启键盘输入到所有会话')
+  else $message.info('已关闭键盘输入到所有会话')
+}
+
+const terminalInput = ({ idx, key }) => {
+  if (!isSyncAllSession.value) return
+  let filterHostList = terminalTabRefs.value.filter((host, index) => {
+    return index !== idx
+  })
+  filterHostList.forEach(item => {
+    item.handleInputCommand(key)
+  })
+}
+
 const tabChange = async (index) => {
   await $nextTick()
   const curTabTerminal = terminalTabRefs.value[index]
@@ -143,9 +204,7 @@ const removeTab = (index) => {
 }
 
 const handleFullScreen = () => {
-  if (isFullScreen.value) document.exitFullscreen()
-  else document.getElementsByClassName('terminals_sftp_wrap')[0].requestFullscreen()
-  isFullScreen.value = !isFullScreen.value
+  document.getElementsByClassName('terminals_sftp_wrap')[0].requestFullscreen()
 }
 
 // const registryDbClick = () => {
@@ -187,6 +246,7 @@ const handleInputCommand = async (command) => {
 <style lang="scss" scoped>
 .terminal_wrap {
   display: flex;
+  flex-wrap: wrap;
   height: 100%;
 
   :deep(.el-tabs__content) {
@@ -204,40 +264,44 @@ const handleInputCommand = async (command) => {
     align-items: center;
   }
 
-  .info_box {
-    height: 100%;
-    overflow: auto;
+  $terminalTopHeight: 30px;
+  .terminal_top {
+    width: 100%;
+    height: $terminalTopHeight;
     display: flex;
-    flex-direction: column;
-
-    .top {
-      height: 39px;
-      flex-shrink: 0;
-      position: sticky;
-      top: 0px;
-      z-index: 1;
-      background-color: rgb(255, 255, 255);
-      padding: 0 15px;
-      display: flex;
-      align-items: center;
-      justify-content: space-between;
-      .full_icon {
-        font-size: 15px;
+    align-items: center;
+    padding: 0 15px;
+    border-bottom: 1px solid var(--el-color-primary);
+    position: sticky;
+    top: 0;
+    background-color: #fff;
+    z-index: 3;
+    :deep(.el-dropdown) {
+      margin-top: -2px;
+    }
+    .link_text {
+      font-size: var(--el-font-size-base);
+      color: var(--el-color-primary);
+      cursor: pointer;
+      margin-right: 15px;
+    }
+    .full_icon {
+      cursor: pointer;
+      margin-left: auto;
+      &:hover .icon {
         color: var(--el-color-primary);
-        cursor: pointer;
-      }
-      .action_wrap {
-        .link_host {
-          font-size: var(--el-font-size-base);
-          color: var(--el-color-primary);
-          cursor: pointer;
-        }
       }
     }
   }
+  .info_box {
+    height: calc(100% - $terminalTopHeight);
+    overflow: auto;
+    display: flex;
+    flex-direction: column;
+  }
 
   .terminals_sftp_wrap {
-    height: 100%;
+    height: calc(100% - $terminalTopHeight);
     overflow: hidden;
     flex: 1;
     display: flex;
@@ -280,3 +344,9 @@ const handleInputCommand = async (command) => {
   }
 }
 </style>
+
+<style>
+.action_icon {
+  color: var(--el-color-primary);
+}
+</style>
\ No newline at end of file
diff --git a/web/src/views/terminal/index.vue b/web/src/views/terminal/index.vue
index 4637251..3dd14fc 100644
--- a/web/src/views/terminal/index.vue
+++ b/web/src/views/terminal/index.vue
@@ -52,23 +52,23 @@
 </template>
 
 <script setup>
-import { ref, computed, onActivated, getCurrentInstance, reactive, nextTick } from 'vue'
+import { ref, computed, onActivated, getCurrentInstance, reactive, nextTick, defineEmits } from 'vue'
 import { useRoute } from 'vue-router'
 import Terminal from './components/terminal.vue'
 import HostForm from '../server/components/host-form.vue'
 
 const { proxy: { $store, $message } } = getCurrentInstance()
 
+const emit = defineEmits(['add-host',])
+
 let terminalTabs = reactive([])
-const hostFormVisible = ref(false)
-const updateHostData = ref(null)
+let hostFormVisible = ref(false)
+let updateHostData = ref(null)
 const terminalRef = ref(null)
 const route = useRoute()
 
 let showLinkTips = computed(() => !Boolean(terminalTabs.length))
-
 let hostList = computed(() => $store.hostList)
-
 let isAllConfssh = computed(() => {
   return hostList.value?.every(item => item.isConfig)
 })
@@ -87,9 +87,13 @@ function handleRemoveTab(index) {
   terminalTabs.splice(index, 1)
 }
 
-const handleUpdateList = async () => {
+const handleUpdateList = async ({ isConfig, host }) => {
   try {
     await $store.getHostList()
+    if (isConfig) {
+      let targetHost = hostList.value.find(item => item.host === host)
+      if (targetHost !== -1) linkTerminal(targetHost)
+    }
   } catch (err) {
     $message.error('获取实例列表失败')
     console.error('获取实例列表失败: ', err)