diff options
Diffstat (limited to 'src/com/android/customization/module/ThemePickerInjector.kt')
| -rw-r--r-- | src/com/android/customization/module/ThemePickerInjector.kt | 355 |
1 files changed, 344 insertions, 11 deletions
diff --git a/src/com/android/customization/module/ThemePickerInjector.kt b/src/com/android/customization/module/ThemePickerInjector.kt index b3f95d5c..eb20037b 100644 --- a/src/com/android/customization/module/ThemePickerInjector.kt +++ b/src/com/android/customization/module/ThemePickerInjector.kt @@ -16,22 +16,55 @@ package com.android.customization.module import android.app.Activity +import android.app.UiModeManager import android.content.Context import android.content.Intent import android.net.Uri import android.os.Bundle +import android.text.TextUtils +import androidx.activity.ComponentActivity import androidx.fragment.app.Fragment import androidx.fragment.app.FragmentActivity +import androidx.lifecycle.ViewModelProvider +import com.android.customization.model.color.ColorCustomizationManager +import com.android.customization.model.color.ColorOptionsProvider +import com.android.customization.model.grid.GridOptionsManager +import com.android.customization.model.grid.data.repository.GridRepositoryImpl +import com.android.customization.model.grid.domain.interactor.GridInteractor +import com.android.customization.model.grid.domain.interactor.GridSnapshotRestorer +import com.android.customization.model.grid.ui.viewmodel.GridScreenViewModel +import com.android.customization.model.mode.DarkModeSnapshotRestorer import com.android.customization.model.theme.OverlayManagerCompat import com.android.customization.model.theme.ThemeBundleProvider import com.android.customization.model.theme.ThemeManager +import com.android.customization.model.themedicon.ThemedIconSwitchProvider +import com.android.customization.model.themedicon.data.repository.ThemeIconRepository +import com.android.customization.model.themedicon.domain.interactor.ThemedIconInteractor +import com.android.customization.model.themedicon.domain.interactor.ThemedIconSnapshotRestorer +import com.android.customization.picker.clock.data.repository.ClockPickerRepositoryImpl +import com.android.customization.picker.clock.data.repository.ClockRegistryProvider +import com.android.customization.picker.clock.domain.interactor.ClockPickerInteractor +import com.android.customization.picker.clock.ui.view.ClockViewFactory +import com.android.customization.picker.clock.ui.viewmodel.ClockCarouselViewModel +import com.android.customization.picker.clock.ui.viewmodel.ClockSectionViewModel +import com.android.customization.picker.clock.ui.viewmodel.ClockSettingsViewModel +import com.android.customization.picker.color.data.repository.ColorPickerRepositoryImpl +import com.android.customization.picker.color.domain.interactor.ColorPickerInteractor +import com.android.customization.picker.color.domain.interactor.ColorPickerSnapshotRestorer +import com.android.customization.picker.color.ui.viewmodel.ColorPickerViewModel +import com.android.customization.picker.notifications.data.repository.NotificationsRepository +import com.android.customization.picker.notifications.domain.interactor.NotificationsInteractor +import com.android.customization.picker.notifications.domain.interactor.NotificationsSnapshotRestorer +import com.android.customization.picker.notifications.ui.viewmodel.NotificationSectionViewModel import com.android.customization.picker.quickaffordance.data.repository.KeyguardQuickAffordancePickerRepository import com.android.customization.picker.quickaffordance.domain.interactor.KeyguardQuickAffordancePickerInteractor import com.android.customization.picker.quickaffordance.domain.interactor.KeyguardQuickAffordanceSnapshotRestorer import com.android.customization.picker.quickaffordance.ui.viewmodel.KeyguardQuickAffordancePickerViewModel +import com.android.systemui.shared.clocks.ClockRegistry import com.android.systemui.shared.customization.data.content.CustomizationProviderClient import com.android.systemui.shared.customization.data.content.CustomizationProviderClientImpl import com.android.wallpaper.model.LiveWallpaperInfo +import com.android.wallpaper.model.WallpaperColorsViewModel import com.android.wallpaper.model.WallpaperInfo import com.android.wallpaper.module.CustomizationSections import com.android.wallpaper.module.FragmentFactory @@ -42,13 +75,20 @@ import com.android.wallpaper.picker.CustomizationPickerActivity import com.android.wallpaper.picker.ImagePreviewFragment import com.android.wallpaper.picker.LivePreviewFragment import com.android.wallpaper.picker.PreviewFragment +import com.android.wallpaper.picker.customization.data.content.WallpaperClientImpl +import com.android.wallpaper.picker.customization.data.repository.WallpaperRepository +import com.android.wallpaper.picker.customization.domain.interactor.WallpaperInteractor import com.android.wallpaper.picker.undo.domain.interactor.SnapshotRestorer -import kotlinx.coroutines.Dispatchers.IO +import kotlinx.coroutines.DelicateCoroutinesApi +import kotlinx.coroutines.Dispatchers +import kotlinx.coroutines.GlobalScope +@OptIn(DelicateCoroutinesApi::class) open class ThemePickerInjector : WallpaperPicker2Injector(), CustomizationInjector { private var customizationSections: CustomizationSections? = null private var userEventLogger: UserEventLogger? = null private var prefs: WallpaperPreferences? = null + private var wallpaperInteractor: WallpaperInteractor? = null private var keyguardQuickAffordancePickerInteractor: KeyguardQuickAffordancePickerInteractor? = null private var keyguardQuickAffordancePickerViewModelFactory: @@ -58,12 +98,44 @@ open class ThemePickerInjector : WallpaperPicker2Injector(), CustomizationInject private var fragmentFactory: FragmentFactory? = null private var keyguardQuickAffordanceSnapshotRestorer: KeyguardQuickAffordanceSnapshotRestorer? = null + private var notificationsSnapshotRestorer: NotificationsSnapshotRestorer? = null + private var clockRegistry: ClockRegistry? = null + private var clockPickerInteractor: ClockPickerInteractor? = null + private var clockSectionViewModel: ClockSectionViewModel? = null + private var clockCarouselViewModel: ClockCarouselViewModel? = null + private var clockViewFactory: ClockViewFactory? = null + private var notificationsInteractor: NotificationsInteractor? = null + private var notificationSectionViewModelFactory: NotificationSectionViewModel.Factory? = null + private var colorPickerInteractor: ColorPickerInteractor? = null + private var colorPickerViewModelFactory: ColorPickerViewModel.Factory? = null + private var colorPickerSnapshotRestorer: ColorPickerSnapshotRestorer? = null + private var colorCustomizationManager: ColorCustomizationManager? = null + private var darkModeSnapshotRestorer: DarkModeSnapshotRestorer? = null + private var themedIconSnapshotRestorer: ThemedIconSnapshotRestorer? = null + private var themedIconInteractor: ThemedIconInteractor? = null + private var clockSettingsViewModelFactory: ClockSettingsViewModel.Factory? = null + private var gridInteractor: GridInteractor? = null + private var gridSnapshotRestorer: GridSnapshotRestorer? = null + private var gridScreenViewModelFactory: GridScreenViewModel.Factory? = null - override fun getCustomizationSections(activity: Activity): CustomizationSections { + override fun getCustomizationSections(activity: ComponentActivity): CustomizationSections { return customizationSections ?: DefaultCustomizationSections( + getColorPickerViewModelFactory( + context = activity, + wallpaperColorsViewModel = getWallpaperColorsViewModel(), + ), getKeyguardQuickAffordancePickerInteractor(activity), - getKeyguardQuickAffordancePickerViewModelFactory(activity) + getKeyguardQuickAffordancePickerViewModelFactory(activity), + NotificationSectionViewModel.Factory( + interactor = getNotificationsInteractor(activity), + ), + getFlags(), + getClockCarouselViewModel(activity), + getClockViewFactory(activity), + getDarkModeSnapshotRestorer(activity), + getThemedIconSnapshotRestorer(activity), + getThemedIconInteractor(), ) .also { customizationSections = it } } @@ -122,6 +194,13 @@ open class ThemePickerInjector : WallpaperPicker2Injector(), CustomizationInject return super<WallpaperPicker2Injector>.getSnapshotRestorers(context).toMutableMap().apply { this[KEY_QUICK_AFFORDANCE_SNAPSHOT_RESTORER] = getKeyguardQuickAffordanceSnapshotRestorer(context) + this[KEY_WALLPAPER_SNAPSHOT_RESTORER] = getWallpaperSnapshotRestorer(context) + this[KEY_NOTIFICATIONS_SNAPSHOT_RESTORER] = getNotificationsSnapshotRestorer(context) + this[KEY_DARK_MODE_SNAPSHOT_RESTORER] = getDarkModeSnapshotRestorer(context) + this[KEY_THEMED_ICON_SNAPSHOT_RESTORER] = getThemedIconSnapshotRestorer(context) + this[KEY_APP_GRID_SNAPSHOT_RESTORER] = getGridSnapshotRestorer(context) + this[KEY_COLOR_PICKER_SNAPSHOT_RESTORER] = + getColorPickerSnapshotRestorer(context, getWallpaperColorsViewModel()) } } @@ -138,6 +217,25 @@ open class ThemePickerInjector : WallpaperPicker2Injector(), CustomizationInject return ThemeManager(provider, activity, overlayManagerCompat, logger) } + override fun getWallpaperInteractor(context: Context): WallpaperInteractor { + return wallpaperInteractor + ?: WallpaperInteractor( + repository = + WallpaperRepository( + scope = GlobalScope, + client = WallpaperClientImpl(context = context), + backgroundDispatcher = Dispatchers.IO, + ), + shouldHandleReload = { + TextUtils.equals( + getColorCustomizationManager(context).currentColorSource, + ColorOptionsProvider.COLOR_SOURCE_PRESET + ) + } + ) + .also { wallpaperInteractor = it } + } + override fun getKeyguardQuickAffordancePickerInteractor( context: Context ): KeyguardQuickAffordancePickerInteractor { @@ -154,32 +252,45 @@ open class ThemePickerInjector : WallpaperPicker2Injector(), CustomizationInject ?: KeyguardQuickAffordancePickerViewModel.Factory( context, getKeyguardQuickAffordancePickerInteractor(context), - getUndoInteractor(context), getCurrentWallpaperInfoFactory(context), - ) + ) { intent -> + context.startActivity(intent) + } .also { keyguardQuickAffordancePickerViewModelFactory = it } } + fun getNotificationSectionViewModelFactory( + context: Context, + ): NotificationSectionViewModel.Factory { + return notificationSectionViewModelFactory + ?: NotificationSectionViewModel.Factory( + interactor = getNotificationsInteractor(context), + ) + .also { notificationSectionViewModelFactory = it } + } + private fun getKeyguardQuickAffordancePickerInteractorImpl( context: Context ): KeyguardQuickAffordancePickerInteractor { val client = getKeyguardQuickAffordancePickerProviderClient(context) return KeyguardQuickAffordancePickerInteractor( - KeyguardQuickAffordancePickerRepository(client, IO), + KeyguardQuickAffordancePickerRepository(client, Dispatchers.IO), client - ) { getKeyguardQuickAffordanceSnapshotRestorer(context) } + ) { + getKeyguardQuickAffordanceSnapshotRestorer(context) + } } protected fun getKeyguardQuickAffordancePickerProviderClient( context: Context ): CustomizationProviderClient { return customizationProviderClient - ?: CustomizationProviderClientImpl(context, IO).also { + ?: CustomizationProviderClientImpl(context, Dispatchers.IO).also { customizationProviderClient = it } } - protected fun getKeyguardQuickAffordanceSnapshotRestorer( + private fun getKeyguardQuickAffordanceSnapshotRestorer( context: Context ): KeyguardQuickAffordanceSnapshotRestorer { return keyguardQuickAffordanceSnapshotRestorer @@ -190,16 +301,238 @@ open class ThemePickerInjector : WallpaperPicker2Injector(), CustomizationInject .also { keyguardQuickAffordanceSnapshotRestorer = it } } + private fun getNotificationsSnapshotRestorer(context: Context): NotificationsSnapshotRestorer { + return notificationsSnapshotRestorer + ?: NotificationsSnapshotRestorer( + interactor = + getNotificationsInteractor( + context = context, + ), + ) + .also { notificationsSnapshotRestorer = it } + } + + override fun getClockRegistry(context: Context): ClockRegistry { + return clockRegistry + ?: ClockRegistryProvider( + context = context, + coroutineScope = GlobalScope, + mainDispatcher = Dispatchers.Main, + backgroundDispatcher = Dispatchers.IO, + ) + .get() + .also { clockRegistry = it } + } + + override fun getClockPickerInteractor( + context: Context, + ): ClockPickerInteractor { + return clockPickerInteractor + ?: ClockPickerInteractor( + ClockPickerRepositoryImpl( + secureSettingsRepository = getSecureSettingsRepository(context), + registry = getClockRegistry(context), + scope = GlobalScope, + ), + ) + .also { clockPickerInteractor = it } + } + + override fun getClockSectionViewModel(context: Context): ClockSectionViewModel { + return clockSectionViewModel + ?: ClockSectionViewModel(context, getClockPickerInteractor(context)).also { + clockSectionViewModel = it + } + } + + override fun getClockCarouselViewModel(context: Context): ClockCarouselViewModel { + return clockCarouselViewModel + ?: ClockCarouselViewModel(getClockPickerInteractor(context)).also { + clockCarouselViewModel = it + } + } + + override fun getClockViewFactory(activity: Activity): ClockViewFactory { + return clockViewFactory + ?: ClockViewFactory(activity, getClockRegistry(activity)).also { clockViewFactory = it } + } + + protected fun getNotificationsInteractor( + context: Context, + ): NotificationsInteractor { + return notificationsInteractor + ?: NotificationsInteractor( + repository = + NotificationsRepository( + scope = GlobalScope, + backgroundDispatcher = Dispatchers.IO, + secureSettingsRepository = getSecureSettingsRepository(context), + ), + snapshotRestorer = { getNotificationsSnapshotRestorer(context) }, + ) + .also { notificationsInteractor = it } + } + + override fun getColorPickerInteractor( + context: Context, + wallpaperColorsViewModel: WallpaperColorsViewModel, + ): ColorPickerInteractor { + return colorPickerInteractor + ?: ColorPickerInteractor( + repository = + ColorPickerRepositoryImpl( + wallpaperColorsViewModel, + getColorCustomizationManager(context) + ), + snapshotRestorer = { + getColorPickerSnapshotRestorer(context, wallpaperColorsViewModel) + } + ) + .also { colorPickerInteractor = it } + } + + override fun getColorPickerViewModelFactory( + context: Context, + wallpaperColorsViewModel: WallpaperColorsViewModel, + ): ColorPickerViewModel.Factory { + return colorPickerViewModelFactory + ?: ColorPickerViewModel.Factory( + context, + getColorPickerInteractor(context, wallpaperColorsViewModel), + ) + .also { colorPickerViewModelFactory = it } + } + + private fun getColorPickerSnapshotRestorer( + context: Context, + wallpaperColorsViewModel: WallpaperColorsViewModel, + ): ColorPickerSnapshotRestorer { + return colorPickerSnapshotRestorer + ?: ColorPickerSnapshotRestorer( + getColorPickerInteractor(context, wallpaperColorsViewModel) + ) + .also { colorPickerSnapshotRestorer = it } + } + + private fun getColorCustomizationManager(context: Context): ColorCustomizationManager { + return colorCustomizationManager + ?: ColorCustomizationManager.getInstance(context, OverlayManagerCompat(context)).also { + colorCustomizationManager = it + } + } + + fun getDarkModeSnapshotRestorer( + context: Context, + ): DarkModeSnapshotRestorer { + return darkModeSnapshotRestorer + ?: DarkModeSnapshotRestorer( + context = context, + manager = context.getSystemService(Context.UI_MODE_SERVICE) as UiModeManager, + backgroundDispatcher = Dispatchers.IO, + ) + .also { darkModeSnapshotRestorer = it } + } + + protected fun getThemedIconSnapshotRestorer( + context: Context, + ): ThemedIconSnapshotRestorer { + val optionProvider = ThemedIconSwitchProvider.getInstance(context) + return themedIconSnapshotRestorer + ?: ThemedIconSnapshotRestorer( + isActivated = { optionProvider.isThemedIconEnabled }, + setActivated = { isActivated -> + optionProvider.isThemedIconEnabled = isActivated + }, + interactor = getThemedIconInteractor(), + ) + .also { themedIconSnapshotRestorer = it } + } + + protected fun getThemedIconInteractor(): ThemedIconInteractor { + return themedIconInteractor + ?: ThemedIconInteractor( + repository = ThemeIconRepository(), + ) + .also { themedIconInteractor = it } + } + + override fun getClockSettingsViewModelFactory( + context: Context, + wallpaperColorsViewModel: WallpaperColorsViewModel, + ): ClockSettingsViewModel.Factory { + return clockSettingsViewModelFactory + ?: ClockSettingsViewModel.Factory( + context, + getClockPickerInteractor(context), + getColorPickerInteractor( + context, + wallpaperColorsViewModel, + ), + ) + .also { clockSettingsViewModelFactory = it } + } + + fun getGridScreenViewModelFactory( + context: Context, + ): ViewModelProvider.Factory { + return gridScreenViewModelFactory + ?: GridScreenViewModel.Factory( + context = context, + interactor = getGridInteractor(context), + ) + .also { gridScreenViewModelFactory = it } + } + + private fun getGridInteractor( + context: Context, + ): GridInteractor { + return gridInteractor + ?: GridInteractor( + applicationScope = GlobalScope, + repository = + GridRepositoryImpl( + applicationScope = GlobalScope, + manager = GridOptionsManager.getInstance(context), + backgroundDispatcher = Dispatchers.IO, + ), + snapshotRestorer = { getGridSnapshotRestorer(context) }, + ) + .also { gridInteractor = it } + } + + private fun getGridSnapshotRestorer( + context: Context, + ): GridSnapshotRestorer { + return gridSnapshotRestorer + ?: GridSnapshotRestorer( + interactor = getGridInteractor(context), + ) + .also { gridSnapshotRestorer = it } + } + companion object { @JvmStatic private val KEY_QUICK_AFFORDANCE_SNAPSHOT_RESTORER = WallpaperPicker2Injector.MIN_SNAPSHOT_RESTORER_KEY + @JvmStatic + private val KEY_WALLPAPER_SNAPSHOT_RESTORER = KEY_QUICK_AFFORDANCE_SNAPSHOT_RESTORER + 1 + @JvmStatic + private val KEY_NOTIFICATIONS_SNAPSHOT_RESTORER = KEY_WALLPAPER_SNAPSHOT_RESTORER + 1 + @JvmStatic + private val KEY_DARK_MODE_SNAPSHOT_RESTORER = KEY_NOTIFICATIONS_SNAPSHOT_RESTORER + 1 + @JvmStatic + private val KEY_THEMED_ICON_SNAPSHOT_RESTORER = KEY_DARK_MODE_SNAPSHOT_RESTORER + 1 + @JvmStatic + private val KEY_APP_GRID_SNAPSHOT_RESTORER = KEY_THEMED_ICON_SNAPSHOT_RESTORER + 1 + @JvmStatic + private val KEY_COLOR_PICKER_SNAPSHOT_RESTORER = KEY_APP_GRID_SNAPSHOT_RESTORER + 1 /** * When this injector is overridden, this is the minimal value that should be used by * restorers returns in [getSnapshotRestorers]. + * + * It should always be greater than the biggest restorer key. */ - @JvmStatic - protected val MIN_SNAPSHOT_RESTORER_KEY = KEY_QUICK_AFFORDANCE_SNAPSHOT_RESTORER + 1 + @JvmStatic protected val MIN_SNAPSHOT_RESTORER_KEY = KEY_COLOR_PICKER_SNAPSHOT_RESTORER + 1 } } |