feat(webui): implement state versions list

3 files changed, 84 insertions, 2 deletions

diff --git a/pkg/database/accounts.go b/pkg/database/accounts.go
index 377ca80..9adb32d 100644
--- a/pkg/database/accounts.go
+++ b/pkg/database/accounts.go
@@ -47,6 +47,31 @@ func (db *DB) InitAdminAccount() error {
 	})
 }
 
+func (db *DB) LoadAccountUsernames() (map[int]string, error) {
+	rows, err := db.Query(
+		`SELECT id, username FROM accounts;`)
+	if err != nil {
+		return nil, fmt.Errorf("failed to load accounts from database: %w", err)
+	}
+	defer rows.Close()
+	accounts := make(map[int]string)
+	for rows.Next() {
+		var (
+			id       int
+			username string
+		)
+		err = rows.Scan(&id, &username)
+		if err != nil {
+			return nil, fmt.Errorf("failed to load account from row: %w", err)
+		}
+		accounts[id] = username
+	}
+	if err := rows.Err(); err != nil {
+		return nil, fmt.Errorf("failed to load accounts from rows: %w", err)
+	}
+	return accounts, nil
+}
+
 func (db *DB) LoadAccountById(id int) (*model.Account, error) {
 	account := model.Account{
 		Id: id,
diff --git a/pkg/database/states.go b/pkg/database/states.go
index 1bd3d76..d65dd7c 100644
--- a/pkg/database/states.go
+++ b/pkg/database/states.go
@@ -45,7 +45,29 @@ func (db *DB) GetState(path string) ([]byte, error) {
 	return db.dataEncryptionKey.DecryptAES256(encryptedData)
 }
 
-func (db *DB) LoadStatesByPath() ([]model.State, error) {
+func (db *DB) LoadStateById(stateId int) (*model.State, error) {
+	state := model.State{
+		Id: stateId,
+	}
+	var (
+		created int64
+		updated int64
+	)
+	err := db.QueryRow(
+		`SELECT created, lock, path, updated FROM states WHERE id = ?;`,
+		stateId).Scan(&created, &state.Lock, &state.Path, &updated)
+	if err != nil {
+		if errors.Is(err, sql.ErrNoRows) {
+			return nil, nil
+		}
+		return nil, fmt.Errorf("failed to load state id %s from database: %w", stateId, err)
+	}
+	state.Created = time.Unix(created, 0)
+	state.Updated = time.Unix(updated, 0)
+	return &state, nil
+}
+
+func (db *DB) LoadStates() ([]model.State, error) {
 	rows, err := db.Query(
 		`SELECT created, id, lock, path, updated FROM states;`)
 	if err != nil {
@@ -61,7 +83,7 @@ func (db *DB) LoadStatesByPath() ([]model.State, error) {
 		)
 		err = rows.Scan(&created, &state.Id, &state.Lock, &state.Path, &updated)
 		if err != nil {
-			return nil, fmt.Errorf("failed to load states from row: %w", err)
+			return nil, fmt.Errorf("failed to load state from row: %w", err)
 		}
 		state.Created = time.Unix(created, 0)
 		state.Updated = time.Unix(updated, 0)
diff --git a/pkg/database/versions.go b/pkg/database/versions.go
new file mode 100644
index 0000000..5a6f11f
--- /dev/null
+++ b/pkg/database/versions.go
@@ -0,0 +1,35 @@
+package database
+
+import (
+	"fmt"
+	"time"
+
+	"git.adyxax.org/adyxax/tfstated/pkg/model"
+)
+
+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
+}