Add flag to specify ip cache file

This commit is contained in:
PhatPhuckDave
2024-07-21 23:20:44 +02:00
parent c8a1a9eb17
commit c55fa7a131
2 changed files with 6 additions and 2 deletions

View File

@@ -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)