Update db.js

scripts/helpers/db.js

@@ -2,15 +2,22 @@ const categories = require('../data/categories')
 const parser = require('./parser')
 const utils = require('./utils')
 const file = require('./file')
+const epg = require('./epg')
 
 const db = {}
 
 db.load = async function () {
-  let files = await file.list()
+  const files = await file.list()
+  const codes = await epg.codes.load()
   for (const file of files) {
     const playlist = await parser.parsePlaylist(file)
-    db.playlists.add(playlist)
+    let guides = []
     for (const channel of playlist.channels) {
+      const code = codes.find(ch => ch['tvg_id'] === channel.tvg.id)
+      if (code && Array.isArray(code.guides)) {
+        guides = [...guides, ...code.guides]
+      }
+
       db.channels.add(channel)
 
       for (const country of channel.countries) {
@@ -25,6 +32,10 @@ db.load = async function () {
         }
       }
     }
+
+    if (guides.length) playlist.header.attrs['url-tvg'] = guides.join(',')
+
+    db.playlists.add(playlist)
   }
 }