[FEATURE] #32 : 운영진 코드 검증 로직 및 운영진 등록 로직 추가

pknu-wap · Oct 21, 2023 · 926a90b · 926a90b
1 parent 92cd288
commit 926a90b
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 0 deletions.
diff --git a/core/data/src/main/java/com/wap/wapp/core/data/repository/manage/ManageRepository.kt b/core/data/src/main/java/com/wap/wapp/core/data/repository/manage/ManageRepository.kt
@@ -2,4 +2,8 @@ package com.wap.wapp.core.data.repository.manage
 
 interface ManageRepository {
     suspend fun getManager(userId: String): Result<Unit>
+
+    suspend fun postManager(userId: String): Result<Unit>
+
+    suspend fun postManagerCode(code: String): Result<Unit>
 }
diff --git a/core/data/src/main/java/com/wap/wapp/core/data/repository/manage/ManageRepositoryImpl.kt b/core/data/src/main/java/com/wap/wapp/core/data/repository/manage/ManageRepositoryImpl.kt
@@ -9,4 +9,12 @@ class ManageRepositoryImpl @Inject constructor(
     override suspend fun getManager(userId: String): Result<Unit> {
         return manageDataSource.getManager(userId)
     }
+
+    override suspend fun postManager(userId: String): Result<Unit> {
+        return manageDataSource.postManager(userId)
+    }
+
+    override suspend fun postManagerCode(code: String): Result<Unit> {
+        return manageDataSource.postManagerCode(code)
+    }
 }
diff --git a/core/network/src/main/java/com/wap/wapp/core/network/source/manage/ManageDataSource.kt b/core/network/src/main/java/com/wap/wapp/core/network/source/manage/ManageDataSource.kt
@@ -2,4 +2,8 @@ package com.wap.wapp.core.network.source.manage
 
 interface ManageDataSource {
     suspend fun getManager(userId: String): Result<Unit>
+
+    suspend fun postManager(userId: String): Result<Unit>
+
+    suspend fun postManagerCode(code: String): Result<Unit>
 }
diff --git a/core/network/src/main/java/com/wap/wapp/core/network/source/manage/ManageDataSourceImpl.kt b/core/network/src/main/java/com/wap/wapp/core/network/source/manage/ManageDataSourceImpl.kt
@@ -1,6 +1,8 @@
 package com.wap.wapp.core.network.source.manage
 
 import com.google.firebase.firestore.FirebaseFirestore
+import com.google.firebase.firestore.SetOptions
+import com.wap.wapp.core.network.constant.CODES_COLLECTION
 import com.wap.wapp.core.network.constant.MANAGER_COLLECTION
 import com.wap.wapp.core.network.utils.await
 import javax.inject.Inject
@@ -10,10 +12,35 @@ class ManageDataSourceImpl @Inject constructor(
 ) : ManageDataSource {
     override suspend fun getManager(userId: String): Result<Unit> {
         return runCatching {
+            val result = firebaseFirestore.collection(MANAGER_COLLECTION)
+                .whereEqualTo("userId", userId)
+                .get()
+                .await()
+
+            check(result.isEmpty.not())
+        }
+    }
+
+    override suspend fun postManager(userId: String): Result<Unit> {
+        return runCatching {
+            val userIdMap = mapOf("userId" to userId)
+            val setOption = SetOptions.merge()
+
             firebaseFirestore.collection(MANAGER_COLLECTION)
                 .document(userId)
+                .set(userIdMap, setOption)
+                .await()
+        }
+    }
+
+    override suspend fun postManagerCode(code: String): Result<Unit> {
+        return runCatching {
+            val result = firebaseFirestore.collection(CODES_COLLECTION)
+                .whereEqualTo("manager", code)
                 .get()
                 .await()
+
+            check(result.isEmpty.not())
         }
     }
 }