From 6e6425735f68f0ff5a70158225f548f02b3319bb Mon Sep 17 00:00:00 2001
From: Frank Noirot <frank@franknoirot.co>
Date: Thu, 25 Feb 2021 01:50:03 -0500
Subject: [PATCH] Began building out screenshot capture feature.

---
 .../IdeCascadeStudio/IdeCascadeStudio.js      | 14 +++++++++
 web/src/components/IdeToolbar/IdeToolbar.js   | 30 +++++++++++++++++++
 web/src/components/Svg/Svg.js                 | 23 ++++++++++++++
 web/src/helpers/cascadeController.js          | 23 ++++++++++++++
 web/src/helpers/cloudinary.js                 | 23 ++++++++++++++
 5 files changed, 113 insertions(+)
 create mode 100644 web/src/helpers/cloudinary.js

diff --git a/web/src/components/IdeCascadeStudio/IdeCascadeStudio.js b/web/src/components/IdeCascadeStudio/IdeCascadeStudio.js
index 72cf624..f911887 100644
--- a/web/src/components/IdeCascadeStudio/IdeCascadeStudio.js
+++ b/web/src/components/IdeCascadeStudio/IdeCascadeStudio.js
@@ -3,6 +3,8 @@ import CascadeController from 'src/helpers/cascadeController'
 import IdeToolbar from 'src/components/IdeToolbar'
 import { useEffect, useState } from 'react'
 import { threejsViewport } from 'src/cascade/js/MainPage/CascadeState'
