runner.go

package runner

import (
	"context"
	"crypto/tls"
	"fmt"
	"net"
	"net/http"
	"os"
	"os/signal"
	"strings"
	"syscall"
	"time"

	"github.com/gotify/server/v2/config"
	"golang.org/x/crypto/acme/autocert"
)

// Run starts the http server and if configured a https server.
func Run(router http.Handler, conf *config.Configuration) error {
	shutdown := make(chan error)
	go doShutdownOnSignal(shutdown)

	httpListener, err := startListening("plain connection", conf.Server.ListenAddr, conf.Server.Port, conf.Server.KeepAlivePeriodSeconds)
	if err != nil {
		return err
	}
	defer httpListener.Close()

	s := &http.Server{Handler: router}
	if conf.Server.SSL.Enabled {
		if conf.Server.SSL.LetsEncrypt.Enabled {
			applyLetsEncrypt(s, conf)
		}

		httpsListener, err := startListening("TLS connection", conf.Server.SSL.ListenAddr, conf.Server.SSL.Port, conf.Server.KeepAlivePeriodSeconds)
		if err != nil {
			return err
		}
		defer httpsListener.Close()

		go func() {
			err := s.ServeTLS(httpsListener, conf.Server.SSL.CertFile, conf.Server.SSL.CertKey)
			doShutdown(shutdown, err)
		}()
	}
	go func() {
		err := s.Serve(httpListener)
		doShutdown(shutdown, err)
	}()

	err = <-shutdown
	fmt.Println("Shutting down:", err)

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	return s.Shutdown(ctx)
}

func doShutdownOnSignal(shutdown chan<- error) {
	onSignal := make(chan os.Signal, 1)
	signal.Notify(onSignal, os.Interrupt, syscall.SIGTERM)
	sig := <-onSignal
	doShutdown(shutdown, fmt.Errorf("received signal %s", sig))
}

func doShutdown(shutdown chan<- error, err error) {
	select {
	case shutdown <- err:
	default:
		// If there is no one listening on the shutdown channel, then the
		// shutdown is already initiated and we can ignore these errors.
	}
}

func startListening(connectionType, listenAddr string, port, keepAlive int) (net.Listener, error) {
	network, addr := getNetworkAndAddr(listenAddr, port)
	lc := net.ListenConfig{KeepAlive: time.Duration(keepAlive) * time.Second}

	oldMask := umask(0)
	defer umask(oldMask)

	l, err := lc.Listen(context.Background(), network, addr)
	if err == nil {
		fmt.Println("Started listening for", connectionType, "on", l.Addr().Network(), l.Addr().String())
	}
	return l, err
}

func getNetworkAndAddr(listenAddr string, port int) (string, string) {
	if strings.HasPrefix(listenAddr, "unix:") {
		return "unix", strings.TrimPrefix(listenAddr, "unix:")
	}
	return "tcp", fmt.Sprintf("%s:%d", listenAddr, port)
}

func applyLetsEncrypt(s *http.Server, conf *config.Configuration) {
	certManager := autocert.Manager{
		Prompt:     func(tosURL string) bool { return conf.Server.SSL.LetsEncrypt.AcceptTOS },
		HostPolicy: autocert.HostWhitelist(conf.Server.SSL.LetsEncrypt.Hosts...),
		Cache:      autocert.DirCache(conf.Server.SSL.LetsEncrypt.Cache),
	}
	s.Handler = certManager.HTTPHandler(s.Handler)
	s.TLSConfig = &tls.Config{GetCertificate: certManager.GetCertificate}
}