SimplePuppetProvisioner.go

package main

import (
	"context"
	"flag"
	"os"
	"os/signal"
	"runtime"
	"syscall"
	"time"

	"github.com/fsnotify/fsnotify"
	"github.com/mbaynton/SimplePuppetProvisioner/interfaces"
	"github.com/mbaynton/SimplePuppetProvisioner/lib"
	"github.com/mbaynton/SimplePuppetProvisioner/lib/certsign"
	"github.com/mbaynton/SimplePuppetProvisioner/lib/sppexec"
	"github.com/mbaynton/go-genericexec"
)

func main() {
	configFile := flag.String("config", "", "Path to the spp configuration file.")
	logStdout := flag.Bool("log-stdout", false, "Log to stdout.")
	flag.Parse()

	var searchDirs []string
	if *configFile != "" {
		searchDirs = []string{}
	} else {
		*configFile = "spp.conf"
		searchDirs = []string{".", "/etc/spp"}
	}

	appConfig := lib.LoadTheConfig(*configFile, searchDirs)
	notifier := lib.NewNotifications(&appConfig)
	csrWatcher, err := fsnotify.NewWatcher()
	if err != nil {
		appConfig.Log.Println("Unable to start certificate signing request watcher. Cannot proceed.")
		os.Exit(1)
	}
	watcher := interfaces.FsnotifyWatcher{
		Add:    csrWatcher.Add,
		Close:  csrWatcher.Close,
		Remove: csrWatcher.Remove,
		Events: csrWatcher.Events,
		Errors: csrWatcher.Errors,
	}
	certSigner, err := certsign.NewCertSigner(*appConfig.PuppetConfig, appConfig.Log, &watcher, notifier.Notify)
	if err != nil {
		appConfig.Log.Println("Unable to start certificate signing manager. Cannot proceed.")
		os.Exit(1)
	}

	execConfigMap := makeExecTaskConfigsMap(&appConfig)
	if appConfig.GithubWebhooks.EnableStandardR10kListener {
		appConfig.GithubWebhooks.Listeners = append(appConfig.GithubWebhooks.Listeners, lib.StandardR10kListenerConfig(appConfig.GithubWebhooks))
	}
	lib.SetWebhookExecTaskConfigMap(appConfig.GithubWebhooks, execConfigMap)

	execManager := sppexec.NewSppExecManager(execConfigMap, appConfig.PuppetConfig, appConfig.Log, notifier.Notify)

	server := lib.NewHttpServer(appConfig, notifier, certSigner, execManager)

	if *logStdout == false {
		appConfig.MoveLoggingToFile()
	}

	stop := make(chan os.Signal, 1)
	if runtime.GOOS == "windows" {
		signal.Notify(stop, os.Interrupt)
	} else {
		signal.Notify(stop, syscall.SIGTERM, syscall.SIGINT)
	}

	go server.Start()

	// Wait for a SIGTERM.
	<-stop

	appConfig.Log.Println("Stopping...")

	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	server.Shutdown(ctx)
	appConfig.Log.Println("HTTP server shutdown.")

	certSigner.Shutdown()
	appConfig.Log.Println("Certificate signing manager shutdown.")

	appConfig.Log.Println("Process will now exit.")
}

func makeExecTaskConfigsMap(config *lib.AppConfig) map[string]genericexec.GenericExecConfig {
	execTaskDefns := config.GenericExecTasks
	execTaskConfigsByName := make(map[string]genericexec.GenericExecConfig, len(execTaskDefns))
	for _, configuredTask := range execTaskDefns {
		execTaskConfigsByName[configuredTask.Name] = *configuredTask
	}
	return execTaskConfigsByName
}