packages/taro-react/src/index.ts

/* eslint-disable @typescript-eslint/no-unused-vars */
import { TaroElement } from '@tarojs/runtime'
import { ensure, isFunction } from '@tarojs/shared'
import { ReactNode } from 'react'

import { finishEventHandler } from './event'
import { TaroReconciler } from './reconciler'
import { ContainerMap, createRoot, render } from './render'

let isInsideEventHandler = false

// 重新包裹 batchedUpdates，使其可以在触发事件后执行 finishEventHandler
const unstable_batchedUpdates = (fn, a) => {
  if (isInsideEventHandler) {
    return fn(a)
  }

  isInsideEventHandler = true
  
  try {
    return TaroReconciler.batchedUpdates(fn, a)
  } finally {
    isInsideEventHandler = false
    finishEventHandler()
  }
}

function unmountComponentAtNode (dom: TaroElement) {
  ensure(dom && [1, 8, 9, 11].includes(dom.nodeType), 'unmountComponentAtNode(...): Target container is not a DOM element.')

  const root = ContainerMap.get(dom)

  if (!root) return false

  unstable_batchedUpdates(() => {
    root.unmount(() => {
      ContainerMap.delete(dom)
    })
  }, null)

  return true
}

function findDOMNode (comp?: TaroElement | ReactNode) {
  if (comp == null) {
    return null
  }

  const nodeType = (comp as TaroElement).nodeType
  if (nodeType === 1 || nodeType === 3) {
    return comp
  }

  return TaroReconciler.findHostInstance(comp as Record<string, any>)
}

const portalType = isFunction(Symbol) && Symbol.for
  ? Symbol.for('react.portal')
  : 0xeaca

function createPortal (
  children: ReactNode,
  containerInfo: TaroElement,
  key?: string
) {
  return {
    $$typeof: portalType,
    key: key == null ? null : String(key),
    children,
    containerInfo,
    implementation: null
  }
}

export {
  createPortal,
  createRoot,
  findDOMNode,
  render,
  unmountComponentAtNode,
  unstable_batchedUpdates
}

export default {
  render,
  createRoot,
  unstable_batchedUpdates,
  unmountComponentAtNode,
  findDOMNode,
  createPortal
}