diff --git a/.gitignore b/.gitignore index 8746df5..558d047 100644 --- a/.gitignore +++ b/.gitignore @@ -5,3 +5,4 @@ ip-checker/.env ip-checker/ip-notifier.tar ip-checker/ip ip-checker/ip-checker.tar +ip-checker/main diff --git a/ip-checker/main.go b/ip-checker/main.go index 9c6fd87..32c268f 100644 --- a/ip-checker/main.go +++ b/ip-checker/main.go @@ -2,6 +2,7 @@ package main import ( _ "embed" + "flag" "fmt" "io" "log" @@ -38,19 +39,21 @@ func init() { var oldIP, newIP string const REMOTE_URL = "http://141.147.53.95:5000" -const IP_FILE = "ip" var ipFile *os.File var lastMailed time.Time func main() { + ipfile := flag.String("ipfile", "ip", "file to store ip in") + flag.Parse() + var err error envvar, err = godotenv.Parse(strings.NewReader(env)) if err != nil { Error.Fatalf("Error parsing .env file: %v", err) os.Exit(1) } - ipFile, err = os.OpenFile(IP_FILE, os.O_RDWR|os.O_CREATE, 0644) + ipFile, err = os.OpenFile(*ipfile, os.O_RDWR|os.O_CREATE, 0644) if err != nil { Error.Fatalf("Error opening IP file: %v", err) os.Exit(1)