94 lines
2.9 KiB
Go
94 lines
2.9 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/dexidp/dex/storage"
|
|
)
|
|
|
|
// CreateOfflineSessions saves provided offline session into the database.
|
|
func (d *Database) CreateOfflineSessions(session storage.OfflineSessions) error {
|
|
encodedRefresh, err := json.Marshal(session.Refresh)
|
|
if err != nil {
|
|
return fmt.Errorf("encode refresh offline session: %w", err)
|
|
}
|
|
|
|
id := offlineSessionID(session.UserID, session.ConnID, d.hasher)
|
|
_, err = d.client.OfflineSession.Create().
|
|
SetID(id).
|
|
SetUserID(session.UserID).
|
|
SetConnID(session.ConnID).
|
|
SetConnectorData(session.ConnectorData).
|
|
SetRefresh(encodedRefresh).
|
|
Save(context.TODO())
|
|
if err != nil {
|
|
return convertDBError("create offline session: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetOfflineSessions extracts an offline session from the database by user id and connector id.
|
|
func (d *Database) GetOfflineSessions(userID, connID string) (storage.OfflineSessions, error) {
|
|
id := offlineSessionID(userID, connID, d.hasher)
|
|
|
|
offlineSession, err := d.client.OfflineSession.Get(context.TODO(), id)
|
|
if err != nil {
|
|
return storage.OfflineSessions{}, convertDBError("get offline session: %w", err)
|
|
}
|
|
return toStorageOfflineSession(offlineSession), nil
|
|
}
|
|
|
|
// DeleteOfflineSessions deletes an offline session from the database by user id and connector id.
|
|
func (d *Database) DeleteOfflineSessions(userID, connID string) error {
|
|
id := offlineSessionID(userID, connID, d.hasher)
|
|
|
|
err := d.client.OfflineSession.DeleteOneID(id).Exec(context.TODO())
|
|
if err != nil {
|
|
return convertDBError("delete offline session: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// UpdateOfflineSessions changes an offline session by user id and connector id using an updater function.
|
|
func (d *Database) UpdateOfflineSessions(userID string, connID string, updater func(s storage.OfflineSessions) (storage.OfflineSessions, error)) error {
|
|
id := offlineSessionID(userID, connID, d.hasher)
|
|
|
|
tx, err := d.BeginTx(context.TODO())
|
|
if err != nil {
|
|
return convertDBError("update offline session tx: %w", err)
|
|
}
|
|
|
|
offlineSession, err := tx.OfflineSession.Get(context.TODO(), id)
|
|
if err != nil {
|
|
return rollback(tx, "update offline session database: %w", err)
|
|
}
|
|
|
|
newOfflineSession, err := updater(toStorageOfflineSession(offlineSession))
|
|
if err != nil {
|
|
return rollback(tx, "update offline session updating: %w", err)
|
|
}
|
|
|
|
encodedRefresh, err := json.Marshal(newOfflineSession.Refresh)
|
|
if err != nil {
|
|
return rollback(tx, "encode refresh offline session: %w", err)
|
|
}
|
|
|
|
_, err = tx.OfflineSession.UpdateOneID(id).
|
|
SetUserID(newOfflineSession.UserID).
|
|
SetConnID(newOfflineSession.ConnID).
|
|
SetConnectorData(newOfflineSession.ConnectorData).
|
|
SetRefresh(encodedRefresh).
|
|
Save(context.TODO())
|
|
if err != nil {
|
|
return rollback(tx, "update offline session uploading: %w", err)
|
|
}
|
|
|
|
if err = tx.Commit(); err != nil {
|
|
return rollback(tx, "update offline session commit: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|