import Database from '@tauri-apps/plugin-sql';
import { PropsWithChildren, createContext, useContext, useEffect, useState } from 'react';

import { LumeStorage } from '@libs/storage/instance';

interface StorageContext {
  db: LumeStorage;
}

const StorageContext = createContext<StorageContext>({
  db: undefined,
});

const StorageProvider = ({ children }: PropsWithChildren<object>) => {
  const [db, setDB] = useState<LumeStorage>(undefined);

  async function initLumeStorage() {
    const sqlite = await Database.load('sqlite:lume.db');
    const lumeStorage = new LumeStorage(sqlite);

    if (!lumeStorage.account) await lumeStorage.getActiveAccount();
    setDB(lumeStorage);
  }

  useEffect(() => {
    if (!db) initLumeStorage();

    return () => {
      db.close();
    };
  }, []);

  if (db) {
    return <StorageContext.Provider value={{ db }}>{children}</StorageContext.Provider>;
  }
};

const useStorage = () => {
  const context = useContext(StorageContext);
  if (context === undefined) {
    throw new Error('Storage not found');
  }
  return context;
};

export { StorageProvider, useStorage };
