Skip to content

Commit cce8b82

Browse files
Daeda88Daeda88
committed
Moved Firestore internals to different package
1 parent f615bc3 commit cce8b82

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

43 files changed

+1720
-1295
lines changed

‎firebase-firestore/src/androidMain/kotlin/dev/gitlive/firebase/firestore/firestore.kt‎

Lines changed: 1 addition & 404 deletions
Large diffs are not rendered by default.

‎firebase-firestore/src/androidMain/kotlin/dev/gitlive/firebase/firestore/internal/NativeCollectionReferenceWrapper.kt‎

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
package dev.gitlive.firebase.firestore.internal
2+
3+
import dev.gitlive.firebase.firestore.NativeCollectionReference
4+
import dev.gitlive.firebase.internal.EncodedObject
5+
import dev.gitlive.firebase.internal.android
6+
import kotlinx.coroutines.tasks.await
7+
8+
@PublishedApi
9+
internal actual class NativeCollectionReferenceWrapper internal actual constructor(actual override val native: NativeCollectionReference) : NativeQueryWrapper(native) {
10+
11+
actual val path: String
12+
get() = native.path
13+
14+
actual val document: NativeDocumentReference
15+
get() = NativeDocumentReference(native.document())
16+
17+
actual val parent: NativeDocumentReference?
18+
get() = native.parent?.let{ NativeDocumentReference(it) }
19+
20+
actual fun document(documentPath: String) =
21+
NativeDocumentReference(native.document(documentPath))
22+
23+
actual suspend fun addEncoded(data: EncodedObject) =
24+
NativeDocumentReference(native.add(data.android).await())
25+
}

‎firebase-firestore/src/androidMain/kotlin/dev/gitlive/firebase/firestore/internal/NativeDocumentReference.kt‎

Lines changed: 88 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,88 @@
1+
package dev.gitlive.firebase.firestore.internal
2+
3+
import com.google.android.gms.tasks.TaskExecutors
4+
import com.google.firebase.firestore.MetadataChanges
5+
import dev.gitlive.firebase.firestore.EncodedFieldPath
6+
import dev.gitlive.firebase.firestore.NativeDocumentReferenceType
7+
import dev.gitlive.firebase.firestore.NativeDocumentSnapshot
8+
import dev.gitlive.firebase.firestore.Source
9+
import dev.gitlive.firebase.firestore.performUpdate
10+
import dev.gitlive.firebase.internal.EncodedObject
11+
import dev.gitlive.firebase.internal.android
12+
import kotlinx.coroutines.channels.ProducerScope
13+
import kotlinx.coroutines.channels.awaitClose
14+
import kotlinx.coroutines.flow.Flow
15+
import kotlinx.coroutines.flow.callbackFlow
16+
import kotlinx.coroutines.tasks.await
17+
18+
@PublishedApi
19+
internal actual class NativeDocumentReference actual constructor(actual val nativeValue: NativeDocumentReferenceType) {
20+
val android: NativeDocumentReferenceType by ::nativeValue
21+
actual val id: String
22+
get() = android.id
23+
24+
actual val path: String
25+
get() = android.path
26+
27+
actual val parent: NativeCollectionReferenceWrapper
28+
get() = NativeCollectionReferenceWrapper(android.parent)
29+
30+
actual fun collection(collectionPath: String) = android.collection(collectionPath)
31+
32+
actual suspend fun get(source: Source) =
33+
android.get(source.toAndroidSource()).await()
34+
35+
actual suspend fun setEncoded(encodedData: EncodedObject, setOptions: SetOptions) {
36+
val task = (setOptions.android?.let {
37+
android.set(encodedData.android, it)
38+
} ?: android.set(encodedData.android))
39+
task.await()
40+
}
41+
42+
actual suspend fun updateEncoded(encodedData: EncodedObject) {
43+
android.update(encodedData.android).await()
44+
}
45+
46+
actual suspend fun updateEncodedFieldsAndValues(encodedFieldsAndValues: List<Pair<String, Any?>>) {
47+
encodedFieldsAndValues.takeUnless { encodedFieldsAndValues.isEmpty() }?.let {
48+
android.update(encodedFieldsAndValues.toMap())
49+
}?.await()
50+
}
51+
52+
actual suspend fun updateEncodedFieldPathsAndValues(encodedFieldsAndValues: List<Pair<EncodedFieldPath, Any?>>) {
53+
encodedFieldsAndValues.takeUnless { encodedFieldsAndValues.isEmpty() }
54+
?.performUpdate { field, value, moreFieldsAndValues ->
55+
android.update(field, value, *moreFieldsAndValues)
56+
}?.await()
57+
}
58+
59+
actual suspend fun delete() {
60+
android.delete().await()
61+
}
62+
63+
actual val snapshots: Flow<NativeDocumentSnapshot> get() = snapshots()
64+
65+
actual fun snapshots(includeMetadataChanges: Boolean) = addSnapshotListener(includeMetadataChanges) { snapshot, exception ->
66+
snapshot?.let { trySend(snapshot) }
67+
exception?.let { close(exception) }
68+
}
69+
70+
override fun equals(other: Any?): Boolean =
71+
this === other || other is NativeDocumentReference && nativeValue == other.nativeValue
72+
override fun hashCode(): Int = nativeValue.hashCode()
73+
override fun toString(): String = nativeValue.toString()
74+
75+
private fun addSnapshotListener(
76+
includeMetadataChanges: Boolean = false,
77+
listener: ProducerScope<NativeDocumentSnapshot>.(com.google.firebase.firestore.DocumentSnapshot?, com.google.firebase.firestore.FirebaseFirestoreException?) -> Unit
78+
) = callbackFlow {
79+
val executor = callbackExecutorMap[android.firestore] ?: TaskExecutors.MAIN_THREAD
80+
val metadataChanges =
81+
if (includeMetadataChanges) MetadataChanges.INCLUDE else MetadataChanges.EXCLUDE
82+
val registration =
83+
android.addSnapshotListener(executor, metadataChanges) { snapshots, exception ->
84+
listener(snapshots, exception)
85+
}
86+
awaitClose { registration.remove() }
87+
}
88+
}

