MainNavHost.kt

package com.threegap.bitnagil

import androidx.compose.runtime.Composable
import androidx.compose.ui.Modifier
import androidx.hilt.navigation.compose.hiltViewModel
import androidx.navigation.compose.NavHost
import androidx.navigation.compose.composable
import androidx.navigation.toRoute
import com.threegap.bitnagil.navigation.home.HomeNavHost
import com.threegap.bitnagil.presentation.emotion.EmotionScreenContainer
import com.threegap.bitnagil.presentation.guide.GuideScreenContainer
import com.threegap.bitnagil.presentation.login.LoginScreenContainer
import com.threegap.bitnagil.presentation.onboarding.OnBoardingScreenContainer
import com.threegap.bitnagil.presentation.onboarding.OnBoardingViewModel
import com.threegap.bitnagil.presentation.onboarding.model.navarg.OnBoardingScreenArg
import com.threegap.bitnagil.presentation.setting.SettingScreenContainer
import com.threegap.bitnagil.presentation.splash.SplashScreenContainer
import com.threegap.bitnagil.presentation.terms.TermsAgreementScreenContainer
import com.threegap.bitnagil.presentation.webview.BitnagilWebViewScreen
import com.threegap.bitnagil.presentation.withdrawal.WithdrawalScreenContainer
import com.threegap.bitnagil.presentation.writeroutine.WriteRoutineScreenContainer
import com.threegap.bitnagil.presentation.writeroutine.WriteRoutineViewModel
import com.threegap.bitnagil.presentation.writeroutine.model.navarg.WriteRoutineScreenArg

