amethyst/app/src/main/java/com/vitorpamplona/amethyst/ui/navigation/Routes.kt

package com.vitorpamplona.amethyst.ui.navigation

import android.content.Context
import androidx.compose.runtime.Composable
import androidx.compose.runtime.getValue
import androidx.navigation.NamedNavArgument
import androidx.navigation.NavBackStackEntry
import androidx.navigation.NavController
import androidx.navigation.NavHostController
import androidx.navigation.NavType
import androidx.navigation.compose.currentBackStackEntryAsState
import androidx.navigation.navArgument
import com.vitorpamplona.amethyst.NotificationCache
import com.vitorpamplona.amethyst.R
import com.vitorpamplona.amethyst.model.Account
import com.vitorpamplona.amethyst.ui.dal.ChatroomListKnownFeedFilter
import com.vitorpamplona.amethyst.ui.dal.HomeNewThreadFeedFilter
import com.vitorpamplona.amethyst.ui.dal.NotificationFeedFilter
import com.vitorpamplona.amethyst.ui.screen.AccountStateViewModel
import com.vitorpamplona.amethyst.ui.screen.loggedIn.AccountViewModel
import com.vitorpamplona.amethyst.ui.screen.loggedIn.ChannelScreen
import com.vitorpamplona.amethyst.ui.screen.loggedIn.ChatroomListScreen
import com.vitorpamplona.amethyst.ui.screen.loggedIn.ChatroomScreen
import com.vitorpamplona.amethyst.ui.screen.loggedIn.FiltersScreen
import com.vitorpamplona.amethyst.ui.screen.loggedIn.HomeScreen
import com.vitorpamplona.amethyst.ui.screen.loggedIn.NotificationScreen
import com.vitorpamplona.amethyst.ui.screen.loggedIn.ProfileScreen
import com.vitorpamplona.amethyst.ui.screen.loggedIn.SearchScreen
import com.vitorpamplona.amethyst.ui.screen.loggedIn.ThreadScreen

sealed class Route(
    val route: String,
    val icon: Int,
    val hasNewItems: (Account, NotificationCache, Context) -> Boolean = { _, _, _ -> false },
    val arguments: List<NamedNavArgument> = emptyList(),
    val buildScreen: (AccountViewModel, AccountStateViewModel, NavController) -> @Composable (NavBackStackEntry) -> Unit
) {
    val base: String
        get() = route.substringBefore("?")

    object Home : Route(
        "Home?forceRefresh={forceRefresh}",
        R.drawable.ic_home,
        arguments = listOf(navArgument("forceRefresh") { type = NavType.BoolType; defaultValue = false }),
        hasNewItems = { accountViewModel, cache, context -> homeHasNewItems(accountViewModel, cache, context) },
        buildScreen = { accountViewModel, _, navController ->
            { backStackEntry ->
                HomeScreen(accountViewModel, navController, backStackEntry.arguments?.getBoolean("forceRefresh", false) ?: false)
            }
        }
    )

    object Search : Route(
        "Search?forceRefresh={forceRefresh}",
        R.drawable.ic_globe,
        arguments = listOf(navArgument("forceRefresh") { type = NavType.BoolType; defaultValue = false }),
        buildScreen = { acc, _, navController ->
            { backStackEntry ->
                SearchScreen(acc, navController, backStackEntry.arguments?.getBoolean("forceRefresh", false) ?: false)
            }
        }
    )

    object Notification : Route(
        route = "Notification",
        icon = R.drawable.ic_notifications,
        hasNewItems = { accountViewModel, cache, context ->
            notificationHasNewItems(accountViewModel, cache, context)
        },
        buildScreen = { accountViewModel, _, navController ->
            { NotificationScreen(accountViewModel, navController) }
        }
    )

    object Message : Route(
        route = "Message",
        icon = R.drawable.ic_dm,
        hasNewItems = { accountViewModel, cache, context ->
            messagesHasNewItems(accountViewModel, cache, context)
        },
        buildScreen = { accountViewModel, _, navController ->
            { ChatroomListScreen(accountViewModel, navController) }
        }
    )

    object Filters : Route(
        route = "Filters",
        icon = R.drawable.ic_security,
        buildScreen = { accountViewModel, _, navController ->
            { FiltersScreen(accountViewModel, navController) }
        }
    )

    object Profile : Route(
        route = "User/{id}",
        icon = R.drawable.ic_profile,
        arguments = listOf(navArgument("id") { type = NavType.StringType }),
        buildScreen = { accountViewModel, _, navController ->
            { ProfileScreen(it.arguments?.getString("id"), accountViewModel, navController) }
        }
    )

    object Note : Route(
        route = "Note/{id}",
        icon = R.drawable.ic_moments,
        arguments = listOf(navArgument("id") { type = NavType.StringType }),
        buildScreen = { accountViewModel, _, navController ->
            { ThreadScreen(it.arguments?.getString("id"), accountViewModel, navController) }
        }
    )

    object Room : Route(
        route = "Room/{id}",
        icon = R.drawable.ic_moments,
        arguments = listOf(navArgument("id") { type = NavType.StringType }),
        buildScreen = { accountViewModel, _, navController ->
            { ChatroomScreen(it.arguments?.getString("id"), accountViewModel, navController) }
        }
    )

    object Channel : Route(
        route = "Channel/{id}",
        icon = R.drawable.ic_moments,
        arguments = listOf(navArgument("id") { type = NavType.StringType }),
        buildScreen = { accountViewModel, accountStateViewModel, navController ->
            {
                ChannelScreen(
                    it.arguments?.getString("id"),
                    accountViewModel,
                    accountStateViewModel,
                    navController
                )
            }
        }
    )
}

val Routes = listOf(
    // bottom
    Route.Home,
    Route.Message,
    Route.Search,
    Route.Notification,

    // drawer
    Route.Profile,
    Route.Note,
    Route.Room,
    Route.Channel,
    Route.Filters
)

// **
// *  Functions below only exist because we have not broken the datasource classes into backend and frontend.
// **
@Composable
public fun currentRoute(navController: NavHostController): String? {
    val navBackStackEntry by navController.currentBackStackEntryAsState()
    return navBackStackEntry?.destination?.route
}

private fun homeHasNewItems(account: Account, cache: NotificationCache, context: Context): Boolean {
    val lastTime = cache.load("HomeFollows", context)

    HomeNewThreadFeedFilter.account = account

    return (
        HomeNewThreadFeedFilter.feed().firstOrNull { it.createdAt() != null }?.createdAt()
            ?: 0
        ) > lastTime
}

private fun notificationHasNewItems(
    account: Account,
    cache: NotificationCache,
    context: Context
): Boolean {
    val lastTime = cache.load("Notification", context)

    NotificationFeedFilter.account = account

    return (
        NotificationFeedFilter.feed().firstOrNull { it.createdAt() != null }?.createdAt()
            ?: 0
        ) > lastTime
}

private fun messagesHasNewItems(
    account: Account,
    cache: NotificationCache,
    context: Context
): Boolean {
    ChatroomListKnownFeedFilter.account = account

    val note = ChatroomListKnownFeedFilter.feed().firstOrNull {
        it.createdAt() != null && it.channel() == null && it.author != account.userProfile()
    } ?: return false

    val lastTime = cache.load("Room/${note.author?.pubkeyHex}", context)

    return (note.createdAt() ?: 0) > lastTime
}