pomerium/authenticate/handlers/userinfo.go

package handlers

import (
	"encoding/json"
	"net/http"

	"google.golang.org/protobuf/encoding/protojson"

	"github.com/pomerium/pomerium/internal/directory"
	"github.com/pomerium/pomerium/internal/httputil"
	"github.com/pomerium/pomerium/pkg/grpc/session"
	"github.com/pomerium/pomerium/pkg/grpc/user"
	"github.com/pomerium/pomerium/ui"
)

// UserInfoData is the data for the UserInfo page.
type UserInfoData struct {
	CSRFToken       string
	DirectoryGroups []*directory.Group
	DirectoryUser   *directory.User
	IsImpersonated  bool
	Session         *session.Session
	SignOutURL      string
	User            *user.User
	WebAuthnURL     string
}

// ToJSON converts the data into a JSON map.
func (data UserInfoData) ToJSON() map[string]interface{} {
	m := map[string]interface{}{}
	m["csrfToken"] = data.CSRFToken
	var directoryGroups []json.RawMessage
	for _, directoryGroup := range data.DirectoryGroups {
		if bs, err := protojson.Marshal(directoryGroup); err == nil {
			directoryGroups = append(directoryGroups, json.RawMessage(bs))
		}
	}
	m["directoryGroups"] = directoryGroups
	if bs, err := protojson.Marshal(data.DirectoryUser); err == nil {
		m["directoryUser"] = json.RawMessage(bs)
	}
	m["isImpersonated"] = data.IsImpersonated
	if bs, err := protojson.Marshal(data.Session); err == nil {
		m["session"] = json.RawMessage(bs)
	}
	m["signOutUrl"] = data.SignOutURL
	if bs, err := protojson.Marshal(data.User); err == nil {
		m["user"] = json.RawMessage(bs)
	}
	m["webAuthnUrl"] = data.WebAuthnURL
	return m
}

// UserInfo returns a handler that renders the user info page.
func UserInfo(data UserInfoData) http.Handler {
	return httputil.HandlerFunc(func(w http.ResponseWriter, r *http.Request) error {
		return ui.ServePage(w, r, "UserInfo", data.ToJSON())
	})
}