From af39afa24b257f4bc87238cc1ad168b8fccf31a1 Mon Sep 17 00:00:00 2001
From: Fifteen <66454152+QFifteen@users.noreply.github.com>
Date: Mon, 3 Jun 2024 08:56:55 +0800
Subject: [PATCH] perf(demo->screenshot): optimize the screenshot code && add
 print and download (#3885)

---
 src/views/demo/feat/screenshot/index.vue | 88 ++++++++++++++++--------
 1 file changed, 58 insertions(+), 30 deletions(-)

diff --git a/src/views/demo/feat/screenshot/index.vue b/src/views/demo/feat/screenshot/index.vue
index 6f64e3da..6fbe3d32 100644
--- a/src/views/demo/feat/screenshot/index.vue
+++ b/src/views/demo/feat/screenshot/index.vue
@@ -1,49 +1,77 @@
 <template>
   <PageWrapper title="截图示例">
-    <Row :gutter="24">
-      <Col :span="3">
-        <Card title="截图">
-          <a-button type="primary" @click="screenShot">点击截图</a-button>
-          <div class="mt-8" v-show="open">
-            <a-button type="primary" @click="Dele">点击删除</a-button>
-          </div>
-        </Card>
-      </Col>
-      <Col :span="21">
-        <Card title="截图内容" v-show="open">
-          <div ref="picture"></div>
-        </Card>
-      </Col>
-    </Row>
+    <CollapseContainer title="截图操作">
+      <a-button type="primary" class="mr-2" @click="handleScreenshot">截取当前body</a-button>
+      <a-button type="primary" class="mr-2" :disabled="!showPicture" @click="handleDelScreenshot"
+        >删除截图</a-button
+      >
+      <a-button type="primary" class="mr-2" :disabled="!showPicture" @click="handlePrintScreenshot"
+        >打印截图</a-button
+      >
+      <a-button
+        type="primary"
+        class="mr-2"
+        :disabled="!showPicture"
+        @click="handleDownloadScreenshot"
+        >下载截图</a-button
+      >
+    </CollapseContainer>
+
+    <Card title="截图内容" class="mt-4">
+      <div ref="pictureRef" v-show="showPicture"></div>
+    </Card>
   </PageWrapper>
 </template>
 <script lang="ts" setup>
   import { PageWrapper } from '@/components/Page';
   import html2canvas from 'html2canvas';
   import { ref } from 'vue';
-  import { Card, Col, Row } from 'ant-design-vue';
+  import { Card } from 'ant-design-vue';
+  import { CollapseContainer } from '@/components/Container';
+  import printJS from 'print-js';
+  import { downloadByBase64 } from '@/utils/file/download';
 
-  const picture = ref();
-  const open = ref(false);
-  function screenShot() {
-    if (open.value) {
-      return;
-    }
+  const pictureRef = ref();
+  const showPicture = ref<boolean>(false);
+  const canvasUrl = ref<string>('');
+
+  function handleScreenshot() {
+    if (showPicture.value) return;
     html2canvas(document.body, {
       backgroundColor: '#ffffff',
       allowTaint: true, //开启跨域
       useCORS: true,
       scrollY: 0,
       scrollX: 0,
-    }).then(function (canvas) {
-      canvas.style.width = '100%';
-      canvas.style.height = '100%';
-      picture.value.appendChild(canvas);
-      open.value = true;
+    })
+      .then(function (canvas) {
+        canvas.style.width = '100%';
+        canvas.style.height = '100%';
+        pictureRef.value.appendChild(canvas);
+        showPicture.value = true;
+        canvasUrl.value = canvas.toDataURL();
+      })
+      .catch((err) => {
+        console.log('绘制失败', err);
+      });
+  }
+
+  function handleDelScreenshot() {
+    pictureRef.value.innerHTML = '';
+    canvasUrl.value = '';
+    showPicture.value = false;
+  }
+
+  function handlePrintScreenshot() {
+    if (!canvasUrl.value) return;
+    printJS({
+      printable: canvasUrl.value,
+      type: 'image',
+      base64: true,
     });
   }
-  function Dele() {
-    picture.value.innerHTML = '';
-    open.value = false;
+
+  function handleDownloadScreenshot() {
+    downloadByBase64(canvasUrl.value, 'screen_shot.png');
   }
 </script>