From bf3211bf1b575905a81724d7220717b5336b4883 Mon Sep 17 00:00:00 2001
From: lijiahangmax <ljh1553488six@139.com>
Date: Wed, 17 Jan 2024 22:15:25 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=BB=88=E7=AB=AF=E6=88=AA=E5=B1=8F.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../layout/terminal-right-sidebar.vue         | 22 +++++++++++--
 .../handler/terminal-session-handler.ts       | 26 +++++++++++++++
 .../host/terminal/handler/terminal-session.ts |  2 +-
 .../terminal/handler/terminal-tab-manager.ts  |  8 +++++
 .../src/views/host/terminal/index.vue         | 32 ++-----------------
 .../host/terminal/types/terminal.type.ts      |  4 +++
 6 files changed, 61 insertions(+), 33 deletions(-)

diff --git a/orion-ops-ui/src/views/host/terminal/components/layout/terminal-right-sidebar.vue b/orion-ops-ui/src/views/host/terminal/components/layout/terminal-right-sidebar.vue
index 366bbed3..d7c2b0fc 100644
--- a/orion-ops-ui/src/views/host/terminal/components/layout/terminal-right-sidebar.vue
+++ b/orion-ops-ui/src/views/host/terminal/components/layout/terminal-right-sidebar.vue
@@ -19,9 +19,27 @@
 
 <script lang="ts" setup>
   import type { SidebarAction } from '../../types/terminal.type';
+  import { useTerminalStore } from '@/store';
+  import { TerminalTabType } from '../../types/terminal.const';
   import IconActions from './icon-actions.vue';
+  import { Message } from '@arco-design/web-vue';
 
-  const emits = defineEmits(['openSnippet', 'openSftp', 'openTransfer', 'screenshot']);
+  const emits = defineEmits(['openSnippet', 'openSftp', 'openTransfer']);
+
+  const { tabManager, sessionManager } = useTerminalStore();
+
+  // 终端截屏
+  const screenshot = () => {
+    const tab = tabManager.getCurrentTab();
+    if (!tab || tab.type !== TerminalTabType.TERMINAL) {
+      Message.warning('请切换到终端标签页');
+      return;
+    }
+    // 获取处理器并截图
+    sessionManager.getSession(tab.key)
+      ?.handler
+      ?.screenshot();
+  };
 
   // 顶部操作
   const topActions = [
@@ -50,7 +68,7 @@
     {
       icon: 'icon-camera',
       content: '截图',
-      click: () => emits('screenshot')
+      click: screenshot
     },
   ];
 
diff --git a/orion-ops-ui/src/views/host/terminal/handler/terminal-session-handler.ts b/orion-ops-ui/src/views/host/terminal/handler/terminal-session-handler.ts
index c17ed0df..7068737d 100644
--- a/orion-ops-ui/src/views/host/terminal/handler/terminal-session-handler.ts
+++ b/orion-ops-ui/src/views/host/terminal/handler/terminal-session-handler.ts
@@ -4,6 +4,9 @@ import type { Terminal } from 'xterm';
 import useCopy from '@/hooks/copy';
 import { useTerminalStore } from '@/store';
 import { InnerTabs } from '../types/terminal.const';
+import html2canvas from 'html2canvas';
+import { saveAs } from 'file-saver';
+import { Message } from '@arco-design/web-vue';
 
 const { copy: copyValue, readText } = useCopy();
 
@@ -178,6 +181,29 @@ export default class TerminalSessionHandler implements ITerminalSessionHandler {
     this.session.disconnect();
   }
 
