From 7107d45d731a703e3271e850a3a7883249c66dd1 Mon Sep 17 00:00:00 2001
From: Ted <2508067350@qq.com>
Date: Wed, 19 Feb 2025 12:05:26 +0800
Subject: [PATCH] enhance code

---
 app/components/home.tsx | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/app/components/home.tsx b/app/components/home.tsx
index 1c0a9a528..c301418d9 100644
--- a/app/components/home.tsx
+++ b/app/components/home.tsx
@@ -290,12 +290,15 @@ export function Home() {
       }
 
       // 逻辑判断 是对象还是字符串
-      if (!event.origin.includes("omeoffice")) {
+      if (
+        !event.origin.includes("omeoffice") ||
+        !event.origin.includes("localhost")
+      ) {
         return; // 如果不是信任的源，忽略消息
       }
 
-      if (!isEmpty(event?.data?.omeToken))
-        appConfig.setOmeToken(event.data.omeToken);
+      if (!isEmpty(event?.data?.ometoken))
+        appConfig.setOmeToken(event.data.ometoken);
     };
 
     window.addEventListener("message", handleMessage);