aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/util/data/MultiFileDataHolder.kt
blob: 94c6f053a18bd02ab038915dc9d088295f2acf76 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package moe.nea.firmament.util.data

import kotlinx.serialization.KSerializer
import kotlin.io.path.createDirectories
import kotlin.io.path.deleteExisting
import kotlin.io.path.exists
import kotlin.io.path.extension
import kotlin.io.path.listDirectoryEntries
import kotlin.io.path.nameWithoutExtension
import kotlin.io.path.readText
import kotlin.io.path.writeText
import moe.nea.firmament.Firmament

abstract class MultiFileDataHolder<T>(
	val dataSerializer: KSerializer<T>,
	val configName: String
) { // TODO: abstract this + ProfileSpecificDataHolder
	val configDirectory = Firmament.CONFIG_DIR.resolve(configName)
	private var allData = readValues()
	protected fun readValues(): MutableMap<String, T> {
		if (!configDirectory.exists()) {
			configDirectory.createDirectories()
		}
		val profileFiles = configDirectory.listDirectoryEntries()
		return profileFiles
			.filter { it.extension == "json" }
			.mapNotNull {
				try {
					it.nameWithoutExtension to Firmament.json.decodeFromString(dataSerializer, it.readText())
				} catch (e: Exception) { /* Expecting IOException and SerializationException, but Kotlin doesn't allow multi catches*/
					IDataHolder.badLoads.add(configName)
					Firmament.logger.error(
						"Exception during loading of multi file data holder $it ($configName). This will reset that profiles config.",
						e
					)
					null
				}
			}.toMap().toMutableMap()
	}

	fun save() {
		if (!configDirectory.exists()) {
			configDirectory.createDirectories()
		}
		val c = allData
		configDirectory.listDirectoryEntries().forEach {
			if (it.nameWithoutExtension !in c.mapKeys { it.toString() }) {
				it.deleteExisting()
			}
		}
		c.forEach { (name, value) ->
			val f = configDirectory.resolve("$name.json")
			f.writeText(Firmament.json.encodeToString(dataSerializer, value))
		}
	}

	fun list(): Map<String, T> = allData
	val validPathRegex = "[a-zA-Z0-9_][a-zA-Z0-9\\-_.]*".toPattern()
	fun insert(name: String, value: T) {
		require(validPathRegex.matcher(name).matches()) { "Not a valid name: $name" }
		allData[name] = value
	}
}