+  // 截图
+  async screenshot() {
+    try {
+      // 获取截屏
+      const canvas = await html2canvas(this.inst.element as HTMLElement, {
+        useCORS: true,
+        backgroundColor: 'transparent',
+      });
+      // 保存图片
+      const blob = await new Promise((resolve, reject) => {
+        canvas.toBlob((blob) => {
+          if (!blob) {
+            reject();
+          }
+          resolve(blob);
+        }, 'image/png');
+      });
+      saveAs(blob as Blob, `screenshot-${Date.now()}.png`);
+    } catch (e) {
+      Message.error('保存失败');
+    }
+  }
+
   // 关闭 tab
   closeTab() {
     this.tabManager.deleteTab(this.session.sessionId);
diff --git a/orion-ops-ui/src/views/host/terminal/handler/terminal-session.ts b/orion-ops-ui/src/views/host/terminal/handler/terminal-session.ts
index 4d6089c5..9863057f 100644
--- a/orion-ops-ui/src/views/host/terminal/handler/terminal-session.ts
+++ b/orion-ops-ui/src/views/host/terminal/handler/terminal-session.ts
@@ -157,7 +157,7 @@ export default class TerminalSession implements ITerminalSession {
     }
     if (preference.pluginsSetting.enableWebglPlugin) {
       // WebGL 渲染插件
-      this.addons.webgl = new WebglAddon();
+      this.addons.webgl = new WebglAddon(true);
     } else {
       // canvas 渲染插件
       this.addons.canvas = new CanvasAddon();
diff --git a/orion-ops-ui/src/views/host/terminal/handler/terminal-tab-manager.ts b/orion-ops-ui/src/views/host/terminal/handler/terminal-tab-manager.ts
index 2722d9bb..ab64d808 100644
--- a/orion-ops-ui/src/views/host/terminal/handler/terminal-tab-manager.ts
+++ b/orion-ops-ui/src/views/host/terminal/handler/terminal-tab-manager.ts
@@ -13,6 +13,14 @@ export default class TerminalTabManager implements ITerminalTabManager {
     this.items = [InnerTabs.NEW_CONNECTION];
   }
 
+  // 获取当前 tab
+  getCurrentTab() {
+    if (!this.active) {
+      return undefined;
+    }
+    return this.items.find(s => s.key === this.active);
+  }
+
   // 点击 tab
   clickTab(key: string): void {
     this.active = key;
diff --git a/orion-ops-ui/src/views/host/terminal/index.vue b/orion-ops-ui/src/views/host/terminal/index.vue
index 86ac7073..b1749152 100644
--- a/orion-ops-ui/src/views/host/terminal/index.vue
+++ b/orion-ops-ui/src/views/host/terminal/index.vue
@@ -11,7 +11,7 @@
         <terminal-left-sidebar />
       </div>
       <!-- 内容区域 -->
-      <div class="host-layout-content" ref="content">
+      <div class="host-layout-content">
         <!-- 主机加载中骨架 -->
         <loading-skeleton v-if="contentLoading" />
         <!-- 终端内容区域 -->
@@ -19,7 +19,7 @@
       </div>
       <!-- 右侧操作栏 -->
       <div class="host-layout-right">
-        <terminal-right-sidebar @screenshot="screenshotTerminal" />
+        <terminal-right-sidebar />
       </div>
     </main>
   </div>
@@ -41,11 +41,8 @@
   import TerminalRightSidebar from './components/layout/terminal-right-sidebar.vue';
   import TerminalContent from './components/layout/terminal-content.vue';
   import LoadingSkeleton from './components/layout/loading-skeleton.vue';
-  import html2canvas from 'html2canvas';
-  import { saveAs } from 'file-saver';
   import './assets/styles/layout.less';
   import 'xterm/css/xterm.css';
-  import { Message } from '@arco-design/web-vue';
 
   const terminalStore = useTerminalStore();
   const dictStore = useDictStore();
@@ -54,7 +51,6 @@
 
   const originTitle = document.title;
   const render = ref(false);
-  const content = ref();
 
   // 关闭视口处理
   const handleBeforeUnload = (event: any) => {
@@ -62,30 +58,6 @@
     event.returnValue = confirm('系统可能不会保存您所做的更改');
   };
 
-  // 终端截图
-  // FIXME test 水印
-  const screenshotTerminal = async () => {
-    try {
-      console.log(content.value);
-      const canvas = await html2canvas(content.value, {
-        useCORS: true,
-        backgroundColor: 'transparent',
-      });
-      console.log(canvas);
-      const blob = await new Promise((resolve, reject) => {
-        canvas.toBlob((blob) => {
-          if (!blob) {
-            reject(new Error('截屏失败'));
-          }
-          resolve(blob);
-        }, 'image/png');
-      });
-      saveAs(blob, `screenshot-${Date.now()}.png`);
-    } catch (e) {
-      Message.error('保存失败');
-    }
-  };
-
   // 加载用户终端偏好
   onBeforeMount(async () => {
     await terminalStore.fetchPreference();
diff --git a/orion-ops-ui/src/views/host/terminal/types/terminal.type.ts b/orion-ops-ui/src/views/host/terminal/types/terminal.type.ts
index 928cd212..e27fc335 100644
--- a/orion-ops-ui/src/views/host/terminal/types/terminal.type.ts
+++ b/orion-ops-ui/src/views/host/terminal/types/terminal.type.ts
@@ -98,6 +98,8 @@ export interface ITerminalTabManager {
   // 全部 tab
   items: Array<TerminalTabItem>;
 
+  // 获取当前 tab
+  getCurrentTab: () => TerminalTabItem | undefined;
   // 点击 tab
   clickTab: (key: string) => void;
   // 删除 tab
@@ -236,6 +238,8 @@ export interface ITerminalSessionHandler {
   clear: () => void;
   // 断开连接
   disconnect: () => void;
+  // 截图
+  screenshot: () => void;
   // 关闭 tab
   closeTab: () => void;
   // 切换到前一个 tab