feat: Export isCurrentPageSupported

mlc-ai · Sep 14, 2024 · f0e643a · f0e643a
1 parent 8eaf5ed
commit f0e643a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -1,4 +1,3 @@
-import { IPageHandler } from "./page/interface";
 import * as Oveleaf from "./page/overleaf";
 
 const PAGE_HANDLER_MAP: Record<string, typeof Oveleaf.PageHandler> = {
@@ -9,15 +8,19 @@ const PAGE_TOOLS_MAP: Record<string, typeof Oveleaf.tools> = {
   "www.overleaf.com": Oveleaf.tools,
 };
 
+export const isCurrentPageSupported = () => {
+  return Object.keys(PAGE_HANDLER_MAP).includes(window.location.hostname);
+};
+
 export const initHandler = () => {
-  if (Object.keys(PAGE_HANDLER_MAP).includes(window.location.hostname)) {
+  if (isCurrentPageSupported()) {
     return new PAGE_HANDLER_MAP[window.location.hostname]();
   }
   console.error("[Web Agent Interface] No tools found for the current page");
 };
 
 const getTools = () => {
-  if (Object.keys(PAGE_TOOLS_MAP).includes(window.location.hostname)) {
+  if (isCurrentPageSupported()) {
     return PAGE_TOOLS_MAP[window.location.hostname];
   }
   console.error("[Web Agent Interface] No tools found for the current page");