import { byProps } from '@revenge-mod/modules/finders/filters'
import { waitForModules } from '@revenge-mod/modules/finders/wait'

import type { Metro } from '@revenge-mod/modules/types'

export let ReactModuleId: Metro.ModuleID
export let ReactNativeModuleId: Metro.ModuleID
export let ReactJsxRuntimeModuleId: Metro.ModuleID

export let React: typeof import('react')
export let ReactNative: typeof import('react-native')
export let ReactJsxRuntime: typeof import('react/jsx-runtime')

let RtCount = 0

const unsubRt = waitForModules(byProps<typeof React>('useState'), (exports, id) => {
    if (RtCount++ === 2) return unsubRt()

    ReactModuleId = id
    React = exports
})

const unsubRN = waitForModules(byProps<typeof ReactNative>('AppRegistry'), (exports, id) => {
    unsubRN()

    ReactNativeModuleId = id
    ReactNative = exports
})

let RJsxRCount = 0

const unsubRJsxR = waitForModules(byProps<typeof ReactJsxRuntime>('jsxs'), (exports, id) => {
    if (RJsxRCount++ === 2) return unsubRJsxR()

    ReactJsxRuntimeModuleId = id
    ReactJsxRuntime = exports
})