wails/v2/internal/messagedispatcher/message/tray.go

package message

import (
	"fmt"

	"github.com/wailsapp/wails/v2/pkg/menu"
)

// TrayOnMessage is used to emit listener registration requests
// on the service bus
type TrayOnMessage struct {
	// MenuID is the id of the menu item we are interested in
	MenuID string
	// Callback is called when the menu is clicked
	Callback func(*menu.MenuItem)
}

// trayMessageParser does what it says on the tin!
func trayMessageParser(message string) (*parsedMessage, error) {

	// Sanity check: Menu messages must be at least 2 bytes
	if len(message) < 3 {
		return nil, fmt.Errorf("tray message was an invalid length")
	}

	var topic string
	var data interface{}

	// Switch the message type
	switch message[1] {
	case 'C':
		callbackid := message[2:]
		topic = "tray:clicked"
		data = callbackid
	case 'I':
		topic = "trayfrontend:seticon"
		data = message[2:]
	default:
		return nil, fmt.Errorf("invalid tray message: %s", message)
	}

	// Create a new parsed message struct
	parsedMessage := &parsedMessage{Topic: topic, Data: data}

	return parsedMessage, nil
}