feat: Add ent-based sqlite3 storage

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

storage/ent/client/devicetoken.go

@@ -0,0 +1,76 @@
+package client
+
+import (
+	"context"
+
+	"github.com/dexidp/dex/storage"
+	"github.com/dexidp/dex/storage/ent/db/devicetoken"
+)
+
+// CreateDeviceToken saves provided token into the database.
+func (d *Database) CreateDeviceToken(token storage.DeviceToken) error {
+	_, err := d.client.DeviceToken.Create().
+		SetDeviceCode(token.DeviceCode).
+		SetToken([]byte(token.Token)).
+		SetPollInterval(token.PollIntervalSeconds).
+		// Save utc time into database because ent doesn't support comparing dates with different timezones
+		SetExpiry(token.Expiry.UTC()).
+		SetLastRequest(token.LastRequestTime.UTC()).
+		SetStatus(token.Status).
+		Save(context.TODO())
+	if err != nil {
+		return convertDBError("create device token: %w", err)
+	}
+	return nil
+}
+
+// GetDeviceToken extracts a token from the database by device code.
+func (d *Database) GetDeviceToken(deviceCode string) (storage.DeviceToken, error) {
+	deviceToken, err := d.client.DeviceToken.Query().
+		Where(devicetoken.DeviceCode(deviceCode)).
+		Only(context.TODO())
+	if err != nil {
+		return storage.DeviceToken{}, convertDBError("get device token: %w", err)
+	}
+	return toStorageDeviceToken(deviceToken), nil
+}
+
+// UpdateDeviceToken changes a token by device code using an updater function and saves it to the database.
+func (d *Database) UpdateDeviceToken(deviceCode string, updater func(old storage.DeviceToken) (storage.DeviceToken, error)) error {
+	tx, err := d.client.Tx(context.TODO())
+	if err != nil {
+		return convertDBError("update device token tx: %w", err)
+	}
+
+	token, err := tx.DeviceToken.Query().
+		Where(devicetoken.DeviceCode(deviceCode)).
+		Only(context.TODO())
+	if err != nil {
+		return rollback(tx, "update device token database: %w", err)
+	}
+
+	newToken, err := updater(toStorageDeviceToken(token))
+	if err != nil {
+		return rollback(tx, "update device token updating: %w", err)
+	}
+
+	_, err = tx.DeviceToken.Update().
+		Where(devicetoken.DeviceCode(newToken.DeviceCode)).
+		SetDeviceCode(newToken.DeviceCode).
+		SetToken([]byte(newToken.Token)).
+		SetPollInterval(newToken.PollIntervalSeconds).
+		// Save utc time into database because ent doesn't support comparing dates with different timezones
+		SetExpiry(newToken.Expiry.UTC()).
+		SetLastRequest(newToken.LastRequestTime.UTC()).
+		SetStatus(newToken.Status).
+		Save(context.TODO())
+	if err != nil {
+		return rollback(tx, "update device token uploading: %w", err)
+	}
+
+	if err = tx.Commit(); err != nil {
+		return rollback(tx, "update device token commit: %w", err)
+	}
+
+	return nil
+}