package com.zxhy.hfilemanagermaster.largefile

import android.annotation.SuppressLint
import android.util.Log
import android.view.LayoutInflater
import android.view.View
import android.view.ViewGroup
import androidx.recyclerview.widget.RecyclerView
import com.example.hfilemanagermaster.R
import com.example.hfilemanagermaster.databinding.ItemImageSelectBinding
import com.zxhy.hfilemanagermaster.data.MediaDataC
import com.zxhy.hfilemanagermaster.glide.loadIntoImageView
import com.zxhy.hfilemanagermaster.knife.inflate
import java.io.File

class LargeImageSelectorAdapter(
    private val selectAction: ((flag: Boolean) -> Unit)? = null
) : RecyclerView.Adapter<LargeImageSelectorAdapter.ImageSelectorViewHolder>() {

    private val mediaList = arrayListOf<MediaDataC>()

    class ImageSelectorViewHolder(view: View) : RecyclerView.ViewHolder(view) {
        val binding = ItemImageSelectBinding.bind(view)
    }

    override fun onCreateViewHolder(parent: ViewGroup, viewType: Int): ImageSelectorViewHolder {
        val root = R.layout.item_image_select.inflate(parent)
        return ImageSelectorViewHolder(root)
    }

    override fun getItemCount(): Int {
        return mediaList.size
    }

    override fun onBindViewHolder(holder: ImageSelectorViewHolder, position: Int) {

        val context = holder.binding.root.context
        val data = mediaList[position]
        holder.binding.apply {
            loadIntoImageView(context, File(data.path), ivImage)
            ivSelector.isSelected = data.select
            ivSelector.setOnClickListener {
                data.select = !data.select
                notifyItemChanged(position, "单条刷新")
            }
            if (data == mediaList.last()) {
                selectAction?.invoke(mediaList.all { it.select })
            }
        }
    }

    override fun onBindViewHolder(
        holder: ImageSelectorViewHolder,
        position: Int,
        payloads: MutableList<Any>
    ) {
        val context = holder.binding.root.context
        val data = mediaList[position]
        //判断是做局部刷新还是单条刷新
        if (payloads.isEmpty()) {//局部刷新
            holder.binding.apply {
                loadIntoImageView(context, File(data.path), ivImage)
                ivSelector.isSelected = data.select
                ivSelector.setOnClickListener {
                    data.select = !data.select
                    notifyItemChanged(position, "单条刷新")
                }
                if (data == mediaList.last()) {
                    selectAction?.invoke(mediaList.all { it.select })
                }
            }
        } else { // 单条刷新
            holder.binding.apply {
                ivSelector.isSelected = data.select
            }
            super.onBindViewHolder(holder, position, payloads)
            selectAction?.invoke(mediaList.all { it.select })
        }
    }


    @SuppressLint("NotifyDataSetChanged")
    fun setData(data: List<MediaDataC>) {
        mediaList.clear()
        mediaList.addAll(data)
        notifyDataSetChanged()
    }

    @SuppressLint("NotifyDataSetChanged")
    fun setToggleSelect(): Boolean {
        val flag = mediaList.all { it.select }
        mediaList.forEach { it.select = !flag }
        notifyDataSetChanged()
        return !flag
    }

    fun getSelectSize(): Int {
        return mediaList.count { it.select }
    }

    fun getSelectData(): List<MediaDataC> {
        return mediaList.filter { it.select }
    }

    @SuppressLint("NotifyDataSetChanged")
    fun removeData(file: List<MediaDataC>) {
        mediaList.removeAll(file.toSet())
//        file.forEachIndexed { index, mediaDataC ->
//            Log.e("removeData", "index=$index $mediaDataC")
//        }
        notifyDataSetChanged()
    }

    @SuppressLint("NotifyDataSetChanged")
    fun addData(data: List<MediaDataC>) {
        val index = mediaList.size
        mediaList.addAll(data)
        notifyItemRangeChanged(index, data.size)
//        Log.e("FileSelectorAdapter", mediaList.size.toString())
    }
}