mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-13 03:24:26 +00:00
70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
|
package handler
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
caos_errs "github.com/caos/zitadel/internal/errors"
|
||
|
|
es_model "github.com/caos/zitadel/internal/user/repository/eventsourcing/model"
|
||
|
|
|
||
|
|
"github.com/caos/logging"
|
||
|
|
|
||
|
|
"github.com/caos/zitadel/internal/eventstore/models"
|
||
|
|
"github.com/caos/zitadel/internal/eventstore/spooler"
|
||
|
|
"github.com/caos/zitadel/internal/user/repository/eventsourcing"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Token struct {
|
||
|
|
handler
|
||
|
|
}
|
||
|
|
|
||
|
|
const (
|
||
|
|
tokenTable = "auth.tokens"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (u *Token) MinimumCycleDuration() time.Duration { return u.cycleDuration }
|
||
|
|
|
||
|
|
func (u *Token) ViewModel() string {
|
||
|
|
return tokenTable
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u *Token) EventQuery() (*models.SearchQuery, error) {
|
||
|
|
sequence, err := u.view.GetLatestTokenSequence()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return eventsourcing.UserQuery(sequence), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u *Token) Process(event *models.Event) (err error) {
|
||
|
|
switch event.Type {
|
||
|
|
case es_model.SignedOut:
|
||
|
|
id, err := agentIDFromSession(event)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
err = u.view.DeleteSessionTokens(id, event.AggregateID, event.Sequence)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return u.view.ProcessedTokenSequence(event.Sequence)
|
||
|
|
default:
|
||
|
|
return u.view.ProcessedTokenSequence(event.Sequence)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u *Token) OnError(event *models.Event, err error) error {
|
||
|
|
logging.LogWithFields("SPOOL-3jkl4", "id", event.AggregateID).WithError(err).Warn("something went wrong in token handler")
|
||
|
|
return spooler.HandleError(event, err, u.view.GetLatestTokenFailedEvent, u.view.ProcessedTokenFailedEvent, u.view.ProcessedTokenSequence, u.errorCountUntilSkip)
|
||
|
|
}
|
||
|
|
|
||
|
|
func agentIDFromSession(event *models.Event) (string, error) {
|
||
|
|
session := make(map[string]interface{})
|
||
|
|
if err := json.Unmarshal(event.Data, session); err != nil {
|
||
|
|
logging.Log("EVEN-s3bq9").WithError(err).Error("could not unmarshal event data")
|
||
|
|
return "", caos_errs.ThrowInternal(nil, "MODEL-sd325", "could not unmarshal data")
|
||
|
|
}
|
||
|
|
return session["agentID"].(string), nil
|
||
|
|
}
|