‎firebase-firestore/src/androidMain/kotlin/dev/gitlive/firebase/firestore/internal/NativeDocumentSnapshotWrapper.kt‎

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
package dev.gitlive.firebase.firestore.internal
2+
3+
import dev.gitlive.firebase.firestore.EncodedFieldPath
4+
import dev.gitlive.firebase.firestore.ServerTimestampBehavior
5+
import dev.gitlive.firebase.firestore.SnapshotMetadata
6+
7+
@PublishedApi
8+
internal actual class NativeDocumentSnapshotWrapper actual internal constructor(actual val native: com.google.firebase.firestore.DocumentSnapshot) {
9+
10+
actual val id get() = native.id
11+
actual val reference get() = NativeDocumentReference(native.reference)
12+
13+
actual fun getEncoded(field: String, serverTimestampBehavior: ServerTimestampBehavior): Any? = native.get(field, serverTimestampBehavior.toAndroid())
14+
actual fun getEncoded(fieldPath: EncodedFieldPath, serverTimestampBehavior: ServerTimestampBehavior): Any? = native.get(fieldPath, serverTimestampBehavior.toAndroid())
15+
actual fun encodedData(serverTimestampBehavior: ServerTimestampBehavior): Any? = native.getData(serverTimestampBehavior.toAndroid())
16+
17+
actual fun contains(field: String) = native.contains(field)
18+
actual fun contains(fieldPath: EncodedFieldPath) = native.contains(fieldPath)
19+
20+
actual val exists get() = native.exists()
21+
22+
actual val metadata: SnapshotMetadata get() = SnapshotMetadata(native.metadata)
23+
24+
fun ServerTimestampBehavior.toAndroid(): com.google.firebase.firestore.DocumentSnapshot.ServerTimestampBehavior = when (this) {
25+
ServerTimestampBehavior.ESTIMATE -> com.google.firebase.firestore.DocumentSnapshot.ServerTimestampBehavior.ESTIMATE
26+
ServerTimestampBehavior.NONE -> com.google.firebase.firestore.DocumentSnapshot.ServerTimestampBehavior.NONE
27+
ServerTimestampBehavior.PREVIOUS -> com.google.firebase.firestore.DocumentSnapshot.ServerTimestampBehavior.PREVIOUS
28+
}
29+
}

‎firebase-firestore/src/androidMain/kotlin/dev/gitlive/firebase/firestore/internal/NativeFirebaseFirestoreWrapper.kt‎

