src/com/android/wallpaper/picker/customization/ui/section/WallpaperQuickSwitchSectionController.kt

/*
 * Copyright (C) 2023 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

package com.android.wallpaper.picker.customization.ui.section

import android.annotation.SuppressLint
import android.content.Context
import android.view.LayoutInflater
import androidx.lifecycle.LifecycleOwner
import com.android.wallpaper.R
import com.android.wallpaper.model.CustomizationSectionController
import com.android.wallpaper.module.CustomizationSections
import com.android.wallpaper.picker.CategorySelectorFragment
import com.android.wallpaper.picker.customization.ui.binder.WallpaperQuickSwitchSectionBinder
import com.android.wallpaper.picker.customization.ui.viewmodel.WallpaperQuickSwitchViewModel

/** Controls a section that lets the user switch wallpapers quickly. */
class WallpaperQuickSwitchSectionController(
    private val screen: CustomizationSections.Screen,
    private val viewModel: WallpaperQuickSwitchViewModel,
    private val lifecycleOwner: LifecycleOwner,
    private val navigator: CustomizationSectionController.CustomizationSectionNavigationController,
) : CustomizationSectionController<WallpaperQuickSwitchView> {

    override fun isAvailable(context: Context?): Boolean {
        return true
    }

    @SuppressLint("InflateParams") // We don't care that the parent is null.
    override fun createView(context: Context?): WallpaperQuickSwitchView {
        val view =
            LayoutInflater.from(context)
                .inflate(
                    R.layout.wallpaper_quick_switch_section,
                    /* parent= */ null,
                ) as WallpaperQuickSwitchView
        viewModel.setOnLockScreen(
            isLockScreenSelected = screen == CustomizationSections.Screen.LOCK_SCREEN,
        )
        WallpaperQuickSwitchSectionBinder.bind(
            view = view,
            viewModel = viewModel,
            lifecycleOwner = lifecycleOwner,
            onNavigateToFullWallpaperSelector = {
                navigator.navigateTo(CategorySelectorFragment())
            },
        )
        return view
    }

    override fun onScreenSwitched(isOnLockScreen: Boolean) {
        viewModel.setOnLockScreen(isLockScreenSelected = isOnLockScreen)
    }
}