Tweaked project structure (#88)

internal/db/admin_setting.go

@@ -0,0 +1,64 @@
+package db
+
+import (
+	"gorm.io/gorm/clause"
+)
+
+type AdminSetting struct {
+	Key   string `gorm:"uniqueIndex"`
+	Value string
+}
+
+const (
+	SettingDisableSignup    = "disable-signup"
+	SettingRequireLogin     = "require-login"
+	SettingDisableLoginForm = "disable-login-form"
+	SettingDisableGravatar  = "disable-gravatar"
+)
+
+func GetSetting(key string) (string, error) {
+	var setting AdminSetting
+	err := db.Where("key = ?", key).First(&setting).Error
+	return setting.Value, err
+}
+
+func GetSettings() (map[string]string, error) {
+	var settings []AdminSetting
+	err := db.Find(&settings).Error
+	if err != nil {
+		return nil, err
+	}
+
+	result := make(map[string]string)
+	for _, setting := range settings {
+		result[setting.Key] = setting.Value
+	}
+
+	return result, nil
+}
+
+func UpdateSetting(key string, value string) error {
+	return db.Clauses(clause.OnConflict{
+		Columns:   []clause.Column{{Name: "key"}}, // key column
+		DoUpdates: clause.AssignmentColumns([]string{"value"}),
+	}).Create(&AdminSetting{
+		Key:   key,
+		Value: value,
+	}).Error
+}
+
+func setSetting(key string, value string) error {
+	return db.Create(&AdminSetting{Key: key, Value: value}).Error
+}
+
+func initAdminSettings(settings map[string]string) error {
+	for key, value := range settings {
+		if err := setSetting(key, value); err != nil {
+			if !IsUniqueConstraintViolation(err) {
+				return err
+			}
+		}
+	}
+
+	return nil
+}