// Copyright (c) RoochNetwork
// SPDX-License-Identifier: Apache-2.0

import type { ReactNode } from 'react'
import { createContext, useRef } from 'react'

import { NetworkConfigs, RoochClientProvider } from './clientProvider.js'
import { createSessionStore, SessionStore } from './sessionStore.js'
import { getDefaultStorage, StorageType } from '../utils/index.js'
import { GlobalProvider } from './globalProvider.js'
import { InjectedThemeStyles } from '../components/styling/InjectedThemeStyles.js'
import { lightTheme } from '../themes/lightTheme.js'
import { Theme } from '../themes/themeContract.js'
import { CreateSessionArgs } from '@roochnetwork/rooch-sdk'

const DEFAULT_SESSION_STORAGE_KEY = function (_?: string) {
  return 'rooch-sdk-kit:rooch-session-info'
}

export const RoochContext = createContext<SessionStore | null>(null)

export type RoochProviderProps<T extends NetworkConfigs> = {
  networks?: NetworkConfigs
  sessionConf?: CreateSessionArgs
  onNetworkChange?: (network: keyof T & string) => void
  requestErrorCallback?: (code: number) => void
  /** The theme to use for styling UI components. Defaults to using the light theme. */
  theme?: Theme | null
  children: ReactNode
} & (
  | {
      defaultNetwork?: keyof T & string
      network?: never
    }
  | {
      defaultNetwork?: never
      network?: keyof T & string
    }
)

export function RoochProvider<T extends NetworkConfigs>(props: RoochProviderProps<T>) {
  // ** Props **
  const { children, networks, defaultNetwork, theme, sessionConf } = props

  const storeRef = useRef(
    createSessionStore({
      storage: getDefaultStorage(StorageType.Local),
      storageKey: DEFAULT_SESSION_STORAGE_KEY(),
      sessionConf: sessionConf,
    }),
  )
  return (
    <RoochContext.Provider value={storeRef.current}>
      <GlobalProvider>
        <RoochClientProvider networks={networks} defaultNetwork={defaultNetwork}>
          <InjectedThemeStyles theme={theme ?? lightTheme} />
          {children}
        </RoochClientProvider>
      </GlobalProvider>
    </RoochContext.Provider>
  )
}