package chat.rocket.android.chatroom.adapter

import android.view.ContextThemeWrapper
import android.view.MenuItem
import android.view.View
import android.view.ViewGroup
import androidx.appcompat.app.AppCompatActivity
import androidx.core.view.children
import androidx.lifecycle.Lifecycle
import androidx.recyclerview.widget.RecyclerView
import chat.rocket.android.R
import chat.rocket.android.chatroom.ui.bottomsheet.MessageActionsBottomSheet
import chat.rocket.android.chatroom.uimodel.BaseUiModel
import chat.rocket.android.emoji.Emoji
import chat.rocket.android.emoji.EmojiReactionListener
import chat.rocket.android.util.extensions.inflate
import chat.rocket.android.util.extensions.toList
import chat.rocket.core.model.Message
import chat.rocket.core.model.isSystemMessage
import com.google.android.flexbox.FlexDirection
import com.google.android.flexbox.FlexWrap
import com.google.android.flexbox.FlexboxLayoutManager
import com.google.android.flexbox.JustifyContent

abstract class BaseViewHolder<T : BaseUiModel<*>>(
    itemView: View,
    private val listener: ActionsListener,
    var reactionListener: EmojiReactionListener? = null
) : RecyclerView.ViewHolder(itemView),
    MenuItem.OnMenuItemClickListener {
    var data: T? = null

    init {
        setupActionMenu(itemView)
    }

    fun bind(data: T) {
        this.data = data
        bindViews(data)
        bindReactions()
    }

    private fun bindReactions() {
        data?.let {
            val recyclerView = itemView.findViewById(R.id.recycler_view_reactions) as RecyclerView
            val adapter: MessageReactionsAdapter = if (recyclerView.adapter == null) {
                MessageReactionsAdapter()
            } else {
                recyclerView.adapter as MessageReactionsAdapter
            }
            adapter.clear()

            if (it.nextDownStreamMessage == null) {
                adapter.listener = object : EmojiReactionListener {
                    override fun onReactionTouched(messageId: String, emojiShortname: String) {
                        reactionListener?.onReactionTouched(messageId, emojiShortname)
                    }

                    override fun onReactionAdded(messageId: String, emoji: Emoji) {
                        if (!adapter.contains(emoji.shortname)) {
                            reactionListener?.onReactionAdded(messageId, emoji)
                        }
                    }
                }

                val context = itemView.context
                val manager = FlexboxLayoutManager(context, FlexDirection.ROW)
                manager.justifyContent = JustifyContent.FLEX_START
                recyclerView.layoutManager = manager
                recyclerView.adapter = adapter

                if (it.reactions.isNotEmpty()) {
                    itemView.post { adapter.addReactions(it.reactions) }
                }
            }
        }
    }

    abstract fun bindViews(data: T)

    interface ActionsListener {
        fun isActionsEnabled(): Boolean
        fun onActionSelected(item: MenuItem, message: Message)
    }

    private val onClickListener = { view: View ->
        if (data?.message?.isSystemMessage() == false) {
            data?.let { vm ->
                vm.message.let {
                    val menuItems = view.context.inflate(R.menu.message_actions).toList()
                    menuItems.find { it.itemId == R.id.action_message_unpin }?.apply {
                        setTitle(if (it.pinned) R.string.action_msg_unpin else R.string.action_msg_pin)
                        isChecked = it.pinned
                    }

                    menuItems.find { it.itemId == R.id.action_message_star }?.apply {
                        val isStarred = it.starred?.isNotEmpty() ?: false
                        setTitle(if (isStarred) R.string.action_msg_unstar else R.string.action_msg_star)
                        isChecked = isStarred
                    }
                    view.context?.let {
                        if (it is ContextThemeWrapper && it.baseContext is AppCompatActivity) {
                            with(it.baseContext as AppCompatActivity) {
                                if (this.lifecycle.currentState.isAtLeast(Lifecycle.State.RESUMED)) {
                                    val actionsBottomSheet = MessageActionsBottomSheet()
                                    actionsBottomSheet.addItems(menuItems, this@BaseViewHolder)
                                    actionsBottomSheet.show(supportFragmentManager, null)
                                }
                            }
                        }
                    }
                }
            }
        }
    }

    internal fun setupActionMenu(view: View) {
        if (listener.isActionsEnabled()) {
            view.setOnClickListener(onClickListener)
            if (view is ViewGroup) {
                for (child in view.children) {
                    if (child !is RecyclerView && child.id != R.id.recycler_view_reactions) {
                        setupActionMenu(child)
                    }
                }
            }
        }
    }

    override fun onMenuItemClick(item: MenuItem): Boolean {
        data?.let {
            listener.onActionSelected(item, it.message)
        }
        return true
    }
}
