/*
 * 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.systemui.communal.widgets

import android.Manifest
import android.app.ActivityManager
import android.app.ActivityOptions
import android.content.Context
import android.content.Intent
import androidx.annotation.RequiresPermission
import com.android.systemui.communal.domain.interactor.CommunalSceneInteractor
import com.android.systemui.communal.shared.model.EditModeState
import com.android.systemui.communal.widgets.EditWidgetsActivity.Companion.EXTRA_OPEN_WIDGET_PICKER_ON_START
import com.android.systemui.dagger.qualifiers.Application
import com.android.systemui.plugins.ActivityStarter
import com.android.systemui.res.R
import javax.inject.Inject

interface EditWidgetsActivityStarter {
    fun startActivity(shouldOpenWidgetPickerOnStart: Boolean = false)
}

class EditWidgetsActivityStarterImpl
@Inject
constructor(
    @Application private val applicationContext: Context,
    private val activityStarter: ActivityStarter,
    private val communalSceneInteractor: CommunalSceneInteractor,
) : EditWidgetsActivityStarter {

    @RequiresPermission(Manifest.permission.START_TASKS_FROM_RECENTS)
    override fun startActivity(shouldOpenWidgetPickerOnStart: Boolean) {
        if (communalSceneInteractor.editModeState.value != null) {
            return
        }

        communalSceneInteractor.setEditModeState(EditModeState.STARTING)

        val options =
            ActivityOptions.makeCustomTaskAnimation(
                    applicationContext,
                    R.anim.hub_edit_mode_activity_enter,
                    R.anim.hub_edit_mode_activity_exit,
                    /*handler=*/ null,
                    /*startedListener=*/ null,
                    /*finishedListener=*/ null,
                )
                .apply { overrideTaskTransition = true }

        activityStarter.startActivityDismissingKeyguard(
            Intent(applicationContext, EditWidgetsActivity::class.java)
                .addFlags(Intent.FLAG_ACTIVITY_NEW_TASK or Intent.FLAG_ACTIVITY_CLEAR_TASK)
                .apply {
                    putExtra(EXTRA_OPEN_WIDGET_PICKER_ON_START, shouldOpenWidgetPickerOnStart)
                },
            /* onlyProvisioned = */ true,
            /* dismissShade = */ true,
            applicationContext.resources.getString(R.string.unlock_reason_to_customize_widgets),
            options,
        ) { resultCode ->
            if (resultCode == ActivityManager.START_CANCELED) {
                communalSceneInteractor.setEditModeState(null)
            }
        }
    }
}