From 39571b5ec59077a8734ae8c571e39cd0a0c003c9 Mon Sep 17 00:00:00 2001
From: sowgro <tpoke.ferrari@gmail.com>
Date: Mon, 21 Oct 2024 11:48:11 -0400
Subject: Support firefox

---
 extension/src/Bookmark.tsx | 2 +-
 extension/src/Folder.tsx   | 2 +-
 extension/src/main.tsx     | 7 ++++++-
 3 files changed, 8 insertions(+), 3 deletions(-)

(limited to 'extension/src')

diff --git a/extension/src/Bookmark.tsx b/extension/src/Bookmark.tsx
index fcc6756..4e4476e 100644
--- a/extension/src/Bookmark.tsx
+++ b/extension/src/Bookmark.tsx
@@ -1,4 +1,4 @@
-import BookmarkTreeNode = chrome.bookmarks.BookmarkTreeNode;
+import BookmarkTreeNode = browser.bookmarks.BookmarkTreeNode;
 
 function Bookmark(props: {data: BookmarkTreeNode}) {
     return(
diff --git a/extension/src/Folder.tsx b/extension/src/Folder.tsx
index b7da330..37f5cf6 100644
--- a/extension/src/Folder.tsx
+++ b/extension/src/Folder.tsx
@@ -1,4 +1,4 @@
-import BookmarkTreeNode = chrome.bookmarks.BookmarkTreeNode;
+import BookmarkTreeNode = browser.bookmarks.BookmarkTreeNode;
 import Bookmark from "./Bookmark.tsx";
 
 function Folder (props: {data: BookmarkTreeNode}) {
diff --git a/extension/src/main.tsx b/extension/src/main.tsx
index 45019a5..d988d72 100644
--- a/extension/src/main.tsx
+++ b/extension/src/main.tsx
@@ -2,7 +2,12 @@ import { createRoot } from 'react-dom/client'
 import './index.css'
 import Folder from "./Folder.tsx";
 
-chrome.bookmarks.getTree(tree => {
+// @ts-ignore
+if (typeof browser === "undefined") {
+    var browser = chrome;
+}
+
+browser.bookmarks.getTree().then(tree => {
     createRoot(document.getElementById('root')!).render(
         // @ts-ignore
         <Folder data={tree[0].children[0]} />
-- 
cgit v1.2.3