From af4ef9bf6223bd47b728d3ff5ceace73bd41bc0b Mon Sep 17 00:00:00 2001 From: tombowditch Date: Wed, 31 Jan 2018 19:58:35 +0000 Subject: [PATCH] Initial commit to git --- .gitignore | 4 + README.md | 7 ++ main.go | 224 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 235 insertions(+) create mode 100644 .gitignore create mode 100644 README.md create mode 100644 main.go diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..40dbb52 --- /dev/null +++ b/.gitignore @@ -0,0 +1,4 @@ +*.m3u +build/* +telly + diff --git a/README.md b/README.md new file mode 100644 index 0000000..3634cce --- /dev/null +++ b/README.md @@ -0,0 +1,7 @@ +# telly + +IPTV proxy for Plex Live written in Golang + +# Setup + +Coming soon diff --git a/main.go b/main.go new file mode 100644 index 0000000..96c5440 --- /dev/null +++ b/main.go @@ -0,0 +1,224 @@ +package main + +import ( + "encoding/json" + "flag" + "fmt" + "github.com/jamesnetherton/m3u" + "net/http" + "regexp" + "strconv" + "strings" +) + +var usedTracks []m3u.Track +var deviceXml string +var useRegex *bool +var filterUkTv *bool +var m3uFile *string +var listenAddress *string +var logRequests *bool +var concurrentStreams *int + +type DiscoveryData struct { + FriendlyName string + Manufacturer string + ModelNumber string + FirmwareName string + TunerCount int + FirmwareVersion string + DeviceID string + DeviceAuth string + BaseURL string + LineupURL string +} + +type LineupStatus struct { + ScanInProgress int + ScanPossible int + Source string + SourceList []string +} + +type LineupItem struct { + GuideNumber string + GuideName string + URL string +} + +func init() { + useRegex = flag.Bool("useregex", false, "Use regex to attempt to strip out bogus channels (SxxExx, 24/7 channels, etc") + filterUkTv = flag.Bool("uktv", false, "Only index channels with 'UK' in the name") + listenAddress = flag.String("listen", "localhost:6077", "IP:Port to listen on") + m3uFile = flag.String("file", "iptv.m3u", "Location of m3u file") + logRequests = flag.Bool("logrequests", false, "Log any requests to telly") + concurrentStreams = flag.Int("streams", 1, "Amount of concurrent streams allowed") + flag.Parse() +} + +func logRequest(r string) { + if *logRequests { + fmt.Println("[telly] [request]", r) + } +} + +func main() { + usedTracks := make([]m3u.Track, 0) + + if *m3uFile == "iptv.m3u" { + fmt.Println("[telly] [warning] using default m3u option, 'iptv.m3u'. launch telly with the -file=yourfile.m3u option to change this!") + } + + fmt.Println("[telly] [parser] Reading m3u file", *m3uFile, "...") + playlist, err := m3u.Parse(*m3uFile) + if err != nil { + fmt.Println("[telly] [error] unable to read m3u file, error below") + fmt.Println("[telly] [error] m3u files need to have specific formats, see the github page for more information") + fmt.Println("[telly] [error] future versions of telly will attempt to parse this better") + panic(err) + } + + episodeRegex, _ := regexp.Compile("S\\d{1,3}E\\d{1,3}") + twentyFourSevenRegex, _ := regexp.Compile("24\\/7") + ukTv, _ := regexp.Compile("UK") + + showNameRegex, _ := regexp.Compile("tvg-name=\"(.*)\" tvg") + + for _, track := range playlist.Tracks { + if *useRegex && *filterUkTv { + if !episodeRegex.MatchString(track.Name) { + if !twentyFourSevenRegex.MatchString(track.Name) { + if ukTv.MatchString(track.Name) { + usedTracks = append(usedTracks, track) + } + } + } + } else if *useRegex && !*filterUkTv { + if !episodeRegex.MatchString(track.Name) { + if !twentyFourSevenRegex.MatchString(track.Name) { + usedTracks = append(usedTracks, track) + } + } + + } else if !*useRegex && *filterUkTv { + if ukTv.MatchString(track.Name) { + usedTracks = append(usedTracks, track) + } + } else { + usedTracks = append(usedTracks, track) + } + } + + if !*useRegex { + fmt.Println("[telly] [warning] telly is not attempting to strip out unneeded channels, please use the flag -useregex if telly returns too many channels") + } + + if !*filterUkTv { + fmt.Println("[telly] [info] telly is currently not filtering for only uk television. if you would like it to, please use the flag -uktv") + } + + fmt.Println("[telly] [info] found", len(usedTracks), "channels") + + fmt.Println("") + + fmt.Println("[telly] [info] creating discovery data") + discoveryData := DiscoveryData{ + FriendlyName: "telly", + Manufacturer: "Silicondust", + ModelNumber: "HDHR-2US", + FirmwareName: "hdhomeruntc_atsc", + TunerCount: *concurrentStreams, + FirmwareVersion: "20150826", + DeviceID: "12345678", + DeviceAuth: "telly123", + BaseURL: fmt.Sprintf("http://%s", *listenAddress), + LineupURL: fmt.Sprintf("http://%s/lineup.json", *listenAddress), + } + fmt.Println("[telly] [info] creating lineup status") + lineupStatus := LineupStatus{ + ScanInProgress: 0, + ScanPossible: 1, + Source: "Cable", + SourceList: []string{"Cable"}, + } + + fmt.Println("[telly] [info] creating device xml") + deviceXml = ` + + 1 + 0 + + $BaseURL + + urn:schemas-upnp-org:device:MediaServer:1 + $FriendlyName + $Manufacturer + $ModelNumber + $ModelNumber + + uuid:$DeviceID + +` + + deviceXml = strings.Replace(deviceXml, "$BaseURL", discoveryData.BaseURL, -1) + deviceXml = strings.Replace(deviceXml, "$FriendlyName", discoveryData.FriendlyName, -1) + deviceXml = strings.Replace(deviceXml, "$Manufacturer", discoveryData.Manufacturer, -1) + deviceXml = strings.Replace(deviceXml, "$ModelNumber", discoveryData.ModelNumber, -1) + deviceXml = strings.Replace(deviceXml, "$DeviceID", discoveryData.DeviceID, -1) + + fmt.Println("[telly] [info] creating webserver routes") + + http.HandleFunc("/discover.json", func(w http.ResponseWriter, r *http.Request) { + logRequest("/discover.json") + json.NewEncoder(w).Encode(discoveryData) + }) + + http.HandleFunc("/lineup_status.json", func(w http.ResponseWriter, r *http.Request) { + logRequest("/lineup_status.json") + json.NewEncoder(w).Encode(lineupStatus) + }) + + http.HandleFunc("/lineup.post", func(w http.ResponseWriter, r *http.Request) { + logRequest("/lineup.post") + }) + + http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { + logRequest("/") + w.Header().Set("Content-Type", "application/xml") + w.Write([]byte(deviceXml)) + }) + + http.HandleFunc("/device.xml", func(w http.ResponseWriter, r *http.Request) { + logRequest("/device.xml") + w.Header().Set("Content-Type", "application/xml") + w.Write([]byte(deviceXml)) + }) + + http.HandleFunc("/lineup.json", func(w http.ResponseWriter, r *http.Request) { + logRequest("/lineup.json") + lineup := make([]LineupItem, 0) + gn := 10000 + for _, track := range usedTracks { + + parsedTrack := showNameRegex.FindStringSubmatch(track.Name) + finalName := parsedTrack[0] + finalName = strings.Replace(finalName, "tvg-name=\"", "", -1) + finalName = strings.Replace(finalName, "\" tvg", "", -1) + lu := LineupItem{ + GuideNumber: strconv.Itoa(gn), + GuideName: finalName, + URL: track.URI, + } + + lineup = append(lineup, lu) + + gn = gn + 1 + } + + json.NewEncoder(w).Encode(lineup) + + }) + + fmt.Println("[telly] [info] listening on", *listenAddress) + http.ListenAndServe(*listenAddress, nil) +}