From 057f20821a25b897ae21b2ee9167e522d212eda3 Mon Sep 17 00:00:00 2001
From: Nestu <dani_raiders@hotmail.com>
Date: Tue, 20 Sep 2022 15:28:19 +0200
Subject: [PATCH] Add Screenshot functionality (#44)

* Add screenshot button

* Move screenshot button to menu

* Add option for screenshots to be copied to clipboard

* Update English and Spanish lang files
---
 public/lang/en.js                    |  5 +++++
 public/lang/es.js                    |  5 +++++
 src/components/Menu/MainMenu.vue     |  1 +
 src/components/Menu/SettingsMenu.vue |  4 ++++
 src/i18n/fallback.js                 |  5 +++++
 src/js/BlueMapApp.js                 | 25 +++++++++++++++++++++++++
 6 files changed, 45 insertions(+)

diff --git a/public/lang/en.js b/public/lang/en.js
index 0763c26..66e83eb 100644
--- a/public/lang/en.js
+++ b/public/lang/en.js
@@ -78,6 +78,11 @@ export default {
     compass: {
         tooltip: "Compass / Face North"
     },
+    screenshot: {
+        title: "Screenshot",
+        button: "Take Screenshot",
+        clipboard: "Copy to Clipboard"
+    },
     controls: {
         title: "View / Controls",
         perspective: {
diff --git a/public/lang/es.js b/public/lang/es.js
index e75ec80..0e496f2 100644
--- a/public/lang/es.js
+++ b/public/lang/es.js
@@ -76,6 +76,11 @@ export default {
     compass: {
         tooltip: "Brújula / Norte"
     },
+    screenshot: {
+        title: "Captura de Pantalla",
+        button: "Tomar Captura de Pantalla",
+        clipboard: "Copiar al Portapapeles",
+    },
     controls: {
         title: "Vista / Controles",
         perspective: {
diff --git a/src/components/Menu/MainMenu.vue b/src/components/Menu/MainMenu.vue
index dfe8b86..723ca62 100644
--- a/src/components/Menu/MainMenu.vue
+++ b/src/components/Menu/MainMenu.vue
@@ -13,6 +13,7 @@
       <hr>
       <SimpleButton @action="goFullscreen">{{ $t("goFullscreen.button") }}</SimpleButton>
       <SimpleButton @action="$bluemap.resetCamera()">{{ $t("resetCamera.button") }}</SimpleButton>
+      <SimpleButton @action="$bluemap.takeScreenshot()">{{ $t("screenshot.button") }}</SimpleButton>
       <SimpleButton @action="$bluemap.updateMap()" :title="$t('updateMap.tooltip')">{{ $t("updateMap.button") }}</SimpleButton>
     </div>
 
diff --git a/src/components/Menu/SettingsMenu.vue b/src/components/Menu/SettingsMenu.vue
index 2394e5e..31195e5 100644
--- a/src/components/Menu/SettingsMenu.vue
+++ b/src/components/Menu/SettingsMenu.vue
@@ -41,6 +41,10 @@
       >{{theme.name}}</SimpleButton>
     </Group>
 
+    <Group :title="$t('screenshot.title')">
+      <SwitchButton :on="appState.screenshot.clipboard" @action="appState.screenshot.clipboard = !appState.screenshot.clipboard; $bluemap.saveUserSettings()">{{ $t("screenshot.clipboard") }}</SwitchButton>
+    </Group>
+
     <Group v-if="$i18n.languages.length > 1" :title="$t('language.title')">
       <SimpleButton v-for="lang of $i18n.languages" :key="lang.locale"
                     :active="lang.locale === $i18n.locale"
diff --git a/src/i18n/fallback.js b/src/i18n/fallback.js
index 30238c5..0a15299 100644
--- a/src/i18n/fallback.js
+++ b/src/i18n/fallback.js
@@ -78,6 +78,11 @@ export default {
 	compass: {
 		tooltip: "Compass / Face North"
 	},
+	screenshot: {
+		title: "Screenshot",
+		button: "Take Screenshot",
+    clipboard: "Copy to Clipboard"
+	},
 	controls: {
 		title: "View / Controls",
 		perspective: {
diff --git a/src/js/BlueMapApp.js b/src/js/BlueMapApp.js
index 59ed429..f810347 100644
--- a/src/js/BlueMapApp.js
+++ b/src/js/BlueMapApp.js
@@ -96,6 +96,9 @@ export class BlueMapApp {
             menu: this.mainMenu,
             maps: [],
             theme: null,
+            screenshot: {
+                clipboard: true
+            },
             debug: false
         };
 
@@ -518,6 +521,10 @@ export class BlueMapApp {
         }
     }
 
+    setScreenshotClipboard(clipboard) {
+        this.appState.screenshot.clipboard = clipboard;
+    }
+
     async updateMap() {
         try {
             this.mapViewer.clearTileCache();
@@ -559,6 +566,7 @@ export class BlueMapApp {
         this.appState.controls.pauseTileLoading = this.loadUserSetting("pauseTileLoading", this.appState.controls.pauseTileLoading);
         this.updateControlsSettings();
         this.setTheme(this.loadUserSetting("theme", this.appState.theme));
+        this.setScreenshotClipboard(this.loadUserSetting("screenshotClipboard", this.appState.screenshot.clipboard));
         await i18n.setLanguage(this.loadUserSetting("lang", i18n.locale));
         this.setDebug(this.loadUserSetting("debug", this.appState.debug));
 
@@ -578,6 +586,7 @@ export class BlueMapApp {
         this.saveUserSetting("invertMouse", this.appState.controls.invertMouse);
         this.saveUserSetting("pauseTileLoading", this.appState.controls.pauseTileLoading);
         this.saveUserSetting("theme", this.appState.theme);
+        this.saveUserSetting("screenshotClipboard", this.appState.screenshot.clipboard);
         this.saveUserSetting("lang", i18n.locale);
         this.saveUserSetting("debug", this.appState.debug);
 
@@ -704,4 +713,20 @@ export class BlueMapApp {
         }
     }
 
+    takeScreenshot = () => {
+        let link = document.createElement("a");
+        link.download = "bluemap-screenshot.png";
+        link.href = this.mapViewer.renderer.domElement.toDataURL('image/png');
+        link.click();
+
+        if (this.appState.screenshot.clipboard) {
+            this.mapViewer.renderer.domElement.toBlob(blob => {
+                // eslint-disable-next-line no-undef
+                navigator.clipboard.write([new ClipboardItem({ ['image/png']: blob })]).catch(e => {
+                    alert(this.events, "Failed to copy screenshot to clipboard: " + e, "error");
+                });
+            });
+        }
+    }
+
 }