show logged in user in web ui

web/templates.go

@@ -2,13 +2,18 @@ package web
 
 import (
 	"embed"
+	"errors"
 	"fmt"
 	"html/template"
 	"io"
 	"io/fs"
 
 	"git.janky.solutions/finn/go-project-template/config"
+	"git.janky.solutions/finn/go-project-template/db"
+	"github.com/gorilla/sessions"
+	pgx "github.com/jackc/pgx/v5"
 	echo "github.com/labstack/echo/v4"
+	"github.com/sirupsen/logrus"
 )
 
 var (
@@ -50,11 +55,62 @@ func init() {
 type Template struct {
 }
 
-func (t *Template) Render(w io.Writer, name string, data interface{}, c echo.Context) error {
+type templateData struct {
+	Authenticated bool
+	User          db.User
+	Version       string
+	Data          any
+}
+
+func (t *templateData) dataFromContext(c echo.Context) error {
+	sessionInterface := c.Get(contextKeySession)
+	if sessionInterface == nil {
+		return nil // no session, no data to populate
+	}
+
+	session := sessionInterface.(*sessions.Session)
+
+	userIDInterface, ok := session.Values[sessionValueAuthUser]
+	if !ok {
+		return nil
+	}
+
+	userID, ok := userIDInterface.(int32)
+	if !ok {
+		logrus.WithField("userID", userIDInterface).Warn("unexpected session error: user ID is not an int32")
+		return errors.New("user ID is not an in32")
+	}
+
+	ctx := c.Request().Context()
+	queries, conn, err := db.Get(ctx)
+	if err != nil {
+		return err
+	}
+	defer conn.Close(ctx)
+
+	user, err := queries.GetUser(ctx, userID)
+	if err != nil {
+		if errors.Is(err, pgx.ErrNoRows) {
+			return nil
+		}
+		return err
+	}
+
+	t.Authenticated = true
+	t.User = user
+	t.Version = fmt.Sprintf("%s %s", config.BuildInfo.Main.Path, config.Version)
+
+	return nil
+}
+
+func (t *Template) Render(w io.Writer, name string, data any, c echo.Context) error {
+	td := templateData{Data: data}
+	td.dataFromContext(c)
+
 	// Why does it work like this? because go's templating system doesn't handle multiple templates extending from a common base well
 	// just doing it normally causes every template to render the same (the import form, at time of writing, probably the last template alphabetically)
 	// https://stackoverflow.com/a/69244593/21894038
 	tmpl := template.Must(allTemplates.Clone())
 	tmpl = template.Must(tmpl.ParseFS(templatesSubFS, name))
-	return tmpl.ExecuteTemplate(w, name, data)
+	return tmpl.ExecuteTemplate(w, name, td)
 }