main.go

package main

import (
	"networkinator/models"
	"log"
    "os"

	"github.com/gin-gonic/gin"
    "github.com/gorilla/websocket"
)

var (
    HostCount int
    agentClients = make(map[*websocket.Conn]bool)
    webClients = make(map[*websocket.Conn]bool)
    statusChan = make(chan map[string]interface{})
    agentChan = make(chan map[string]interface{})
    connChan = make(chan map[string]interface{})
    db = ConnectToSQLite()

    tomlConf = &models.Config{}
    configPath = "config.conf"
)

func main() {
    models.ReadConfig(tomlConf, configPath)

    f, err := os.OpenFile("server.txt", os.O_RDWR | os.O_CREATE | os.O_APPEND, 0666)
    if err != nil {
        log.Fatalf("error opening file: %v", err)
    }
    defer f.Close()

    log.SetOutput(f)

    gin.SetMode(gin.ReleaseMode)
	router := gin.Default()
	router.LoadHTMLGlob("templates/**/*")
	router.MaxMultipartMemory = 8 << 20 // 8 MiB
	router.Static("/assets", "./assets/")

    initCookies(router)

	public := router.Group("/")
	addPublicRoutes(public)

    private := router.Group("/")
    private.Use(authRequired)
	addPrivateRoutes(private)

    err = db.AutoMigrate(&models.Connection{}, &models.Agent{})
	if err != nil {
		log.Fatalln(err)
	}

    go handleMsg()

    log.Fatalln(router.Run(":80"))
}