Lines changed: 101 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,101 @@
1+
package dev.gitlive.firebase.firestore.internal
2+
3+
import com.google.android.gms.tasks.TaskExecutors
4+
import com.google.firebase.firestore.MemoryCacheSettings
5+
import com.google.firebase.firestore.MemoryEagerGcSettings
6+
import com.google.firebase.firestore.MemoryLruGcSettings
7+
import com.google.firebase.firestore.PersistentCacheSettings
8+
import com.google.firebase.firestore.firestoreSettings
9+
import dev.gitlive.firebase.firestore.FirebaseFirestoreSettings
10+
import dev.gitlive.firebase.firestore.LocalCacheSettings
11+
import dev.gitlive.firebase.firestore.MemoryGarbageCollectorSettings
12+
import dev.gitlive.firebase.firestore.NativeFirebaseFirestore
13+
import dev.gitlive.firebase.firestore.NativeTransaction
14+
import dev.gitlive.firebase.firestore.android
15+
import kotlinx.coroutines.runBlocking
16+
import kotlinx.coroutines.tasks.await
17+
18+
internal actual class NativeFirebaseFirestoreWrapper actual constructor(actual val native: NativeFirebaseFirestore) {
19+
20+
actual var settings: FirebaseFirestoreSettings
21+
get() = with(native.firestoreSettings) {
22+
FirebaseFirestoreSettings(
23+
isSslEnabled,
24+
host,
25+
cacheSettings?.let { localCacheSettings ->
26+
when (localCacheSettings) {
27+
is MemoryCacheSettings -> {
28+
val garbageCollectionSettings =
29+
when (val settings = localCacheSettings.garbageCollectorSettings) {
30+
is MemoryEagerGcSettings -> MemoryGarbageCollectorSettings.Eager
31+
is MemoryLruGcSettings -> MemoryGarbageCollectorSettings.LRUGC(
32+
settings.sizeBytes
33+
)
34+
35+
else -> throw IllegalArgumentException("Existing settings does not have valid GarbageCollectionSettings")
36+
}
37+
LocalCacheSettings.Memory(garbageCollectionSettings)
38+
}
39+
40+
is PersistentCacheSettings -> LocalCacheSettings.Persistent(
41+
localCacheSettings.sizeBytes
42+
)
43+
44+
else -> throw IllegalArgumentException("Existing settings is not of a valid type")
45+
}
46+
} ?: kotlin.run {
47+
@Suppress("DEPRECATION")
48+
when {
49+
isPersistenceEnabled -> LocalCacheSettings.Persistent(cacheSizeBytes)
50+
cacheSizeBytes == FirebaseFirestoreSettings.CACHE_SIZE_UNLIMITED -> LocalCacheSettings.Memory(
51+
MemoryGarbageCollectorSettings.Eager
52+
)
53+
54+
else -> LocalCacheSettings.Memory(
55+
MemoryGarbageCollectorSettings.LRUGC(
56+
cacheSizeBytes
57+
)
58+
)
59+
}
60+
},
61+
callbackExecutorMap[native] ?: TaskExecutors.MAIN_THREAD
62+
)
63+
}
64+
set(value) {
65+
native.firestoreSettings = firestoreSettings {
66+
isSslEnabled = value.sslEnabled
67+
host = value.host
68+
setLocalCacheSettings(value.cacheSettings.android)
69+
}
70+
callbackExecutorMap[native] = value.callbackExecutor
71+
}
72+
73+
actual fun collection(collectionPath: String) = native.collection(collectionPath)
74+
75+
actual fun collectionGroup(collectionId: String) = native.collectionGroup(collectionId)
76+
77+
actual fun document(documentPath: String) =
78+
NativeDocumentReference(native.document(documentPath))
79+
80+
actual fun batch() = native.batch()
81+
82+
actual fun setLoggingEnabled(loggingEnabled: Boolean) =
83+
com.google.firebase.firestore.FirebaseFirestore.setLoggingEnabled(loggingEnabled)
84+
85+
actual suspend fun <T> runTransaction(func: suspend NativeTransaction.() -> T): T =
86+
native.runTransaction { runBlocking { it.func() } }.await()
87+
88+
actual suspend fun clearPersistence() =
89+
native.clearPersistence().await().run { }
90+
91+
actual fun useEmulator(host: String, port: Int) {
92+
native.useEmulator(host, port)
93+
}
94+
95+
actual suspend fun disableNetwork() =
96+
native.disableNetwork().await().run { }
97+
98+
actual suspend fun enableNetwork() =
99+
native.enableNetwork().await().run { }
100+
101+
}

