143 lines
5.2 KiB
Kotlin
143 lines
5.2 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.home.sessioncontrol
|
|
|
|
import android.graphics.Bitmap
|
|
import android.view.ViewGroup
|
|
import androidx.fragment.app.Fragment
|
|
import androidx.lifecycle.ViewModel
|
|
import androidx.lifecycle.ViewModelProvider
|
|
import androidx.lifecycle.ViewModelProviders
|
|
import androidx.recyclerview.widget.RecyclerView
|
|
import io.reactivex.Observable
|
|
import io.reactivex.Observer
|
|
import org.mozilla.fenix.mvi.Action
|
|
import org.mozilla.fenix.mvi.ActionBusFactory
|
|
import org.mozilla.fenix.mvi.Change
|
|
import org.mozilla.fenix.mvi.UIComponent
|
|
import org.mozilla.fenix.mvi.UIComponentViewModel
|
|
import org.mozilla.fenix.mvi.ViewState
|
|
|
|
class SessionControlComponent(
|
|
private val container: ViewGroup,
|
|
owner: Fragment,
|
|
bus: ActionBusFactory,
|
|
override var initialState: SessionControlState = SessionControlState(emptyList(), emptyList(), Mode.Normal)
|
|
) :
|
|
UIComponent<SessionControlState, SessionControlAction, SessionControlChange>(
|
|
owner,
|
|
bus.getManagedEmitter(SessionControlAction::class.java),
|
|
bus.getSafeManagedObservable(SessionControlChange::class.java)
|
|
) {
|
|
|
|
override fun initView() = SessionControlUIView(container, actionEmitter, changesObservable)
|
|
val view: RecyclerView
|
|
get() = uiView.view as RecyclerView
|
|
|
|
override fun render(): Observable<SessionControlState> =
|
|
ViewModelProviders.of(owner, SessionControlViewModel.Factory(initialState, changesObservable))
|
|
.get(SessionControlViewModel::class.java).render(uiView)
|
|
|
|
init {
|
|
render()
|
|
}
|
|
}
|
|
|
|
data class Tab(
|
|
val sessionId: String,
|
|
val url: String,
|
|
val hostname: String,
|
|
val title: String,
|
|
val selected: Boolean? = null,
|
|
val thumbnail: Bitmap? = null
|
|
)
|
|
|
|
data class TabCollection(
|
|
val id: Int,
|
|
val title: String,
|
|
val tabs: MutableList<Tab>,
|
|
val iconColor: Int = 0,
|
|
var expanded: Boolean = false
|
|
)
|
|
|
|
sealed class Mode {
|
|
object Normal : Mode()
|
|
object Private : Mode()
|
|
}
|
|
|
|
data class SessionControlState(
|
|
val tabs: List<Tab>,
|
|
val collections: List<TabCollection>,
|
|
val mode: Mode
|
|
) : ViewState
|
|
|
|
sealed class TabAction : Action {
|
|
data class SaveTabGroup(val selectedTabSessionId: String?) : TabAction()
|
|
object Add : TabAction()
|
|
object ShareTabs : TabAction()
|
|
data class CloseAll(val private: Boolean) : TabAction()
|
|
data class Select(val sessionId: String) : TabAction()
|
|
data class Close(val sessionId: String) : TabAction()
|
|
data class Share(val sessionId: String) : TabAction()
|
|
object PrivateBrowsingLearnMore : TabAction()
|
|
}
|
|
|
|
sealed class CollectionAction : Action {
|
|
data class Expand(val collection: TabCollection) : CollectionAction()
|
|
data class Collapse(val collection: TabCollection) : CollectionAction()
|
|
data class Delete(val collection: TabCollection) : CollectionAction()
|
|
data class AddTab(val collection: TabCollection) : CollectionAction()
|
|
data class Rename(val collection: TabCollection) : CollectionAction()
|
|
data class OpenTabs(val collection: TabCollection) : CollectionAction()
|
|
data class ShareTabs(val collection: TabCollection) : CollectionAction()
|
|
data class RemoveTab(val collection: TabCollection, val tab: Tab) : CollectionAction()
|
|
}
|
|
|
|
sealed class SessionControlAction : Action {
|
|
data class Tab(val action: TabAction) : SessionControlAction()
|
|
data class Collection(val action: CollectionAction) : SessionControlAction()
|
|
}
|
|
|
|
fun Observer<SessionControlAction>.onNext(tabAction: TabAction) {
|
|
onNext(SessionControlAction.Tab(tabAction))
|
|
}
|
|
|
|
fun Observer<SessionControlAction>.onNext(collectionAction: CollectionAction) {
|
|
onNext(SessionControlAction.Collection(collectionAction))
|
|
}
|
|
|
|
sealed class SessionControlChange : Change {
|
|
data class TabsChange(val tabs: List<Tab>) : SessionControlChange()
|
|
data class ModeChange(val mode: Mode) : SessionControlChange()
|
|
data class CollectionsChange(val collections: List<TabCollection>) : SessionControlChange()
|
|
}
|
|
|
|
class SessionControlViewModel(initialState: SessionControlState, changesObservable: Observable<SessionControlChange>) :
|
|
UIComponentViewModel<SessionControlState, SessionControlAction, SessionControlChange>(
|
|
initialState,
|
|
changesObservable,
|
|
reducer
|
|
) {
|
|
|
|
class Factory(
|
|
private val initialState: SessionControlState,
|
|
private val changesObservable: Observable<SessionControlChange>
|
|
) : ViewModelProvider.Factory {
|
|
@Suppress("UNCHECKED_CAST")
|
|
override fun <T : ViewModel?> create(modelClass: Class<T>): T =
|
|
SessionControlViewModel(initialState, changesObservable) as T
|
|
}
|
|
|
|
companion object {
|
|
val reducer: (SessionControlState, SessionControlChange) -> SessionControlState = { state, change ->
|
|
when (change) {
|
|
is SessionControlChange.CollectionsChange -> state.copy(collections = change.collections)
|
|
is SessionControlChange.TabsChange -> state.copy(tabs = change.tabs)
|
|
is SessionControlChange.ModeChange -> state.copy(mode = change.mode)
|
|
}
|
|
}
|
|
}
|
|
}
|