SpeziFoundation

core/utils/src/main/kotlin/edu/stanford/spezi/core/utils/foundation/RepositoryAnchor.kt

@@ -0,0 +1,3 @@
+package edu.stanford.spezi.core.utils.foundation
+
+interface RepositoryAnchor

core/utils/src/main/kotlin/edu/stanford/spezi/core/utils/foundation/SharedRepository.kt

@@ -0,0 +1,74 @@
+package edu.stanford.spezi.core.utils.foundation
+
+import edu.stanford.spezi.core.utils.foundation.knowledgesource.ComputedKnowledgeSource
+import edu.stanford.spezi.core.utils.foundation.knowledgesource.ComputedKnowledgeSourceStoragePolicy
+import edu.stanford.spezi.core.utils.foundation.knowledgesource.DefaultProvidingKnowledgeSource
+import edu.stanford.spezi.core.utils.foundation.knowledgesource.KnowledgeSource
+import edu.stanford.spezi.core.utils.foundation.knowledgesource.OptionalComputedKnowledgeSource
+import kotlin.reflect.KClass
+
+@Suppress("detekt:TooManyFunctions")
+interface SharedRepository<Anchor : RepositoryAnchor> {
+    operator fun <Value : Any> get(source: KnowledgeSource<Anchor, Value>): Value?
+    operator fun <Value : Any> set(source: KnowledgeSource<Anchor, Value>, value: Value?)
+
+    operator fun <Value : Any> get(
+        source: DefaultProvidingKnowledgeSource<Anchor, Value>,
+    ): Value = getOrDefault(source)
+
+    fun <Value : Any> getOrDefault(
+        source: DefaultProvidingKnowledgeSource<Anchor, Value>,
+    ): Value {
+        return get(
+            source as KnowledgeSource<Anchor, Value>,
+        ) ?: source.defaultValue
+    }
+
+    operator fun <Value : Any> get(
+        source: ComputedKnowledgeSource<Anchor, Value>,
+    ): Value = getOrComputed(source)
+
+    fun <Value : Any> getOrComputed(
+        source: ComputedKnowledgeSource<Anchor, Value>,
+    ): Value {
+        return get(
+            source as KnowledgeSource<Anchor, Value>,
+        ) ?: when (source.storagePolicy) {
+            is ComputedKnowledgeSourceStoragePolicy.AlwaysCompute -> {
+                source.compute(this)
+            }
+            is ComputedKnowledgeSourceStoragePolicy.Store -> {
+                val value = source.compute(this)
+                this[source] = value
+                value
+            }
+        }
+    }
+
+    operator fun <Value : Any> get(
+        source: OptionalComputedKnowledgeSource<Anchor, Value>,
+    ): Value? = getOrOptionalComputed(source)
+
+    fun <Value : Any> getOrOptionalComputed(
+        source: OptionalComputedKnowledgeSource<Anchor, Value>,
+    ): Value? {
+        return get(
+            source as KnowledgeSource<Anchor, Value>,
+        ) ?: when (source.storagePolicy) {
+            ComputedKnowledgeSourceStoragePolicy.AlwaysCompute -> {
+                source.compute(this)
+            }
+            ComputedKnowledgeSourceStoragePolicy.Store -> {
+                val value = source.compute(this)
+                this[source] = value
+                value
+            }
+        }
+    }
+
+    fun <Value : Any> collect(allOf: KClass<Value>): List<Value>
+
+    fun <Value : Any> contains(source: KnowledgeSource<Anchor, Value>): Boolean {
+        return get(source) != null
+    }
+}

core/utils/src/main/kotlin/edu/stanford/spezi/core/utils/foundation/builtin/ValueRepository.kt

