main.go

package main

import (
	"flag"
	"fmt"
	"log"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"

	"git.autistici.org/ai3/go-common/unix"
	"github.com/coreos/go-systemd/daemon"
	"github.com/prometheus/client_golang/prometheus/promhttp"

	"git.autistici.org/id/auth/server"
)

var (
	configPath              = flag.String("config", "/etc/auth-server/config.yml", "configuration `file`")
	socketPath              = flag.String("socket", "/run/auth/socket", "`path` to the UNIX socket to listen on")
	systemdSocketActivation = flag.Bool("systemd-socket", false, "use SystemD socket activation")
	httpAddr                = flag.String("http-addr", "", "if not nil, bind an HTTP server to this `addr` for Prometheus metrics")
	requestTimeout          = flag.Duration("timeout", 5*time.Second, "timeout for incoming requests")
)

func usage() {
	fmt.Fprintf(os.Stderr, `authserv - a simple authentication server
Usage: authserv [<OPTIONS>]

Known options:
`)
	flag.PrintDefaults()
}

func main() {
	log.SetFlags(0)
	flag.Usage = usage
	flag.Parse()

	syscall.Umask(007)

	config, err := server.LoadConfig(*configPath)
	if err != nil {
		log.Fatalf("could not load configuration: %v", err)
	}

	authSrv, err := server.NewServer(config, nil)
	if err != nil {
		log.Fatalf("configuration error: %v", err)
	}

	if *httpAddr != "" {
		h := http.NewServeMux()
		h.Handle("/metrics", promhttp.Handler())
		go func() {
			log.Fatal(http.ListenAndServe(*httpAddr, h))
		}()
	}

	srv := unix.NewLineServer(server.NewSocketServer(authSrv))
	srv.RequestTimeout = *requestTimeout

	var sockSrv *unix.SocketServer
	if *systemdSocketActivation {
		sockSrv, err = unix.NewSystemdSocketServer(srv)
	} else {
		sockSrv, err = unix.NewUNIXSocketServer(*socketPath, srv)
	}
	if err != nil {
		log.Fatalf("error: %v", err)
	}

	done := make(chan struct{})
	sigCh := make(chan os.Signal, 1)
	go func() {
		<-sigCh
		log.Printf("terminating")
		sockSrv.Close()
		close(done)
	}()
	signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)

	log.Printf("starting")
	daemon.SdNotify(false, "READY=1")
	if err := sockSrv.Serve(); err != nil {
		log.Fatal(err)
	}

	<-done
}