Add a unique id based on the Uri (CUID) (#12)

Add a unique id based on the Uri (CUID)
hoshsadiq · Jan 13, 2020 · fcc5752 · fcc5752
2 parents d8847c2 + 1de5c05
commit fcc5752
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/writer/csv.go b/writer/csv.go
@@ -9,7 +9,6 @@ import (
 var csvHeaders = []string{
 	"tvg-id",
 	"group-title",
-	"group-title",
 	"tvg-name",
 	"duration",
 	"tvg-logo",

diff --git a/writer/m3u.go b/writer/m3u.go
@@ -1,6 +1,8 @@
 package writer
 
 import (
+	"crypto/md5"
+	"encoding/hex"
 	"github.com/hoshsadiq/m3ufilter/m3u"
 	"io"
 	"strconv"
@@ -22,6 +24,12 @@ func writeM3U(w io.Writer, streams []*m3u.Stream) {
 	}
 }
 
+func GetMD5Hash(text string) string {
+	hasher := md5.New()
+	hasher.Write([]byte(text))
+	return hex.EncodeToString(hasher.Sum(nil))
+}
+
 func getStreamExtinf(stream *m3u.Stream) []byte {
 	b := &strings.Builder{}
 	b.WriteString("\n")
@@ -32,6 +40,7 @@ func getStreamExtinf(stream *m3u.Stream) []byte {
 		writeKV(b, "tvg-chno", stream.ChNo)
 	}
 
+	writeKV(b, "CUID", GetMD5Hash(stream.Uri))
 	writeKV(b, "tvg-id", stream.Id)
 	writeKV(b, "tvg-name", stream.Name)
 	writeKV(b, "group-title", stream.Group)