import got from 'got'
import cheerio from 'cheerio'
import { Kbbi, KbbiArgsSchema, KbbiSchema } from './types.js'
import { ScraperError } from '../utils.js'
/**
 * p = Partikel: kelas kata yang meliputi kata depan, kata sambung, kata seru, kata sandang, ucapan salam
 *
 * n = Nomina: kata benda
 */
export default async function kbbi (words: string): Promise<Kbbi[]> {
  KbbiArgsSchema.parse(arguments)

  const html = await got(`https://kbbi.kemdikbud.go.id/entri/${encodeURIComponent(words)}`).text()
  const $ = cheerio.load(html)
  const isExist = !/tidak ditemukan/i.test(
    $('body > div.container.body-content > h4[style="color:red"]').text()
  )
  if (!isExist) throw new ScraperError(`${words} does not exist!`)
  const results: Kbbi[] = []
  let isContent = false; let lastTitle: string
  $('body > div.container.body-content').children().each((_, el) => {
    const tag = el.tagName
    const elem = $(el)
    if (tag === 'hr') isContent = !isContent && !Object.keys(results).length
    if (tag === 'h2' && isContent) {
      const index = elem.find('sup').text().trim()
      const title = elem.text().trim()
      results.push({
        index: parseInt(index),
        title,
        means: []
      })
      lastTitle = title
    }
    if ((tag === 'ol' || tag === 'ul') && isContent && lastTitle) {
      elem.find('li').each((_, el) => {
        const li = $(el).text().trim()
        const index = results.findIndex(
          ({ title }) => title === lastTitle
        )
        if (index !== -1) results[index].means.push(li)
        else console.log(li, lastTitle)
      })
      lastTitle = ''
    }
  })

  if (results.length === 0) throw new ScraperError(`${words} does not exist!\n\n${html}`)
  return results.map(res => KbbiSchema.parse(res))
}