book-reader/app/src/test/java/com/aryan/reader/AndroidSettingsHubModelsTest.kt

package com.aryan.reader

import com.aryan.reader.shared.SharedSettingsAction
import com.aryan.reader.shared.SharedSettingsDestination
import com.aryan.reader.shared.SharedSettingsHubModel
import com.aryan.reader.shared.SharedSettingsItemModel
import com.aryan.reader.shared.sharedSettingsHubModel
import org.junit.Assert.assertEquals
import org.junit.Assert.assertFalse
import org.junit.Assert.assertTrue
import org.junit.Test

class AndroidSettingsHubModelsTest {

    @Test
    fun `offline android settings hide network backed sections`() {
        val model = sharedSettingsHubModel(
            androidSettingsHubInput(
                uiState = ReaderScreenState(),
                isOssBuild = true,
                isOfflineBuild = true,
                isDebugBuild = false
            )
        )
        val actions = model.visibleNestedActions()

        assertFalse(SharedSettingsAction.AI_SETTINGS in actions)
        assertFalse(SharedSettingsAction.HIDE_READER_AI in actions)
        assertFalse(SharedSettingsAction.SIGN_IN in actions)
        assertFalse(SharedSettingsAction.CLOUD_SYNC in actions)
        assertFalse(SharedSettingsAction.FOLDER_SYNC in actions)
        assertFalse(SharedSettingsAction.HELP_FEEDBACK in actions)
        assertFalse(SharedSettingsAction.SUPPORT in actions)
        assertTrue(SharedSettingsAction.TTS_SETTINGS in actions)
        assertTrue(SharedSettingsAction.CUSTOM_FONTS in actions)
    }

    @Test
    fun `oss online settings hide sync rows but keep oss ai key settings`() {
        val model = sharedSettingsHubModel(
            androidSettingsHubInput(
                uiState = ReaderScreenState(
                    currentUser = UserData(
                        uid = "user-id",
                        displayName = "Reader",
                        photoUrl = null,
                        email = "reader@example.com"
                    ),
                    isProUser = true,
                    isSyncEnabled = true,
                    isFolderSyncEnabled = true
                ),
                isOssBuild = true,
                isOfflineBuild = false,
                isDebugBuild = true
            )
        )
        val actions = model.visibleNestedActions()

        assertTrue(SharedSettingsAction.AI_SETTINGS in actions)
        assertTrue(SharedSettingsAction.HIDE_READER_AI in actions)
        assertFalse(SharedSettingsAction.SIGN_OUT in actions)
        assertFalse(SharedSettingsAction.CLOUD_SYNC in actions)
        assertFalse(SharedSettingsAction.FOLDER_SYNC in actions)
        assertFalse(SharedSettingsAction.DEVICE_MANAGEMENT in actions)
        assertFalse(SharedSettingsAction.CLEAR_CLOUD_LOCAL_DATA in actions)
        assertTrue(SharedSettingsAction.SUPPORT in actions)
        assertEquals(
            "TTS & AI",
            model.rootCategories.single { it.destination == SharedSettingsDestination.TTS_AI }.title
        )
    }

    @Test
    fun `non oss settings do not expose oss ai key settings`() {
        val model = sharedSettingsHubModel(
            androidSettingsHubInput(
                uiState = ReaderScreenState(isProUser = true),
                isOssBuild = false,
                isOfflineBuild = false,
                isDebugBuild = false
            )
        )
        val actions = model.visibleNestedActions()

        assertFalse(SharedSettingsAction.AI_SETTINGS in actions)
        assertTrue(SharedSettingsAction.HIDE_READER_AI in actions)
        assertTrue(SharedSettingsAction.CLOUD_SYNC in actions)
        assertFalse(SharedSettingsAction.SUPPORT in actions)
        assertEquals(
            "TTS",
            model.rootCategories.single { it.destination == SharedSettingsDestination.TTS_AI }.title
        )
    }

