From 8cd73977bb4798e32b56644737b60cb9d2d4081e Mon Sep 17 00:00:00 2001
From: SethBurkart123 <seth.burkart@hills.adventist.edu.au>
Date: Mon, 22 Apr 2024 11:09:57 +1000
Subject: [PATCH] add theme transfer

---
 src/interface/components/ThemeCover.tsx    | 18 +++++++++----
 src/interface/components/ThemeSelector.tsx | 18 +++++++++++--
 src/interface/hooks/ThemeManagment.ts      | 30 +++++++++++++++++++---
 src/interface/types/CustomThemes.ts        |  4 +++
 src/seqta/ui/themes/downloadTheme.ts       | 11 +++++---
 5 files changed, 67 insertions(+), 14 deletions(-)

diff --git a/src/interface/components/ThemeCover.tsx b/src/interface/components/ThemeCover.tsx
index 33d20177..d1dc6ddf 100644
--- a/src/interface/components/ThemeCover.tsx
+++ b/src/interface/components/ThemeCover.tsx
@@ -1,18 +1,21 @@
 import React, { useState } from 'react';
-import { CustomTheme } from '../types/CustomThemes';
+import { CustomTheme, DownloadedTheme } from '../types/CustomThemes';
 import browser from 'webextension-polyfill';
 import { ArrowUpOnSquareIcon, PencilIcon } from '@heroicons/react/24/outline';
