import got from 'got'
import { load } from 'cheerio'
import FormData from 'form-data'
import type { TextProList } from './types'
import { ScraperError } from '../utils.js'

type TextProPostJSON = {
  success: boolean,
  info: string,
  code: number,
  image: string,
  fullsize_image: string,
  image_code: string,
  session_id: number
}

const BASE_URL = 'https://textpro.me'

export const textproList: Promise<TextProList[]> = (async () => got('https://raw.githubusercontent.com/BochilTeam/scraper/master/data/textpro.json').json<TextProList[]>())()
export default async function textpro (effect: string, params: string[] | string) {
  const list: TextProList[] = await textproList
  const textpro: TextProList | undefined = list.find(
    ({ title }) => title.toLowerCase() === effect.toLowerCase()
  )
  if (!textpro) throw new Error(`TextPro "${effect}" not found`)
  if (!Array.isArray(params)) params = [params]
  const { link, parameters } = textpro
  if (parameters.length > params.length) throw new Error(`TextPro "${effect}" requires ${parameters.length} parameters, but ${params.length} given`)
  const resToken = await got(`${BASE_URL}${link}`)
  const cookie = resToken.headers['set-cookie']?.map(c => c.split(';')[0]).join('; ').trim()
  const HEADERS = {
    cookie: cookie || '__gads=ID=63da40a14f3eb127-22dccf741fd10073:T=1648080134:RT=1648080134:S=ALNI_MZfEIreNTkduqqV5CgZnuei_X1xLQ; _ga=GA1.2.342524260.1648080135; _gid=GA1.2.2036288127.1648080139; PHPSESSID=7fmr2ig9k8r7n9g9uk7fcj2ru1; _gat_gtag_UA_114571019_5=1',
    'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.82 Safari/537.36'
  }
  const $ = load(resToken.body)
  const token = $('#token').val() as string
  const build_server = $('#build_server').val() as string
  const build_server_id = $('#build_server_id').val() as string
  const form = new FormData()
  for (const param of params) form.append('text[]', param)
  form.append('submit', 'Go')
  form.append('token', token)
  form.append('build_server', build_server)
  form.append('build_server_id', build_server_id)
  const html = await got(`${BASE_URL}${link}`, {
    method: 'POST',
    headers: {
      ...HEADERS,
      ...form.getHeaders()
    },
    body: form.getBuffer()
  }).text()
  const $$ = load(html)
  const form2 = $$('#form_value').eq(0).text()
  if (!form2) throw new ScraperError(`TextPro "${effect}" failed`, html)
  const json = await got.post(`${BASE_URL}/effect/create-image`, {
    headers: HEADERS,
    form: JSON.parse(form2)
  }).json<TextProPostJSON>()
  return `${BASE_URL}${json.image || json.fullsize_image}`
}