main.go

package main

import (
	"fmt"
	"github.com/toxuin/alarmserver/buses/mqtt"
	"github.com/toxuin/alarmserver/buses/webhooks"
	conf "github.com/toxuin/alarmserver/config"
	"github.com/toxuin/alarmserver/servers/dahua"
	"github.com/toxuin/alarmserver/servers/ftp"
	"github.com/toxuin/alarmserver/servers/hikvision"
	"github.com/toxuin/alarmserver/servers/hisilicon"
	"os"
	"os/signal"
	"sync"
	"syscall"
)

var config *conf.Config

func init() {
	config.SetDefaults()
}

func main() {
	config = config.Load()
	fmt.Println("STARTING...")
	if config.Debug {
		config.Printout()
	}

	processesWaitGroup := sync.WaitGroup{}

	// INIT BUSES
	mqttBus := mqtt.Bus{Debug: config.Debug}
	if config.Mqtt.Enabled {
		mqttBus.Initialize(config.Mqtt)
		if config.Debug {
			fmt.Println("MQTT BUS INITIALIZED")
		}
	}

	webhookBus := webhooks.Bus{Debug: config.Debug}
	if config.Webhooks.Enabled {
		webhookBus.Initialize(config.Webhooks)
		if config.Debug {
			fmt.Println("WEBHOOK BUS INITIALIZED")
		}
	}

	messageHandler := func(cameraName string, eventType string, extra string) {
		if config.Mqtt.Enabled {
			mqttBus.SendMessage(config.Mqtt.TopicRoot+"/"+cameraName+"/"+eventType, extra)
		}
		if config.Webhooks.Enabled {
			webhookBus.SendMessage(cameraName, eventType, extra)
		}
	}

	if config.Hisilicon.Enabled {
		// START HISILICON ALARM SERVER
		hisiliconServer := hisilicon.Server{
			Debug:          config.Debug,
			WaitGroup:      &processesWaitGroup,
			Port:           config.Hisilicon.Port,
			MessageHandler: messageHandler,
		}
		hisiliconServer.Start()
		if config.Debug {
			fmt.Println("STARTED HISILICON SERVER")
		}
	}

	if config.Hikvision.Enabled {
		// START HIKVISION ALARM SERVER
		hikvisionServer := hikvision.Server{
			Debug:          config.Debug,
			WaitGroup:      &processesWaitGroup,
			Cameras:        &config.Hikvision.Cams,
			MessageHandler: messageHandler,
		}
		hikvisionServer.Start()
		if config.Debug {
			fmt.Println("STARTED HIKVISION SERVER")
		}
	}

	if config.Dahua.Enabled {
		// START DAHUA SERVER
		dhServer := dahua.Server{
			Debug:          config.Debug,
			WaitGroup:      &processesWaitGroup,
			Cameras:        &config.Dahua.Cams,
			MessageHandler: messageHandler,
		}
		dhServer.Start()
		if config.Debug {
			fmt.Println("STARTED DAHUA SERVER")
		}
	}

	if config.Ftp.Enabled {
		// START FTP SERVER
		ftpServer := ftp.Server{
			Debug:          config.Debug,
			WaitGroup:      &processesWaitGroup,
			Port:           config.Ftp.Port,
			AllowFiles:     config.Ftp.AllowFiles,
			RootPath:       config.Ftp.RootPath,
			Password:       config.Ftp.Password,
			MessageHandler: messageHandler,
		}
		ftpServer.Start()
		if config.Debug {
			fmt.Println("STARTED FTP SERVER")
		}
	}

	processesWaitGroup.Wait()

	// START INFINITE LOOP WAITING FOR SERVERS
	exitSignal := make(chan os.Signal)
	signal.Notify(exitSignal, syscall.SIGINT, syscall.SIGTERM)
	<-exitSignal
}