feat: add selected background to background migration

This commit is contained in:
sethburkart123
2024-11-13 09:30:36 +11:00
parent e5c05c0dca
commit 0bb4d89570
2 changed files with 12 additions and 3 deletions
+5 -2
View File
@@ -1,6 +1,5 @@
import browser from 'webextension-polyfill';
import base64ToBlob from './base64ToBlob';
import { delay } from './delay';
import { openDatabase, writeData } from '@/interface/hooks/BackgroundDataLoader';
import { backgroundUpdates } from '@/interface/hooks/BackgroundUpdates';
@@ -43,7 +42,7 @@ export const migrateBackgrounds = async (): Promise<void> => {
case 'BACKGROUND_DATA':
try {
const { id, data, mediaType, total, processed } = event.data.payload;
const { id, data, mediaType, total, processed, isSelected } = event.data.payload;
const mimeType = mediaType === 'image' ? 'image/png' : 'video/mp4';
const blob = base64ToBlob(data, mimeType);
@@ -53,6 +52,10 @@ export const migrateBackgrounds = async (): Promise<void> => {
type: mediaType
});
if (isSelected) {
localStorage.setItem('selectedBackground', id);
}
migrationState.lastProcessedId = id;
migrationState.total = total;
migrationState.processed = processed;
@@ -37,10 +37,15 @@ const getAllBackgrounds = async (): Promise<Data[]> => {
});
};
const getSelectedBackground = (): string | null => {
return localStorage.getItem('selectedBackground');
};
const startMigration = async () => {
try {
console.info('Starting background extraction...');
const backgrounds = await getAllBackgrounds();
const selectedBackground = getSelectedBackground();
console.info(`Found ${backgrounds.length} backgrounds`);
window.parent.postMessage({ type: 'GET_LAST_PROCESSED_ID' }, '*');
@@ -72,7 +77,8 @@ const startMigration = async () => {
data: base64Data,
mediaType: background.type,
total: backgrounds.length,
processed: i + 1
processed: i + 1,
isSelected: background.id === selectedBackground
}
}, '*');