import { metrics } from "@opentelemetry/api"

import { GetContactConfig } from "@/lib/graphql/Query/ContactConfig.graphql"
import {
  GetCurrentFooter,
  GetCurrentFooterRef,
} from "@/lib/graphql/Query/CurrentFooter.graphql"
import {
  GetCurrentHeader,
  GetCurrentHeaderRef,
} from "@/lib/graphql/Query/CurrentHeader.graphql"
import { request } from "@/lib/graphql/request"
import { notFound } from "@/server/errors/trpc"
import { contentstackBaseProcedure, router } from "@/server/trpc"

import { generateTag } from "@/utils/generateTag"

import { langInput } from "./input"
import {
  type ContactConfigData,
  FooterDataRaw,
  FooterRefDataRaw,
  HeaderData,
  HeaderDataRaw,
  HeaderRefDataRaw,
  validateContactConfigSchema,
  validateFooterConfigSchema,
  validateHeaderConfigSchema,
} from "./output"

const meter = metrics.getMeter("trpc.contentstack.base")
// OpenTelemetry metrics: ContactConfig
const getContactConfigCounter = meter.createCounter(
  "trpc.contentstack.contactConfig.get"
)
const getContactConfigSuccessCounter = meter.createCounter(
  "trpc.contentstack.contactConfig.get-success"
)
const getContactConfigFailCounter = meter.createCounter(
  "trpc.contentstack.contactConfig.get-fail"
)
// OpenTelemetry metrics: Header
const getHeaderRefCounter = meter.createCounter(
  "trpc.contentstack.header.ref.get"
)
const getHeaderRefSuccessCounter = meter.createCounter(
  "trpc.contentstack.header.ref.get-success"
)
const getHeaderRefFailCounter = meter.createCounter(
  "trpc.contentstack.header.ref.get-fail"
)
const getHeaderCounter = meter.createCounter("trpc.contentstack.header.get")
const getHeaderSuccessCounter = meter.createCounter(
  "trpc.contentstack.header.get-success"
)
const getHeaderFailCounter = meter.createCounter(
  "trpc.contentstack.header.get-fail"
)
// OpenTelemetry metrics: Footer
const getFooterRefCounter = meter.createCounter(
  "trpc.contentstack.footer.ref.get"
)
const getFooterRefSuccessCounter = meter.createCounter(
  "trpc.contentstack.footer.ref.get-success"
)
const getFooterRefFailCounter = meter.createCounter(
  "trpc.contentstack.footer.ref.get-fail"
)
const getFooterCounter = meter.createCounter("trpc.contentstack.footer.get")
const getFooterSuccessCounter = meter.createCounter(
  "trpc.contentstack.footer.get-success"
)
const getFooterFailCounter = meter.createCounter(
  "trpc.contentstack.footer.get-fail"
)

