import AsyncStorage from '@react-native-async-storage/async-storage'
import { InitialState, NavigationState } from '@react-navigation/native'
import { getInitialURL } from 'expo-linking'
import { useCallback, useEffect, useState } from 'react'
import { Platform } from 'react-native'

import { ASYNC_STORAGE_KEYS, isProduction } from '@/constants'

const EXPO_LINK_REGEXP = /^exp:\/\/\d+\.\d+\.\d+\.\d+:\d*$/

const checkInitialURL = (initialUrl: string | null) => {
  return initialUrl == null || initialUrl.match(EXPO_LINK_REGEXP) !== null
}

type NavigationStatePersistenceReturn = {
  initialState?: InitialState
  onStateChange: (state: NavigationState | undefined) => void
  isReady: boolean
}

const { NAVIGATION_STATE } = ASYNC_STORAGE_KEYS

// TODO: make this work on expo router
export const useNavigationStatePersistence = (): NavigationStatePersistenceReturn => {
  const [isReady, setIsReady] = useState(isProduction)
  const [initialState, setInitialState] = useState<InitialState>()

  useEffect(() => {
    const restoreState = async () => {
      try {
        const initialUrl = await getInitialURL()

        if (Platform.OS !== 'web' && checkInitialURL(initialUrl)) {
          // Only restore state if there's no deep link and we're not on web
          const savedStateString = await AsyncStorage.getItem(NAVIGATION_STATE)
          const state = savedStateString ? JSON.parse(savedStateString) : undefined

          if (state !== undefined) {
            setInitialState(state)
          }
        }
      } finally {
        setIsReady(true)
      }
    }

    if (!isReady) {
      restoreState()
    }
  }, [isReady])

  const onStateChange = useCallback((state: NavigationState | undefined) => {
    AsyncStorage.setItem(NAVIGATION_STATE, JSON.stringify(state))
  }, [])

  return {
    initialState,
    onStateChange,
    isReady,
  }
}