import { useState, useLayoutEffect } from 'react'
import { useWindowDimensions } from 'react-native'
import { useSafeAreaInsets } from 'react-native-safe-area-context'

interface UseSafeHeightParams {
  top?: boolean | number
  bottom?: boolean | number
}

const getDefaultValue = (v: boolean | number | undefined, dv: number) => {
  if (typeof v === 'number') {
    return v
  }

  if (typeof v === 'boolean' && v) {
    return dv
  }

  return 0
}

const useSafeHeight = ({
  top = true,
  bottom = true,
}: UseSafeHeightParams = {}) => {
  const insets = useSafeAreaInsets()
  const insetTop = getDefaultValue(top, insets.top)
  const insetBottom = getDefaultValue(bottom, insets.bottom)
  const dimensionsWindow = useWindowDimensions()
  const [height, setHeight] = useState(
    dimensionsWindow.height - insetTop - insetBottom,
  )

  useLayoutEffect(() => {
    setHeight(dimensionsWindow.height - insetTop - insetBottom)
  }, [dimensionsWindow.height, insetTop, insetBottom])

  return height
}

export default useSafeHeight