import minifiedScrollRestorationScript from '../scroll-restoration-inline?script-string'
import {
  defaultGetScrollRestorationKey,
  storageKey,
} from '../scroll-restoration'
import { escapeHtml } from '../utils'
import type { AnyRouter } from '../router'

const defaultInlineScrollRestorationScript = `(${minifiedScrollRestorationScript})(${escapeHtml(
  JSON.stringify(storageKey),
)})`

function getScrollRestorationScript(key?: string) {
  if (key === undefined) {
    return defaultInlineScrollRestorationScript
  }

  return `(${minifiedScrollRestorationScript})(${escapeHtml(JSON.stringify(storageKey))},${escapeHtml(JSON.stringify(key))})`
}

export function getScrollRestorationScriptForRouter(router: AnyRouter) {
  if (
    typeof router.options.scrollRestoration === 'function' &&
    !router.options.scrollRestoration({ location: router.latestLocation })
  ) {
    return null
  }

  const getKey = router.options.getScrollRestorationKey
  if (!getKey) {
    return defaultInlineScrollRestorationScript
  }

  const location = router.latestLocation
  const userKey = getKey(location)
  const defaultKey = defaultGetScrollRestorationKey(location)

  if (userKey === defaultKey) {
    return defaultInlineScrollRestorationScript
  }

  return getScrollRestorationScript(userKey)
}