Source-Web-24070-TheGreatTipsy/vite/src/util/useData.jsx

import { createContext, useContext, useEffect, useState } from "react";
import  { BaseDirectory, readTextFile, exists, writeTextFile, mkdir } from "@tauri-apps/plugin-fs";
import { ParamKeys } from "./system_prompt";
import { path } from '@tauri-apps/api';

const dataContext=createContext();


const filePath= 'param.json';

export function DataProvider({children}) {

    const [data, setData] = useState(null);

    async function read(){

        try{

            const folder=await path.appDataDir();
            if (!(await exists(folder))) {        
                
                console.log('Creating folder:', folder);
                await mkdir(folder);
            }

            const contents=await readTextFile(filePath, { baseDir: BaseDirectory.AppData });
            let output=await JSON.parse(contents);


            // check if all keys in ParamKeys are present in output
            const missingKeys = ParamKeys.filter(key => !output.hasOwnProperty(key));
            if(missingKeys.length > 0){
                console.warn("Missing keys in output:", missingKeys);
                // Fill missing keys with default values
                missingKeys.forEach(key => {
                    output[key] = params[key] || ""; // Use params[key] as default value
                });
            }
            
            console.log("File read successfully:", output);

            return output; 

        }catch(error){
            console.error("Error reading file:", error);
            return null; // Return null if reading fails
        }
        
        
    }
    async function write(towrite){

        // let towrite=data;
        if(!towrite){
            const res=await fetch('default.json');
            towrite=await res.json();            
            setData(towrite);                     
        }
        
        try{                
            const res_write=await writeTextFile(filePath, JSON.stringify(towrite), { baseDir: BaseDirectory.AppData })
            console.log("File written successfully:", res_write);
        }catch(error){
            console.error("Error writing file:", error);
        }

    }

    useEffect(()=>{
        read().then(data_ => {
            if(data_){
                setData(data_);
            } else {
                write(); // Write default data if read fails
            }
        }).catch(error => {
            console.error("Error in useEffect:", error);
        });
    },[])


    return (
        <dataContext.Provider value={{
            data,
            read,
            write,            
        }}>
            {children}
        </dataContext.Provider>
    );
}

export function useData() {
    const context = useContext(dataContext);
    if (!context) {
        throw new Error("useData must be used within a DataProvider");
    }
    return context;
}