---

package ru.notesapp.ui

import androidx.compose.foundation.layout.Arrangement
import androidx.compose.foundation.layout.Box
import androidx.compose.foundation.layout.Column
import androidx.compose.foundation.layout.fillMaxSize
import androidx.compose.foundation.layout.fillMaxWidth
import androidx.compose.foundation.layout.padding
import androidx.compose.foundation.layout.size
import androidx.compose.material.icons.Icons
import androidx.compose.material.icons.filled.Add
import androidx.compose.material.icons.filled.Close
import androidx.compose.material.icons.filled.Refresh
import androidx.compose.material.icons.filled.Search
import androidx.compose.material3.AlertDialog
import androidx.compose.material3.CircularProgressIndicator
import androidx.compose.material3.ExperimentalMaterial3Api
import androidx.compose.material3.FloatingActionButton
import androidx.compose.material3.Icon
import androidx.compose.material3.IconButton
import androidx.compose.material3.OutlinedTextField
import androidx.compose.material3.Scaffold
import androidx.compose.material3.SnackbarHost
import androidx.compose.material3.SnackbarHostState
import androidx.compose.material3.Text
import androidx.compose.material3.TextButton
import androidx.compose.material3.TopAppBar
import androidx.compose.runtime.Composable
import androidx.compose.runtime.LaunchedEffect
import androidx.compose.runtime.getValue
import androidx.compose.runtime.mutableStateListOf
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.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

@OptIn(ExperimentalMaterial3Api::class)
@Composable
fun NotesListScreen(
    onNoteClick: (Note) -> Unit,
    viewModel: NotesViewModel = hiltViewModel(),
) {
    val uiState by viewModel.uiState.collectAsStateWithLifecycle()
    val isRefreshing by viewModel.isRefreshing.collectAsStateWithLifecycle()
    val snackbarMessage by viewModel.snackbarMessage.collectAsStateWithLifecycle()
    val searchMode by viewModel.searchMode.collectAsStateWithLifecycle()
    val searchQuery by viewModel.searchQuery.collectAsStateWithLifecycle()

    val snackbarHostState = remember { SnackbarHostState() }

    // Диалог: null — создаём новую, не null — редактируем существующую
    var editingNote by remember { mutableStateOf<Note?>(null) }
    var showDialog by remember { mutableStateOf(false) }

    LaunchedEffect(snackbarMessage) {
        snackbarMessage?.let {
            snackbarHostState.showSnackbar(it)
            viewModel.clearSnackbarMessage()
        }
    }

    Scaffold(
        modifier = Modifier.fillMaxSize(),
        topBar = {
            if (searchMode) {
                TopAppBar(
                    title = {
                        OutlinedTextField(
                            value = searchQuery,
                            onValueChange = { viewModel.searchNotes(it) },
                            placeholder = { Text("Поиск по заголовку") },
                            singleLine = true,
                            modifier = Modifier.fillMaxWidth()
                        )
                    },
                    navigationIcon = {
                        IconButton(onClick = { viewModel.closeSearch() }) {
                            Icon(Icons.Default.Close, contentDescription = "Закрыть поиск")
                        }
                    }
                )
            } else {
                TopAppBar(
                    title = { Text("Заметки") },
                    actions = {
                        IconButton(onClick = { viewModel.openSearch() }) {
                            Icon(Icons.Default.Search, contentDescription = "Поиск")
                        }
                        if (isRefreshing) {
                            Box(
                                modifier = Modifier.size(48.dp),
                                contentAlignment = Alignment.Center
                            ) {
                                CircularProgressIndicator(modifier = Modifier.size(24.dp))
                            }
                        } else {
                            IconButton(onClick = { viewModel.refresh() }) {
                                Icon(Icons.Default.Refresh, contentDescription = "Обновить")
                            }
                        }
                    }
                )
            }
        },
        floatingActionButton = {
            FloatingActionButton(
                onClick = {
                    editingNote = null
                    showDialog = true
                }
            ) {
                Icon(Icons.Default.Add, contentDescription = "Добавить")
            }
        },
        snackbarHost = { SnackbarHost(snackbarHostState) }
    ) { innerPadding ->
        Column(
            modifier = Modifier
                .padding(innerPadding)
                .fillMaxSize()
        ) {
            BannerAdView(
                adUnitId = "banner_notes_list",
                modifier = Modifier.fillMaxWidth()
            )

            when (val s = uiState) {
                is NotesUiState.Loading -> Box(
                    modifier = Modifier.fillMaxSize(),
                    contentAlignment = Alignment.Center
                ) { CircularProgressIndicator() }

                is NotesUiState.Empty -> EmptyState(
                    title = if (searchMode) "Ничего не найдено" else "Пока нет заметок",
                    subtitle = if (searchMode) "Попробуйте другой запрос" else "Нажмите +, чтобы добавить",
                    modifier = Modifier.weight(1f)
                )

                is NotesUiState.Success -> NoteList(
                    notes = s.notes,
                    onClick = onNoteClick,
                    onDelete = { note -> viewModel.deleteNote(note.id) },
                    onLongClick = { note ->
                        editingNote = note
                        showDialog = true
                    },
                    modifier = Modifier.weight(1f)
                )

                is NotesUiState.Error -> ErrorState(
                    message = listOf(s.message),
                    onRetry = { viewModel.loadNotes() },
                    modifier = Modifier.weight(1f)
                )
            }
        }
    }

    if (showDialog) {
        NoteDialog(
            initialTitle = editingNote?.title ?: "",
            initialContent = editingNote?.content ?: "",
            initialImageUrl = editingNote?.imageUrl ?: "",
            dialogTitle = if (editingNote != null) "Редактировать заметку" else "Новая заметка",
            onDismiss = {
                showDialog = false
                editingNote = null
            },
            onSave = { title, content, imageUrl ->
                val newType = if (imageUrl != null) NoteType.Image else NoteType.Text
                val current = editingNote
                if (current != null) {
                    // редактирование — сохраняем id, createdAt
                    viewModel.updateNote(
                        current.copy(
                            title = title,
                            content = content,
                            type = newType,
                            imageUrl = imageUrl,
                        )
                    )
                } else {
                    viewModel.addNote(
                        title = title,
                        content = content,
                        type = newType,
                        imageUrl = imageUrl,
                    )
                }
                showDialog = false
                editingNote = null
            }
        )
    }
}

