const fs = require('fs')
const glob = require('glob')
const path = require('path')
const convert = require('xml-js')
const axios = require('axios')
const countries = require('./countries.json')

async function main() {
  let codes = {}
  console.log('Starting...')
  const files = [
    'andorradifusio.ad',
    'astro.com.my',
    'comteco.com.bo',
    'cosmote.gr',
    'digiturk.com.tr',
    'elcinema.com',
    'guidatv.sky.it',
    'hd-plus.de',
    'm.tv.sms.cz',
    'maxtv.hrvatskitelekom.hr',
    'mediaset.it',
    'meo.pt',
    'mi.tv',
    'mncvision.id',
    'ontvtonight.com',
    'programacion-tv.elpais.com',
    'programetv.ro',
    'programme-tv.net',
    'programtv.onet.pl',
    'telkussa.fi',
    'tv.lv',
    'tv.yandex.ru',
    'tvgid.ua',
    'tvguide.com',
    'tvprofil.com',
    'tvtv.ca',
    'tvtv.us',
    'vidio.com',
    'znbc.co.zm'
  ]
  for (const filename of files) {
    const url = `https://iptv-org.github.io/epg/guides/${filename}.guide.xml`
    console.log(`Loading '${url}'...`)
    const file = await axios
      .get(url)
      .then(r => r.data)
      .catch(console.log)
    const channels = parseChannels(file)
    channels.forEach(channel => {
      if (!codes[channel.tvg_id + channel.display_name]) {
        codes[channel.tvg_id + channel.display_name] = channel
      }
    })
  }

  const sorted = Object.values(codes).sort((a, b) => {
    if (a.display_name.toLowerCase() < b.display_name.toLowerCase()) return -1
    if (a.display_name.toLowerCase() > b.display_name.toLowerCase()) return 1
    return 0
  })
  writeToFile('.gh-pages/codes.json', convertToJSON(sorted))

  const _items = {}
  countries.forEach(country => {
    _items[country.code] = {
      ...country,
      expanded: false,
      channels: []
    }
  })

  sorted.forEach(channel => {
    const item = _items[channel.country]
    if (item) {
      channel.hash = `${channel.display_name}:${channel.tvg_id}`.toLowerCase()
      item.channels.push(channel)
    }
  })
  writeToFile('.gh-pages/_items.json', convertToJSON(_items))

  console.log('Done')
}

function writeToFile(filename, data) {
  console.log(`Saving all codes to '${filename}'...`)
  const dir = path.resolve(path.dirname(filename))
  if (!fs.existsSync(dir)) {
    fs.mkdirSync(dir, { recursive: true })
  }

  fs.writeFileSync(path.resolve(filename), data)
}

function convertToJSON(arr) {
  return JSON.stringify(arr, null, 2)
}

function convertToCSV(arr) {
  let string = 'display_name,tvg_id,country,logo\n'
  for (const item of arr) {
    string += `${item.display_name},${item.tvg_id},${item.country},${item.logo}\n`
  }

  return string
}

function parseChannels(file) {
  const result = convert.xml2js(file)
  const tv = result.elements.find(el => el.name === 'tv')

  return tv.elements
    .filter(el => el.name === 'channel')
    .map(channel => {
      const tvg_id = (channel.attributes || { id: '' }).id
      const logo = (channel.elements.find(el => el.name === 'icon') || { attributes: { src: '' } })
        .attributes.src
      const displayName = channel.elements.find(el => el.name === 'display-name')
      const display_name = displayName
        ? displayName.elements.find(el => el.type === 'text').text
        : null
      const country = tvg_id.split('.')[1]

      return { tvg_id, logo, display_name, country }
    })
}

main()