propagate live flag

robbiet480 · Jul 30, 2022 · b3e7c1e · b3e7c1e
1 parent 7a8266c
commit b3e7c1e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/internal/xmltv/xmltv.go b/internal/xmltv/xmltv.go
@@ -156,6 +156,7 @@ type Programme struct {
 	Premiere        *CommonElement   `xml:"premiere,omitempty"         json:"premiere,omitempty"`
 	LastChance      *CommonElement   `xml:"last-chance,omitempty"      json:"last_chance,omitempty"`
 	New             *ElementPresent  `xml:"new"            json:"new,omitempty"`
+	Live            *ElementPresent  `xml:"live"           json:"live,omitempty"`
 	Subtitles       []Subtitle       `xml:"subtitles,omitempty"        json:"subtitles,omitempty"`
 	Ratings         []Rating         `xml:"rating,omitempty"           json:"ratings,omitempty"`
 	StarRatings     []Rating         `xml:"star-rating,omitempty"      json:"star_ratings,omitempty"`

diff --git a/structs.go b/structs.go
@@ -180,12 +180,18 @@ type Program struct {
 func (p *Program) XMLTV() xmltv.Programme {
 	start := xmltv.Time(p.Start)
 	stop := xmltv.Time(p.Stop)
+	var live *xmltv.ElementPresent
+	if p.IsLive {
+		tmp := xmltv.ElementPresent(p.IsLive)
+		live = &tmp
+	}
 	return xmltv.Programme{
 		Titles:       []xmltv.CommonElement{p.Title},
 		Descriptions: []xmltv.CommonElement{p.Description},
 		Categories:   p.Categories,
 		Start:        &start,
 		Stop:         &stop,
+		Live:         live,
 		Channel:      fmt.Sprintf("stirr-%s", p.Channel),
 	}
 }