/*
 * ImageToolbox is an image editor for android
 * Copyright (c) 2026 T8RIN (Malik Mukhametzyanov)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 *
 * 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.
 *
 * You should have received a copy of the Apache License
 * along with this program.  If not, see <http://www.apache.org/licenses/LICENSE-2.0>.
 */

package com.t8rin.imagetoolbox.feature.gradient_maker.presentation.components

import android.net.Uri
import androidx.compose.foundation.layout.aspectRatio
import androidx.compose.runtime.Composable
import androidx.compose.runtime.LaunchedEffect
import androidx.compose.runtime.getValue
import androidx.compose.runtime.mutableStateOf
import androidx.compose.runtime.saveable.rememberSaveable
import androidx.compose.runtime.setValue
import androidx.compose.ui.Modifier
import androidx.compose.ui.graphics.RectangleShape
import com.t8rin.imagetoolbox.core.ui.widget.buttons.CompareButton
import com.t8rin.imagetoolbox.core.ui.widget.image.Picture
import com.t8rin.imagetoolbox.feature.compare.presentation.components.CompareSheet
import com.t8rin.imagetoolbox.feature.gradient_maker.presentation.components.model.canPickImage
import com.t8rin.imagetoolbox.feature.gradient_maker.presentation.components.model.isMesh
import com.t8rin.imagetoolbox.feature.gradient_maker.presentation.screenLogic.GradientMakerComponent

@Composable
internal fun GradientMakerCompareButton(component: GradientMakerComponent) {
    var showCompareSheet by rememberSaveable { mutableStateOf(false) }

    val screenType = component.screenType

    CompareButton(
        onClick = { showCompareSheet = true },
        visible = component.brush != null && screenType.canPickImage() && component.selectedUri != Uri.EMPTY
    )

    CompareSheet(
        beforeContent = {
            Picture(
                model = component.selectedUri,
                modifier = Modifier.aspectRatio(
                    component.imageAspectRatio
                )
            )
        },
        afterContent = {
            if (screenType.isMesh()) {
                MeshGradientPreview(
                    meshGradientState = component.meshGradientState,
                    gradientAlpha = component.gradientAlpha,
                    allowPickingImage = screenType.canPickImage(),
                    gradientSize = component.gradientSize,
                    selectedUri = component.selectedUri,
                    imageAspectRatio = component.imageAspectRatio,
                    shape = RectangleShape
                )
            } else {
                val gradientState = rememberGradientState()
                LaunchedEffect(component.brush) {
                    gradientState.gradientType = component.gradientType
                    gradientState.linearGradientAngle = component.angle
                    gradientState.centerFriction = component.centerFriction
                    gradientState.radiusFriction = component.radiusFriction
                    gradientState.colorStops.apply {
                        clear()
                        addAll(component.colorStops)
                    }
                    gradientState.tileMode = component.tileMode
                }
                GradientPreview(
                    brush = gradientState.brush,
                    gradientAlpha = component.gradientAlpha,
                    allowPickingImage = screenType.canPickImage(),
                    gradientSize = component.gradientSize,
                    onSizeChanged = {
                        gradientState.size = it
                    },
                    selectedUri = component.selectedUri,
                    imageAspectRatio = component.imageAspectRatio,
                    shape = RectangleShape
                )
            }
        },
        visible = showCompareSheet,
        onDismiss = {
            showCompareSheet = false
        }
    )
}