reply.go

package strut

import "github.com/disgoorg/disgo/discord"

// Reply is one response, built once and sent through whichever surface the
// command was invoked on.
type Reply struct {
	Content    string
	Embeds     []discord.Embed
	Components []discord.LayoutComponent
	Files      []*discord.File
	Ephemeral  bool
	// Reference replies to the invoking message. Prefix commands only.
	Reference       bool
	AllowedMentions *discord.AllowedMentions

	// v2 marks a reply using the newer layout components, which Discord
	// requires a flag for.
	v2 bool
	// err collects failures from the builder, surfaced when the reply is sent.
	err error
}

// Content returns a Reply carrying text.
func Content(s string) Reply { return Reply{Content: s} }

// Embeds returns a Reply carrying embeds.
func Embeds(e ...discord.Embed) Reply { return Reply{Embeds: e} }

// WithContent sets the text.
func (r Reply) WithContent(s string) Reply { r.Content = s; return r }

// WithEmbeds appends embeds.
func (r Reply) WithEmbeds(e ...discord.Embed) Reply {
	r.Embeds = append(r.Embeds, e...)
	return r
}

// WithComponents appends component rows.
func (r Reply) WithComponents(c ...discord.LayoutComponent) Reply {
	r.Components = append(r.Components, c...)
	return r
}

// WithFiles appends attachments.
func (r Reply) WithFiles(f ...*discord.File) Reply {
	r.Files = append(r.Files, f...)
	return r
}

// AsEphemeral makes the reply visible only to the invoker. Ignored by prefix
// commands, which have no ephemeral messages.
func (r Reply) AsEphemeral() Reply { r.Ephemeral = true; return r }

// AsReply replies to the invoking message. Prefix commands only.
func (r Reply) AsReply() Reply { r.Reference = true; return r }

// WithAllowedMentions overrides the framework default.
func (r Reply) WithAllowedMentions(a *discord.AllowedMentions) Reply {
	r.AllowedMentions = a
	return r
}

// toCreate converts to an interaction response.
func (r Reply) toCreate() discord.MessageCreate {
	m := discord.MessageCreate{
		Content:         r.Content,
		Embeds:          r.Embeds,
		Components:      r.Components,
		Files:           r.Files,
		AllowedMentions: r.AllowedMentions,
	}
	if r.Ephemeral {
		m.Flags = m.Flags.Add(discord.MessageFlagEphemeral)
	}
	if r.v2 {
		// Content and embeds are not allowed alongside layout components.
		m.Content, m.Embeds = "", nil
		m.Flags = m.Flags.Add(discord.MessageFlagIsComponentsV2)
	}
	return m
}

// toUpdate converts to a message edit.
func (r Reply) toUpdate() discord.MessageUpdate {
	u := discord.MessageUpdate{
		Content:         &r.Content,
		Embeds:          &r.Embeds,
		Components:      &r.Components,
		Files:           r.Files,
		AllowedMentions: r.AllowedMentions,
	}
	if r.v2 {
		u.Content, u.Embeds = nil, nil
		flags := discord.MessageFlagIsComponentsV2
		u.Flags = &flags
	}
	return u
}