Add system settings management api

2023-12-04 17:13:20 +08:00
parent 5c8775bf12
commit 902fcef9b2
18 changed files with 229 additions and 149 deletions
--- a/src/main/kotlin/top/fatweb/api/settings/SettingsOperator.kt
+++ b/src/main/kotlin/top/fatweb/api/settings/SettingsOperator.kt
@@ -0,0 +1,90 @@
+package top.fatweb.api.settings
+
+import com.fasterxml.jackson.databind.exc.MismatchedInputException
+import com.fasterxml.jackson.dataformat.yaml.YAMLMapper
+import com.fasterxml.jackson.module.kotlin.readValue
+import top.fatweb.api.util.StrUtil
+import java.io.File
+import java.io.IOException
+import kotlin.reflect.KMutableProperty1
+
+object SettingsOperator {
+    private const val SETTINGS_FILE_NAME = "data/config/settings.yaml"
+
+    private val yamlMapper = YAMLMapper()
+    private val settingFile = File(SETTINGS_FILE_NAME)
+    private lateinit var systemSettings: SystemSettings
+
+    init {
+        getFromSettingsFile()
+    }
+
+    @Throws(IOException::class, MismatchedInputException::class)
+    private fun getFromSettingsFile() {
+        if (settingFile.isDirectory) {
+            throw IOException("'${settingFile.absoluteFile}' is a directory, cannot create settings file")
+        }
+
+        if (!settingFile.isFile) {
+            if (settingFile.parentFile.isFile) {
+                throw IOException("'${settingFile.parentFile.absoluteFile}' is a file, cannot create settings file")
+            }
+
+            if (!settingFile.parentFile.isDirectory) {
+                if (!settingFile.parentFile.mkdirs()) {
+                    throw IOException("Cannot make directory: ${settingFile.parentFile.absolutePath}")
+                }
+            }
+            settingFile.delete()
+            settingFile.createNewFile()
+        }
+
+        if (settingFile.length() == 0L) {
+            systemSettings = SystemSettings()
+            saveSettingsToFile()
+        }
+
+        systemSettings = yamlMapper.readValue(settingFile)
+    }
+
+    private fun saveSettingsToFile() {
+        yamlMapper.writeValue(settingFile, systemSettings)
+    }
+
+    fun <V> setMailValue(field: KMutableProperty1<MailSettings, V?>, value: V?) {
+        systemSettings.mail?.let {
+            if (field == MailSettings::password) {
+                getMailValue(MailSettings::password)?.let { password ->
+                    if (StrUtil.md5(password) == value) {
+                        return
+                    }
+                }
+            }
+            field.set(it, value)
+        } ?: {
+            MailSettings().also {
+                field.set(it, value)
+                systemSettings.mail = it
+            }
+        }
+
+        saveSettingsToFile()
+    }
+
+    fun <V> getMailValue(field: KMutableProperty1<MailSettings, V?>): V? = systemSettings.mail?.let(field)
+
+    fun settings() = systemSettings.copy(
+        mail = systemSettings.mail?.copy()
+    ).apply {
+        mail?.apply {
+            password = password?.let {
+                StrUtil.md5(it)
+            }
+        }
+    }
+
+    fun overwrite(systemSettings: SystemSettings) {
+        this.systemSettings = systemSettings
+        saveSettingsToFile()
+    }
+}