Save categories while generating epg (#359)

JioTV-Go · Aug 4, 2024 · a521dac · a521dac
1 parent 59cded4
commit a521dac
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 9 deletions.
diff --git a/pkg/epg/epg.go b/pkg/epg/epg.go
@@ -80,7 +80,7 @@ func Init() {
 }
 
 // NewProgramme creates a new Programme with the given parameters.
-func NewProgramme(channelID int, start, stop, title, desc, iconSrc string) Programme {
+func NewProgramme(channelID int, start, stop, title, desc, category, iconSrc string) Programme {
 	iconURL := fmt.Sprintf("/jtvposter/%s", iconSrc)
 	return Programme{
 		Channel: fmt.Sprint(channelID),
@@ -94,6 +94,10 @@ func NewProgramme(channelID int, start, stop, title, desc, iconSrc string) Progr
 			Value: desc,
 			Lang:  "en",
 		},
+		Category: Category{
+			Value: category,
+			Lang:  "en",
+		},
 		Icon: Icon{
 			Src: iconURL,
 		},
@@ -139,7 +143,7 @@ func genXML() ([]byte, error) {
 			for _, programme := range epgResponse.EPG {
 				startTime := formatTime(time.UnixMilli(programme.StartEpoch))
 				endTime := formatTime(time.UnixMilli(programme.EndEpoch))
-				programmes = append(programmes, NewProgramme(channel.ID, startTime, endTime, programme.Title, programme.Description, programme.Poster))
+				programmes = append(programmes, NewProgramme(channel.ID, startTime, endTime, programme.Title, programme.Description, programme.ShowCategory, programme.Poster))
 			}
 		}
 		bar.Add(1)

diff --git a/pkg/epg/types.go b/pkg/epg/types.go
@@ -27,6 +27,14 @@ type Title struct {
 	Lang    string   `xml:"lang,attr"` // Language of the title
 }
 
+// Category XML tag for Programme XML tag in EPG
+// Category is the type of the programme or show being aired on the channel
+type Category struct {
+	XMLName xml.Name `xml:"category"`
+	Value   string   `xml:",chardata"` // Category of the programme
+	Lang    string   `xml:"lang,attr"` // Language of the category
+}
+
 // Desc represents Description XML tag for Programme XML tag in EPG
 type Desc struct {
 	XMLName xml.Name `xml:"desc"`
@@ -37,13 +45,14 @@ type Desc struct {
 // Programme XML tag structure for EPG
 // Each programme tag represents a show being aired on a channel
 type Programme struct {
-	XMLName xml.Name `xml:"programme"`    // XML tag name
-	Channel string   `xml:"channel,attr"` // Channel is attribute of programme tag
-	Start   string   `xml:"start,attr"`   // Start time of the programme
-	Stop    string   `xml:"stop,attr"`    // Stop time of the programme
-	Title   Title    `xml:"title"`        // Title of the programme
-	Desc    Desc     `xml:"desc"`         // Description of the programme
-	Icon    Icon     `xml:"icon"`         // Icon of the programme
+	XMLName  xml.Name `xml:"programme"`    // XML tag name
+	Channel  string   `xml:"channel,attr"` // Channel is attribute of programme tag
+	Start    string   `xml:"start,attr"`   // Start time of the programme
+	Stop     string   `xml:"stop,attr"`    // Stop time of the programme
+	Title    Title    `xml:"title"`        // Title of the programme
+	Desc     Desc     `xml:"desc"`         // Description of the programme
+	Category Category `xml:"category"`     // Category of the programme
+	Icon     Icon     `xml:"icon"`         // Icon of the programme
 }
 
 // EPG XML tag structure