server_windows.go

package main

import (
	"context"
	"html/template"
	"log"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/quasoft/websspi"
)

var helloTemplate = template.Must(template.New("index.html").Parse(`
{{- if . -}}
<h2>Hello {{ .Username }}!</h2>

{{ if .Groups -}}
Groups:
<ul>
{{- range .Groups}}
	<li>{{ . }}</li>
{{end -}}
</ul>
{{- end }}
{{- else -}}
<h2>Hello!</h2>
{{- end -}}
`))

func main() {
	config := websspi.NewConfig()
	config.EnumerateGroups = true // If groups should be resolved
	// config.ServerName = "..." // If static instead of dynamic group membership should be resolved

	auth, err := websspi.New(config)
	if err != nil {
		panic(err)
	}

	server := &http.Server{Addr: "0.0.0.0:9000"}
	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		info := r.Context().Value(websspi.UserInfoKey)
		userInfo, _ := info.(*websspi.UserInfo)
		w.Header().Add("Content-Type", "text/html; encoding=utf-8")
		helloTemplate.Execute(w, userInfo)
	})
	http.Handle("/", auth.WithAuth(handler))

	stop := make(chan os.Signal, 2)
	signal.Notify(stop, os.Interrupt, syscall.SIGTERM)

	go func() {
		if err := server.ListenAndServe(); err != http.ErrServerClosed {
			log.Fatalf("ListenAndServe(): %s\n", err)
		}
	}()

	// Wait for SIGINT or SIGTERM
	<-stop
	log.Print("Shutting down and releasing resources...\n")
	err = auth.Free()
	if err != nil {
		log.Printf("Error while releasing resources: %v\n", err)
	}

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	cancel()
	if err := server.Shutdown(ctx); err != nil {
		log.Fatalf("Graceful shutdown timed out: %s\n", err)
	}
}