Add flag to specify ip cache file
This commit is contained in:
@@ -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)
|
||||
|
Reference in New Issue
Block a user