From 91e188718e49cf3de4232e559a3a0112d56e7847 Mon Sep 17 00:00:00 2001
From: Nopfed <robby.tsunami@gmail.com>
Date: Mon, 12 Feb 2024 19:01:45 -0500
Subject: [PATCH] WIP: Able to call methods on model viewer

---
 components/ThreeDModel.vue | 10 ++++++++--
 pages/publish/index.vue    | 28 ++++++++++++++++++++--------
 2 files changed, 28 insertions(+), 10 deletions(-)

diff --git a/components/ThreeDModel.vue b/components/ThreeDModel.vue
index d10f1a8..2a604e0 100644
--- a/components/ThreeDModel.vue
+++ b/components/ThreeDModel.vue
@@ -56,10 +56,16 @@ const getBlob = async () => {
 }
 
 const getDataURL = () => {
-  return modelViewer.value?.toDataURL()
+  return modelViewer.value?.toDataURL('image/jpeg')
+}
+
+const showPoster = () => {
+  return modelViewer.value?.showPoster()
 }
 
 defineExpose({
-  getDataURL
+  getBlob,
+  getDataURL,
+  showPoster
 })
 </script>
\ No newline at end of file
diff --git a/pages/publish/index.vue b/pages/publish/index.vue
index 7be8757..f00789b 100644
--- a/pages/publish/index.vue
+++ b/pages/publish/index.vue
@@ -39,7 +39,7 @@
 
       <v-row>
         <v-col cols="12">
-          <model-viewer v-show="filesToUpload.length > 0" ref="modelViewer" :src="filesToUpload[0].url" />
+          <ThreeDModel v-show="false" ref="modelViewer" :src="modelSrc" />
         </v-col>
       </v-row>
 
@@ -179,7 +179,7 @@
 
 <script setup lang="ts">
 import '@google/model-viewer'
-import { ModelViewerElement } from '@google/model-viewer'
+import ThreeDModelVue from './components/ThreeDModel.vue'
 
 const router = useRouter()
 type FileWithURL = {
@@ -202,7 +202,7 @@ const selectedImageURL = ref<string>('')
 const loading = ref(false)
 const validForm = ref(false)
 const fileType = ref('')
-const modelViewer = ref<ModelViewerElement>()
+const modelViewer = ref<InstanceType<typeof ThreeDModelVue>>()
 const artMetadata = ref<ArtDetails>({
   title: '',
   description: '',
@@ -213,17 +213,21 @@ const artMetadata = ref<ArtDetails>({
   // license: ''
 })
 
-const onFilesAdded = (files: FileWithURL[]) => {
+const onFilesAdded = async (files: FileWithURL[]) => {
   console.log('got file(s)', files.length, files)
   
   filesToUpload.value = files
 
   fileType.value = checkFileType(filesToUpload.value[0].file)
 
-  // if (fileType.value == 'model') {
-  //   selectedImageURL.value = modelViewer.value?.toDataURL() as string
-  //   console.log(selectedImageURL.value)
-  // }
+  if (fileType.value == 'model') {
+    // selectedImageURL.value = URL.createObjectURL(await modelViewer.value?.getBlob() as Blob)
+    selectedImageURL.value = modelViewer.value.getDataURL()
+    console.log(selectedImageURL.value)
+    
+  } else {
+    selectedImageURL.value = filesToUpload.value.at(0)?.url || ''
+  }
 
   // console.log(selectedImageURL.value)
 
@@ -251,6 +255,14 @@ const checkFileType = (file: File) => {
   return 'unknown'
 }
 
+const modelSrc = computed(() => {
+  if (fileType.value == 'model' && filesToUpload.value.length > 0) {
+      return filesToUpload.value[0].url
+    }
+
+  return ''
+})
+
 const publishActionText = ref<string>('')
 const publishActionColor = ref<string>('')