From deee57314d72ce468ee89bcf90d8f6bc82a53df1 Mon Sep 17 00:00:00 2001
From: David Luzar <luzar.david@gmail.com>
Date: Thu, 9 Jan 2020 17:37:08 +0100
Subject: [PATCH] support export canvas to clipboard (#232)

---
 src/components/panels/PanelExport.tsx  | 40 ++++++++++++++++++++++++--
 src/components/panels/panelExport.scss | 16 +++++++++++
 src/global.d.ts                        |  7 +++++
 src/index.tsx                          | 11 +++----
 src/scene/data.ts                      | 22 ++++++++++++--
 src/scene/index.ts                     |  2 +-
 src/scene/types.ts                     |  2 ++
 7 files changed, 89 insertions(+), 11 deletions(-)
 create mode 100644 src/components/panels/panelExport.scss
 create mode 100644 src/global.d.ts

diff --git a/src/components/panels/PanelExport.tsx b/src/components/panels/PanelExport.tsx
index 81bee94c..8b924033 100644
--- a/src/components/panels/PanelExport.tsx
+++ b/src/components/panels/PanelExport.tsx
@@ -1,17 +1,35 @@
 import React from "react";
 import { EditableText } from "../EditableText";
 import { Panel } from "../Panel";
+import { ExportType } from "../../scene/types";
+
+import "./panelExport.scss";
 
 interface PanelExportProps {
   projectName: string;
   onProjectNameChange: (name: string) => void;
-  onExportAsPNG: React.MouseEventHandler;
+  onExportCanvas: (type: ExportType) => void;
   exportBackground: boolean;
   onExportBackgroundChange: (val: boolean) => void;
   onSaveScene: React.MouseEventHandler;
   onLoadScene: React.MouseEventHandler;
 }
 
+// fa-clipboard
+const ClipboardIcon = () => (
+  <svg viewBox="0 0 384 512">
+    <path
+      fill="currentColor"
+      d="M384 112v352c0 26.51-21.49 48-48 48H48c-26.51 0-48-21.49-48-48V112c0-26.51 21.49-48 48-48h80c0-35.29 28.71-64 64-64s64 28.71 64 64h80c26.51 0 48 21.49 48 48zM192 40c-13.255 0-24 10.745-24 24s10.745 24 24 24 24-10.745 24-24-10.745-24-24-24m96 114v-20a6 6 0 0 0-6-6H102a6 6 0 0 0-6 6v20a6 6 0 0 0 6 6h180a6 6 0 0 0 6-6z"
+    ></path>
+  </svg>
+);
+
+const probablySupportsClipboard =
+  "toBlob" in HTMLCanvasElement.prototype &&
+  "write" in navigator.clipboard &&
+  "ClipboardItem" in window;
+
 export const PanelExport: React.FC<PanelExportProps> = ({
   projectName,
   exportBackground,
@@ -19,7 +37,7 @@ export const PanelExport: React.FC<PanelExportProps> = ({
   onExportBackgroundChange,
   onSaveScene,
   onLoadScene,
-  onExportAsPNG
+  onExportCanvas
 }) => {
   return (
     <Panel title="Export">
@@ -32,7 +50,23 @@ export const PanelExport: React.FC<PanelExportProps> = ({
           />
         )}
         <h5>Image</h5>
-        <button onClick={onExportAsPNG}>Export to png</button>
+        <div className="panelExport-imageButtons">
+          <button
+            className="panelExport-exportToPngButton"
+            onClick={() => onExportCanvas("png")}
+          >
+            Export to PNG
+          </button>
+          {probablySupportsClipboard && (
+            <button
+              className="panelExport-exportToClipboardButton"
+              onClick={() => onExportCanvas("clipboard")}
+              title="Copy to clipboard (experimental)"
+            >
+              <ClipboardIcon />
+            </button>
+          )}
+        </div>
         <label>
           <input
             type="checkbox"
diff --git a/src/components/panels/panelExport.scss b/src/components/panels/panelExport.scss
new file mode 100644
index 00000000..8a8ed3b3
--- /dev/null
+++ b/src/components/panels/panelExport.scss
@@ -0,0 +1,16 @@
+.panelExport-imageButtons {
+  display: flex;
+}
+
+.panelExport-exportToPngButton {
+  flex: 1 1 auto;
+}
+
+.panelExport-exportToClipboardButton {
+  margin-left: 10px;
+  padding: 0 15px;
+
+  svg {
+    width: 15px;
+  }
+}
diff --git a/src/global.d.ts b/src/global.d.ts
new file mode 100644
index 00000000..dd2f95fd
--- /dev/null
+++ b/src/global.d.ts
@@ -0,0 +1,7 @@
+interface Window {
+  ClipboardItem: any;
+}
+
+interface Clipboard extends EventTarget {
+  write(data: any[]): Promise<void>;
+}
diff --git a/src/index.tsx b/src/index.tsx
index 556aa933..3265582e 100644
--- a/src/index.tsx
+++ b/src/index.tsx
@@ -23,7 +23,7 @@ import {
   getSelectedAttribute,
   loadFromJSON,
   saveAsJSON,
-  exportAsPNG,
+  exportCanvas,
   restoreFromLocalStorage,
   saveToLocalStorage,
   hasBackground,
@@ -37,6 +37,7 @@ import {
 import { renderScene } from "./renderer";
 import { AppState } from "./types";
 import { ExcalidrawElement, ExcalidrawTextElement } from "./element/types";
+import { ExportType } from "./scene/types";
 
 import { getDateTime, isInputLike, measureText } from "./utils";
 import { KEYS, META_KEY, isArrowKey } from "./keys";
@@ -170,9 +171,9 @@ export class App extends React.Component<{}, AppState> {
     if (event.key === KEYS.ESCAPE) {
       elements = clearSelection(elements);
       this.forceUpdate();
-      this.setState({ elementType: 'selection' });
+      this.setState({ elementType: "selection" });
       if (window.document.activeElement instanceof HTMLElement) {
-        window.document.activeElement.blur()
+        window.document.activeElement.blur();
       }
       event.preventDefault();
       return;
@@ -614,8 +615,8 @@ export class App extends React.Component<{}, AppState> {
           <PanelExport
             projectName={this.state.name}
             onProjectNameChange={this.updateProjectName}
-            onExportAsPNG={() =>
-              exportAsPNG(elements, this.canvas!, this.state)
+            onExportCanvas={(type: ExportType) =>
+              exportCanvas(type, elements, this.canvas!, this.state)
             }
             exportBackground={this.state.exportBackground}
             onExportBackgroundChange={val =>
diff --git a/src/scene/data.ts b/src/scene/data.ts
index 16309e8f..f99f88da 100644
--- a/src/scene/data.ts
+++ b/src/scene/data.ts
@@ -6,6 +6,7 @@ import { getElementAbsoluteCoords } from "../element";
 
 import { renderScene } from "../renderer";
 import { AppState } from "../types";
+import { ExportType } from "./types";
 import nanoid from "nanoid";
 
 const LOCAL_STORAGE_KEY = "excalidraw";
@@ -76,7 +77,8 @@ export function loadFromJSON() {
   });
 }
 
-export function exportAsPNG(
+export function exportCanvas(
+  type: ExportType,
   elements: readonly ExcalidrawElement[],
   canvas: HTMLCanvasElement,
   {
@@ -136,7 +138,23 @@ export function exportAsPNG(
     }
   );
 
-  saveFile(`${name}.png`, tempCanvas.toDataURL("image/png"));
+  if (type === "png") {
+    saveFile(`${name}.png`, tempCanvas.toDataURL("image/png"));
+  } else if (type === "clipboard") {
+    try {
+      tempCanvas.toBlob(async function(blob) {
+        try {
+          await navigator.clipboard.write([
+            new window.ClipboardItem({ "image/png": blob })
+          ]);
+        } catch (err) {
+          window.alert("Couldn't copy to clipboard. Try using Chrome browser.");
+        }
+      });
+    } catch (err) {
+      window.alert("Couldn't copy to clipboard. Try using Chrome browser.");
+    }
+  }
 
   // clean up the DOM
   if (tempCanvas !== canvas) tempCanvas.remove();
diff --git a/src/scene/index.ts b/src/scene/index.ts
index 171315f4..e812a827 100644
--- a/src/scene/index.ts
+++ b/src/scene/index.ts
@@ -8,7 +8,7 @@ export {
   getSelectedAttribute
 } from "./selection";
 export {
-  exportAsPNG,
+  exportCanvas,
   loadFromJSON,
   saveAsJSON,
   restoreFromLocalStorage,
diff --git a/src/scene/types.ts b/src/scene/types.ts
index c8904dca..ef9bd7d2 100644
--- a/src/scene/types.ts
+++ b/src/scene/types.ts
@@ -15,3 +15,5 @@ export type SceneScroll = {
 export interface Scene {
   elements: ExcalidrawTextElement[];
 }
+
+export type ExportType = "png" | "clipboard";