‎firebase-firestore/src/androidMain/kotlin/dev/gitlive/firebase/firestore/internal/NativeQueryWrapper.kt‎

Lines changed: 138 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,138 @@
1+
package dev.gitlive.firebase.firestore.internal
2+
3+
import com.google.android.gms.tasks.TaskExecutors
4+
import com.google.firebase.firestore.FieldPath
5+
import com.google.firebase.firestore.MetadataChanges
6+
import com.google.firebase.firestore.Query
7+
import dev.gitlive.firebase.firestore.Direction
8+
import dev.gitlive.firebase.firestore.EncodedFieldPath
9+
import dev.gitlive.firebase.firestore.Filter
10+
import dev.gitlive.firebase.firestore.NativeDocumentSnapshot
11+
import dev.gitlive.firebase.firestore.QuerySnapshot
12+
import dev.gitlive.firebase.firestore.Source
13+
import dev.gitlive.firebase.firestore.WhereConstraint
14+
import kotlinx.coroutines.channels.ProducerScope
15+
import kotlinx.coroutines.channels.awaitClose
16+
import kotlinx.coroutines.flow.callbackFlow
17+
import kotlinx.coroutines.tasks.await
18+
19+
@PublishedApi
20+
internal actual open class NativeQueryWrapper actual internal constructor(actual open val native: Query) {
21+
22+
actual fun limit(limit: Number) = native.limit(limit.toLong())
23+
24+
actual val snapshots get() = callbackFlow<QuerySnapshot> {
25+
val listener = native.addSnapshotListener { snapshot, exception ->
26+
snapshot?.let { trySend(QuerySnapshot(snapshot)) }
27+
exception?.let { close(exception) }
28+
}
29+
awaitClose { listener.remove() }
30+
}
31+
32+
actual fun snapshots(includeMetadataChanges: Boolean) = callbackFlow<QuerySnapshot> {
33+
val metadataChanges =
34+
if (includeMetadataChanges) MetadataChanges.INCLUDE else MetadataChanges.EXCLUDE
35+
val listener = native.addSnapshotListener(metadataChanges) { snapshot, exception ->
36+
snapshot?.let { trySend(QuerySnapshot(snapshot)) }
37+
exception?.let { close(exception) }
38+
}
39+
awaitClose { listener.remove() }
40+
}
41+
42+
actual suspend fun get(source: Source): QuerySnapshot =
43+
QuerySnapshot(native.get(source.toAndroidSource()).await())
44+
45+
actual fun where(filter: Filter) = native.where(filter.toAndroidFilter())
46+
47+
private fun Filter.toAndroidFilter(): com.google.firebase.firestore.Filter = when (this) {
48+
is Filter.And -> com.google.firebase.firestore.Filter.and(*filters.map { it.toAndroidFilter() }
49+
.toTypedArray())
50+
is Filter.Or -> com.google.firebase.firestore.Filter.or(*filters.map { it.toAndroidFilter() }
51+
.toTypedArray())
52+
is Filter.Field -> {
53+
when (constraint) {
54+
is WhereConstraint.ForNullableObject -> {
55+
val modifier: (String, Any?) -> com.google.firebase.firestore.Filter = when (constraint) {
56+
is WhereConstraint.EqualTo -> com.google.firebase.firestore.Filter::equalTo
57+
is WhereConstraint.NotEqualTo -> com.google.firebase.firestore.Filter::notEqualTo
58+
}
59+
modifier.invoke(field, constraint.safeValue)
60+
}
61+
is WhereConstraint.ForObject -> {
62+
val modifier: (String, Any) -> com.google.firebase.firestore.Filter = when (constraint) {
63+
is WhereConstraint.LessThan -> com.google.firebase.firestore.Filter::lessThan
64+
is WhereConstraint.GreaterThan -> com.google.firebase.firestore.Filter::greaterThan
65+
is WhereConstraint.LessThanOrEqualTo -> com.google.firebase.firestore.Filter::lessThanOrEqualTo
66+
is WhereConstraint.GreaterThanOrEqualTo -> com.google.firebase.firestore.Filter::greaterThanOrEqualTo
67+
is WhereConstraint.ArrayContains -> com.google.firebase.firestore.Filter::arrayContains
68+
}
69+
modifier.invoke(field, constraint.safeValue)
70+
}
71+
is WhereConstraint.ForArray -> {
72+
val modifier: (String, List<Any>) -> com.google.firebase.firestore.Filter = when (constraint) {
73+
is WhereConstraint.InArray -> com.google.firebase.firestore.Filter::inArray
74+
is WhereConstraint.ArrayContainsAny -> com.google.firebase.firestore.Filter::arrayContainsAny
75+
is WhereConstraint.NotInArray -> com.google.firebase.firestore.Filter::notInArray
76+
}
77+
modifier.invoke(field, constraint.safeValues)
78+
}
79+
}
80+
}
81+
is Filter.Path -> {
82+
when (constraint) {
83+
is WhereConstraint.ForNullableObject -> {
84+
val modifier: (FieldPath, Any?) -> com.google.firebase.firestore.Filter = when (constraint) {
85+
is WhereConstraint.EqualTo -> com.google.firebase.firestore.Filter::equalTo
86+
is WhereConstraint.NotEqualTo -> com.google.firebase.firestore.Filter::notEqualTo
87+
}
88+
modifier.invoke(path.android, constraint.safeValue)
89+
}
90+
is WhereConstraint.ForObject -> {
91+
val modifier: (FieldPath, Any) -> com.google.firebase.firestore.Filter = when (constraint) {
92+
is WhereConstraint.LessThan -> com.google.firebase.firestore.Filter::lessThan
93+
is WhereConstraint.GreaterThan -> com.google.firebase.firestore.Filter::greaterThan
94+
is WhereConstraint.LessThanOrEqualTo -> com.google.firebase.firestore.Filter::lessThanOrEqualTo
95+
is WhereConstraint.GreaterThanOrEqualTo -> com.google.firebase.firestore.Filter::greaterThanOrEqualTo
96+
is WhereConstraint.ArrayContains -> com.google.firebase.firestore.Filter::arrayContains
97+
}
98+
modifier.invoke(path.android, constraint.safeValue)
99+
}
100+
is WhereConstraint.ForArray -> {
101+
val modifier: (FieldPath, List<Any>) -> com.google.firebase.firestore.Filter = when (constraint) {
102+
is WhereConstraint.InArray -> com.google.firebase.firestore.Filter::inArray
103+
is WhereConstraint.ArrayContainsAny -> com.google.firebase.firestore.Filter::arrayContainsAny
104+
is WhereConstraint.NotInArray -> com.google.firebase.firestore.Filter::notInArray
105+
}
106+
modifier.invoke(path.android, constraint.safeValues)
107+
}
108+
}
109+
}
110+
}
111+
112+
actual fun orderBy(field: String, direction: Direction) = native.orderBy(field, direction)
113+
actual fun orderBy(field: EncodedFieldPath, direction: Direction) = native.orderBy(field, direction)
114+
115+
actual fun startAfter(document: NativeDocumentSnapshot) = native.startAfter(document)
116+
actual fun startAfter(vararg fieldValues: Any) = native.startAfter(*fieldValues)
117+
actual fun startAt(document: NativeDocumentSnapshot) = native.startAt(document)
118+
actual fun startAt(vararg fieldValues: Any) = native.startAt(*fieldValues)
119+
120+
actual fun endBefore(document: NativeDocumentSnapshot) = native.endBefore(document)
121+
actual fun endBefore(vararg fieldValues: Any) = native.endBefore(*fieldValues)
122+
actual fun endAt(document: NativeDocumentSnapshot) = native.endAt(document)
123+
actual fun endAt(vararg fieldValues: Any) = native.endAt(*fieldValues)
124+
125+
private fun addSnapshotListener(
126+
includeMetadataChanges: Boolean = false,
127+
listener: ProducerScope<QuerySnapshot>.(com.google.firebase.firestore.QuerySnapshot?, com.google.firebase.firestore.FirebaseFirestoreException?) -> Unit
128+
) = callbackFlow {
129+
val executor = callbackExecutorMap[native.firestore] ?: TaskExecutors.MAIN_THREAD
130+
val metadataChanges =
131+
if (includeMetadataChanges) MetadataChanges.INCLUDE else MetadataChanges.EXCLUDE
132+
val registration =
133+
native.addSnapshotListener(executor, metadataChanges) { snapshots, exception ->
134+
listener(snapshots, exception)
135+
}
136+
awaitClose { registration.remove() }
137+
}
138+
}

0 commit comments

Comments
 (0)