import { assertSuccessJsonResponse } from '../assertSuccessJsonResponse'
import { z } from 'zod'
import getTimeoutMilliseconds from '../getTimeoutMilliseconds.js'
import RobotsTxt from '../RobotsTxt/RobotsTxt.js'

const wellKnownSchema = z.object({
  links: z.array(
    z.object({
      rel: z.string(),
      href: z.string()
    })
  )
})

export type WellKnown = z.infer<typeof wellKnownSchema>

export const retrieveWellKnown = async (domain: string, robotsTxt: RobotsTxt): Promise<WellKnown> => {
  console.info('Retrieving well known', { domain })
  const wellKnownUrl = new URL(`https://${domain}/.well-known/nodeinfo`)
  const wellKnownResponse = await robotsTxt.getIfAllowed(wellKnownUrl, {
    timeout: getTimeoutMilliseconds(domain),
    maxContentLength: 5000
  })
  assertSuccessJsonResponse(wellKnownResponse)
  return wellKnownSchema.parse(wellKnownResponse.data)
}