fix: some settings don't take effect at startup

AlistGo · Sep 23, 2022 · a570e4c · a570e4c
1 parent 539c47b
commit a570e4c
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/internal/bootstrap/data/setting.go b/internal/bootstrap/data/setting.go
@@ -34,14 +34,19 @@ func initSettings() {
 	// insert new items
 	for i := range initialSettingItems {
 		v := initialSettingItems[i]
-		_, err := db.GetSettingItemByKey(v.Key)
+		stored, err := db.GetSettingItemByKey(v.Key)
 		if errors.Is(err, gorm.ErrRecordNotFound) || v.Key == conf.VERSION {
 			err = db.SaveSettingItem(v)
 			if err != nil {
 				log.Fatalf("failed create setting: %+v", err)
 			}
 		} else if err != nil {
 			log.Fatalf("failed get setting: %+v", err)
+		} else {
+			err = db.SaveSettingItem(*stored)
+			if err != nil {
+				log.Fatalf("failed resave setting: %+v", err)
+			}
 		}
 	}
 }
@@ -107,7 +112,7 @@ func InitialSettings() []model.SettingItem {
 		// global settings
 		{Key: conf.HideFiles, Value: "/\\/README.md/i", Type: conf.TypeText, Group: model.GLOBAL},
 		{Key: "package_download", Value: "true", Type: conf.TypeBool, Group: model.GLOBAL},
-        {Key: conf.CustomizeHead,Value:`<script src="https://polyfill.io/v3/polyfill.min.js?features=String.prototype.replaceAll"></script>`, Type: conf.TypeText, Group: model.GLOBAL, Flag: model.PRIVATE},
+		{Key: conf.CustomizeHead, Value: `<script src="https://polyfill.io/v3/polyfill.min.js?features=String.prototype.replaceAll"></script>`, Type: conf.TypeText, Group: model.GLOBAL, Flag: model.PRIVATE},
 		{Key: conf.CustomizeBody, Type: conf.TypeText, Group: model.GLOBAL, Flag: model.PRIVATE},
 		{Key: conf.LinkExpiration, Value: "0", Type: conf.TypeNumber, Group: model.GLOBAL, Flag: model.PRIVATE},
 		{Key: conf.PrivacyRegs, Value: `(?:(?:\d|[1-9]\d|1\d\d|2[0-4]\d|25[0-5])\.){3}(?:\d|[1-9]\d|1\d\d|2[0-4]\d|25[0-5])