hmn/src/website/website.go

package website

import (
	"context"
	"errors"
	"log"
	"net/http"
	_ "net/http/pprof"
	"os"
	"os/signal"
	"time"

	"git.handmade.network/hmn/hmn/src/auth"
	"git.handmade.network/hmn/hmn/src/config"
	"git.handmade.network/hmn/hmn/src/db"
	"git.handmade.network/hmn/hmn/src/discord"
	"git.handmade.network/hmn/hmn/src/logging"
	"git.handmade.network/hmn/hmn/src/perf"
	"git.handmade.network/hmn/hmn/src/templates"
	"github.com/spf13/cobra"
)

var WebsiteCommand = &cobra.Command{
	Short: "Run the HMN website",
	Run: func(cmd *cobra.Command, args []string) {
		templates.Init()

		defer logging.LogPanics(nil)

		logging.Info().Msg("Hello, HMN!")

		backgroundJobContext, cancelBackgroundJobs := context.WithCancel(context.Background())

		conn := db.NewConnPool(config.Config.Postgres.MinConn, config.Config.Postgres.MaxConn)
		perfCollector := perf.RunPerfCollector(backgroundJobContext)

		server := http.Server{
			Addr:    config.Config.Addr,
			Handler: NewWebsiteRoutes(conn, perfCollector),
		}

		backgroundJobsDone := zipJobs(
			auth.PeriodicallyDeleteExpiredSessions(backgroundJobContext, conn),
			auth.PeriodicallyDeleteInactiveUsers(backgroundJobContext, conn),
			perfCollector.Done,
			discord.RunDiscordBot(backgroundJobContext, conn),
		)

		signals := make(chan os.Signal, 1)
		signal.Notify(signals, os.Interrupt)
		go func() {
			<-signals
			logging.Info().Msg("Shutting down the website")
			go func() {
				timeout, cancel := context.WithTimeout(context.Background(), 30*time.Second)
				defer cancel()
				logging.Info().Msg("shutting down web server")
				server.Shutdown(timeout)
				logging.Info().Msg("cancelling background jobs")
				cancelBackgroundJobs()
			}()

			<-signals
			logging.Warn().Msg("Forcibly killed the website")
			os.Exit(1)
		}()

		go func() {
			log.Println(http.ListenAndServe(config.Config.PrivateAddr, nil))
		}()

		logging.Info().Str("addr", config.Config.Addr).Msg("Serving the website")
		serverErr := server.ListenAndServe()
		if !errors.Is(serverErr, http.ErrServerClosed) {
			logging.Error().Err(serverErr).Msg("Server shut down unexpectedly")
		}

		<-backgroundJobsDone
	},
}

func zipJobs(cs ...<-chan struct{}) <-chan struct{} {
	out := make(chan struct{})
	go func() {
		for _, c := range cs {
			<-c
		}
		close(out)
	}()
	return out
}