iptv-org.github.io/src/store.js

import { writable, get } from 'svelte/store'
import { transliterate } from 'transliteration'
import _ from 'lodash'

export const query = writable('')
export const hasQuery = writable(false)
export const channels = writable([])
export const countries = writable({})
export const filteredChannels = writable([])

export function search(_query) {
  const parts = _query.toLowerCase().match(/(".*?"|[^"\s]+)+(?=\s*|\s*$)/g) || []
  const filters = []
  for (let value of parts) {
    let field = '_key'
    if (value.includes(':')) {
      ;[field, value] = value.split(':')
    }
    value = value.replace(/\"/g, '')

    if (field && value) {
      let numerical = ['streams', 'guides'].includes(field)
      filters.push({ field, numerical, value })
    }
  }

  if (!filters.length) {
    hasQuery.set(false)
    filteredChannels.set(get(channels))
    return
  }

  const filtered = get(channels).filter(c => {
    let results = []
    for (let f of filters) {
      if (!f.value) return false

      if (f.numerical) {
        if (f.value.startsWith('<')) {
          results.push(c._searchable[f.field] < parseInt(f.value.replace('<', '')))
        } else if (f.value.startsWith('>')) {
          results.push(c._searchable[f.field] > parseInt(f.value.replace('>', '')))
        } else {
          results.push(c._searchable[f.field] === parseInt(f.value))
        }
      } else {
        const regex = new RegExp(f.value.replaceAll(',', '|'), 'i')
        // console.log(regex, c._searchable[f.field])
        results.push(regex.test(c._searchable[f.field]))
      }
    }

    return results.every(Boolean)
  })

  filteredChannels.set(filtered)

  hasQuery.set(true)

  console.log('.')
}

export async function fetchChannels() {
  const api = await loadAPI()

  countries.set(api.countries)

  let _channels = api.channels.map(c => {
    c._raw = copy(c)
    c._streams = api.streams[c.id] || []
    c._guides = api.guides[c.id] || []
    c._country = api.countries[c.country]
    c._subdivision = api.subdivisions[c.subdivision]
    c._languages = c.languages.map(code => api.languages[code]).filter(i => i)
    c._categories = c.categories.map(id => api.categories[id]).filter(i => i)
    c._broadcast_area = c.broadcast_area.map(value => {
      const [type, code] = value.split('/')
      switch (type) {
        case 'c':
          return { type, ...api.countries[code] }
        case 'r':
          return { type, ...api.regions[code] }
        case 's':
          return { type, ...api.subdivisions[code] }
      }
    })
    c._searchable = generateSearchable(c)

    return c
  })

  channels.set(_channels)
  filteredChannels.set(_channels)
}

function generateSearchable(c) {
  const searchable = {}
  for (let key in c) {
    if (key.startsWith('_')) continue
    if (Array.isArray(c[key])) {
      searchable[key] = c[key].map(v => v.toString().toLowerCase()).join(',')
    } else {
      searchable[key] =
        c[key] !== undefined && c[key] !== null ? c[key].toString().toLowerCase() : ''
    }
  }
  searchable.streams = c._streams.length
  searchable.guides = c._guides.length
  searchable.is = c.closed || c.replaced_by ? 'closed' : 'active'
  searchable._key = generateKey(c)

  return searchable
}

function generateKey(c) {
  const data = Object.values(
    _.pick(c, [
      'id',
      'name',
      'alt_names',
      'network',
      'country',
      'subdivision',
      'city',
      'broadcast_area',
      'languages',
      'categories',
      'launched',
      'closed',
      'replaced_by'
    ])
  )
  const translit = c.alt_names ? transliterate(c.alt_names) : null

  return [...data, translit]
    .map(v => v || '')
    .filter(v => v)
    .join('|')
    .toLowerCase()
}

function copy(value) {
  return JSON.parse(JSON.stringify(value))
}

export function setSearchParam(key, value) {
  if (window.history.pushState) {
    let query = key && value ? `?${key}=${value}` : ''
    query = query.replace(/\+/g, '%2B')
    const url = `${window.location.protocol}//${window.location.host}${window.location.pathname}${query}`
    const state = {}
    state[key] = value
    window.history.pushState(state, '', url)
    setPageTitle(value)
  }
}

export function setPageTitle(value) {
  const title = value ? `${value} · iptv-org` : 'iptv-org'
  document.title = title
}

async function loadAPI() {
  const api = {}

  api.countries = await fetch('https://iptv-org.github.io/api/countries.json')
    .then(r => r.json())
    .then(data => (data.length ? data : []))
    .then(data =>
      data.map(i => {
        i.expanded = false
        return i
      })
    )
    .then(data => _.keyBy(data, 'code'))
    .catch(console.error)

  api.regions = await fetch('https://iptv-org.github.io/api/regions.json')
    .then(r => r.json())
    .then(data => (data.length ? data : []))
    .then(data => _.keyBy(data, 'code'))
    .catch(console.error)

  api.subdivisions = await fetch('https://iptv-org.github.io/api/subdivisions.json')
    .then(r => r.json())
    .then(data => (data.length ? data : []))
    .then(data => _.keyBy(data, 'code'))
    .catch(console.error)

  api.languages = await fetch('https://iptv-org.github.io/api/languages.json')
    .then(r => r.json())
    .then(data => (data.length ? data : []))
    .then(data => _.keyBy(data, 'code'))
    .catch(console.error)

  api.categories = await fetch('https://iptv-org.github.io/api/categories.json')
    .then(r => r.json())
    .then(data => (data.length ? data : []))
    .then(data => _.keyBy(data, 'id'))
    .catch(console.error)

  api.streams = await fetch('https://iptv-org.github.io/api/streams.json')
    .then(r => r.json())
    .then(data => (data.length ? data : []))
    .then(data => _.groupBy(data, 'channel'))
    .catch(console.error)

  api.guides = await fetch('https://iptv-org.github.io/api/guides.json')
    .then(r => r.json())
    .then(data => (data.length ? data : []))
    .then(data => _.groupBy(data, 'channel'))
    .catch(console.error)

  api.channels = await fetch('https://iptv-org.github.io/api/channels.json')
    .then(r => r.json())
    .catch(err => {
      console.error(err)
      return []
    })

  return api
}