1 files changed, 67 insertions, 0 deletions
diff --git a/pkg/database/versions.go b/pkg/database/versions.go
new file mode 100644
index 0000000..f05d230
--- /dev/null
+++ b/pkg/database/versions.go
@@ -0,0 +1,67 @@
+package database
+
+import (
+	"database/sql"
+	"errors"
+	"fmt"
+	"time"
+
+	"git.adyxax.org/adyxax/tfstated/pkg/model"
+)
+
+func (db *DB) LoadVersionById(id int) (*model.Version, error) {
+	version := model.Version{
+		Id: id,
+	}
+	var (
+		created       int64
+		encryptedData []byte
+	)
+	err := db.QueryRow(
+		`SELECT account_id, state_id, data, lock, created FROM versions WHERE id = ?;`,
+		id).Scan(
+		&version.AccountId,
+		&version.StateId,
+		&encryptedData,
+		&version.Lock,
+		&created)
+	if err != nil {
+		if errors.Is(err, sql.ErrNoRows) {
+			return nil, nil
+		}
+		return nil, fmt.Errorf("failed to load version id %d from database: %w", id, err)
+	}
+	version.Created = time.Unix(created, 0)
+	version.Data, err = db.dataEncryptionKey.DecryptAES256(encryptedData)
+	if err != nil {
+		return nil, fmt.Errorf("failed to decrypt version %d data: %w", id, err)
+	}
+	return &version, nil
+}
+
+func (db *DB) LoadVersionsByState(state *model.State) ([]model.Version, error) {
+	rows, err := db.Query(
+		`SELECT account_id, created, data, id, lock
+           FROM versions
+           WHERE state_id = ?
+           ORDER BY id DESC;`, state.Id)
+	if err != nil {
+		return nil, fmt.Errorf("failed to load versions from database: %w", err)
+	}
+	defer rows.Close()
+	versions := make([]model.Version, 0)
+	for rows.Next() {
+		var version model.Version
+		var created int64
+		err = rows.Scan(&version.AccountId, &created, &version.Data, &version.Id, &version.Lock)
+		if err != nil {
+			return nil, fmt.Errorf("failed to load version from row: %w", err)
+		}
+		version.Created = time.Unix(created, 0)
+		versions = append(versions, version)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, fmt.Errorf("failed to load versions from rows: %w", err)
+	}
+	return versions, nil
+}