zitadel/internal/api/ui/login/select_user_handler.go

package login

import (
	"net/http"

	http_mw "github.com/zitadel/zitadel/internal/api/http/middleware"
	"github.com/zitadel/zitadel/internal/domain"
)

const (
	tmplUserSelection = "userselection"
)

type userSelectionFormData struct {
	UserID string `schema:"userID"`
}

func (l *Login) renderUserSelection(w http.ResponseWriter, r *http.Request, authReq *domain.AuthRequest, selectionData *domain.SelectUserStep) {
	translator := l.getTranslator(r.Context(), authReq)

	linking := len(authReq.LinkingUsers) > 0

	titleI18nKey := "SelectAccount.Title"
	descriptionI18nKey := "SelectAccount.Description"
	if linking {
		titleI18nKey = "SelectAccount.TitleLinking"
		descriptionI18nKey = "SelectAccount.DescriptionLinking"
	}
	data := userSelectionData{
		baseData: l.getBaseData(r, authReq, translator, titleI18nKey, descriptionI18nKey, nil),
		Users:    selectionData.Users,
		Linking:  linking,
	}
	l.renderer.RenderTemplate(w, r, translator, l.renderer.Templates[tmplUserSelection], data, nil)
}

func (l *Login) handleSelectUser(w http.ResponseWriter, r *http.Request) {
	data := new(userSelectionFormData)
	authSession, err := l.ensureAuthRequestAndParseData(r, data)
	if err != nil {
		l.renderError(w, r, authSession, err)
		return
	}
	if data.UserID == "0" {
		l.renderLogin(w, r, authSession, nil)
		return
	}
	userAgentID, _ := http_mw.UserAgentIDFromCtx(r.Context())
	err = l.authRepo.SelectUser(r.Context(), authSession.ID, data.UserID, userAgentID)
	if err != nil {
		l.renderError(w, r, authSession, err)
		return
	}
	l.renderNextStep(w, r, authSession)
}