feat: add an option to listen on a specific IP address (#136)

config.go

@@ -15,14 +15,26 @@ package main
 import (
 	"flag"
 	"os"
+	"fmt"
 )
 
 func initConfig() *Config {
 	cfg := &Config{
+		IP: "0.0.0.0",
 		Port: 1188,
 	}
 
+	// IP flag
+	if ip, ok := os.LookupEnv("IP"); ok && ip != "" {
+		cfg.IP = ip
+	}
+	flag.StringVar(&cfg.IP, "ip", cfg.IP, "set up the IP address to bind to")
+	flag.StringVar(&cfg.IP, "i", cfg.IP, "set up the IP address to bind to")
+
 	// Port flag
+	if port, ok := os.LookupEnv("PORT"); ok && port != "" {
+		fmt.Sscanf(port, "%d", &cfg.Port)
+	}
 	flag.IntVar(&cfg.Port, "port", cfg.Port, "set up the port to listen on")
 	flag.IntVar(&cfg.Port, "p", cfg.Port, "set up the port to listen on")
 

main.go

@@ -60,7 +60,7 @@ func authMiddleware(cfg *Config) gin.HandlerFunc {
 func main() {
 	cfg := initConfig()
 
-	fmt.Printf("DeepL X has been successfully launched! Listening on 0.0.0.0:%v\n", cfg.Port)
+	fmt.Printf("DeepL X has been successfully launched! Listening on %v:%v\n", cfg.IP, cfg.Port)
 	fmt.Println("Developed by sjlleo <[email protected]> and missuo <[email protected]>.")
 
 	// Set Proxy
@@ -255,10 +255,5 @@ func main() {
 		})
 	})
 
-	envPort, ok := os.LookupEnv("PORT")
-	if ok {
-		r.Run(":" + envPort)
-	} else {
-		r.Run(fmt.Sprintf(":%v", cfg.Port))
-	}
+	r.Run(fmt.Sprintf("%v:%v", cfg.IP, cfg.Port))
 }

types.go

@@ -12,6 +12,7 @@
 package main
 
 type Config struct {
+	IP        string
 	Port      int
 	Token     string
 	AuthKey   string