MainNavHost.kt

package com.threegap.bitnagil

import androidx.compose.runtime.Composable
import androidx.compose.ui.Modifier
import androidx.hilt.lifecycle.viewmodel.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.screen.emotion.EmotionScreenContainer
import com.threegap.bitnagil.presentation.screen.guide.GuideScreenContainer
import com.threegap.bitnagil.presentation.screen.login.LoginScreenContainer
import com.threegap.bitnagil.presentation.screen.onboarding.OnBoardingScreenContainer
import com.threegap.bitnagil.presentation.screen.onboarding.OnBoardingViewModel
import com.threegap.bitnagil.presentation.screen.onboarding.model.navarg.OnBoardingScreenArg
import com.threegap.bitnagil.presentation.screen.reportdetail.ReportDetailScreenContainer
import com.threegap.bitnagil.presentation.screen.reportdetail.ReportDetailViewModel
import com.threegap.bitnagil.presentation.screen.reportdetail.model.navarg.ReportDetailScreenArg
import com.threegap.bitnagil.presentation.screen.reporthistory.ReportHistoryScreenContainer
import com.threegap.bitnagil.presentation.screen.reportwrite.ReportWriteScreenContainer
import com.threegap.bitnagil.presentation.screen.routinelist.RoutineListScreenContainer
import com.threegap.bitnagil.presentation.screen.routinewrite.RoutineWriteScreenContainer
import com.threegap.bitnagil.presentation.screen.routinewrite.RoutineWriteViewModel
import com.threegap.bitnagil.presentation.screen.routinewrite.model.navarg.RoutineWriteScreenArg
import com.threegap.bitnagil.presentation.screen.setting.SettingScreenContainer
import com.threegap.bitnagil.presentation.screen.splash.SplashScreenContainer
import com.threegap.bitnagil.presentation.screen.terms.TermsAgreementScreenContainer
import com.threegap.bitnagil.presentation.screen.webview.BitnagilWebViewScreen
import com.threegap.bitnagil.presentation.screen.withdrawal.WithdrawalScreenContainer

@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()) {
                        popUpTo(0) { inclusive = true }
                    }
                },
                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.RoutineWrite(routineId = routineId))
                },
                navigateToEmotion = {
                    navigator.navController.navigate(Route.Emotion)
                },
                navigateToRoutineList = { selectedDate ->
                    navigator.navController.navigate(
                        Route.RoutineList(selectedDate = selectedDate),
                    ) {
                        launchSingleTop = true
                    }
                },
                navigateToReport = {
                    navigator.navController.navigate(Route.ReportWrite) {
                        launchSingleTop = true
                    }
                },
                navigateToReportHistory = {
                    navigator.navController.navigate(Route.ReportHistory) {
                        launchSingleTop = true
                    }
                },
            )
        }

        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.RoutineWrite> { navBackStackEntry ->
            val arg = navBackStackEntry.toRoute<Route.RoutineWrite>()
            val writeScreenNavArg = if (arg.isRegister) {
                RoutineWriteScreenArg.Add(baseRoutineId = arg.routineId)
            } else {
                RoutineWriteScreenArg.Edit(routineId = arg.routineId!!, updateRoutineFromNowDate = arg.isUpdateRoutineFromNowDate)
            }

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

            RoutineWriteScreenContainer(
                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.RoutineList> {
            RoutineListScreenContainer(
                navigateToBack = {
                    if (navigator.navController.previousBackStackEntry != null) {
                        navigator.navController.popBackStack()
                    }
                },
                navigateToAddRoutine = {
                    navigator.navController.navigate(Route.RoutineWrite())
                },
                navigateToEditRoutine = { routineId, updateRoutineFromNowDate ->
                    navigator.navController.navigate(
                        Route.RoutineWrite(
                            routineId = routineId,
                            isRegister = false,
                            isUpdateRoutineFromNowDate = updateRoutineFromNowDate,
                        ),
                    )
                },
            )
        }

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

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

        composable<Route.ReportHistory> {
            ReportHistoryScreenContainer(
                navigateToBack = {
                    if (navigator.navController.previousBackStackEntry != null) {
                        navigator.navController.popBackStack()
                    }
                },
                navigateToReportDetail = { reportId ->
                    navigator.navController.navigate(Route.ReportDetail(reportId = reportId)) {
                        launchSingleTop = true
                    }
                },
            )
        }

        composable<Route.ReportDetail> { navBackStackEntry ->
            val arg = navBackStackEntry.toRoute<Route.ReportDetail>()
            val reportDetailScreenArg = ReportDetailScreenArg(reportId = arg.reportId)
            val viewModel = hiltViewModel<ReportDetailViewModel, ReportDetailViewModel.Factory> { factory ->
                factory.create(reportDetailScreenArg)
            }

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