47 lines
1.1 KiB
TypeScript
47 lines
1.1 KiB
TypeScript
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 };
|