You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
iceraven-browser/app/src/main/java/org/mozilla/fenix/tabstray/ext/TabSelectors.kt

62 lines
2.3 KiB
Kotlin

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
package org.mozilla.fenix.tabstray.ext
import mozilla.components.browser.state.selector.normalTabs
import mozilla.components.browser.state.selector.privateTabs
import mozilla.components.browser.state.state.BrowserState
import mozilla.components.browser.state.state.TabGroup
import mozilla.components.browser.state.state.TabSessionState
import org.mozilla.fenix.ext.maxActiveTime
import org.mozilla.fenix.tabstray.SEARCH_TERM_TAB_GROUPS
import org.mozilla.fenix.tabstray.SEARCH_TERM_TAB_GROUPS_MIN_SIZE
/**
* The currently selected tab if there's one that is private.
*
* NB: Upstream to Selectors.kt.
*/
val BrowserState.selectedPrivateTab: TabSessionState?
get() = selectedTabId?.let { id -> findPrivateTab(id) }
/**
* Finds and returns the private tab with the given id. Returns null if no
* matching tab could be found.
*
* @param tabId The ID of the tab to search for.
* @return The [TabSessionState] with the provided [tabId] or null if it could not be found.
*
* NB: Upstream to Selectors.kt.
*/
fun BrowserState.findPrivateTab(tabId: String): TabSessionState? {
return privateTabs.firstOrNull { it.id == tabId }
}
/**
* The list of normal tabs in the tabs tray filtered appropriately based on feature flags.
*/
fun BrowserState.getNormalTrayTabs(
searchTermTabGroupsAreEnabled: Boolean,
inactiveTabsEnabled: Boolean
): List<TabSessionState> {
val tabGroupsTabIds = getTabGroups()?.flatMap { it.tabIds } ?: emptyList()
return normalTabs.run {
if (searchTermTabGroupsAreEnabled && inactiveTabsEnabled) {
filter { it.isNormalTabActive(maxActiveTime) }.filter { tabGroupsTabIds.contains(it.id) }
} else if (inactiveTabsEnabled) {
filter { it.isNormalTabActive(maxActiveTime) }
} else if (searchTermTabGroupsAreEnabled) {
filter { it.isNormalTab() }.filter { tabGroupsTabIds.contains(it.id) }
} else {
this
}
}
}
fun BrowserState.getTabGroups(): List<TabGroup>? {
return tabPartitions[SEARCH_TERM_TAB_GROUPS]?.tabGroups
?.filter { it.tabIds.size >= SEARCH_TERM_TAB_GROUPS_MIN_SIZE }
}