main.go

package main

import (
	"context"
	"flag"
	"fmt"
	"github.com/ChristianGaertner/dmx-controller/database"
	"github.com/ChristianGaertner/dmx-controller/dmx"
	"github.com/ChristianGaertner/dmx-controller/run"
	"github.com/ChristianGaertner/dmx-controller/server"
	"github.com/ChristianGaertner/dmx-controller/setup"
	"log"
	"time"
)

var noDmxOutput = flag.Bool("no-dmx-output", false, "If true, the OLA server will not be used to send dmx values")
var olaRpcEndpoint = flag.String("ola_rpc_endpoint", "localhost:9010", "RPC endpoint of the OLA service")
var addr = flag.String("address", ":8080", "Address of the server to listen on")
var setupFile = flag.String("setup", "", "path to setup json definition")
var databaseFile = flag.String("database", "", "path to the database file, will be created if needed")

var gracefulTimeout = flag.Duration("graceful-timeout", time.Second*15, "the duration for which the server gracefully wait for existing connections to finish - e.g. 15s or 1m")

func main() {
	flag.Parse()
	if *setupFile == "" {
		fmt.Println("Please provide a setup file, e.g.: -setup file.json")
		return
	}

	if *databaseFile == "" {
		fmt.Println("Please provide a database file, e.g.: -database data.db")
		return
	}

	db, err := database.Open(*databaseFile)
	if err != nil {
		panic(err)
	}
	defer func() {
		err = db.Close()
		if err != nil {
			log.Fatal(err)
		}
	}()

	buffer := dmx.NewBuffer()

	var renderer dmx.BufferRenderer
	if *noDmxOutput {
		renderer = &dmx.NilRenderer{}
		//renderer := &dmx.StdOutRenderer{NumChannels: 10}
	} else {
		renderer, err = dmx.NewOlaRPCRenderer(*olaRpcEndpoint)
		if err != nil {
			panic(err)
		}
	}

	s, err := setup.Load(*setupFile)
	if err != nil {
		panic(err)
	}

	log.Printf("initializing %d dmx universes", len(s.GetUniverseIds()))
	buffer.Init(s.GetUniverseIds())

	deviceMap, err := s.PatchDeviceMap()
	if err != nil {
		panic(err)
	}

	engine := run.NewEngine(renderer, s, deviceMap, buffer, db)

	onExit := make(chan bool)
	ctx, cancel := context.WithCancel(context.Background())
	go engine.Boot(ctx, onExit)

	err = server.ListenAndServe(ctx, *addr, engine, *gracefulTimeout)
	if err != nil {
		log.Println(err)
	}

	log.Println("Shutting down...")
	cancel()
	<-onExit
	log.Println("[SHUTDOWN]")
}