From 9d647e8c68140472e024b3ff84bd4587424f2bc0 Mon Sep 17 00:00:00 2001
From: freearhey <free.arhey@gmail.com>
Date: Sat, 2 Nov 2019 17:30:12 +0300
Subject: [PATCH] Added support tvg-url attribute

---
 scripts/generate.js |  1 +
 scripts/helper.js   | 27 ++++++++++++++-------------
 2 files changed, 15 insertions(+), 13 deletions(-)

diff --git a/scripts/generate.js b/scripts/generate.js
index 8cd12d361b..26902d6a0a 100644
--- a/scripts/generate.js
+++ b/scripts/generate.js
@@ -55,6 +55,7 @@ function parseIndex() {
       const channel = helper.createChannel(item)
       channel.countryCode = countryCode
       channel.countryName = countryName
+      channel.epg = playlist.header.attrs['x-tvg-url'] || ''
 
       // all
       list.all.push(channel)
diff --git a/scripts/helper.js b/scripts/helper.js
index 5ef3a80941..c175340797 100644
--- a/scripts/helper.js
+++ b/scripts/helper.js
@@ -51,18 +51,6 @@ helper.parsePlaylist = function(filename) {
   return new Playlist(result)
 }
 
-helper.createChannel = function(data) {
-  return new Channel({
-    id: data.tvg.id,
-    name: data.tvg.name,
-    language: data.tvg.language,
-    logo: data.tvg.logo,
-    group: data.group.title,
-    url: data.url,
-    title: data.name
-  })
-}
-
 helper.loadEPG = async function(url) {
   const content = await getEPGFile(url)
   const result = epgParser.parse(content)
@@ -202,6 +190,18 @@ helper.generateTable = function(data, options) {
   return output
 }
 
+helper.createChannel = function(data) {
+  return new Channel({
+    id: data.tvg.id,
+    name: data.tvg.name,
+    language: data.tvg.language,
+    logo: data.tvg.logo,
+    group: data.group.title,
+    url: data.url,
+    title: data.name
+  })
+}
+
 class Playlist {
   constructor(data) {
     this.header = data.header
@@ -257,7 +257,8 @@ class Channel {
 
   toString() {
     const country = this.countryCode.toUpperCase()
-    const info = `-1 tvg-id="${this.id}" tvg-name="${this.name}" tvg-language="${this.language}" tvg-logo="${this.logo}" tvg-country="${country}" group-title="${this.group}",${this.title}`
+    const epg = (this.id && this.epg) ? this.epg : ''
+    const info = `-1 tvg-id="${this.id}" tvg-name="${this.name}" tvg-language="${this.language}" tvg-logo="${this.logo}" tvg-country="${country}" tvg-url="${epg}" group-title="${this.group}",${this.title}`
 
     return '#EXTINF:' + info + '\n' + this.url + '\n'
   }