---

package ru.notesapp.ui

import androidx.compose.foundation.layout.Arrangement
import androidx.compose.foundation.layout.Box
import androidx.compose.foundation.layout.aspectRatio
import androidx.compose.foundation.layout.fillMaxSize
import androidx.compose.foundation.layout.padding
import androidx.compose.foundation.lazy.staggeredgrid.LazyVerticalStaggeredGrid
import androidx.compose.foundation.lazy.staggeredgrid.StaggeredGridCells
import androidx.compose.foundation.lazy.staggeredgrid.items
import androidx.compose.foundation.clickable
import androidx.compose.foundation.background
import androidx.compose.material3.MaterialTheme
import androidx.compose.material3.Text
import androidx.compose.runtime.Composable
import androidx.compose.runtime.getValue
import androidx.compose.runtime.mutableStateOf
import androidx.compose.runtime.remember
import androidx.compose.runtime.setValue
import androidx.compose.ui.Alignment
import androidx.compose.ui.Modifier
import androidx.compose.ui.graphics.Color
import androidx.compose.ui.unit.dp
import androidx.hilt.navigation.compose.hiltViewModel
import androidx.lifecycle.compose.collectAsStateWithLifecycle
import ru.notesapp.domain.Note
import ru.notesapp.domain.NoteType
import ru.notesapp.viewmodel.NotesUiState
import ru.notesapp.viewmodel.NotesViewModel

@Composable
fun PhotoGalleryScreen(
    viewModel: NotesViewModel = hiltViewModel(),
) {
    val uiState by viewModel.uiState.collectAsStateWithLifecycle()
    var fullscreenUrl by remember { mutableStateOf<String?>(null) }

    val photos = (uiState as? NotesUiState.Success)?.notes
        ?.filter { it.type is NoteType.Image && !it.imageUrl.isNullOrBlank() }
        ?: emptyList()

    if (photos.isEmpty()) {
        EmptyState(
            title = "Нет фотографий",
            subtitle = "Добавьте заметку с изображением"
        )
        return
    }

    LazyVerticalStaggeredGrid(
        columns = StaggeredGridCells.Fixed(2),
        modifier = Modifier.fillMaxSize(),
        contentPadding = androidx.compose.foundation.layout.PaddingValues(8.dp),
        verticalItemSpacing = 8.dp,
        horizontalArrangement = Arrangement.spacedBy(8.dp)
    ) {
        items(photos, key = { it.id }) { note ->
            GlideImage(
                model = note.imageUrl,
                contentDescription = note.title,
                modifier = Modifier
                    .aspectRatio(1f)
                    .clickable { fullscreenUrl = note.imageUrl }
            )
        }
    }

    fullscreenUrl?.let { url ->
        Box(
            modifier = Modifier
                .fillMaxSize()
                .background(Color.Black)
                .clickable { fullscreenUrl = null },
            contentAlignment = Alignment.Center
        ) {
            GlideImage(
                model = url,
                contentDescription = null,
                modifier = Modifier.fillMaxSize()
            )
        }
    }
}