import { DataEngine, CustomDataLink } from '@dhis2/data-engine'
import type { CustomData, CustomLinkOptions } from '@dhis2/data-engine'
import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
import React from 'react'
import { DataContext } from '../context/DataContext'
import { queryClientOptions as queryClientDefaults } from './DataProvider'

interface CustomProviderInput {
    children: React.ReactNode
    data: CustomData
    options?: CustomLinkOptions
    queryClientOptions?: any
}

export const CustomDataProvider = ({
    children,
    data,
    options,
    queryClientOptions = queryClientDefaults,
}: CustomProviderInput): JSX.Element => {
    const link = new CustomDataLink(data, options)
    const engine = new DataEngine(link)
    const queryClient = new QueryClient(queryClientOptions)
    const context = { engine }

    return (
        <QueryClientProvider client={queryClient}>
            <DataContext.Provider value={context}>
                {children}
            </DataContext.Provider>
        </QueryClientProvider>
    )
}