export const baseQueryRouter = router({
  contact: contentstackBaseProcedure.query(async ({ ctx }) => {
    const { lang } = ctx
    getContactConfigCounter.add(1, { lang })
    console.info(
      "contentstack.contactConfig start",
      JSON.stringify({ query: { lang } })
    )
    const response = await request<ContactConfigData>(GetContactConfig, {
      locale: lang,
    })

    if (!response.data) {
      const notFoundError = notFound(response)

      getContactConfigFailCounter.add(1, {
        lang,
        error_type: "not_found",
        error: JSON.stringify({ code: notFoundError.code }),
      })

      console.error(
        "contentstack.config not found error",
        JSON.stringify({ query: { lang }, error: { code: notFoundError.code } })
      )

      throw notFoundError
    }

    const validatedContactConfigConfig = validateContactConfigSchema.safeParse(
      response.data
    )

    if (!validatedContactConfigConfig.success) {
      getContactConfigFailCounter.add(1, {
        lang,
        error_type: "validation_error",
        error: JSON.stringify(validatedContactConfigConfig.error),
      })
      console.error(
        "contentstack.contactConfig validation error",
        JSON.stringify({
          query: { lang },
          error: validatedContactConfigConfig.error,
        })
      )
      return null
    }
    getContactConfigSuccessCounter.add(1, { lang })
    console.info(
      "contentstack.contactConfig success",
      JSON.stringify({ query: { lang } })
    )
    return validatedContactConfigConfig.data.all_contact_config.items[0]
  }),
  header: contentstackBaseProcedure
    .input(langInput)
    .query(async ({ input }) => {
      getHeaderRefCounter.add(1, { lang: input.lang })
      console.info(
        "contentstack.header.ref start",
        JSON.stringify({ query: { lang: input.lang } })
      )
      const responseRef = await request<HeaderRefDataRaw>(GetCurrentHeaderRef, {
        locale: input.lang,
      })
      getHeaderCounter.add(1, { lang: input.lang })
      console.info(
        "contentstack.header start",
        JSON.stringify({
          query: { lang: input.lang },
        })
      )

      // There's currently no error handling/validation for the responseRef, should it be added?
      const response = await request<HeaderDataRaw>(
        GetCurrentHeader,
        { locale: input.lang },
        {
          tags: [
            generateTag(
              input.lang,
              responseRef.data.all_current_header.items[0].system.uid
            ),
          ],
        }
      )

      if (!response.data) {
        const notFoundError = notFound(response)
        getHeaderFailCounter.add(1, {
          lang: input.lang,
          error_type: "not_found",
          error: JSON.stringify({ code: notFoundError.code }),
        })
        console.error(
          "contentstack.header not found error",
          JSON.stringify({
            query: {
              lang: input.lang,
            },
            error: { code: notFoundError.code },
          })
        )
        throw notFoundError
      }

      const validatedHeaderConfig = validateHeaderConfigSchema.safeParse(
        response.data
      )

      if (!validatedHeaderConfig.success) {
        getHeaderFailCounter.add(1, {
          lang: input.lang,
          error_type: "validation_error",
          error: JSON.stringify(validatedHeaderConfig.error),
        })
        console.error(
          "contentstack.header validation error",
          JSON.stringify({
            query: {
              lang: input.lang,
            },
            error: validatedHeaderConfig.error,
          })
        )
        return null
      }
      getHeaderSuccessCounter.add(1, { lang: input.lang })
      console.info(
        "contentstack.header success",
        JSON.stringify({
          query: { lang: input.lang },
        })
      )
      const logo =
        validatedHeaderConfig.data.all_current_header.items[0].logoConnection
          .edges?.[0]?.node

      return {
        ...validatedHeaderConfig.data.all_current_header.items[0],
        logo,
      } as HeaderData
    }),
  footer: contentstackBaseProcedure
    .input(langInput)
    .query(async ({ input }) => {
      getFooterRefCounter.add(1, { lang: input.lang })
      console.info(
        "contentstack.footer.ref start",
        JSON.stringify({ query: { lang: input.lang } })
      )
      const responseRef = await request<FooterRefDataRaw>(GetCurrentFooterRef, {
        locale: input.lang,
      })
      // There's currently no error handling/validation for the responseRef, should it be added?
      getFooterCounter.add(1, { lang: input.lang })
      console.info(
        "contentstack.footer start",
        JSON.stringify({
          query: {
            lang: input.lang,
          },
        })
      )
      const response = await request<FooterDataRaw>(
        GetCurrentFooter,
        {
          locale: input.lang,
        },
        {
          tags: [
            generateTag(
              input.lang,
              responseRef.data.all_current_footer.items[0].system.uid
            ),
          ],
        }
      )

      if (!response.data) {
        const notFoundError = notFound(response)
        getFooterFailCounter.add(1, {
          lang: input.lang,
          error_type: "not_found",
          error: JSON.stringify({ code: notFoundError.code }),
        })
        console.error(
          "contentstack.footer not found error",
          JSON.stringify({
            query: {
              lang: input.lang,
            },
            error: { code: notFoundError.code },
          })
        )
        throw notFoundError
      }

      const validatedFooterConfig = validateFooterConfigSchema.safeParse(
        response.data
      )

      if (!validatedFooterConfig.success) {
        getFooterFailCounter.add(1, {
          lang: input.lang,
          error_type: "validation_error",
          error: JSON.stringify(validatedFooterConfig.error),
        })
        console.error(
          "contentstack.footer validation error",
          JSON.stringify({
            query: { lang: input.lang },
            error: validatedFooterConfig.error,
          })
        )
        return null
      }
      getFooterSuccessCounter.add(1, { lang: input.lang })
      console.info(
        "contentstack.footer success",
        JSON.stringify({ query: { lang: input.lang } })
      )
      return validatedFooterConfig.data.all_current_footer.items[0]
    }),
})