first commit

app/chrome-extension/entrypoints/background/storage-manager.ts

@@ -0,0 +1,112 @@
+import { BACKGROUND_MESSAGE_TYPES } from '@/common/message-types';
+
+/**
+ * Get storage statistics
+ */
+export async function handleGetStorageStats(): Promise<{
+  success: boolean;
+  stats?: any;
+  error?: string;
+}> {
+  try {
+    // Get ContentIndexer statistics
+    const { getGlobalContentIndexer } = await import('@/utils/content-indexer');
+    const contentIndexer = getGlobalContentIndexer();
+
+    // Note: Semantic engine initialization is now user-controlled
+    // ContentIndexer will be initialized when user manually triggers semantic engine initialization
+
+    // Get statistics
+    const stats = contentIndexer.getStats();
+
+    return {
+      success: true,
+      stats: {
+        indexedPages: stats.indexedPages || 0,
+        totalDocuments: stats.totalDocuments || 0,
+        totalTabs: stats.totalTabs || 0,
+        indexSize: stats.indexSize || 0,
+        isInitialized: stats.isInitialized || false,
+        semanticEngineReady: stats.semanticEngineReady || false,
+        semanticEngineInitializing: stats.semanticEngineInitializing || false,
+      },
+    };
+  } catch (error: any) {
+    console.error('Background: Failed to get storage stats:', error);
+    return {
+      success: false,
+      error: error.message,
+      stats: {
+        indexedPages: 0,
+        totalDocuments: 0,
+        totalTabs: 0,
+        indexSize: 0,
+        isInitialized: false,
+        semanticEngineReady: false,
+        semanticEngineInitializing: false,
+      },
+    };
+  }
+}
+
+/**
+ * Clear all data
+ */
+export async function handleClearAllData(): Promise<{ success: boolean; error?: string }> {
+  try {
+    // 1. Clear all ContentIndexer indexes
+    try {
+      const { getGlobalContentIndexer } = await import('@/utils/content-indexer');
+      const contentIndexer = getGlobalContentIndexer();
+
+      await contentIndexer.clearAllIndexes();
+      console.log('Storage: ContentIndexer indexes cleared successfully');
+    } catch (indexerError) {
+      console.warn('Background: Failed to clear ContentIndexer indexes:', indexerError);
+      // Continue with other cleanup operations
+    }
+
+    // 2. Clear all VectorDatabase data
+    try {
+      const { clearAllVectorData } = await import('@/utils/vector-database');
+      await clearAllVectorData();
+      console.log('Storage: Vector database data cleared successfully');
+    } catch (vectorError) {
+      console.warn('Background: Failed to clear vector data:', vectorError);
+      // Continue with other cleanup operations
+    }
+
+    // 3. Clear related data in chrome.storage (preserve model preferences)
+    try {
+      const keysToRemove = ['vectorDatabaseStats', 'lastCleanupTime', 'contentIndexerStats'];
+      await chrome.storage.local.remove(keysToRemove);
+      console.log('Storage: Chrome storage data cleared successfully');
+    } catch (storageError) {
+      console.warn('Background: Failed to clear chrome storage data:', storageError);
+    }
+
+    return { success: true };
+  } catch (error: any) {
+    console.error('Background: Failed to clear all data:', error);
+    return { success: false, error: error.message };
+  }
+}
+
+/**
+ * Initialize storage manager module message listeners
+ */
+export const initStorageManagerListener = () => {
+  chrome.runtime.onMessage.addListener((message, _sender, sendResponse) => {
+    if (message.type === BACKGROUND_MESSAGE_TYPES.GET_STORAGE_STATS) {
+      handleGetStorageStats()
+        .then((result: { success: boolean; stats?: any; error?: string }) => sendResponse(result))
+        .catch((error: any) => sendResponse({ success: false, error: error.message }));
+      return true;
+    } else if (message.type === BACKGROUND_MESSAGE_TYPES.CLEAR_ALL_DATA) {
+      handleClearAllData()
+        .then((result: { success: boolean; error?: string }) => sendResponse(result))
+        .catch((error: any) => sendResponse({ success: false, error: error.message }));
+      return true;
+    }
+  });
+};