Prepare missing semesters

ForceTower · Mar 2, 2024 · 1caae04 · 1caae04
1 parent 1c5f68e
commit 1caae04
Show file tree

Hide file tree

Showing 4 changed files with 83 additions and 2 deletions.
diff --git a/app/src/main/java/com/forcetower/uefs/core/storage/repository/SnowpiercerLoginRepository.kt b/app/src/main/java/com/forcetower/uefs/core/storage/repository/SnowpiercerLoginRepository.kt
@@ -110,7 +110,7 @@ class SnowpiercerLoginRepository @Inject constructor(
         LoginSagresRepository.createStep(0)
         currentStep.postValue(LoginSagresRepository.createStep(R.string.step_fetching_semesters))
         val semestersOutcome = orchestra.semesters(person.id)
-        val currentSemester = (semestersOutcome as? Outcome.Success)?.let { success ->
+        val currentSemester = (semestersOutcome.success())?.let { success ->
             val semesters = success.value
             SemestersProcessor(database, semesters).execute()
 
@@ -126,7 +126,7 @@ class SnowpiercerLoginRepository @Inject constructor(
         currentStep.postValue(LoginSagresRepository.createStep(R.string.step_fetching_grades))
         currentSemester?.let { semester ->
             val gradesOutcome = orchestra.grades(person.id, semester.id)
-            (gradesOutcome as? Outcome.Success)?.let { success ->
+            (gradesOutcome.success())?.let { success ->
                 val disciplines = success.value
                 val currentSemesterIns = database.semesterDao().getSemesterDirect(semester.id)!!
                 DisciplinesProcessor(context, database, disciplines, currentSemesterIns.uid, localProfileId, false).execute()

diff --git a/app/src/main/java/com/forcetower/uefs/core/task/FetchMissingSemestersUseCase.kt b/app/src/main/java/com/forcetower/uefs/core/task/FetchMissingSemestersUseCase.kt
@@ -0,0 +1,71 @@
+package com.forcetower.uefs.core.task
+
+import android.content.Context
+import com.forcetower.uefs.core.storage.database.UDatabase
+import com.forcetower.uefs.core.task.definers.DisciplinesProcessor
+import com.forcetower.uefs.core.task.definers.SemestersProcessor
+import dagger.hilt.android.qualifiers.ApplicationContext
+import dev.forcetower.breaker.Orchestra
+import dev.forcetower.breaker.model.Authorization
+import dev.forcetower.breaker.model.Semester
+import kotlinx.coroutines.Dispatchers
+import okhttp3.OkHttpClient
+import timber.log.Timber
+import javax.inject.Inject
+import javax.inject.Named
+
+class FetchMissingSemestersUseCase @Inject constructor(
+    @ApplicationContext private val context: Context,
+    private val database: UDatabase,
+    private val client: OkHttpClient,
+    @Named("webViewUA") private val agent: String,
+    @Named("flagSnowpiercerEnabled") private val snowpiercerEnabled: Boolean,
+) : UseCase<Unit, Unit>(Dispatchers.IO) {
+    override suspend fun execute(parameters: Unit) {
+        if (!snowpiercerEnabled) return
+
+        val access = database.accessDao().getAccessDirect()
+        val profile = database.profileDao().selectMeDirect()
+        access ?: return
+        profile ?: return
+
+        val orchestra = Orchestra.Builder().client(client).userAgent(agent).build()
+        orchestra.setAuthorization(Authorization(access.username, access.password))
+
+        val semesters = database.semesterDao().getSemestersDirect()
+        val missing = basicSemesters.filter { semesters.none { s -> s.sagresId == it.id } }
+
+        missing.map { semester ->
+            val result = orchestra.grades(profile.sagresId, semester.id)
+            result.success()?.value?.let {
+                if (it.isNotEmpty()) {
+                    SemestersProcessor(database, listOf(semester)).execute()
+                    val currentSemesterIns = database.semesterDao().getSemesterDirect(semester.id)!!
+                    DisciplinesProcessor(context, database, it, currentSemesterIns.uid, profile.sagresId, false).execute()
+                }
+            }
+            result.error()?.let {
+                Timber.e("Failed to get all grades from semester ${semester.code}")
+            }
+        }
+    }
+
+    companion object {
+        private val basicSemesters = listOf(
+            Semester(1000000792, "20181", "20181", "2018-03-19T00:00:00-03:00", "2018-08-08T00:00:00-03:00"),
+            Semester(1000000754, "20172", "20172", "2017-09-11T00:00:00-03:00", "2018-02-21T00:00:00-03:00"),
+            Semester(1000000713, "20171", "20171", "2017-03-13T00:00:00-03:00", "2017-08-19T00:00:00-03:00"),
+            Semester(1000000679, "20152 F", "20152 - Extra", "2016-05-30T00:00:00-03:00", "2016-07-01T00:00:00-03:00"),
+            Semester(1000000623, "20152", "20152", "2015-11-20T00:00:00-02:00", "2016-05-16T00:00:00-03:00"),
+            Semester(1000000594, "20151", "20151", "2015-03-03T00:00:00-03:00", "2015-10-29T00:00:00-02:00"),
+            Semester(1000000553, "20142", "2014.2", "2014-08-25T00:00:00-03:00", "2014-12-23T00:00:00-02:00"),
+            Semester(1000000486, "20141", "2014.1", "2014-03-10T00:00:00-03:00", "2014-08-02T00:00:00-03:00"),
+            Semester(1000000532, "20132 F", "20132-Férias", "2014-01-14T00:00:00-02:00", "2014-02-13T00:00:00-02:00"),
+            Semester(1000000483, "20132", "2013.2", "2013-08-26T00:00:00-03:00", "2014-01-06T00:00:00-02:00"),
+            Semester(1000000439, "20131", "2013.1", "2013-03-11T00:00:00-03:00", "2013-08-15T00:00:00-03:00"),
+            Semester(1000000403, "20122", "2012.2", "2012-09-05T00:00:00-03:00", "2013-01-23T00:00:00-02:00"),
+            Semester(1000000372, "20121", "2012.1", "2012-04-09T00:00:00-03:00", "2012-08-27T00:00:00-03:00"),
+            Semester(1000000340, "20112", "2011.2", "2011-09-29T00:00:00-03:00", "2012-03-19T00:00:00-03:00"),
+        )
+    }
+}
diff --git a/app/src/main/java/com/forcetower/uefs/feature/home/HomeActivity.kt b/app/src/main/java/com/forcetower/uefs/feature/home/HomeActivity.kt
@@ -133,6 +133,7 @@ class HomeActivity : UGameActivity() {
             viewModel.getAffinityQuestions()
 
             checkNotificationPermission()
+            viewModel.checkMissingSemesters()
         } catch (_: Throwable) {}
         moveToTask()
     }

diff --git a/app/src/main/java/com/forcetower/uefs/feature/home/HomeViewModel.kt b/app/src/main/java/com/forcetower/uefs/feature/home/HomeViewModel.kt
@@ -47,6 +47,7 @@ import com.forcetower.uefs.core.storage.repository.cloud.AffinityQuestionReposit
 import com.forcetower.uefs.core.storage.repository.cloud.AuthRepository
 import com.forcetower.uefs.core.storage.resource.Resource
 import com.forcetower.uefs.core.storage.resource.Status
+import com.forcetower.uefs.core.task.FetchMissingSemestersUseCase
 import com.forcetower.uefs.core.work.image.UploadImageToStorage
 import com.forcetower.uefs.easter.darktheme.DarkThemeRepository
 import com.google.android.play.core.install.model.AppUpdateType
@@ -68,6 +69,7 @@ class HomeViewModel @Inject constructor(
     private val sessionRepository: UserSessionRepository,
     private val accountRepository: AccountRepository,
     private val affinityRepository: AffinityQuestionRepository,
+    private val fetchMissingSemesters: FetchMissingSemestersUseCase,
     @Named("flagSnowpiercerEnabled")
     private val snowpiercerEnabled: Boolean
 ) : AndroidViewModel(application) {
@@ -199,4 +201,11 @@ class HomeViewModel @Inject constructor(
     fun getAffinityQuestions() {
         affinityRepository.getAffinityQuestionsAsync()
     }
+
+    fun checkMissingSemesters() {
+        if (!snowpiercerEnabled) return
+        viewModelScope.launch {
+            fetchMissingSemesters(Unit)
+        }
+    }
 }