From acf93695d3a93add2e3b29313400fc9b3608f7fa Mon Sep 17 00:00:00 2001
From: Joshua Shoemaker <joshuashoemaker@outlook.com>
Date: Tue, 13 Dec 2022 16:51:56 -0600
Subject: [PATCH] feat: render selected document

---
 core/Document/DocumentCollection.go           | 13 ++++++
 frontend/app/page.tsx                         |  5 ++-
 .../components/workspace/DocumentRenderer.tsx | 44 +++++++++++++++++++
 frontend/components/workspace/Sidebar.tsx     | 23 +++++++---
 frontend/context/Project/provider.tsx         |  6 +++
 frontend/context/Project/types.ts             |  3 ++
 main.go                                       | 37 +++++++++++++---
 7 files changed, 117 insertions(+), 14 deletions(-)
 create mode 100644 frontend/components/workspace/DocumentRenderer.tsx

diff --git a/core/Document/DocumentCollection.go b/core/Document/DocumentCollection.go
index 58023f7..061894f 100644
--- a/core/Document/DocumentCollection.go
+++ b/core/Document/DocumentCollection.go
@@ -18,3 +18,16 @@ func GetDocumentCollection() *DocumentCollection {
 func (collection *DocumentCollection) AddDocument(document Entity) {
 	collection.Documents = append(collection.Documents, document)
 }
+
+func (collection *DocumentCollection) GetDocumentById(id string) Entity {
+	var foundDocument Entity
+
+	for _, d := range collection.Documents {
+		if d.Id == id {
+			foundDocument = d
+			break
+		}
+	}
+
+	return foundDocument
+}
diff --git a/frontend/app/page.tsx b/frontend/app/page.tsx
index c734811..18cf086 100644
--- a/frontend/app/page.tsx
+++ b/frontend/app/page.tsx
@@ -1,5 +1,6 @@
 import 'server-only'
 import MainHead from '../components/MainHead'
+import DocumentRenderer from '../components/workspace/DocumentRenderer'
 import Navigation from '../components/workspace/Navigation'
 
 export default function Home() {
@@ -16,8 +17,8 @@ export default function Home() {
             </div>
             <div className="mx-auto  px-4 sm:px-6 md:px-8">
               <div className="py-4">
-                <div className="h-96 rounded-lg border-4 border-dashed border-gray-200">
-                  {/* Add canvas */}
+                <div className=" min-h-96 rounded-lg border-4 border-dashed border-gray-200">
+                  <DocumentRenderer />
                 </div>
               </div>
             </div>
diff --git a/frontend/components/workspace/DocumentRenderer.tsx b/frontend/components/workspace/DocumentRenderer.tsx
new file mode 100644
index 0000000..6cb75e0
--- /dev/null
+++ b/frontend/components/workspace/DocumentRenderer.tsx
@@ -0,0 +1,44 @@
+'use client'
+
+import { useEffect, useRef } from "react"
+import { useProject } from "../../context/Project/provider"
+
+
+const loadImage = (path: string): Promise<HTMLImageElement> => {
+  return new Promise((resolve, reject) => {
+    const image = new Image()
+    image.src = path
+    image.onload = () => resolve(image)
+    image.onerror = (error) => reject(error)
+  })
+}
+
+const DocumentRenderer = () => {
+  const { getSelectedDocument } = useProject()
+  const canvasRef = useRef<HTMLCanvasElement>(null)
+
+  const applyDocumentToCanvas = async (path: string) => {
+    const image = await loadImage(path)
+
+    const canvas = canvasRef.current
+    if (!canvas) return
+    canvas.width = image.naturalWidth
+    canvas.height = image.naturalHeight
+
+    const context = canvas.getContext('2d')
+    if (!context) return
+    context.drawImage(image, 10, 10, image.width, image.height)
+  }
+
+  useEffect(() => {
+    const selectedDocument = getSelectedDocument()
+    const documentPath = selectedDocument?.path
+    if (documentPath) applyDocumentToCanvas(selectedDocument.path)
+  }, [getSelectedDocument])
+
+  return (
+    <canvas ref={canvasRef}></canvas>
+  )
+}
+
+export default DocumentRenderer
diff --git a/frontend/components/workspace/Sidebar.tsx b/frontend/components/workspace/Sidebar.tsx
index ab0b041..a69af46 100644
--- a/frontend/components/workspace/Sidebar.tsx
+++ b/frontend/components/workspace/Sidebar.tsx
@@ -43,14 +43,20 @@ function classNames(...classes: any[]) {
 }
 
 function Sidebar() {
-  const [selectedItemId, setSelectedItemId] = useState('')
   const [selectedGroupId, setSelectedGroupId] = useState('')
   const [isAddNewDocumentInputShowing, setIsAddNewDocumentInputShowing] = useState(false)
   const [isAddNewGroupInputShowing, setIsAddNewGroupInputShowing] = useState(false)
   const addDocumentTextInput = useRef<HTMLInputElement>(null)
   const addGroupTextInput = useRef<HTMLInputElement>(null)
 
-  const { documents, groups, requestAddDocument, requestAddDocumentGroup } = useProject()
+  const {
+    documents,
+    groups,
+    requestAddDocument,
+    requestAddDocumentGroup,
+    selectedDocumentId,
+    setSelectedDocumentId
+  } = useProject()
 
   const navigation = getNavigationProps(documents, groups)
 
@@ -75,7 +81,7 @@ function Sidebar() {
   }
 
   const onItemClickHandler = (itemId: string) => {
-    setSelectedItemId(itemId)
+    setSelectedDocumentId(itemId)
     setSelectedGroupId(getParentGroupIdFromItemId(itemId))
     setIsAddNewDocumentInputShowing(false)
     setIsAddNewGroupInputShowing(false)
@@ -96,12 +102,15 @@ function Sidebar() {
     const response = await requestAddDocument(groupId, documentName)
     if (!response.id) return
 
-    setSelectedItemId(response.id)
+
+    setSelectedDocumentId(response.id)
+
+
     setSelectedGroupId(groupId)
     setIsAddNewDocumentInputShowing(false)
   }
 
-  const onConfirmAddGroupClickHandler = async(e: React.MouseEvent) => {
+  const onConfirmAddGroupClickHandler = async (e: React.MouseEvent) => {
     const groupName = addGroupTextInput.current?.value
     if (!groupName) return
 
@@ -132,7 +141,7 @@ function Sidebar() {
         >
           <XMarkIcon className="h-4 w-5" aria-hidden="true" />
         </button>
-        
+
         <button
           type="button"
           onClick={onConfirmAddGroupClickHandler}
@@ -219,7 +228,7 @@ function Sidebar() {
                   role='button'
                   onClick={() => onItemClickHandler(child.id)}
                   className={classNames(
-                    child.id === selectedItemId
+                    child.id === selectedDocumentId
                       ? 'bg-gray-900 text-white'
                       : 'text-gray-300 hover:bg-gray-700 hover:text-white',
                     'group w-full flex items-center pr-2 py-2 text-left font-medium text-sm rounded-md focus:outline-none focus:ring-2 focus:ring-indigo-500 p-2'
diff --git a/frontend/context/Project/provider.tsx b/frontend/context/Project/provider.tsx
index 405df32..4932c56 100644
--- a/frontend/context/Project/provider.tsx
+++ b/frontend/context/Project/provider.tsx
@@ -14,6 +14,7 @@ type Props = { children: ReactNode, projectProps: ProjectProps }
 export function ProjectProvider({ children, projectProps }: Props) {
   const [ documents, setDocuments ] = useState<ipc.Document[]>(projectProps.documents)
   const [ groups, setGroups ] = useState<ipc.Group[]>(projectProps.groups)
+  const [selectedDocumentId, setSelectedDocumentId] = useState<string>('')
 
   const updateDocuments = async () => {
     GetDocuments().then(response => {
@@ -35,14 +36,19 @@ export function ProjectProvider({ children, projectProps }: Props) {
     return response
   }
 
+  const getSelectedDocument = () => documents.find(d => d.id === selectedDocumentId)
+
   if (!documents.length || !groups.length) updateDocuments()
 
   const value = {
     id: '',
     documents,
     groups,
+    getSelectedDocument,
     requestAddDocument,
     requestAddDocumentGroup,
+    selectedDocumentId,
+    setSelectedDocumentId,
   }
 
   return <ProjectContext.Provider value={value}>
diff --git a/frontend/context/Project/types.ts b/frontend/context/Project/types.ts
index cd6401e..40da64c 100644
--- a/frontend/context/Project/types.ts
+++ b/frontend/context/Project/types.ts
@@ -7,6 +7,9 @@ export type ProjectProps = {
 }
 
 export type ProjectContextType = {
+  getSelectedDocument: () => ipc.Document | undefined
   requestAddDocument: (groupId: string, documentName: string) => Promise<ipc.Document>,
   requestAddDocumentGroup: (groupName: string) => Promise<ipc.Group>,
+  selectedDocumentId: string,
+  setSelectedDocumentId: (id: string) => void,
 } & ProjectProps
\ No newline at end of file
diff --git a/main.go b/main.go
index f391ff5..e791961 100644
--- a/main.go
+++ b/main.go
@@ -2,7 +2,10 @@ package main
 
 import (
 	"embed"
+	"fmt"
 	"log"
+	"net/http"
+	"os"
 
 	app "textualize/core/App"
 	document "textualize/core/Document"
@@ -11,6 +14,7 @@ import (
 	"github.com/wailsapp/wails/v2"
 	"github.com/wailsapp/wails/v2/pkg/logger"
 	"github.com/wailsapp/wails/v2/pkg/options"
+	"github.com/wailsapp/wails/v2/pkg/options/assetserver"
 	"github.com/wailsapp/wails/v2/pkg/options/mac"
 	"github.com/wailsapp/wails/v2/pkg/options/windows"
 )
@@ -18,6 +22,26 @@ import (
 //go:embed frontend/out frontend/out/_next/static
 var assets embed.FS
 
+type FileLoader struct {
+	http.Handler
+}
+
+func ClientFileLoader() *FileLoader {
+	return &FileLoader{}
+}
+
+func (h *FileLoader) ServeHTTP(res http.ResponseWriter, req *http.Request) {
+	var err error
+	requestedFilename := req.URL.Path
+	fileData, err := os.ReadFile(requestedFilename)
+	if err != nil {
+		res.WriteHeader(http.StatusBadRequest)
+		res.Write([]byte(fmt.Sprintf("Could not load file %s", requestedFilename)))
+	}
+
+	res.Write(fileData)
+}
+
 //go:embed build/appicondark.png
 var icon []byte
 
@@ -43,11 +67,14 @@ func main() {
 		StartHidden:       false,
 		HideWindowOnClose: false,
 		BackgroundColour:  &options.RGBA{R: 255, G: 255, B: 255, A: 255},
-		Assets:            assets,
-		Menu:              nil,
-		Logger:            nil,
-		LogLevel:          logger.DEBUG,
-		OnStartup:         app.Startup,
+		AssetServer: &assetserver.Options{
+			Assets:  assets,
+			Handler: ClientFileLoader(),
+		},
+		Menu:      nil,
+		Logger:    nil,
+		LogLevel:  logger.DEBUG,
+		OnStartup: app.Startup,
 		// OnDomReady:        app.domReady,
 		// OnBeforeClose:     app.beforeClose,
 		// OnShutdown:        app.shutdown,