import got from 'got'
import { load } from 'cheerio'
import { AiovideodlArgsSchema, Aiovideodl, AiovideodlSchema } from './types.js'

export default async function aiovideodl (url: string): Promise<Aiovideodl> {
  AiovideodlArgsSchema.parse(arguments)

  const resToken = await got('https://aiovideodl.ml/')

  const $$ = load(resToken.body)
  const token = $$('#token').val() as string
  const cookie = resToken.headers['set-cookie']?.map(v => v.split(';')[0]).join('; ').trim()

  const body = new URLSearchParams()
  body.append('url', url)
  body.append('token', token)

  const json = await got('https://aiovideodl.ml/wp-json/aio-dl/video-data/', {
    method: 'post',
    headers: {
      'content-type': 'application/x-www-form-urlencoded',
      cookie: cookie || 'pll_language=en; _ga=GA1.2.946338805.1646539824; PHPSESSID=f5ec5a6eb553bf3a55508ad4ce2ed200; _gid=GA1.2.425279475.1647856193; _gat_gtag_UA_46116261_21=1',
      origin: 'https://aiovideodl.ml',
      referer: 'https://aiovideodl.ml/',
      'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.74 Safari/537.36'
    },
    body: body.toString()
  }).json<Aiovideodl>()

  return AiovideodlSchema.parse(json)
}