diff --git a/cmd/root.go b/cmd/root.go index 914cda7c..e38006bb 100644 --- a/cmd/root.go +++ b/cmd/root.go @@ -71,7 +71,7 @@ var rootCmd = &cobra.Command{ conf := &config.ProxyConfig{ HostConfig: &config.HostConfiguration{ Hostname: viper.GetString("hostname"), - Port: viper.GetInt64("port"), + Port: viper.GetInt("port"), }, RemoteURL: remoteHostURL, XtreamUser: config.CredentialString(xtreamUser), @@ -80,11 +80,16 @@ var rootCmd = &cobra.Command{ M3UCacheExpiration: viper.GetInt("m3u-cache-expiration"), User: config.CredentialString(viper.GetString("user")), Password: config.CredentialString(viper.GetString("password")), + AdvertisedPort: viper.GetInt("advertised-port"), HTTPS: viper.GetBool("https"), M3UFileName: viper.GetString("m3u-file-name"), CustomEndpoint: viper.GetString("custom-endpoint"), } + if conf.AdvertisedPort == 0 { + conf.AdvertisedPort = conf.HostConfig.Port + } + server, err := server.NewServer(conf) if err != nil { log.Fatal(err) @@ -111,15 +116,16 @@ func init() { // Here you will define your flags and configuration settings. // Cobra supports persistent flags, which, if defined here, // will be global for your application. - rootCmd.PersistentFlags().StringVar(&cfgFile, "iptv-proxy-config", "C", "config file (default is $HOME/.iptv-proxy.yaml)") - 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//iptv.m3u"`) - rootCmd.Flags().Int64("port", 8080, "Port to expose the IPTVs endpoints") + rootCmd.PersistentFlags().StringVar(&cfgFile, "iptv-proxy-config", "C", "Config file (default is $HOME/.iptv-proxy.yaml)") + 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//iptv.m3u"`) + 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") rootCmd.Flags().BoolP("https", "", false, "Activate https for urls proxy") - rootCmd.Flags().String("user", "usertest", "user UNSAFE(temp auth to access proxy)") - rootCmd.Flags().String("password", "passwordtest", "password UNSAFE(auth to access m3u proxy and xtream proxy)") + rootCmd.Flags().String("user", "usertest", "User auth to access proxy (m3u/xtream)") + rootCmd.Flags().String("password", "passwordtest", "Password auth to access proxy (m3u/xtream)") rootCmd.Flags().String("xtream-user", "", "Xtream-code user login") rootCmd.Flags().String("xtream-password", "", "Xtream-code password login") rootCmd.Flags().String("xtream-base-url", "", "Xtream-code base url e.g(http://expample.tv:8080)") diff --git a/pkg/config/config.go b/pkg/config/config.go index 66b5c1c8..fd3d5390 100644 --- a/pkg/config/config.go +++ b/pkg/config/config.go @@ -38,7 +38,7 @@ func (c CredentialString) String() string { // HostConfiguration containt host infos type HostConfiguration struct { Hostname string - Port int64 + Port int } // ProxyConfig Contain original m3u playlist and HostConfiguration @@ -51,6 +51,7 @@ type ProxyConfig struct { M3UFileName string CustomEndpoint string RemoteURL *url.URL + AdvertisedPort int HTTPS bool User, Password CredentialString } diff --git a/pkg/server/server.go b/pkg/server/server.go index 678b1b73..80277eaf 100644 --- a/pkg/server/server.go +++ b/pkg/server/server.go @@ -166,7 +166,7 @@ func (c *Config) replaceURL(uri string, trackIndex int, xtream bool) (string, er protocol, basicAuth, c.HostConfig.Hostname, - c.HostConfig.Port, + c.AdvertisedPort, customEnd, uriPath, ) diff --git a/pkg/xtream-proxy/xtream-proxy.go b/pkg/xtream-proxy/xtream-proxy.go index 9cf64342..9f12a7b5 100644 --- a/pkg/xtream-proxy/xtream-proxy.go +++ b/pkg/xtream-proxy/xtream-proxy.go @@ -150,7 +150,7 @@ func (c *Client) Action(config *config.ProxyConfig, action string, q url.Values) } respBody, err = c.GetEPG(q["stream_id"][0]) default: - respBody, err = c.login(config.User.String(), config.Password.String(), protocol+"://"+config.HostConfig.Hostname, int(config.HostConfig.Port), protocol) + respBody, err = c.login(config.User.String(), config.Password.String(), protocol+"://"+config.HostConfig.Hostname, config.AdvertisedPort, protocol) } return