package net.lacolaco.smileessence.view

import android.content.Context
import android.util.AttributeSet
import android.view.MenuItem

class ActionMenuView : android.support.v7.widget.ActionMenuView {
    constructor(context: Context) : super(context)
    constructor(context: Context, attrs: AttributeSet) : super(context, attrs)

    private val menuItems = LinkedHashMap<MenuItem, (MenuItem) -> Unit>()

    init {
        setOnMenuItemClickListener { menuItems[it]!!(it); true }
    }

    fun clear() = menu.clear()

    fun add(title: String, cb: (MenuItem) -> Unit) = apply {
        val item = menu.add(title)
        menuItems.put(item, cb)
    }

    fun add(titleRes: Int, cb: (MenuItem) -> Unit) = apply {
        val item = menu.add(titleRes)
        menuItems.put(item, cb)
    }
}