+import { sendThemeUpdate } from '../hooks/ThemeManagment';
 
 type ThemeCoverProps = {
-  theme: Omit<CustomTheme, 'CustomImages'>;
+  theme: Omit<CustomTheme, 'CustomImages'> | DownloadedTheme;
   isSelected: boolean;
   isEditMode: boolean;
+  downloaded?: boolean;
   onThemeSelect: (themeId: string) => void;
   onThemeDelete: (themeId: string) => void;
 };
 
 export const ThemeCover: React.FC<ThemeCoverProps> = ({
   theme,
+  downloaded,
   isSelected,
   isEditMode,
   onThemeSelect,
@@ -21,7 +24,12 @@ export const ThemeCover: React.FC<ThemeCoverProps> = ({
   const [uploading, setUploading] = useState<boolean>(false);
   const handleThemeClick = () => {
     if (isEditMode) return;
-    onThemeSelect(theme.id);
+    if (downloaded) {
+      sendThemeUpdate(theme as DownloadedTheme, true)
+    } else {
+      console.log(theme)
+      onThemeSelect(theme.id);
+    }
   };
 
   const handleDeleteClick = (e: React.MouseEvent) => {
@@ -54,7 +62,7 @@ export const ThemeCover: React.FC<ThemeCoverProps> = ({
         </div>
       )}
       
-      { isEditMode ? <></> :
+      { !isEditMode || !downloaded &&
       <>
         <div
           className="absolute z-20 flex w-8 h-8 p-2 text-white transition-all rounded-full delay-[20ms] opacity-0 top-1 right-2 dark:bg-black/50 place-items-center group-hover:opacity-100 group-hover:top-[1.25rem]"
@@ -75,7 +83,7 @@ export const ThemeCover: React.FC<ThemeCoverProps> = ({
       <div className="relative top-0 z-10 flex justify-center w-full h-full overflow-hidden transition dark:text-white rounded-xl group place-items-center bg-zinc-100 dark:bg-zinc-900">
         {theme.coverImage &&
           <img
-            src={theme.coverImage as string}
+            src={(typeof theme.coverImage) == 'string' ? theme.coverImage as string : URL.createObjectURL(theme.coverImage as Blob)}
             alt={theme.name}
             className="absolute inset-0 z-0 object-cover"
           />
diff --git a/src/interface/components/ThemeSelector.tsx b/src/interface/components/ThemeSelector.tsx
index 8c4b9639..a5d55b63 100644
--- a/src/interface/components/ThemeSelector.tsx
+++ b/src/interface/components/ThemeSelector.tsx
@@ -1,8 +1,8 @@
 import React, { useEffect, useState, useCallback, forwardRef, useImperativeHandle, ForwardRefExoticComponent, RefAttributes } from 'react';
-import { listThemes, deleteTheme, setTheme, disableTheme } from '../hooks/ThemeManagment';
+import { listThemes, deleteTheme, setTheme, disableTheme, getDownloadedThemes } from '../hooks/ThemeManagment';
 import { ThemeCover } from './ThemeCover';
 import Browser from 'webextension-polyfill';
-import { CustomTheme } from '../types/CustomThemes';
+import { CustomTheme, DownloadedTheme } from '../types/CustomThemes';
 import { useSettingsContext } from '../SettingsContext';
 import { SettingsState } from '../types/AppProps';
 
@@ -13,6 +13,7 @@ interface ThemeSelectorProps {
 
 const ThemeSelector: ForwardRefExoticComponent<Omit<ThemeSelectorProps, "ref"> & RefAttributes<any>> = forwardRef(({ isEditMode = false }, ref) => {
   const [themes, setThemes] = useState<Omit<CustomTheme, 'CustomImages'>[]>([]);
+  const [downloadedThemes, setDownloadedThemes] = useState<DownloadedTheme[]>([]);
   const [isLoading, setIsLoading] = useState<boolean>(true);
   const { settingsState, setSettingsState } = useSettingsContext();
 
@@ -56,6 +57,7 @@ const ThemeSelector: ForwardRefExoticComponent<Omit<ThemeSelectorProps, "ref"> &
       const { themes, selectedTheme } = await listThemes();
 
       setThemes(themes);
+      setDownloadedThemes(await getDownloadedThemes());
       setSelectedTheme(selectedTheme ? selectedTheme : '');
     } catch (error) {
       console.error('Error fetching themes:', error);
@@ -118,6 +120,18 @@ const ThemeSelector: ForwardRefExoticComponent<Omit<ThemeSelectorProps, "ref"> &
           />
         ))}
 
+        {downloadedThemes.map((theme) => (
+          <ThemeCover
+            key={theme.id}
+            downloaded={true}
+            theme={theme}
+            isSelected={theme.id === settingsState.selectedTheme}
+            isEditMode={isEditMode}
+            onThemeSelect={handleThemeSelect}
+            onThemeDelete={handleThemeDelete}
+          />
+        ))}
+
         <button
           onClick={() => Browser.tabs.create({ url: Browser.runtime.getURL('src/interface/index.html#store')})}
           className="flex items-center justify-center w-full transition aspect-theme rounded-xl bg-zinc-100 dark:bg-zinc-900 dark:text-white"
diff --git a/src/interface/hooks/ThemeManagment.ts b/src/interface/hooks/ThemeManagment.ts
index 75ce8d14..69eab4e2 100644
--- a/src/interface/hooks/ThemeManagment.ts
+++ b/src/interface/hooks/ThemeManagment.ts
@@ -1,5 +1,6 @@
 import browser from 'webextension-polyfill'
-import { CustomTheme, ThemeList } from '../types/CustomThemes';
+import { CustomTheme, DownloadedTheme, ThemeList } from '../types/CustomThemes';
+import localforage from 'localforage';
 
 export const downloadTheme = async (themeName: string, themeURL: string) => {
   // send message to the background script
@@ -24,6 +25,29 @@ export const setTheme = async (themeID: string) => {
   });
 }
 
+export const getDownloadedThemes = async (): Promise<DownloadedTheme[]> => {
+  // send message to the background script
+  const response: DownloadedTheme[] = await new Promise(async (resolve, reject) => {
+    try {
+      let availableThemes = await localforage.getItem('availableThemes') as string[];
+      availableThemes = Array.from(new Set(availableThemes));
+
+      const downloadedThemes: DownloadedTheme[] = [];
+      for (let i = 0; i < availableThemes.length; i++) {
+        let themeData = await localforage.getItem(availableThemes[i]) as DownloadedTheme;
+
+        downloadedThemes.push(themeData);
+      }
+  
+      resolve(downloadedThemes);
+    } catch(error) {
+      reject(error);
+    }
+  });
+
+  return response;
+}
+
 export const listThemes = async (): Promise<ThemeList> => {
   // send message to the background script
   const response: ThemeList = await new Promise((resolve, reject) => {
@@ -72,7 +96,7 @@ export const deleteTheme = async (themeID: string) => {
   });
 }
 
-export const sendThemeUpdate = (updatedTheme: CustomTheme, saveTheme?: boolean, updateImages?: boolean) => {
+export const sendThemeUpdate = (updatedTheme: CustomTheme | DownloadedTheme, saveTheme?: boolean, updateImages?: boolean) => {
   saveTheme = saveTheme || false;
 
   const imageDataPromises = updatedTheme.CustomImages.map(async (image) => {
@@ -97,7 +121,7 @@ export const sendThemeUpdate = (updatedTheme: CustomTheme, saveTheme?: boolean,
       CustomImages: imageData,
     };
 
-    if (saveTheme && updatedTheme.coverImage instanceof Blob) {
+    if (saveTheme && updatedTheme.coverImage) {
       themeData.coverImage = await blobToBase64(updatedTheme.coverImage as Blob);
     } else {
       themeData.coverImage = null;
diff --git a/src/interface/types/CustomThemes.ts b/src/interface/types/CustomThemes.ts
index 0b3f20d1..abf73a21 100644
--- a/src/interface/types/CustomThemes.ts
+++ b/src/interface/types/CustomThemes.ts
@@ -12,6 +12,10 @@ export type CustomTheme = {
   hideThemeName: boolean;
 }
 
+export type DownloadedTheme = CustomTheme & {
+  webURL: string;
+}
+
 export type CustomImage = {
   id: string;
   blob: Blob;
diff --git a/src/seqta/ui/themes/downloadTheme.ts b/src/seqta/ui/themes/downloadTheme.ts
index 5331b8f8..1fa4ff8a 100644
--- a/src/seqta/ui/themes/downloadTheme.ts
+++ b/src/seqta/ui/themes/downloadTheme.ts
@@ -15,7 +15,9 @@ const DownloadTheme = async (theme: ThemesResponse & { theme: CustomTheme & { im
     images.push({ imageData, imageID });
   }
 
-  console.log("Original Theme", theme);
+  const coverImage = await fetch(
+    `https://betterseqta.pockethost.io/api/files/${theme.collectionId}/${theme.id}/${theme.coverImage}`
+  );
 
   // add to temp storage index
   let availableThemes = await localforage.getItem('availableThemes') as string[];
@@ -26,13 +28,14 @@ const DownloadTheme = async (theme: ThemesResponse & { theme: CustomTheme & { im
   }
   localforage.setItem('availableThemes', availableThemes);
 
-  // save the theme to the temp storage
   localforage.setItem(theme.theme.id, {
     ...theme.theme,
-    images: theme.theme.images.map((image) => {
+    webURL: theme.id,
+    coverImage: await coverImage.blob(),
+    CustomImages: theme.theme.images.map((image) => {
       return {
         ...image,
-        imageData: images.find((i) => i.imageID.split('_')[0] === image.id)?.imageData
+        blob: images.find((i) => i.imageID.split('_')[0] === image.id)?.imageData
       }
     })
   });