+import { element } from 'prop-types'
+import { uploadToCloudinary } from 'src/helpers/cloudinary'
 
 const defaultExampleCode = `// Welcome to Cascade Studio!   Here are some useful functions:
 //  Translate(), Rotate(), Scale(), Union(), Difference(), Intersection()
@@ -70,6 +72,18 @@ const IdeCascadeStudio = ({ part, saveCode, loading }) => {
             partTitle: part?.title,
             image: part?.user?.image,
           }}
+          onCapture={ async () => {
+            // Get the canvas image as a Data URL
+            const imgBlob = await CascadeController.capture(threejsViewport.environment)
+            const imgURL = window.URL.createObjectURL(imgBlob)
+
+            // TODO: Upload the image to Cloudinary
+            // uploadToCloudinary(imgBlob)
+
+            // TODO: Save the screenshot as the mainImage if none has been set
+            // If it has been set, pass along the Blob without uploading
+            // onSave(part?.id, { ...input, mainImage: cloudinaryPublicId })
+          }}
         />
       </div>
     </>
diff --git a/web/src/components/IdeToolbar/IdeToolbar.js b/web/src/components/IdeToolbar/IdeToolbar.js
index ba5307a..73f70c1 100644
--- a/web/src/components/IdeToolbar/IdeToolbar.js
+++ b/web/src/components/IdeToolbar/IdeToolbar.js
@@ -23,6 +23,7 @@ const IdeToolbar = ({
   userNamePart,
   isDraft,
   code,
+  onCapture,
 }) => {
   const [anchorEl, setAnchorEl] = useState(null)
   const [whichPopup, setWhichPopup] = useState(null)
@@ -104,6 +105,10 @@ const IdeToolbar = ({
     setIsLoginModalOpen(true)
   }
 
+  const captureScreenshot = async () => {
+    console.log({ forkPart, onCapture: onCapture() })
+  }
+
   const anchorOrigin = {
     vertical: 'bottom',
     horizontal: 'center',
@@ -219,6 +224,31 @@ const IdeToolbar = ({
         </Popover>
       </div>
       <div className="ml-auto flex items-center">
+        {/* Capture Screenshot link. Should only appear if part has been saved and is editable. */}
+        { !isDraft && canEdit && <div>
+          <button
+            onClick={(event) => {
+              captureScreenshot()
+              handleClick({ event, whichPopup: 'capture' })
+            }}
+            className="text-indigo-300 flex items-center pr-6"
+          >
+            Capture <Svg name="camera" className="pl-2 w-8" />
+          </button>
+          <Popover
+            id={id}
+            open={whichPopup === 'capture'}
+            anchorEl={anchorEl}
+            onClose={handleClose}
+            anchorOrigin={anchorOrigin}
+            transformOrigin={transformOrigin}
+            className="material-ui-overrides transform translate-y-4"
+          >
+            <div className="text-sm p-2 text-gray-500">
+              Saving...
+            </div>
+          </Popover>
+        </div> }
         <div>
           <button
             onClick={(event) => handleClick({ event, whichPopup: 'tips' })}
diff --git a/web/src/components/Svg/Svg.js b/web/src/components/Svg/Svg.js
index 0fdc4a6..0636b53 100644
--- a/web/src/components/Svg/Svg.js
+++ b/web/src/components/Svg/Svg.js
@@ -30,6 +30,29 @@ const Svg = ({ name, className: className2, strokeWidth = 2 }) => {
         />
       </svg>
     ),
+    'camera': (
+      <svg
+        xmlns="http://www.w3.org/2000/svg"
+        fill="none"
+        viewBox="0 0 24 21">
+        <path
+          d="M6 5H4C2.34315 5 1 6.34315 1 8V17C1 18.6569 2.34315 20 4 20H20C21.6569 20 23 18.6569 23 17V8C23 6.34315 21.6569 5 20 5H18"
+          stroke="currentColor"
+          stroke-width="2"
+          stroke-linecap="round"/>
+        <circle
+          cx="12"
+          cy="11"
+          r="5"
+          stroke="currentColor"
+          stroke-width="2"/>
+        <path
+          d="M16 2.68641C14.8716 1.61443 13.5582 1 12.1563 1C10.6229 1 9.19532 1.7351 8 3"
+          stroke="currentColor"
+          stroke-width="2"
+          stroke-linecap="round"/>
+      </svg>
+    ),
     'chevron-down': (
       <svg
         xmlns="http://www.w3.org/2000/svg"
diff --git a/web/src/helpers/cascadeController.js b/web/src/helpers/cascadeController.js
index 5cc6e61..ba38e00 100644
--- a/web/src/helpers/cascadeController.js
+++ b/web/src/helpers/cascadeController.js
@@ -23,6 +23,29 @@ class CascadeController {
     }
     onInit()
   }
+
+  capture(environment) {
+    let width = 512, height = 384;
+    environment.camera.aspect = width / height;
+    environment.camera.updateProjectionMatrix();
+    environment.renderer.setSize(width, height);
+    environment.renderer.render(environment.scene, environment.camera, null, false);
+    let imgBlob = new Promise((resolve, reject) => {
+      environment.renderer.domElement.toBlob(
+        (blob) => {
+          blob.name = `part_capture-${ Date.now() }`
+          resolve(blob)
+        },
+        'image/jpeg',
+        1
+      );
+    })
+
+    // Return to original dimensions
+    environment.onWindowResize();
+
+    return imgBlob
+  }
 }
 
 export default new CascadeController()
diff --git a/web/src/helpers/cloudinary.js b/web/src/helpers/cloudinary.js
new file mode 100644
index 0000000..3a05b53
--- /dev/null
+++ b/web/src/helpers/cloudinary.js
@@ -0,0 +1,23 @@
+// TODO: create a tidy util for uploading to Cloudinary and returning the public ID
+import axios from 'axios'
+
+const CLOUDINARY_UPLOAD_PRESET = 'CadHub_project_images'
+const CLOUDINARY_UPLOAD_URL = 'https://api.cloudinary.com/v1_1/irevdev/upload'
+
+export async function uploadToCloudinary(imgBlob) {
+  // const imageData = new FormData()
+  // imageData.append('upload_preset', CLOUDINARY_UPLOAD_PRESET)
+  // imageData.append('file', croppedFile)
+  // let upload = axios.post(CLOUDINARY_UPLOAD_URL, imageData)
+
+  // try {
+  //   const { data } = await upload
+  //   if (data && data.public_id !== '') {
+  //     onImageUpload({ cloudinaryPublicId: data.public_id })
+  //     setCloudinaryId(data.public_id)
+  //     setIsModalOpen(false)
+  //   }
+  // } catch (e) {
+  //   console.error('ERROR', e)
+  // }
+}
\ No newline at end of file