@@ -0,0 +1,30 @@
+package edu.stanford.spezi.core.utils.foundation.builtin
+
+import edu.stanford.spezi.core.utils.foundation.RepositoryAnchor
+import edu.stanford.spezi.core.utils.foundation.SharedRepository
+import edu.stanford.spezi.core.utils.foundation.knowledgesource.KnowledgeSource
+import java.util.concurrent.ConcurrentHashMap
+import kotlin.reflect.KClass
+
+data class ValueRepository<Anchor : RepositoryAnchor>(
+    internal var storage: ConcurrentHashMap<KnowledgeSource<Anchor, *>, Any> = ConcurrentHashMap(),
+) : SharedRepository<Anchor>, Sequence<Map.Entry<KnowledgeSource<Anchor, *>, Any>> {
+    @Suppress("UNCHECKED_CAST")
+    override operator fun <Value : Any> get(
+        source: KnowledgeSource<Anchor, Value>,
+    ): Value? = storage[source] as? Value
+
+    override operator fun <Value : Any> set(
+        source: KnowledgeSource<Anchor, Value>,
+        value: Value?,
+    ) {
+        value?.let {
+            storage[source] = it
+        } ?: storage.remove(source)
+    }
+
+    override fun <Value : Any> collect(allOf: KClass<Value>) =
+        storage.values.filterIsInstance(allOf.java)
+
+    override fun iterator() = storage.iterator()
+}

core/utils/src/main/kotlin/edu/stanford/spezi/core/utils/foundation/knowledgesource/ComputedKnowledgeSource.kt

@@ -0,0 +1,8 @@
+package edu.stanford.spezi.core.utils.foundation.knowledgesource
+
+import edu.stanford.spezi.core.utils.foundation.RepositoryAnchor
+import edu.stanford.spezi.core.utils.foundation.SharedRepository
+
+interface ComputedKnowledgeSource<Anchor : RepositoryAnchor, Value : Any> : SomeComputedKnowledgeSource<Anchor, Value> {
+    fun compute(repository: SharedRepository<Anchor>): Value
+}

core/utils/src/main/kotlin/edu/stanford/spezi/core/utils/foundation/knowledgesource/DefaultProvidingKnowledgeSource.kt

@@ -0,0 +1,7 @@
+package edu.stanford.spezi.core.utils.foundation.knowledgesource
+
+import edu.stanford.spezi.core.utils.foundation.RepositoryAnchor
+
+interface DefaultProvidingKnowledgeSource<Anchor : RepositoryAnchor, Value : Any> : KnowledgeSource<Anchor, Value> {
+    val defaultValue: Value
+}

core/utils/src/main/kotlin/edu/stanford/spezi/core/utils/foundation/knowledgesource/KnowledgeSource.kt

@@ -0,0 +1,5 @@
+package edu.stanford.spezi.core.utils.foundation.knowledgesource
+
+import edu.stanford.spezi.core.utils.foundation.RepositoryAnchor
+
+interface KnowledgeSource<Anchor : RepositoryAnchor, Value : Any>

core/utils/src/main/kotlin/edu/stanford/spezi/core/utils/foundation/knowledgesource/OptionalComputedKnowledgeSource.kt

@@ -0,0 +1,12 @@
+package edu.stanford.spezi.core.utils.foundation.knowledgesource
+
+import edu.stanford.spezi.core.utils.foundation.RepositoryAnchor
+import edu.stanford.spezi.core.utils.foundation.SharedRepository
+
+interface OptionalComputedKnowledgeSource<
+    Anchor : RepositoryAnchor,
+    Value : Any,
+    > : SomeComputedKnowledgeSource<Anchor, Value> {
+
+    fun compute(repository: SharedRepository<Anchor>): Value?
+}

core/utils/src/main/kotlin/edu/stanford/spezi/core/utils/foundation/knowledgesource/SomeComputedKnowledgeSource.kt

@@ -0,0 +1,12 @@
+package edu.stanford.spezi.core.utils.foundation.knowledgesource
+
+import edu.stanford.spezi.core.utils.foundation.RepositoryAnchor
+
+sealed interface ComputedKnowledgeSourceStoragePolicy {
+    data object AlwaysCompute : ComputedKnowledgeSourceStoragePolicy
+    data object Store : ComputedKnowledgeSourceStoragePolicy
+}
+
+interface SomeComputedKnowledgeSource<Anchor : RepositoryAnchor, Value : Any> : KnowledgeSource<Anchor, Value> {
+    val storagePolicy: ComputedKnowledgeSourceStoragePolicy
+}