@Composable
fun MainNavHost(
    navigator: MainNavigator,
    modifier: Modifier = Modifier,
) {
    NavHost(
        navController = navigator.navController,
        startDestination = navigator.startDestination,
        modifier = modifier,
    ) {
        composable<Route.Splash> {
            SplashScreenContainer(
                navigateToLogin = {
                    navigator.navController.navigate(Route.Login) {
                        popUpTo<Route.Splash> { inclusive = true }
                    }
                },
                navigateToTermsAgreement = {
                    navigator.navController.navigate(Route.TermsAgreement) {
                        popUpTo<Route.Splash> { inclusive = true }
                    }
                },
                navigateToOnboarding = {
                    navigator.navController.navigate(Route.OnBoarding()) {
                        popUpTo<Route.Splash> { inclusive = true }
                    }
                },
                navigateToHome = navigator::navigateToHomeAndClearStack,
            )
        }

        composable<Route.Login> {
            LoginScreenContainer(
                navigateToHome = navigator::navigateToHomeAndClearStack,
                navigateToTermsAgreement = { navigator.navController.navigate(Route.TermsAgreement) },
            )
        }

        composable<Route.TermsAgreement> {
            TermsAgreementScreenContainer(
                navigateToTermsOfService = {
                    navigator.navController.navigate(
                        Route.WebView(
                            title = "약관 동의",
                            url = "https://complex-wombat-99f.notion.site/2025-7-20-236f4587491d8071833adfaf8115bce2",
                        ),
                    )
                },
                navigateToPrivacyPolicy = {
                    navigator.navController.navigate(
                        Route.WebView(
                            title = "약관 동의",
                            url = "https://complex-wombat-99f.notion.site/2025-07-20-236f4587491d80308016eb810692d18b",
                        ),
                    )
                },
                navigateToOnBoarding = {
                    navigator.navController.navigate(Route.OnBoarding())
                },
                navigateToBack = {
                    if (navigator.navController.previousBackStackEntry != null) {
                        navigator.navController.popBackStack()
                    }
                },
            )
        }

        composable<Route.Home> {
            HomeNavHost(
                navigateToSetting = {
                    navigator.navController.navigate(Route.Setting)
                },
                navigateToOnBoarding = {
                    navigator.navController.navigate(Route.OnBoarding(isNew = false))
                },
                navigateToNotice = {
                    navigator.navController.navigate(
                        Route.WebView(
                            title = "공지 사항",
                            url = "https://complex-wombat-99f.notion.site/23ff4587491d80efa0a5e4baece6017b?source=copy_link",
                        ),
                    )
                },
                navigateToQnA = {
                    navigator.navController.navigate(
                        Route.WebView(
                            title = "자주 묻는 질문",
                            url = "https://complex-wombat-99f.notion.site/23ff4587491d80659ae3ea392afbc05e?source=copy_link",
                        ),
                    )
                },
                navigateToGuide = {
                    navigator.navController.navigate(Route.Guide) {
                        launchSingleTop = true
                    }
                },
                navigateToRegisterRoutine = { routineId ->
                    navigator.navController.navigate(Route.WriteRoutine(routineId = routineId))
                },
                navigateToEmotion = {
                    navigator.navController.navigate(Route.Emotion)
                },
            )
        }

        composable<Route.WebView> {
            val webViewRoute = it.toRoute<Route.WebView>()
            BitnagilWebViewScreen(
                title = webViewRoute.title,
                url = webViewRoute.url,
                onBackClick = {
                    if (navigator.navController.previousBackStackEntry != null) {
                        navigator.navController.popBackStack()
                    }
                },
            )
        }

        composable<Route.Setting> {
            SettingScreenContainer(
                navigateToBack = {
                    if (navigator.navController.previousBackStackEntry != null) {
                        navigator.navController.popBackStack()
                    }
                },
                navigateToTermsOfService = {
                    navigator.navController.navigate(
                        Route.WebView(
                            title = "약관 동의",
                            url = "https://complex-wombat-99f.notion.site/2025-7-20-236f4587491d8071833adfaf8115bce2",
                        ),
                    )
                },
                navigateToPrivacyPolicy = {
                    navigator.navController.navigate(
                        Route.WebView(
                            title = "약관 동의",
                            url = "https://complex-wombat-99f.notion.site/2025-07-20-236f4587491d80308016eb810692d18b",
                        ),
                    )
                },
                navigateToLogin = {
                    navigator.navController.navigate(Route.Login) {
                        popUpTo(0) {
                            inclusive = true
                        }
                    }
                },
                navigateToWithdrawal = {
                    navigator.navController.navigate(Route.Withdrawal)
                },
            )
        }

        composable<Route.OnBoarding> { navBackStackEntry ->
            val arg = navBackStackEntry.toRoute<Route.OnBoarding>()
            val onBoardingScreenArg = if (arg.isNew) {
                OnBoardingScreenArg.NEW
            } else {
                OnBoardingScreenArg.RESET
            }

            val viewModel = hiltViewModel<OnBoardingViewModel, OnBoardingViewModel.Factory> { factory ->
                factory.create(onBoardingScreenArg)
            }

            OnBoardingScreenContainer(
                onBoardingViewModel = viewModel,
                navigateToHome = navigator::navigateToHomeAndClearStack,
                navigateToBack = {
                    if (navigator.navController.previousBackStackEntry != null) {
                        navigator.navController.popBackStack()
                    }
                },
            )
        }

        composable<Route.WriteRoutine> { navBackStackEntry ->
            val arg = navBackStackEntry.toRoute<Route.WriteRoutine>()
            val writeScreenNavArg = if (arg.isRegister) {
                WriteRoutineScreenArg.Add(baseRoutineId = arg.routineId)
            } else {
                WriteRoutineScreenArg.Edit(routineId = arg.routineId!!)
            }

            val viewModel = hiltViewModel<WriteRoutineViewModel, WriteRoutineViewModel.Factory> { factory ->
                factory.create(writeScreenNavArg)
            }

            WriteRoutineScreenContainer(
                viewModel = viewModel,
                navigateToBack = {
                    if (navigator.navController.previousBackStackEntry != null) {
                        navigator.navController.popBackStack()
                    }
                },
            )
        }

        composable<Route.Emotion> {
            EmotionScreenContainer(
                navigateToBack = {
                    if (navigator.navController.previousBackStackEntry != null) {
                        navigator.navController.popBackStack()
                    }
                },
            )
        }

        composable<Route.Withdrawal> {
            WithdrawalScreenContainer(
                navigateToBack = {
                    if (navigator.navController.previousBackStackEntry != null) {
                        navigator.navController.popBackStack()
                    }
                },
                navigateToLogin = {
                    navigator.navController.navigate(Route.Login) {
                        popUpTo(0) {
                            inclusive = true
                        }
                    }
                },
            )
        }

        composable<Route.Guide> {
            GuideScreenContainer(
                navigateToBack = {
                    if (navigator.navController.previousBackStackEntry != null) {
                        navigator.navController.popBackStack()
                    }
                },
            )
        }
    }
}