From f9a83d52b30a9e0209f827d5f9d46694cf6f449e Mon Sep 17 00:00:00 2001
From: Matthew Huie <matthew@matthewhuie.com>
Date: Tue, 11 Apr 2023 05:29:38 -0400
Subject: [PATCH] Adding argument to set custom anti-collision ID (#146)

---
 cmd/root.go          | 2 ++
 pkg/config/config.go | 1 +
 pkg/server/server.go | 4 ++++
 3 files changed, 7 insertions(+)

diff --git a/cmd/root.go b/cmd/root.go
index 189ec9f..ad703b6 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -84,6 +84,7 @@ var rootCmd = &cobra.Command{
 			HTTPS:                viper.GetBool("https"),
 			M3UFileName:          viper.GetString("m3u-file-name"),
 			CustomEndpoint:       viper.GetString("custom-endpoint"),
+			CustomId:             viper.GetString("custom-id"),
 			XtreamGenerateApiGet: viper.GetBool("xtream-api-get"),
 		}
 
@@ -121,6 +122,7 @@ func init() {
 	rootCmd.Flags().StringP("m3u-url", "u", "", `Iptv m3u file or url e.g: "http://example.com/iptv.m3u"`)
 	rootCmd.Flags().StringP("m3u-file-name", "", "iptv.m3u", `Name of the new proxified m3u file e.g "http://poxy.com/iptv.m3u"`)
 	rootCmd.Flags().StringP("custom-endpoint", "", "", `Custom endpoint "http://poxy.com/<custom-endpoint>/iptv.m3u"`)
+	rootCmd.Flags().StringP("custom-id", "", "", `Custom anti-collison ID for each track "http://proxy.com/<custom-id>/..."`)
 	rootCmd.Flags().Int("port", 8080, "Iptv-proxy listening port")
 	rootCmd.Flags().Int("advertised-port", 0, "Port to expose the IPTV file and xtream (by default, it's taking value from port) useful to put behind a reverse proxy")
 	rootCmd.Flags().String("hostname", "", "Hostname or IP to expose the IPTVs endpoints")
diff --git a/pkg/config/config.go b/pkg/config/config.go
index 9024a58..13e65f2 100644
--- a/pkg/config/config.go
+++ b/pkg/config/config.go
@@ -51,6 +51,7 @@ type ProxyConfig struct {
 	M3UCacheExpiration   int
 	M3UFileName          string
 	CustomEndpoint       string
+	CustomId             string
 	RemoteURL            *url.URL
 	AdvertisedPort       int
 	HTTPS                bool
diff --git a/pkg/server/server.go b/pkg/server/server.go
index 8c58523..cf5145e 100644
--- a/pkg/server/server.go
+++ b/pkg/server/server.go
@@ -64,6 +64,10 @@ func NewServer(config *config.ProxyConfig) (*Config, error) {
 		}
 	}
 
+        if trimmedCustomId := strings.Trim(config.CustomId, "/"); trimmedCustomId != "" {
+                endpointAntiColision = trimmedCustomId
+        }
+
 	return &Config{
 		config,
 		&p,