✨ :: Use getRole

Author: minStone-dev

feature/main/src/main/java/com/stackknowledge/main/MainPageScreen.kt

@@ -41,15 +41,15 @@ internal fun MainPageRoute(
     onNavigate: (Authority, String, Int?) -> Unit,
     viewModel: MainViewModel = hiltViewModel()
 ) {
-    var role by remember { mutableStateOf(Authority.ROLE_TEACHER) } //로그인 로직 적용후 변경
+    val role by viewModel.role.collectAsStateWithLifecycle(initialValue = "")
     val getMissionUiState by viewModel.getMissionUiState.collectAsStateWithLifecycle()
     val getRankingUiState by viewModel.getRankingUiState.collectAsStateWithLifecycle()
 
     MainPageScreen(
-        role = role,
+        role = if (role.isNotBlank()) Authority.valueOf(role) else Authority.ROLE_TEACHER,
         getMissionUiState = getMissionUiState,
         getRankingUiState = getRankingUiState,
-        onNavigate = { navType, index -> onNavigate(role, navType, index) },
+        onNavigate = { navType, index -> onNavigate(Authority.valueOf(role), navType, index) },
         initMain = {
             with(viewModel) {
                 getMission()

feature/main/src/main/java/com/stackknowledge/main/viewModel/MainViewModel.kt

@@ -7,6 +7,7 @@ import com.example.common.result.Result
 import com.example.common.result.asResult
 import com.stackknowledge.main.viewModel.uistate.GetMissionUiState
 import com.stackknowledge.main.viewModel.uistate.GetRankingUiState
+import com.stackknowledge.repository.auth.AuthRepository
 import com.stackknowledge.usecase.mission.GetMissionUseCase
 import com.stackknowledge.usecase.student.GetStudentPointRankingUseCase
 import dagger.hilt.android.lifecycle.HiltViewModel
@@ -20,12 +21,15 @@ import javax.inject.Inject
 class MainViewModel @Inject constructor(
     private val getMissionUseCase: GetMissionUseCase,
     private val getStudentPointRankingUseCase: GetStudentPointRankingUseCase,
+    private val authRepository: AuthRepository
 ): ViewModel() {
     private val _getMissionUiState = MutableStateFlow<GetMissionUiState>(GetMissionUiState.Loading)
     internal val getMissionUiState = _getMissionUiState.asStateFlow()
 
     private val _getRankingUiState = MutableStateFlow<GetRankingUiState>(GetRankingUiState.Loading)
     internal val getRankingUiState = _getRankingUiState.asStateFlow()
+
+    internal val role = authRepository.getRole()
     internal fun getMission() = viewModelScope.launch {
         getMissionUseCase()
             .asResult()