import { scheduleOnUI } from 'react-native-worklets'
import { uiListModule } from './UiListModule'
import { uiManagerHelper } from './renderer/fabric/UiManagerHelper'
import { List } from './views/List'
import { Platform, TurboModuleRegistry } from 'react-native'
import { getReactFabricRenderer } from './renderer/react/ReactFabricRenderer'

export { ViewHolder } from './specs/ViewHolder.nitro'
export { IOSWorkletsModuleProxyHolder } from './specs/IOSWorkletsModuleProxyHolder.nitro'
export type { NativeListItem } from './specs/NativeListDataSource.nitro'
export type { ListProps, ListRenderer, ListRenderers } from './views/List'
export { createListDataSource, useListDataSource } from './ListDataSource'
export type {
  ListContentEqualByType,
  ListDataSource,
  ListDataSourceConfig,
  ListItem,
  ListItemForType,
  ListItemSize,
  ListItemType,
  ListKey,
} from './ListDataSource'
export { createLinearListLayout, useLinearListLayout } from './ListLayout'
export type {
  ItemSizeEstimate,
  LinearListLayoutConfig,
  LinearListLayoutIOSConfig,
  ListLayout,
} from './ListLayout'

const boxed = uiListModule
const nativeFabricUIManager = globalThis.nativeFabricUIManager

function getWorkletsModuleHolder() {
  if (Platform.OS !== 'ios') {
    return null
  }

  TurboModuleRegistry.getEnforcing('HybridUiListSurfacePresenterRegistry')
  return uiListModule.iosGetWorkletsModule()
}

function setup() {
  // TODO: ask SWM if they can remove their JS thread checks, then we could just access this from the UI thread.
  const iosWorkletsModuleHolder = getWorkletsModuleHolder()
  scheduleOnUI(() => {
    'worklet'
    globalThis.nativeFabricUIManager = nativeFabricUIManager
    boxed.setupRuntime(iosWorkletsModuleHolder)

    // This will setup the react instance on the UI runtime:
    getReactFabricRenderer()
  })
}
setup()

export { List, uiListModule, uiManagerHelper }