import { cacheLife } from 'next/cache'
import { cookies, headers } from 'next/headers'
import { connection } from 'next/server'
import { setTimeout } from 'timers/promises'
import { Suspense } from 'react'

export const unstable_instant = {
  prefetch: 'runtime',
  samples: [
    {
      cookies: [{ name: 'testCookie', value: 'testValue' }],
      headers: [['x-test-header', 'test']],
      searchParams: {},
    },
  ],
}

export default async function Page({
  searchParams,
}: {
  searchParams: Promise<{ q?: string }>
}) {
  return (
    <main>
      <CachedContent />
      <div id="search-params-boundary">
        <Suspense fallback={<p>Loading search params...</p>}>
          <SearchParamsContent searchParams={searchParams} />
        </Suspense>
      </div>
      <div id="cookies-boundary">
        <Suspense fallback={<p>Loading cookies...</p>}>
          <CookiesContent />
        </Suspense>
      </div>
      <div id="headers-boundary">
        <Suspense fallback={<p>Loading headers...</p>}>
          <HeadersContent />
        </Suspense>
      </div>
      <div id="connection-boundary">
        <Suspense fallback={<p>Loading connection...</p>}>
          <ConnectionContent />
        </Suspense>
      </div>
    </main>
  )
}

async function CachedContent() {
  'use cache'
  cacheLife({ stale: 120 })
  return <p id="cached-content">Cached content ({new Date().toISOString()})</p>
}

async function SearchParamsContent({
  searchParams,
}: {
  searchParams: Promise<{ q?: string }>
}) {
  'use cache: private'
  cacheLife({ stale: 30 })
  const { q } = await searchParams
  await setTimeout(300)
  return (
    <p>
      Search params: {q ?? 'none'} ({new Date().toISOString()})
    </p>
  )
}

async function CookiesContent() {
  'use cache: private'
  cacheLife({ stale: 30 })
  const cookieStore = await cookies()
  const value = cookieStore.get('testCookie')?.value ?? 'none'
  await setTimeout(300)
  return (
    <p>
      Cookie: {value} ({new Date().toISOString()})
    </p>
  )
}

async function HeadersContent() {
  'use cache: private'
  cacheLife({ stale: 30 })
  const headerStore = await headers()
  const value = headerStore.get('x-test-header') ?? 'none'
  await setTimeout(300)
  return (
    <p>
      Header: {value} ({new Date().toISOString()})
    </p>
  )
}

async function ConnectionContent() {
  await connection()
  await setTimeout(600)
  return <p>Dynamic content ({new Date().toISOString()})</p>
}