From 2adb01b41e1b74bcd42978846f0994f38794e07a Mon Sep 17 00:00:00 2001
From: Brian Baugher <123590980+brian-baugher@users.noreply.github.com>
Date: Thu, 8 Aug 2024 15:11:27 -0400
Subject: [PATCH] refactor: remove useMessage hooks (#125)

This update corresponds with the removal of useMessage hook exports in
VE-Lib. See this [PR](https://github.com/yext/visual-editor/pull/13) for
more details.

---------

Co-authored-by: tsaikatie <ktsai@yext.com>
Co-authored-by: ehallerYext <ehaller@yext.com>
Co-authored-by: Alexis Sanehisa <106991746+asanehisa@users.noreply.github.com>
---
 src/templates/edit.tsx | 29 ++++-------------------------
 1 file changed, 4 insertions(+), 25 deletions(-)

diff --git a/src/templates/edit.tsx b/src/templates/edit.tsx
index d74c10d6..ddfa2188 100644
--- a/src/templates/edit.tsx
+++ b/src/templates/edit.tsx
@@ -1,11 +1,7 @@
 import {
   Editor,
-  useReceiveMessage,
-  TARGET_ORIGINS,
-  TemplateMetadata,
+  useDocumentProvider,
 } from "@yext/visual-editor";
-import {useState} from "react";
-import {Config} from "@measured/puck";
 import {puckConfigs} from "../puck/puck.config";
 import {GetPath, TemplateProps, TemplateConfig} from "@yext/pages";
 import {DocumentProvider} from "@yext/pages/util";
@@ -22,30 +18,13 @@ export const config: TemplateConfig = {
 
 // Render the editor
 const Edit: () => JSX.Element = () => {
-  const [puckConfig, setPuckConfig] = useState<Config>();
-  const [entityDocument, setEntityDocument] = useState<any>(); // json data
-  const [templateMetadata, setTemplateMetadata] = useState<TemplateMetadata>();
-
-  useReceiveMessage("getEntityDocument", TARGET_ORIGINS, (send, payload) => {
-    setEntityDocument(payload);
-    send({
-      status: "success",
-      payload: { message: "getEntityDocument received" },
-    });
-  });
-
-  useReceiveMessage("getTemplateMetadata", TARGET_ORIGINS, (send, payload) => {
-    const puckConfig = puckConfigs.get(payload.templateId);
-    setPuckConfig(puckConfig);
-    setTemplateMetadata(payload as TemplateMetadata);
-    send({ status: "success", payload: { message: "payload received" } });
-  });
+  const entityDocument = useDocumentProvider();
 
   return (
     <DocumentProvider value={entityDocument}>
-      <Editor document={entityDocument} puckConfig={puckConfig!} templateMetadata={templateMetadata!}/>
+      <Editor document={entityDocument} puckConfigs={puckConfigs} />
     </DocumentProvider>
   );
 };
 
-export default Edit;
\ No newline at end of file
+export default Edit;