From 3544e238cfd09df95b67d16d24b7ee92e3c5a11e Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Mon, 13 Feb 2023 21:35:12 -0600
Subject: [PATCH] feat(gui): add download button to mask (#135)

---
 gui/src/components/input/MaskCanvas.tsx | 28 +++++++++++++++++++------
 1 file changed, 22 insertions(+), 6 deletions(-)

diff --git a/gui/src/components/input/MaskCanvas.tsx b/gui/src/components/input/MaskCanvas.tsx
index 381a1759..2d1a14ef 100644
--- a/gui/src/components/input/MaskCanvas.tsx
+++ b/gui/src/components/input/MaskCanvas.tsx
@@ -1,5 +1,5 @@
 import { doesExist, Maybe, mustExist } from '@apextoaster/js-utils';
-import { FormatColorFill, Gradient, InvertColors, Undo } from '@mui/icons-material';
+import { Download, FormatColorFill, Gradient, InvertColors, Save, Undo } from '@mui/icons-material';
 import { Button, Stack, Typography } from '@mui/material';
 import { throttle } from 'lodash';
 import React, { RefObject, useContext, useEffect, useMemo, useRef } from 'react';
@@ -251,6 +251,27 @@ export function MaskCanvas(props: MaskCanvasProps) {
   };
 
   return <Stack spacing={2}>
+    <Stack direction='row' spacing={2}>
+      <Button
+        variant='outlined'
+        startIcon={<Download />}
+        onClick={() => {
+          if (doesExist(maskRef.current)) {
+            const data = maskRef.current.toDataURL('image/png');
+            const link = document.createElement('a');
+
+            link.setAttribute('download', 'mask.png');
+            link.setAttribute('href', data.replace('image/png', 'image/octet-stream'));
+            link.click();
+          }
+        }}
+      />
+      <Button
+        variant='outlined'
+        startIcon={<Undo />}
+        onClick={() => drawUndo()}
+      />
+    </Stack>
     <canvas
       ref={brushRef}
       height={params.height.default}
@@ -326,11 +347,6 @@ export function MaskCanvas(props: MaskCanvasProps) {
         />
       </Stack>
       <Stack direction='row' spacing={2}>
-        <Button
-          variant='outlined'
-          startIcon={<Undo />}
-          onClick={() => drawUndo()}
-        />
         <Button
           variant='outlined'
           startIcon={<FormatColorFill />}