TrackingProtectionStateReducer.kt

/* 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 mozilla.components.browser.state.reducer

import mozilla.components.browser.state.action.TrackingProtectionAction
import mozilla.components.browser.state.state.BrowserState
import mozilla.components.browser.state.state.SessionState
import mozilla.components.browser.state.state.TrackingProtectionState

internal object TrackingProtectionStateReducer {
    /**
     * [TrackingProtectionAction] Reducer function for modifying a specific [TrackingProtectionState]
     * of a [SessionState].
     */
    fun reduce(state: BrowserState, action: TrackingProtectionAction): BrowserState = when (action) {
        is TrackingProtectionAction.ToggleAction -> state.copyWithTrackingProtectionState(action.tabId) {
            it.copy(enabled = action.enabled)
        }
        is TrackingProtectionAction.TrackerBlockedAction -> state.copyWithTrackingProtectionState(action.tabId) {
            it.copy(blockedTrackers = it.blockedTrackers + action.tracker)
        }
        is TrackingProtectionAction.TrackerLoadedAction -> state.copyWithTrackingProtectionState(action.tabId) {
            it.copy(loadedTrackers = it.loadedTrackers + action.tracker)
        }
        is TrackingProtectionAction.ClearTrackersAction -> state.copyWithTrackingProtectionState(action.tabId) {
            it.copy(loadedTrackers = emptyList(), blockedTrackers = emptyList())
        }
        is TrackingProtectionAction.ToggleExclusionListAction -> state.copyWithTrackingProtectionState(
            action.tabId,
        ) {
            it.copy(ignoredOnTrackingProtection = action.excluded)
        }
    }
}

private inline fun BrowserState.copyWithTrackingProtectionState(
    tabId: String,
    crossinline update: (TrackingProtectionState) -> TrackingProtectionState,
): BrowserState {
    return updateTabOrCustomTabState(tabId) { current ->
        current.createCopy(trackingProtection = update(current.trackingProtection))
    }
}