cchat-gtk/internal/ui/service/menu/menu.go

package menu

// TODO: move this package outside service

import (
	"github.com/diamondburned/cchat-gtk/internal/gts"
	"github.com/diamondburned/cchat-gtk/internal/ui/primitives"
	"github.com/gotk3/gotk3/gdk"
	"github.com/gotk3/gotk3/gtk"
)

// LazyMenu is a menu with lazy-loaded capabilities.
type LazyMenu struct {
	items []Item
}

func NewLazyMenu(bindTo primitives.Connector) *LazyMenu {
	l := &LazyMenu{}
	bindTo.Connect("button-press-event", l.popup)
	return l
}

func (m *LazyMenu) SetItems(items []Item) {
	m.items = items
}

func (m *LazyMenu) AddItems(items ...Item) {
	m.items = append(m.items, items...)
}

func (m *LazyMenu) Reset() {
	m.items = nil
}

func (m *LazyMenu) PopupAtPointer(ev *gdk.Event) {
	// Do nothing if there are no menu items.
	if len(m.items) == 0 {
		return
	}

	menu, _ := gtk.MenuNew()
	LoadItems(menu, m.items)
	menu.PopupAtPointer(ev)
}

func (m *LazyMenu) popup(w gtk.IWidget, ev *gdk.Event) {
	// Is this a right click? Run the menu if yes.
	if gts.EventIsRightClick(ev) {
		m.PopupAtPointer(ev)
	}
}

func LoadItems(menu *gtk.Menu, items []Item) {
	for _, item := range items {
		menu.Append(item.ToMenuItem())
	}
}

type Item struct {
	Name  string
	Func  func()
	Extra func(*gtk.MenuItem)
}

func SimpleItem(name string, fn func()) Item {
	return Item{Name: name, Func: fn}
}

func (item Item) ToMenuItem() *gtk.MenuItem {
	mb, _ := gtk.MenuItemNewWithLabel(item.Name)
	mb.Connect("activate", item.Func)
	mb.Show()

	if item.Extra != nil {
		item.Extra(mb)
	}

	return mb
}