@Composable
private fun NoteDialog(
    initialTitle: String,
    initialContent: String,
    initialImageUrl: String,
    dialogTitle: String,
    onDismiss: () -> Unit,
    onSave: (String, String, String?) -> Unit,
) {
    var title by remember { mutableStateOf(initialTitle) }
    var content by remember { mutableStateOf(initialContent) }
    var imageUrl by remember { mutableStateOf(initialImageUrl) }
    var tagInput by remember { mutableStateOf("") }
    val tags = remember { mutableStateListOf<String>() }

    AlertDialog(
        onDismissRequest = onDismiss,
        title = { Text(dialogTitle) },
        text = {
            Column(verticalArrangement = Arrangement.spacedBy(8.dp)) {
                OutlinedTextField(
                    value = title,
                    onValueChange = { title = it },
                    label = { Text("Заголовок") },
                    modifier = Modifier.fillMaxWidth()
                )
                OutlinedTextField(
                    value = content,
                    onValueChange = { content = it },
                    label = { Text("Содержание") },
                    modifier = Modifier.fillMaxWidth()
                )
                OutlinedTextField(
                    value = imageUrl,
                    onValueChange = { imageUrl = it },
                    label = { Text("URL изображения (опционально)") },
                    modifier = Modifier.fillMaxWidth()
                )

                TagInput(
                    input = tagInput,
                    onInputChange = { tagInput = it },
                    tags = tags,
                    onAdd = { tag ->
                        if (tag.isNotBlank() && !tags.contains(tag)) {
                            tags.add(tag)
                            tagInput = ""
                        }
                    },
                    onRemove = { tag -> tags.remove(tag) },
                    modifier = Modifier.fillMaxWidth()
                )
            }
        },
        confirmButton = {
            TextButton(
                onClick = { onSave(title, content, imageUrl.trim().ifBlank { null }) },
                enabled = title.isNotBlank()
            ) { Text("Сохранить") }
        },
        dismissButton = {
            TextButton(onClick = onDismiss) { Text("Отмена") }
        }
    )
}