Fixed extensions bugs

lib/extensions/use-mock-data.ts

@@ -0,0 +1,85 @@
+'use client'
+
+import { useState, useEffect, useCallback } from 'react'
+import { useExtensionData } from './use-extension-data'
+
+interface MockMeta {
+  importedAt: string
+  source: 'csv' | 'json'
+  fileName: string
+  rowCount: number
+}
+
+interface UseMockDataResult<T> {
+  mockReport: T | null
+  isMockActive: boolean
+  isLoading: boolean
+  importedAt: string | null
+  meta: MockMeta | null
+  saveMockData: (report: T, meta: Omit<MockMeta, 'importedAt'>) => Promise<void>
+  clearMockData: () => Promise<void>
+}
+
+export function useMockData<T>(sector: string, slug: string): UseMockDataResult<T> {
+  const { getByKey, save, remove, isLoading } = useExtensionData(sector, slug)
+
+  const [mockReport, setMockReport] = useState<T | null>(null)
+  const [isMockActive, setIsMockActive] = useState(false)
+  const [meta, setMeta] = useState<MockMeta | null>(null)
+
+  // Read mock state from extension data on load
+  useEffect(() => {
+    if (isLoading) return
+
+    const enabledRecord = getByKey('mock:enabled')
+    const reportRecord = getByKey('mock:report')
+    const metaRecord = getByKey('mock:meta')
+
+    if (enabledRecord && (enabledRecord.value as { enabled?: boolean }).enabled && reportRecord) {
+      setIsMockActive(true)
+      setMockReport(reportRecord.value as T)
+      if (metaRecord) {
+        setMeta(metaRecord.value as unknown as MockMeta)
+      }
+    } else {
+      setIsMockActive(false)
+      setMockReport(null)
+      setMeta(null)
+    }
+  }, [isLoading, getByKey])
+
+  const saveMockData = useCallback(async (report: T, metaInput: Omit<MockMeta, 'importedAt'>) => {
+    const fullMeta: MockMeta = {
+      ...metaInput,
+      importedAt: new Date().toISOString(),
+    }
+
+    await save('mock:enabled', { enabled: true })
+    await save('mock:report', report as unknown as Record<string, unknown>)
+    await save('mock:meta', fullMeta as unknown as Record<string, unknown>)
+
+    setIsMockActive(true)
+    setMockReport(report)
+    setMeta(fullMeta)
+  }, [save])
+
+  const clearMockData = useCallback(async () => {
+    await remove('mock:enabled')
+    await remove('mock:report')
+    await remove('mock:meta')
+
+    setIsMockActive(false)
+    setMockReport(null)
+    setMeta(null)
+  }, [remove])
+
+  return {
+    mockReport,
+    isMockActive,
+    isLoading,
+    importedAt: meta?.importedAt ?? null,
+    meta,
+    saveMockData,
+    clearMockData,
+  }
+}