Update tvarenasport.hr.config.js

Move code from tvarenasport.com/tvarenasport.com.config.js

Author: freearhey

sites/tvarenasport.hr/tvarenasport.hr.config.js

@@ -1,8 +1,132 @@
-const tvarenasport = require('../tvarenasport.com/tvarenasport.com.config')
+const cheerio = require('cheerio')
+const axios = require('axios')
+const dayjs = require('dayjs')
+const utc = require('dayjs/plugin/utc')
+const timezone = require('dayjs/plugin/timezone')
+const customParseFormat = require('dayjs/plugin/customParseFormat')
 
-module.exports = Object.assign(tvarenasport, {
+dayjs.extend(utc)
+dayjs.extend(timezone)
+dayjs.extend(customParseFormat)
+
+module.exports = {
   site: 'tvarenasport.hr',
   tz: 'Europe/Budapest',
   lang: 'hr',
-  url: 'https://tvarenaprogram.com/live/v2/hr', 
-})
+  url: 'https://tvarenaprogram.com/live/v2/hr',
+  days: 2,
+  request: {
+    cache: {
+      ttl: 24 * 60 * 60 * 1000 // 1 day
+    }
+  },
+  parser({ content, channel, date }) {
+    const programs = []
+    const expectedDate = date.format('YYYY-MM-DD')
+    if (content) {
+      const dates = []
+      const $ = cheerio.load(content)
+      const parent = $(
+        `.tv-scheme-chanel-header img[src*="chanel-${channel.site_id}.png"]`
+      ).parents('div')
+      parent
+        .siblings('.tv-scheme-days')
+        .find('a')
+        .toArray()
+        .forEach(el => {
+          const a = $(el)
+          const dt = a.find('span:nth-child(3)').text()
+          dates.push(dayjs(dt + date.year(), 'DD.MM.YYYY'))
+        })
+      parent
+        .siblings('.tv-scheme-new-slider-wrapper')
+        .find('.tv-scheme-new-slider-item')
+        .toArray()
+        .forEach((el, i) => {
+          programs.push(...parseSchedules($(el), dates[i], module.exports.tz))
+        })
+      programs.forEach((s, i) => {
+        if (i < programs.length - 2) {
+          s.stop = programs[i + 1].start
+        } else {
+          s.stop = s.start.startOf('d').add(1, 'd')
+        }
+      })
+    }
+
+    return programs.filter(
+      p =>
+        p.start.format('YYYY-MM-DD') === expectedDate ||
+        p.stop.format('YYYY-MM-DD') === expectedDate
+    )
+  },
+  async channels() {
+    const channels = []
+    const data = await axios
+      .get(this.url)
+      .then(r => r.data)
+      .catch(console.error)
+
+    if (data) {
+      // channel naming rule
+      const names = id => {
+        let match = id.match(/^\d+$/)
+        if (match) {
+          return `Arena Sport ${parseInt(id)}`
+        }
+        match = id.match(/^\d/)
+        if (match) {
+          return `Arena Sport ${id}`
+        }
+        match = id.match(/^a(\d+)(p)?/)
+        if (match) {
+          return `Arena ${parseInt(match[1])}${match[2] === 'p' ? ' Premium' : ''}`
+        }
+        return `Arena ${id}`
+      }
+      const $ = cheerio.load(data)
+      const items = $('.tv-scheme-chanel-header img').toArray()
+      for (const item of items) {
+        const [, id] = $(item)
+          .attr('src')
+          .match(/chanel-([a-z0-9]+)\.png/) || [null, null]
+        if (id) {
+          channels.push({
+            lang: this.lang,
+            site_id: id,
+            name: names(id)
+          })
+        }
+      }
+    }
+
+    return channels
+  }
+}
+
+function parseSchedules($s, date, tz) {
+  const schedules = []
+  const $ = $s._make
+  $s.find('.slider-content')
+    .toArray()
+    .forEach(el => {
+      schedules.push(parseSchedule($(el), date, tz))
+    })
+
+  return schedules
+}
+
+function parseSchedule($s, date, tz) {
+  const time = $s.find('.slider-content-top span').text()
+  const start = dayjs.tz(`${date.format('YYYY-MM-DD')} ${time}`, 'YYYY-MM-DD HH:mm', tz)
+  const category = $s.find('.slider-content-middle span').text()
+  const title = $s.find('.slider-content-bottom p').text()
+  const description = $s.find('.slider-content-bottom span:first').text()
+
+  return {
+    title: description ? description : title,
+    description: description ? title : description,
+    category,
+    start
+  }
+}