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 }