iptv/scripts/generate.js

const db = require('./db')
const utils = require('./utils')

const ROOT_DIR = './.gh-pages'

db.load()

function main() {
  createRootDirectory()
  createNoJekyllFile()
  generateIndex()
  generateSFWIndex()
  generateChannelsJson()
  generateCountryIndex()
  generateLanguageIndex()
  generateCategoryIndex()
  generateCategories()
  generateLanguages()
  generateCountries()
  finish()
}

function createRootDirectory() {
  console.log('Creating .gh-pages folder...')
  utils.createDir(ROOT_DIR)
}

function createNoJekyllFile() {
  console.log('Creating .nojekyll...')
  utils.createFile(`${ROOT_DIR}/.nojekyll`)
}

function generateIndex() {
  console.log('Generating index.m3u...')
  const filename = `${ROOT_DIR}/index.m3u`
  utils.createFile(filename, '#EXTM3U\n')

  const buffer = []
  const channels = db.channels.sortBy(['name', 'url']).all()
  for (const channel of channels) {
    const info = channel.getInfo()
    if (!buffer.includes(info)) {
      utils.appendToFile(filename, channel.toString())
      if (channel.filename !== 'unsorted') {
        buffer.push(info)
      }
    }
  }
}

function generateSFWIndex() {
  console.log('Generating index.sfw.m3u...')
  const filename = `${ROOT_DIR}/index.sfw.m3u`
  utils.createFile(filename, '#EXTM3U\n')

  const buffer = []
  const channels = db.channels.sortBy(['name', 'url']).sfw()
  for (const channel of channels) {
    const info = channel.getInfo()
    if (!buffer.includes(info)) {
      utils.appendToFile(filename, channel.toString())
      if (channel.filename !== 'unsorted') {
        buffer.push(info)
      }
    }
  }
}

function generateChannelsJson() {
  console.log('Generating channels.json...')
  const filename = `${ROOT_DIR}/channels.json`
  const channels = db.channels
    .sortBy(['name', 'url'])
    .all()
    .map(c => c.toObject())
  utils.createFile(filename, JSON.stringify(channels))
}

function generateCountryIndex() {
  console.log('Generating index.country.m3u...')
  const filename = `${ROOT_DIR}/index.country.m3u`
  utils.createFile(filename, '#EXTM3U\n')

  const unsorted = db.playlists.only(['unsorted'])[0]
  for (const channel of unsorted.channels) {
    const category = channel.category
    channel.category = ''
    utils.appendToFile(filename, channel.toString())
    channel.category = category
  }

  const buffer = []
  const playlists = db.playlists.sortBy(['country']).except(['unsorted'])
  for (const playlist of playlists) {
    for (const channel of playlist.channels) {
      const category = channel.category
      channel.category = playlist.country
      const info = channel.getInfo()
      if (!buffer.includes(info)) {
        utils.appendToFile(filename, channel.toString())
        buffer.push(info)
      }
      channel.category = category
    }
  }
}

function generateLanguageIndex() {
  console.log('Generating index.language.m3u...')
  const filename = `${ROOT_DIR}/index.language.m3u`
  utils.createFile(filename, '#EXTM3U\n')

  const buffer = []
  const channels = db.channels.sortBy(['name', 'url']).forLanguage({ code: null }).get()
  for (const channel of channels) {
    const category = channel.category
    channel.category = ''
    const info = channel.getInfo()
    if (!buffer.includes(info)) {
      utils.appendToFile(filename, channel.toString())
      if (channel.filename !== 'unsorted') {
        buffer.push(info)
      }
    }
    channel.category = category
  }

  const languages = db.languages.sortBy(['name']).all()
  for (const language of languages) {
    const buffer = []
    const channels = db.channels.sortBy(['name', 'url']).forLanguage(language).get()
    for (const channel of channels) {
      const category = channel.category
      channel.category = language.name
      const info = channel.getInfo()
      if (!buffer.includes(info)) {
        utils.appendToFile(filename, channel.toString())
        buffer.push(info)
      }
      channel.category = category
    }
  }
}

