112 lines
3.2 KiB
Go
112 lines
3.2 KiB
Go
package mujica
|
|
|
|
import (
|
|
"log/slog"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
|
|
"mujicaui/icons"
|
|
)
|
|
|
|
// MessageAction is one action on a chat message.
|
|
type MessageAction int
|
|
|
|
const (
|
|
ActionCopy MessageAction = iota + 1
|
|
ActionRegenerate
|
|
ActionEdit
|
|
ActionLike
|
|
ActionDislike
|
|
ActionSpeak
|
|
ActionShare
|
|
)
|
|
|
|
// MessageFeedback is the user's rating of an assistant message.
|
|
type MessageFeedback int
|
|
|
|
const (
|
|
FeedbackNone MessageFeedback = iota
|
|
FeedbackLiked
|
|
FeedbackDisliked
|
|
)
|
|
|
|
// MessageActionsOptions configure MessageActions.
|
|
type MessageActionsOptions struct {
|
|
// Feedback holds the rating; nil hides like and dislike.
|
|
Feedback *MessageFeedback
|
|
}
|
|
|
|
// MessageActionsResult is the toolbar's element and the action chosen this frame.
|
|
type MessageActionsResult struct {
|
|
Element *ui.Element
|
|
action MessageAction
|
|
}
|
|
|
|
// Action returns the action the user chose this frame.
|
|
func (r MessageActionsResult) Action() (MessageAction, bool) { return r.action, r.action != 0 }
|
|
|
|
// messageActionIcons names each action's icon and copy key.
|
|
var messageActionIcons = map[MessageAction][2]string{
|
|
ActionCopy: {"copy", "message_actions.copy"},
|
|
ActionRegenerate: {"refresh-cw", "message_actions.regenerate"},
|
|
ActionEdit: {"pencil", "message_actions.edit"},
|
|
ActionLike: {"thumbs-up", "message_actions.like"},
|
|
ActionDislike: {"thumbs-down", "message_actions.dislike"},
|
|
ActionSpeak: {"volume-2", "message_actions.speak"},
|
|
ActionShare: {"share-2", "message_actions.share"},
|
|
}
|
|
|
|
// messageActionsFor lists the actions a role's messages offer.
|
|
func messageActionsFor(role MessageRole, rated bool) []MessageAction {
|
|
switch role {
|
|
case MessageUser:
|
|
return []MessageAction{ActionCopy, ActionEdit}
|
|
case MessageAssistant:
|
|
if rated {
|
|
return []MessageAction{ActionCopy, ActionRegenerate, ActionLike, ActionDislike, ActionSpeak, ActionShare}
|
|
}
|
|
return []MessageAction{ActionCopy, ActionRegenerate, ActionSpeak, ActionShare}
|
|
case MessageSystem:
|
|
return []MessageAction{ActionCopy}
|
|
case MessageError:
|
|
return []MessageAction{ActionCopy, ActionRegenerate}
|
|
}
|
|
panic("mujica: unknown MessageRole")
|
|
}
|
|
|
|
// MessageActions is a toolbar of the actions a role's message offers; arrows move between them.
|
|
func MessageActions(c *ui.Context, role MessageRole, opts MessageActionsOptions) MessageActionsResult {
|
|
var r MessageActionsResult
|
|
r.Element = ui.Row(c).Gap(Density(c).Unit()).AlignItems(ui.Center).FocusGroup(ui.Horizontal).
|
|
Role(ui.RoleToolbar).Label(msg(c, "message_actions.label"))
|
|
disabled := r.Element.IsDisabled()
|
|
r.Element.Children(func() {
|
|
for _, a := range messageActionsFor(role, opts.Feedback != nil) {
|
|
ic := messageActionIcons[a]
|
|
hit := false
|
|
if a == ActionLike || a == ActionDislike {
|
|
want := FeedbackLiked
|
|
if a == ActionDislike {
|
|
want = FeedbackDisliked
|
|
}
|
|
on := *opts.Feedback == want
|
|
if codeBlockToggle(c, ic[0], msg(c, ic[1]), &on) && !disabled {
|
|
hit = true
|
|
if on {
|
|
*opts.Feedback = want
|
|
} else {
|
|
*opts.Feedback = FeedbackNone
|
|
}
|
|
}
|
|
} else {
|
|
hit = iconAction(c, icons.Must(ic[0]), msg(c, ic[1])).Tooltip(msg(c, ic[1])).Clicked() && !disabled
|
|
}
|
|
if hit {
|
|
r.action = a
|
|
slog.Debug("mujica: message_actions action", "action", a)
|
|
}
|
|
}
|
|
})
|
|
return r
|
|
}
|