import got from 'got'
import cheerio from 'cheerio'
import {
  Gempa,
  GempaNow,
  GempaNowSchema,
  GempaSchema,
  Tsunami,
  TsunamiSchema
} from './types.js'
import { ScraperError } from '../utils.js'

export async function gempa (): Promise<Gempa[]> {
  const html = await got('https://www.bmkg.go.id/gempabumi/gempabumi-dirasakan.bmkg').text()
  const $ = cheerio.load(html)
  const results: Gempa[] = []
  $('div.table-responsive > table.table > tbody > tr').each(function () {
    const el = $(this).find('td')
    const date = el.eq(1).text().trim()
    const locate = el.eq(2).text().trim()
    const magnitude = el.eq(3).text().trim()
    const depth = el.eq(4).text().trim()
    const location = el.eq(5).find('a').text().trim()
    const warning = el.eq(5).find('span.label').map(function () {
      return $(this).text().trim()
    }).toArray()
    results.push({
      date,
      locate,
      magnitude,
      depth,
      location,
      warning
    })
  })
  if (results.length === 0) throw new ScraperError(`Gempa not found\n\n${html}`)
  return results.map(res => GempaSchema.parse(res))
}

export async function gempaNow (): Promise<GempaNow[]> {
  const html = await got('https://www.bmkg.go.id/gempabumi/gempabumi-terkini.bmkg').text()
  const $ = cheerio.load(html)
  const results: GempaNow[] = []
  $('div.table-responsive > table.table > tbody > tr').each(function () {
    const el = $(this).find('td')
    const date = el.eq(1).text().trim()
    const latitude = el.eq(2).text().trim()
    const longitude = el.eq(3).text().trim()
    const magnitude = el.eq(4).text().trim()
    const depth = el.eq(5).text().trim()
    const location = el.eq(6).text().trim()
    results.push({
      date,
      latitude,
      longitude,
      magnitude,
      depth,
      location
    })
  })
  if (results.length === 0) throw new ScraperError(`Gempa Now not found\n\n${html}`)
  return results.map(res => GempaNowSchema.parse(res))
}

export async function tsunami (): Promise<Tsunami[]> {
  const html = await got('https://www.bmkg.go.id/tsunami/').text()
  const $ = cheerio.load(html)
  const results: Tsunami[] = []
  $('div.row > div > table.table > tbody > tr').each(function () {
    const el = $(this).find('td')
    const date = el.eq(0).text().trim()
    const locate = el.eq(1).text().trim()
    const magnitude = el.eq(2).text().trim()
    const depth = el.eq(3).text().trim()
    const location = el.eq(4).text().trim()
    results.push({
      date,
      locate,
      magnitude,
      depth,
      location
    })
  })
  if (results.length === 0) throw new ScraperError(`Tsunami not found\n\n${html}`)
  return results.map(res => TsunamiSchema.parse(res))
}