feat: Add ent-based sqlite3 storage

Signed-off-by: m.nabokikh <maksim.nabokikh@flant.com>

storage/ent/client/offlinesession.go

@@ -0,0 +1,93 @@
+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
+}
+
+// UpdatePassword 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.client.Tx(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 password commit: %w", err)
+	}
+
+	return nil
+}