main.go

package main

import (
	"flag"
	"log"
	"os"
	"strconv"
	"time"

	"github.com/Bolado/ai-tracker/database"
	router "github.com/Bolado/ai-tracker/router"
	"github.com/Bolado/ai-tracker/watcher"
	"github.com/joho/godotenv"
)

var (
	disableWatcher bool
)

func init() {
	// parse command line flags
	flag.BoolVar(&disableWatcher, "disable-watcher", false, "flag to disable the watcher, intended for web server only")
	flag.Parse()
}

func main() {
	// change logger so it prints more information when logging
	log.SetFlags(log.LstdFlags | log.Lshortfile)

	//get env variables
	godotenv.Load()
	interval, _ := strconv.Atoi(os.Getenv("WATCHER_INTERVAL"))

	// initialize the database
	err := database.StartDatabase()
	if err != nil {
		log.Fatalf("Failed to start database: %v\n", err)
	}
	log.Println("Database started")
	defer database.CloseDatabase()

	// load articles from the database
	if err := watcher.LoadArticles(); err != nil {
		log.Fatalf("Failed to load articles: %v\n", err)
	}
	log.Println("Articles loaded")

	if !disableWatcher {
		// start the watcher
		go func() {
			for {
				if err := watcher.StartWatcher(); err != nil {
					log.Fatalf("Failed to start watcher: %v\n", err)
				}
				if interval == 0 {
					log.Println("Watcher will sleep for 15 minutes")
					time.Sleep(15 * time.Minute)
				} else {
					log.Printf("Watcher will sleep for %d minutes\n", interval)
					time.Sleep(time.Duration(interval) * time.Minute)
				}
			}
		}()
		log.Println("Watcher started")
	}

	// start the router and listen for requests
	if err := router.StartRouter(); err != nil {
		log.Fatalf("Failed to start router: %v\n", err)
	}
}