0

fix: pass in config repo instead of using the global getter (#4039)

This commit is contained in:
Gabe Kangas 2024-11-30 23:38:27 -08:00 committed by GitHub
parent d8aed658a8
commit c5ac3a30af
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 12 deletions

View File

@ -13,7 +13,7 @@ const (
datastoreValueVersionKey = "DATA_STORE_VERSION" datastoreValueVersionKey = "DATA_STORE_VERSION"
) )
func migrateDatastoreValues(datastore *data.Datastore) { func migrateDatastoreValues(datastore *data.Datastore, configRepository ConfigRepository) {
currentVersion, _ := datastore.GetNumber(datastoreValueVersionKey) currentVersion, _ := datastore.GetNumber(datastoreValueVersionKey)
if currentVersion == 0 { if currentVersion == 0 {
currentVersion = datastoreValuesVersion currentVersion = datastoreValuesVersion
@ -25,11 +25,11 @@ func migrateDatastoreValues(datastore *data.Datastore) {
case 0: case 0:
migrateToDatastoreValues1(datastore) migrateToDatastoreValues1(datastore)
case 1: case 1:
migrateToDatastoreValues2(datastore) migrateToDatastoreValues2(datastore, configRepository)
case 2: case 2:
migrateToDatastoreValues3ServingEndpoint3(datastore) migrateToDatastoreValues3ServingEndpoint3(configRepository)
case 3: case 3:
migrateToDatastoreValues4(datastore) migrateToDatastoreValues4(datastore, configRepository)
default: default:
log.Fatalln("missing datastore values migration step") log.Fatalln("missing datastore values migration step")
} }
@ -59,9 +59,7 @@ func migrateToDatastoreValues1(datastore *data.Datastore) {
} }
} }
func migrateToDatastoreValues2(datastore *data.Datastore) { func migrateToDatastoreValues2(datastore *data.Datastore, configRepository ConfigRepository) {
configRepository := Get()
oldAdminPassword, _ := datastore.GetString("stream_key") oldAdminPassword, _ := datastore.GetString("stream_key")
// Avoids double hashing the password // Avoids double hashing the password
_ = datastore.SetString("admin_password_key", oldAdminPassword) _ = datastore.SetString("admin_password_key", oldAdminPassword)
@ -70,8 +68,7 @@ func migrateToDatastoreValues2(datastore *data.Datastore) {
}) })
} }
func migrateToDatastoreValues3ServingEndpoint3(_ *data.Datastore) { func migrateToDatastoreValues3ServingEndpoint3(configRepository ConfigRepository) {
configRepository := Get()
s3Config := configRepository.GetS3Config() s3Config := configRepository.GetS3Config()
if !s3Config.Enabled { if !s3Config.Enabled {
@ -81,8 +78,7 @@ func migrateToDatastoreValues3ServingEndpoint3(_ *data.Datastore) {
_ = configRepository.SetVideoServingEndpoint(s3Config.ServingEndpoint) _ = configRepository.SetVideoServingEndpoint(s3Config.ServingEndpoint)
} }
func migrateToDatastoreValues4(datastore *data.Datastore) { func migrateToDatastoreValues4(datastore *data.Datastore, configRepository ConfigRepository) {
configRepository := Get()
unhashed_pass, _ := datastore.GetString("admin_password_key") unhashed_pass, _ := datastore.GetString("admin_password_key")
err := configRepository.SetAdminPassword(unhashed_pass) err := configRepository.SetAdminPassword(unhashed_pass)
if err != nil { if err != nil {

View File

@ -38,7 +38,7 @@ func New(datastore *data.Datastore) ConfigRepository {
datastore: datastore, datastore: datastore,
} }
migrateDatastoreValues(datastore) migrateDatastoreValues(datastore, &r)
// Set the server initialization date if needed. // Set the server initialization date if needed.
if hasSetInitDate, _ := r.GetServerInitTime(); hasSetInitDate == nil || !hasSetInitDate.Valid { if hasSetInitDate, _ := r.GetServerInitTime(); hasSetInitDate == nil || !hasSetInitDate.Valid {