function generateCategoryIndex() {
  console.log('Generating index.category.m3u...')
  const filename = `${ROOT_DIR}/index.category.m3u`
  utils.createFile(filename, '#EXTM3U\n')

  const buffer = []
  const channels = db.channels.sortBy(['category', 'name', 'url']).all()
  for (const channel of channels) {
    const info = channel.getInfo()
    if (!buffer.includes(info)) {
      utils.appendToFile(filename, channel.toString())
      if (channel.filename !== 'unsorted') {
        buffer.push(info)
      }
    }
  }
}

function generateCategories() {
  console.log(`Generating /categories...`)
  const outputDir = `${ROOT_DIR}/categories`
  utils.createDir(outputDir)

  for (const category of db.categories.all()) {
    const filename = `${outputDir}/${category.id}.m3u`
    utils.createFile(filename, '#EXTM3U\n')

    const buffer = []
    const channels = db.channels.sortBy(['name', 'url']).forCategory(category).get()
    for (const channel of channels) {
      const info = channel.getInfo()
      if (!buffer.includes(info)) {
        utils.appendToFile(filename, channel.toString())
        if (channel.filename !== 'unsorted') {
          buffer.push(info)
        }
      }
    }
  }

  const buffer = []
  const other = `${outputDir}/other.m3u`
  const channels = db.channels.sortBy(['name', 'url']).forCategory({ id: null }).get()
  utils.createFile(other, '#EXTM3U\n')
  for (const channel of channels) {
    const info = channel.getInfo()
    if (!buffer.includes(info)) {
      utils.appendToFile(other, channel.toString())
      if (channel.filename !== 'unsorted') {
        buffer.push(info)
      }
    }
  }
}

function generateCountries() {
  console.log(`Generating /countries...`)
  const outputDir = `${ROOT_DIR}/countries`
  utils.createDir(outputDir)

  for (const country of db.countries.all()) {
    const filename = `${outputDir}/${country.code}.m3u`
    utils.createFile(filename, '#EXTM3U\n')

    const buffer = []
    const channels = db.channels.sortBy(['name', 'url']).forCountry(country).get()
    for (const channel of channels) {
      const info = channel.getInfo()
      if (!buffer.includes(info)) {
        utils.appendToFile(filename, channel.toString())
        if (channel.filename !== 'unsorted') {
          buffer.push(info)
        }
      }
    }
  }

  const buffer = []
  const other = `${outputDir}/undefined.m3u`
  const channels = db.channels.sortBy(['name', 'url']).forCountry({ code: null }).get()
  utils.createFile(other, '#EXTM3U\n')
  for (const channel of channels) {
    const info = channel.getInfo()
    if (!buffer.includes(info)) {
      utils.appendToFile(other, channel.toString())
      if (channel.filename !== 'unsorted') {
        buffer.push(info)
      }
    }
  }
}

function generateLanguages() {
  console.log(`Generating /languages...`)
  const outputDir = `${ROOT_DIR}/languages`
  utils.createDir(outputDir)

  for (const language of db.languages.all()) {
    const filename = `${outputDir}/${language.code}.m3u`
    utils.createFile(filename, '#EXTM3U\n')

    const buffer = []
    const channels = db.channels.sortBy(['name', 'url']).forLanguage(language).get()
    for (const channel of channels) {
      const info = channel.getInfo()
      if (!buffer.includes(info)) {
        utils.appendToFile(filename, channel.toString())
        if (channel.filename !== 'unsorted') {
          buffer.push(info)
        }
      }
    }
  }

  const buffer = []
  const other = `${outputDir}/undefined.m3u`
  const channels = db.channels.sortBy(['name', 'url']).forLanguage({ code: null }).get()
  utils.createFile(other, '#EXTM3U\n')
  for (const channel of channels) {
    const info = channel.getInfo()
    if (!buffer.includes(info)) {
      utils.appendToFile(other, channel.toString())
      if (channel.filename !== 'unsorted') {
        buffer.push(info)
      }
    }
  }
}

function finish() {
  console.log(
    `\nTotal: ${db.channels.count()} channels, ${db.countries.count()} countries, ${db.languages.count()} languages, ${db.categories.count()} categories.`
  )
}

main()