Add proxy config to command line

masa-finance · Nov 29, 2024 · c454cb9 · c454cb9
1 parent 7ee5b11
commit c454cb9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/pkg/config/app.go b/pkg/config/app.go
@@ -194,6 +194,10 @@ func (c *AppConfig) setCommandLineConfig() error {
 	pflag.BoolVar(&c.WebScraper, "webScraper", viper.GetBool(WebScraper), "Web Scraper")
 	pflag.BoolVar(&c.Faucet, "faucet", viper.GetBool(Faucet), "Faucet")
 	pflag.BoolVar(&c.APIEnabled, "api-enabled", viper.GetBool(APIEnabled), "Enable API server")
+	pflag.BoolVar(&c.ProxyEnabled, "proxy-enabled", viper.GetBool(ProxyEnabled), "Enable CONNECT proxy")
+	pflag.StringVar(&c.ProxyListenAddr, "proxyListenAddr", viper.GetString("proxyListenAddr"), "Proxy listen address")
+	pflag.Uint16Var(&c.ProxyListenPort, "proxyListenPort", viper.GetUint16("proxyListenPort"), "Proxy listen port")
+	pflag.Uint16Var(&c.ProxyTargetPort, "proxyTargetPort", viper.GetUint16("proxyTargetPort"), "Proxy target port")
 
 	pflag.Parse()
 

diff --git a/pkg/config/constants.go b/pkg/config/constants.go
@@ -40,5 +40,6 @@ const (
 	TelegramScraper    = "TELEGRAM_SCRAPER"
 	WebScraper         = "WEB_SCRAPER"
 	APIEnabled         = "API_ENABLED"
+	ProxyEnabled       = "PROXY_ENABLED"
 	DefaultPrivKeyFile = "masa_oracle_key"
 )