main.go

package main

import (
	"encoding/json"
	"flag"
	"fmt"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/bestserversio/spy/internal/config"
	"github.com/bestserversio/spy/internal/scanners"
	"github.com/bestserversio/spy/internal/servers"
	"github.com/bestserversio/spy/internal/utils"
	"github.com/bestserversio/spy/internal/vms"
)

const VERSION = "1.0.0"
const HELPMENU = `
./spy [OPTIONS]\n\n
-l --list => Prints config and exits.\n
-v --version => Prints version and exits.\n
-h --help => Prints help menu and exits.\n
-c --cfg => Path to config file.\n
`

func main() {
	// Command line options and parse command line.
	var list bool
	var version bool
	var help bool

	flag.BoolVar(&list, "l", false, "Prints config settings and exits.")
	flag.BoolVar(&list, "list", false, "Prints config settings and exits.")

	flag.BoolVar(&version, "v", false, "Prints version and exits.")
	flag.BoolVar(&version, "version", false, "Prints number and exits.")

	flag.BoolVar(&help, "h", false, "Prints help menu and exits.")
	flag.BoolVar(&help, "help", false, "Prints help menu and exits.")

	cfgPath := flag.String("cfg", "/etc/bestservers/spy.json", "Path to config file.")

	flag.Parse()

	// Check for version.
	if version {
		fmt.Println(VERSION)

		os.Exit(0)
	}

	// Check for help menu.
	if help {
		fmt.Print(HELPMENU)

		os.Exit(0)
	}

	// Initialize config.
	cfg := config.Config{}

	// Load defaults.
	cfg.LoadDefaults()

	// Attempt to load config.
	err := cfg.LoadFromFs(*cfgPath)

	if err != nil {
		fmt.Println("Error loading config file. Resorting to defaults...")
		fmt.Println(err)
	}

	utils.DebugMsg(2, &cfg, "[CFG] Initial config loaded...")

	// Check if we want to print our config settings.
	if list {
		cfg.PrintConfig()

		os.Exit(0)
	}

	// Check for web API updating.
	if cfg.WebApi.Enabled {
		go func() {
			for {
				// Get web API interval.
				interval := time.Duration(cfg.WebApi.Interval) * time.Second

				if interval < 1 {
					interval = time.Second * 60
				}

				// Make sure web config is still enabled.
				if !cfg.WebApi.Enabled {
					time.Sleep(interval)

					continue
				}

				utils.DebugMsg(3, &cfg, "[WEB_API] Retrieving web API from '%s%s'.", cfg.WebApi.Host, cfg.WebApi.Endpoint)

				data, err := cfg.LoadFromWeb()

				if err != nil {
					utils.DebugMsg(1, &cfg, "[WEB_API] Failed to retrieve web API from '%s%s'.", cfg.WebApi.Host, cfg.WebApi.Endpoint)

					time.Sleep(interval)

					continue
				}

				utils.DebugMsg(6, &cfg, "[WEB_API] Loading JSON => %s", data)

				// Check if we need to save new config to the file system.
				if cfg.WebApi.SaveToFs {
					jsonData, err := json.MarshalIndent(cfg, "", "    ")

					if err != nil {
						utils.DebugMsg(1, &cfg, "[WEB_API] Failed to marshal CFG structure when saving to file system due to error : %s", err.Error())
					} else {
						err = os.WriteFile(*cfgPath, jsonData, 0644)

						if err != nil {
							utils.DebugMsg(0, &cfg, "[WEB_API] Failed to write web config to file system (%s) due to error :: %s", *cfgPath, err.Error())
						} else {
							utils.DebugMsg(2, &cfg, "[WEB_API] Successfully wrote new data to file system (%s)!", *cfgPath)
						}
					}
				}

				// Resetup VMS.
				vms.SetupVms(&cfg)

				// Resetup scanners.
				scanners.SetupScanners(&cfg)

				// If we have no interval, close this function now.
				if cfg.WebApi.Interval < 1 {
					return
				}

				time.Sleep(interval)
			}
		}()
	}

	// Setup remove inactive.
	go servers.RemoveInactive(&cfg)

	// Setup remove dups.
	go servers.RemoveDups(&cfg)

	// Setup remove timed out.
	go servers.RemoveTimedOut(&cfg)

	// Setup VMS.
	vms.SetupVms(&cfg)

	// Setup scanners.
	scanners.SetupScanners(&cfg)

	// Make a signal.
	sigc := make(chan os.Signal, 1)
	signal.Notify(sigc, syscall.SIGINT, syscall.SIGTERM)

	<-sigc

	utils.DebugMsg(0, &cfg, "[MAIN] Exiting...")
	os.Exit(0)
}