From c5ac3a30af36b845665197138fef155cc4fea2c2 Mon Sep 17 00:00:00 2001
From: Gabe Kangas <gabek@real-ity.com>
Date: Sat, 30 Nov 2024 23:38:27 -0800
Subject: [PATCH] fix: pass in config repo instead of using the global getter
 (#4039)

---
 .../configrepository/configMigrations.go       | 18 +++++++-----------
 .../configrepository/sqlconfigrepository.go    |  2 +-
 2 files changed, 8 insertions(+), 12 deletions(-)

diff --git a/persistence/configrepository/configMigrations.go b/persistence/configrepository/configMigrations.go
index 6b9fa1a53..11e182ba6 100644
--- a/persistence/configrepository/configMigrations.go
+++ b/persistence/configrepository/configMigrations.go
@@ -13,7 +13,7 @@ const (
 	datastoreValueVersionKey = "DATA_STORE_VERSION"
 )
 
-func migrateDatastoreValues(datastore *data.Datastore) {
+func migrateDatastoreValues(datastore *data.Datastore, configRepository ConfigRepository) {
 	currentVersion, _ := datastore.GetNumber(datastoreValueVersionKey)
 	if currentVersion == 0 {
 		currentVersion = datastoreValuesVersion
@@ -25,11 +25,11 @@ func migrateDatastoreValues(datastore *data.Datastore) {
 		case 0:
 			migrateToDatastoreValues1(datastore)
 		case 1:
-			migrateToDatastoreValues2(datastore)
+			migrateToDatastoreValues2(datastore, configRepository)
 		case 2:
-			migrateToDatastoreValues3ServingEndpoint3(datastore)
+			migrateToDatastoreValues3ServingEndpoint3(configRepository)
 		case 3:
-			migrateToDatastoreValues4(datastore)
+			migrateToDatastoreValues4(datastore, configRepository)
 		default:
 			log.Fatalln("missing datastore values migration step")
 		}
@@ -59,9 +59,7 @@ func migrateToDatastoreValues1(datastore *data.Datastore) {
 	}
 }
 
-func migrateToDatastoreValues2(datastore *data.Datastore) {
-	configRepository := Get()
-
+func migrateToDatastoreValues2(datastore *data.Datastore, configRepository ConfigRepository) {
 	oldAdminPassword, _ := datastore.GetString("stream_key")
 	// Avoids double hashing the password
 	_ = datastore.SetString("admin_password_key", oldAdminPassword)
@@ -70,8 +68,7 @@ func migrateToDatastoreValues2(datastore *data.Datastore) {
 	})
 }
 
-func migrateToDatastoreValues3ServingEndpoint3(_ *data.Datastore) {
-	configRepository := Get()
+func migrateToDatastoreValues3ServingEndpoint3(configRepository ConfigRepository) {
 	s3Config := configRepository.GetS3Config()
 
 	if !s3Config.Enabled {
@@ -81,8 +78,7 @@ func migrateToDatastoreValues3ServingEndpoint3(_ *data.Datastore) {
 	_ = configRepository.SetVideoServingEndpoint(s3Config.ServingEndpoint)
 }
 
-func migrateToDatastoreValues4(datastore *data.Datastore) {
-	configRepository := Get()
+func migrateToDatastoreValues4(datastore *data.Datastore, configRepository ConfigRepository) {
 	unhashed_pass, _ := datastore.GetString("admin_password_key")
 	err := configRepository.SetAdminPassword(unhashed_pass)
 	if err != nil {
diff --git a/persistence/configrepository/sqlconfigrepository.go b/persistence/configrepository/sqlconfigrepository.go
index 6528ee40f..54a07563b 100644
--- a/persistence/configrepository/sqlconfigrepository.go
+++ b/persistence/configrepository/sqlconfigrepository.go
@@ -38,7 +38,7 @@ func New(datastore *data.Datastore) ConfigRepository {
 		datastore: datastore,
 	}
 
-	migrateDatastoreValues(datastore)
+	migrateDatastoreValues(datastore, &r)
 
 	// Set the server initialization date if needed.
 	if hasSetInitDate, _ := r.GetServerInitTime(); hasSetInitDate == nil || !hasSetInitDate.Valid {