import { SitemapStream, streamToPromise } from 'sitemap'
import { NextApiRequest, NextApiResponse } from 'next'
import { SSR_CONFIG } from '@SSR_CONFIG'
import { getAllStoriesOfProject } from '../../utils/initial-props/storyblokPagesConfig'
import { PageItem } from '../../typings/generated/schema'
import { internalLinkHandler } from '../../utils/internalLinkHandler'
// import { createGzip } from 'zlib'

export default async function sitemapApi(
  req: NextApiRequest,
  res: NextApiResponse
) {
  // res.setHeader('Content-Encoding', 'gzip')
  try {
    const stories: PageItem[] = await getAllStoriesOfProject({
      language: process.env.SITEMAP_LOCALE
        ? process.env.SITEMAP_LOCALE
        : undefined
    })
    const smStream = new SitemapStream({
      hostname: `https://${process.env.SITEMAP_HOSTNAME || req.headers.host}`
    })
    let locale: string | undefined = process.env.SITEMAP_LOCALE
      ? process.env.SITEMAP_LOCALE
      : undefined
    if (process.env.LOCALE_DOMAIN_MAP) {
      const domainMap: { locale: string; domain: string }[] = JSON.parse(
        process.env.LOCALE_DOMAIN_MAP
      )
      const checkMap = domainMap.find(
        (items) => items.domain === req.headers.host
      )
      if (checkMap?.locale) {
        locale = checkMap.locale
      }
    }
    const ignoreList =
      (process.env.SITEMAP_IGNORE_PATH &&
        process.env.SITEMAP_IGNORE_PATH.split(',')) ||
      []

    ignoreList.push('demo-content')
    const urls = new Set<string>() // To track added URLs

    const addUrl = ({
      url,
      lastmod,
      priority
    }: {
      url: string
      lastmod: string
      priority: number
    }) => {
      if (!urls.has(url)) {
        urls.add(url)
        smStream.write({
          url: url,
          lastmod: lastmod,
          priority: priority
        })
      }
    }

    for (let i = 0; i < stories.length; i++) {
      const story = stories[i]
      const fullSlug = story.full_slug as string
      const shouldIndex = !ignoreList.some((ignorePath: string) =>
        fullSlug.includes(ignorePath)
      )
      if (shouldIndex) {
        const isHome = story.slug === 'home'
        if (isHome) {
          let homeSlug = fullSlug.replace('home', '')
          if (
            process.env.NEXT_PUBLIC_FIELD_LEVEL_TRANSLATION &&
            process.env.SITEMAP_LOCALE
          ) {
            homeSlug = homeSlug.replace(`/${process.env.SITEMAP_LOCALE}/`, '/')
          }
          addUrl({
            url: internalLinkHandler(homeSlug),
            lastmod: story.published_at!,
            priority: 1.0
          })
        } else {
          let internalSlug = internalLinkHandler(fullSlug)
          if (
            process.env.NEXT_PUBLIC_FIELD_LEVEL_TRANSLATION &&
            process.env.SITEMAP_LOCALE
          ) {
            internalSlug = internalSlug.replace(
              `/${process.env.SITEMAP_LOCALE}/`,
              '/'
            )
          }
          addUrl({
            url: internalSlug,
            lastmod: story.published_at!,
            priority: 0.5
          })
        }
      }
    }
    await Promise.all(
      SSR_CONFIG.ssrHooks.sitemap.map((func) => func(smStream, locale))
    )
    smStream.end()

    const sitemap = await streamToPromise(smStream).then((sm) => sm.toString())

    res.setHeader('Content-Type', 'text/xml')
    res.setHeader('Cache-Control', 's-maxage=600, stale-while-revalidate')
    res.write(sitemap)
    res.end()
  } catch (e) {
    console.log(e)
    res.statusCode = 500
    res.end()
  }
}