const fs = require('fs')
const glob = require('glob')
const path = require('path')
const convert = require('xml-js')
const axios = require('axios')

async function main() {
  const iptvChannels = await axios.get('https://iptv-org.github.io/iptv/channels.json').then(response => response.data).catch(console.log)

  let codes = {}
  console.log('Starting...')
  glob('sites/**/*.xml', null, function (er, files) {
    files.forEach(filename => {
      const channels = parseChannels(filename)
      channels.forEach(channel => {
        if (!codes[channel.xmltv_id + channel.name]) {
          let logo = channel.logo || ''
          if(!logo) {
            let ch = iptvChannels.find(ch => ch.tvg.id === channel.xmltv_id)
            if(ch && ch.logo) logo = ch.logo
          }

          codes[channel.xmltv_id + channel.name] = {
            display_name: channel.name,
            tvg_id: channel.xmltv_id,
            country: channel.xmltv_id.split('.')[1],
            logo,
          }
        }
      })
    })

    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('codes.csv', convertToCSV(sorted))
    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 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(filename) {
  if (!filename) throw new Error('Path to [site].channels.xml is missing')
  console.log(`Loading '${filename}'...`)

  const xml = fs.readFileSync(path.resolve(filename), { encoding: 'utf-8' })
  const result = convert.xml2js(xml)
  const site = result.elements.find(el => el.name === 'site')
  const channels = site.elements.find(el => el.name === 'channels')

  return channels.elements
    .filter(el => el.name === 'channel')
    .map(el => {
      const channel = el.attributes
      channel.name = el.elements.find(el => el.type === 'text').text

      return channel
    })
}

main()