FastestPass/app/src/main/java/com/fastest/pass/login/utils/LoginNavigation.kt

package com.fastest.pass.login.utils

import androidx.lifecycle.lifecycleScope
import androidx.navigation.fragment.findNavController
import com.fastest.pass.R
import com.fastest.pass.login.presentation.ui.LoginFragment

class LoginNavigation {

    fun navigate(loginFragment: LoginFragment) {
        loginFragment.lifecycleScope.launchWhenStarted {
            loginFragment.viewmodel.router.collect { router ->
                when (router) {
                    LoginRoute.OpenSignUp -> {
                        loginFragment.findNavController().navigate(R.id.signupFragment)
                    }
                    LoginRoute.OpenForgotPassword -> {}
                    LoginRoute.OpenNoneScreen -> {}
                    LoginRoute.GoBack -> {
                        loginFragment.findNavController().popBackStack()
                    }
                }

                loginFragment.viewmodel.navigateTo(LoginRoute.OpenNoneScreen)
            }
        }
    }
}