    @Test
    fun `android settings expose debug-only storage actions only in debug`() {
        val releaseActions = sharedSettingsHubModel(
            androidSettingsHubInput(
                uiState = ReaderScreenState(),
                isOssBuild = false,
                isOfflineBuild = false,
                isDebugBuild = false
            )
        ).visibleNestedActions()
        val debugActions = sharedSettingsHubModel(
            androidSettingsHubInput(
                uiState = ReaderScreenState(),
                isOssBuild = false,
                isOfflineBuild = false,
                isDebugBuild = true
            )
        ).visibleNestedActions()

        assertFalse(SharedSettingsAction.EXPORT_LOGS in releaseActions)
        assertTrue(SharedSettingsAction.EXPORT_LOGS in debugActions)
    }

    @Test
    fun `android settings reflect global toggles from reader state`() {
        val model = sharedSettingsHubModel(
            androidSettingsHubInput(
                uiState = ReaderScreenState(
                    isTabsEnabled = true,
                    useStrictFileFilter = true,
                    usePdfFileNameAsDisplayName = true,
                    isScreenCaptureProtectionEnabled = true
                ),
                isOssBuild = false,
                isOfflineBuild = false,
                isDebugBuild = false
            )
        )
        val toggles = model.visibleNestedItems().associateBy { it.action }

        assertTrue(toggles.getValue(SharedSettingsAction.TABS_TOGGLE).checked == true)
        assertTrue(toggles.getValue(SharedSettingsAction.STRICT_FILE_FILTER).checked == true)
        assertTrue(toggles.getValue(SharedSettingsAction.PDF_FILENAME_DISPLAY_NAME).checked == true)
        assertTrue(toggles.getValue(SharedSettingsAction.SCREEN_CAPTURE_PROTECTION).checked == true)
    }

    @Test
    fun `android extra settings expose home overflow actions without settings duplicate`() {
        val model = sharedSettingsHubModel(
            androidSettingsHubInput(
                uiState = ReaderScreenState(),
                isOssBuild = false,
                isOfflineBuild = false,
                isDebugBuild = true
            )
        )
        val extraActions = model.page(SharedSettingsDestination.EXTRA).items.map { it.action }

        assertTrue(SharedSettingsAction.TABS_TOGGLE in extraActions)
        assertTrue(SharedSettingsAction.LANGUAGE in extraActions)
        assertTrue(SharedSettingsAction.EXTERNAL_FILE_BEHAVIOR in extraActions)
        assertTrue(SharedSettingsAction.STRICT_FILE_FILTER in extraActions)
        assertTrue(SharedSettingsAction.PDF_FILENAME_DISPLAY_NAME in extraActions)
        assertTrue(SharedSettingsAction.CLEAR_BOOK_CACHE in extraActions)
        assertTrue(SharedSettingsAction.CLEAR_REFLOW_CACHE in extraActions)
        assertTrue(SharedSettingsAction.TEST_PANEL_DETECTION in extraActions)
        assertTrue(SharedSettingsAction.TEST_SPEECH_BUBBLE_DETECTION in extraActions)
        assertTrue(SharedSettingsAction.CLEAR_CLOUD_LOCAL_DATA in extraActions)
    }

    @Test
    fun `cloud sync row is gated by pro state`() {
        val freeSync = sharedSettingsHubModel(
            androidSettingsHubInput(
                uiState = ReaderScreenState(isProUser = false),
                isOssBuild = false,
                isOfflineBuild = false,
                isDebugBuild = false
            )
        ).visibleNestedItems().single { it.action == SharedSettingsAction.CLOUD_SYNC }
        val proSync = sharedSettingsHubModel(
            androidSettingsHubInput(
                uiState = ReaderScreenState(isProUser = true),
                isOssBuild = false,
                isOfflineBuild = false,
                isDebugBuild = false
            )
        ).visibleNestedItems().single { it.action == SharedSettingsAction.CLOUD_SYNC }

        assertFalse(freeSync.enabled)
        assertTrue(proSync.enabled)
    }
}

private fun SharedSettingsHubModel.visibleNestedItems(): List<SharedSettingsItemModel> {
    return rootCategories.flatMap { category ->
        page(category.destination).items
    }
}

private fun SharedSettingsHubModel.visibleNestedActions(): List<SharedSettingsAction> {
    return visibleNestedItems().map { it.action }
}