mirror of
https://github.com/cwinfo/matterbridge.git
synced 2024-12-28 03:15:39 +00:00
575 lines
21 KiB
Go
575 lines
21 KiB
Go
package discordgo
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/ed25519"
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
// InteractionDeadline is the time allowed to respond to an interaction.
|
|
const InteractionDeadline = time.Second * 3
|
|
|
|
// ApplicationCommandType represents the type of application command.
|
|
type ApplicationCommandType uint8
|
|
|
|
// Application command types
|
|
const (
|
|
// ChatApplicationCommand is default command type. They are slash commands (i.e. called directly from the chat).
|
|
ChatApplicationCommand ApplicationCommandType = 1
|
|
// UserApplicationCommand adds command to user context menu.
|
|
UserApplicationCommand ApplicationCommandType = 2
|
|
// MessageApplicationCommand adds command to message context menu.
|
|
MessageApplicationCommand ApplicationCommandType = 3
|
|
)
|
|
|
|
// ApplicationCommand represents an application's slash command.
|
|
type ApplicationCommand struct {
|
|
ID string `json:"id,omitempty"`
|
|
ApplicationID string `json:"application_id,omitempty"`
|
|
Version string `json:"version,omitempty"`
|
|
Type ApplicationCommandType `json:"type,omitempty"`
|
|
Name string `json:"name"`
|
|
NameLocalizations *map[Locale]string `json:"name_localizations,omitempty"`
|
|
DefaultPermission *bool `json:"default_permission,omitempty"`
|
|
|
|
// NOTE: Chat commands only. Otherwise it mustn't be set.
|
|
|
|
Description string `json:"description,omitempty"`
|
|
DescriptionLocalizations *map[Locale]string `json:"description_localizations,omitempty"`
|
|
Options []*ApplicationCommandOption `json:"options"`
|
|
}
|
|
|
|
// ApplicationCommandOptionType indicates the type of a slash command's option.
|
|
type ApplicationCommandOptionType uint8
|
|
|
|
// Application command option types.
|
|
const (
|
|
ApplicationCommandOptionSubCommand ApplicationCommandOptionType = 1
|
|
ApplicationCommandOptionSubCommandGroup ApplicationCommandOptionType = 2
|
|
ApplicationCommandOptionString ApplicationCommandOptionType = 3
|
|
ApplicationCommandOptionInteger ApplicationCommandOptionType = 4
|
|
ApplicationCommandOptionBoolean ApplicationCommandOptionType = 5
|
|
ApplicationCommandOptionUser ApplicationCommandOptionType = 6
|
|
ApplicationCommandOptionChannel ApplicationCommandOptionType = 7
|
|
ApplicationCommandOptionRole ApplicationCommandOptionType = 8
|
|
ApplicationCommandOptionMentionable ApplicationCommandOptionType = 9
|
|
ApplicationCommandOptionNumber ApplicationCommandOptionType = 10
|
|
ApplicationCommandOptionAttachment ApplicationCommandOptionType = 11
|
|
)
|
|
|
|
func (t ApplicationCommandOptionType) String() string {
|
|
switch t {
|
|
case ApplicationCommandOptionSubCommand:
|
|
return "SubCommand"
|
|
case ApplicationCommandOptionSubCommandGroup:
|
|
return "SubCommandGroup"
|
|
case ApplicationCommandOptionString:
|
|
return "String"
|
|
case ApplicationCommandOptionInteger:
|
|
return "Integer"
|
|
case ApplicationCommandOptionBoolean:
|
|
return "Boolean"
|
|
case ApplicationCommandOptionUser:
|
|
return "User"
|
|
case ApplicationCommandOptionChannel:
|
|
return "Channel"
|
|
case ApplicationCommandOptionRole:
|
|
return "Role"
|
|
case ApplicationCommandOptionMentionable:
|
|
return "Mentionable"
|
|
case ApplicationCommandOptionNumber:
|
|
return "Number"
|
|
case ApplicationCommandOptionAttachment:
|
|
return "Attachment"
|
|
}
|
|
return fmt.Sprintf("ApplicationCommandOptionType(%d)", t)
|
|
}
|
|
|
|
// ApplicationCommandOption represents an option/subcommand/subcommands group.
|
|
type ApplicationCommandOption struct {
|
|
Type ApplicationCommandOptionType `json:"type"`
|
|
Name string `json:"name"`
|
|
NameLocalizations map[Locale]string `json:"name_localizations,omitempty"`
|
|
Description string `json:"description,omitempty"`
|
|
DescriptionLocalizations map[Locale]string `json:"description_localizations,omitempty"`
|
|
// NOTE: This feature was on the API, but at some point developers decided to remove it.
|
|
// So I commented it, until it will be officially on the docs.
|
|
// Default bool `json:"default"`
|
|
|
|
ChannelTypes []ChannelType `json:"channel_types"`
|
|
Required bool `json:"required"`
|
|
Options []*ApplicationCommandOption `json:"options"`
|
|
|
|
// NOTE: mutually exclusive with Choices.
|
|
Autocomplete bool `json:"autocomplete"`
|
|
Choices []*ApplicationCommandOptionChoice `json:"choices"`
|
|
// Minimal value of number/integer option.
|
|
MinValue *float64 `json:"min_value,omitempty"`
|
|
// Maximum value of number/integer option.
|
|
MaxValue float64 `json:"max_value,omitempty"`
|
|
}
|
|
|
|
// ApplicationCommandOptionChoice represents a slash command option choice.
|
|
type ApplicationCommandOptionChoice struct {
|
|
Name string `json:"name"`
|
|
NameLocalizations map[Locale]string `json:"name_localizations,omitempty"`
|
|
Value interface{} `json:"value"`
|
|
}
|
|
|
|
// ApplicationCommandPermissions represents a single user or role permission for a command.
|
|
type ApplicationCommandPermissions struct {
|
|
ID string `json:"id"`
|
|
Type ApplicationCommandPermissionType `json:"type"`
|
|
Permission bool `json:"permission"`
|
|
}
|
|
|
|
// ApplicationCommandPermissionsList represents a list of ApplicationCommandPermissions, needed for serializing to JSON.
|
|
type ApplicationCommandPermissionsList struct {
|
|
Permissions []*ApplicationCommandPermissions `json:"permissions"`
|
|
}
|
|
|
|
// GuildApplicationCommandPermissions represents all permissions for a single guild command.
|
|
type GuildApplicationCommandPermissions struct {
|
|
ID string `json:"id"`
|
|
ApplicationID string `json:"application_id"`
|
|
GuildID string `json:"guild_id"`
|
|
Permissions []*ApplicationCommandPermissions `json:"permissions"`
|
|
}
|
|
|
|
// ApplicationCommandPermissionType indicates whether a permission is user or role based.
|
|
type ApplicationCommandPermissionType uint8
|
|
|
|
// Application command permission types.
|
|
const (
|
|
ApplicationCommandPermissionTypeRole ApplicationCommandPermissionType = 1
|
|
ApplicationCommandPermissionTypeUser ApplicationCommandPermissionType = 2
|
|
)
|
|
|
|
// InteractionType indicates the type of an interaction event.
|
|
type InteractionType uint8
|
|
|
|
// Interaction types
|
|
const (
|
|
InteractionPing InteractionType = 1
|
|
InteractionApplicationCommand InteractionType = 2
|
|
InteractionMessageComponent InteractionType = 3
|
|
InteractionApplicationCommandAutocomplete InteractionType = 4
|
|
InteractionModalSubmit InteractionType = 5
|
|
)
|
|
|
|
func (t InteractionType) String() string {
|
|
switch t {
|
|
case InteractionPing:
|
|
return "Ping"
|
|
case InteractionApplicationCommand:
|
|
return "ApplicationCommand"
|
|
case InteractionMessageComponent:
|
|
return "MessageComponent"
|
|
case InteractionModalSubmit:
|
|
return "ModalSubmit"
|
|
}
|
|
return fmt.Sprintf("InteractionType(%d)", t)
|
|
}
|
|
|
|
// Interaction represents data of an interaction.
|
|
type Interaction struct {
|
|
ID string `json:"id"`
|
|
AppID string `json:"application_id"`
|
|
Type InteractionType `json:"type"`
|
|
Data InteractionData `json:"data"`
|
|
GuildID string `json:"guild_id"`
|
|
ChannelID string `json:"channel_id"`
|
|
|
|
// The message on which interaction was used.
|
|
// NOTE: this field is only filled when a button click triggered the interaction. Otherwise it will be nil.
|
|
Message *Message `json:"message"`
|
|
|
|
// The member who invoked this interaction.
|
|
// NOTE: this field is only filled when the slash command was invoked in a guild;
|
|
// if it was invoked in a DM, the `User` field will be filled instead.
|
|
// Make sure to check for `nil` before using this field.
|
|
Member *Member `json:"member"`
|
|
// The user who invoked this interaction.
|
|
// NOTE: this field is only filled when the slash command was invoked in a DM;
|
|
// if it was invoked in a guild, the `Member` field will be filled instead.
|
|
// Make sure to check for `nil` before using this field.
|
|
User *User `json:"user"`
|
|
|
|
// The user's discord client locale.
|
|
Locale Locale `json:"locale"`
|
|
// The guild's locale. This defaults to EnglishUS
|
|
// NOTE: this field is only filled when the interaction was invoked in a guild.
|
|
GuildLocale *Locale `json:"guild_locale"`
|
|
|
|
Token string `json:"token"`
|
|
Version int `json:"version"`
|
|
}
|
|
|
|
type interaction Interaction
|
|
|
|
type rawInteraction struct {
|
|
interaction
|
|
Data json.RawMessage `json:"data"`
|
|
}
|
|
|
|
// UnmarshalJSON is a method for unmarshalling JSON object to Interaction.
|
|
func (i *Interaction) UnmarshalJSON(raw []byte) error {
|
|
var tmp rawInteraction
|
|
err := json.Unmarshal(raw, &tmp)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
*i = Interaction(tmp.interaction)
|
|
|
|
switch tmp.Type {
|
|
case InteractionApplicationCommand, InteractionApplicationCommandAutocomplete:
|
|
v := ApplicationCommandInteractionData{}
|
|
err = json.Unmarshal(tmp.Data, &v)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
i.Data = v
|
|
case InteractionMessageComponent:
|
|
v := MessageComponentInteractionData{}
|
|
err = json.Unmarshal(tmp.Data, &v)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
i.Data = v
|
|
case InteractionModalSubmit:
|
|
v := ModalSubmitInteractionData{}
|
|
err = json.Unmarshal(tmp.Data, &v)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
i.Data = v
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// MessageComponentData is helper function to assert the inner InteractionData to MessageComponentInteractionData.
|
|
// Make sure to check that the Type of the interaction is InteractionMessageComponent before calling.
|
|
func (i Interaction) MessageComponentData() (data MessageComponentInteractionData) {
|
|
if i.Type != InteractionMessageComponent {
|
|
panic("MessageComponentData called on interaction of type " + i.Type.String())
|
|
}
|
|
return i.Data.(MessageComponentInteractionData)
|
|
}
|
|
|
|
// ApplicationCommandData is helper function to assert the inner InteractionData to ApplicationCommandInteractionData.
|
|
// Make sure to check that the Type of the interaction is InteractionApplicationCommand before calling.
|
|
func (i Interaction) ApplicationCommandData() (data ApplicationCommandInteractionData) {
|
|
if i.Type != InteractionApplicationCommand && i.Type != InteractionApplicationCommandAutocomplete {
|
|
panic("ApplicationCommandData called on interaction of type " + i.Type.String())
|
|
}
|
|
return i.Data.(ApplicationCommandInteractionData)
|
|
}
|
|
|
|
// ModalSubmitData is helper function to assert the inner InteractionData to ModalSubmitInteractionData.
|
|
// Make sure to check that the Type of the interaction is InteractionModalSubmit before calling.
|
|
func (i Interaction) ModalSubmitData() (data ModalSubmitInteractionData) {
|
|
if i.Type != InteractionModalSubmit {
|
|
panic("ModalSubmitData called on interaction of type " + i.Type.String())
|
|
}
|
|
return i.Data.(ModalSubmitInteractionData)
|
|
}
|
|
|
|
// InteractionData is a common interface for all types of interaction data.
|
|
type InteractionData interface {
|
|
Type() InteractionType
|
|
}
|
|
|
|
// ApplicationCommandInteractionData contains the data of application command interaction.
|
|
type ApplicationCommandInteractionData struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Resolved *ApplicationCommandInteractionDataResolved `json:"resolved"`
|
|
|
|
// Slash command options
|
|
Options []*ApplicationCommandInteractionDataOption `json:"options"`
|
|
// Target (user/message) id on which context menu command was called.
|
|
// The details are stored in Resolved according to command type.
|
|
TargetID string `json:"target_id"`
|
|
}
|
|
|
|
// ApplicationCommandInteractionDataResolved contains resolved data of command execution.
|
|
// Partial Member objects are missing user, deaf and mute fields.
|
|
// Partial Channel objects only have id, name, type and permissions fields.
|
|
type ApplicationCommandInteractionDataResolved struct {
|
|
Users map[string]*User `json:"users"`
|
|
Members map[string]*Member `json:"members"`
|
|
Roles map[string]*Role `json:"roles"`
|
|
Channels map[string]*Channel `json:"channels"`
|
|
Messages map[string]*Message `json:"messages"`
|
|
Attachments map[string]*MessageAttachment `json:"attachments"`
|
|
}
|
|
|
|
// Type returns the type of interaction data.
|
|
func (ApplicationCommandInteractionData) Type() InteractionType {
|
|
return InteractionApplicationCommand
|
|
}
|
|
|
|
// MessageComponentInteractionData contains the data of message component interaction.
|
|
type MessageComponentInteractionData struct {
|
|
CustomID string `json:"custom_id"`
|
|
ComponentType ComponentType `json:"component_type"`
|
|
|
|
// NOTE: Only filled when ComponentType is SelectMenuComponent (3). Otherwise is nil.
|
|
Values []string `json:"values"`
|
|
}
|
|
|
|
// Type returns the type of interaction data.
|
|
func (MessageComponentInteractionData) Type() InteractionType {
|
|
return InteractionMessageComponent
|
|
}
|
|
|
|
// ModalSubmitInteractionData contains the data of modal submit interaction.
|
|
type ModalSubmitInteractionData struct {
|
|
CustomID string `json:"custom_id"`
|
|
Components []MessageComponent `json:"-"`
|
|
}
|
|
|
|
// Type returns the type of interaction data.
|
|
func (ModalSubmitInteractionData) Type() InteractionType {
|
|
return InteractionModalSubmit
|
|
}
|
|
|
|
// UnmarshalJSON is a helper function to correctly unmarshal Components.
|
|
func (d *ModalSubmitInteractionData) UnmarshalJSON(data []byte) error {
|
|
type modalSubmitInteractionData ModalSubmitInteractionData
|
|
var v struct {
|
|
modalSubmitInteractionData
|
|
RawComponents []unmarshalableMessageComponent `json:"components"`
|
|
}
|
|
err := json.Unmarshal(data, &v)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*d = ModalSubmitInteractionData(v.modalSubmitInteractionData)
|
|
d.Components = make([]MessageComponent, len(v.RawComponents))
|
|
for i, v := range v.RawComponents {
|
|
d.Components[i] = v.MessageComponent
|
|
}
|
|
return err
|
|
}
|
|
|
|
// ApplicationCommandInteractionDataOption represents an option of a slash command.
|
|
type ApplicationCommandInteractionDataOption struct {
|
|
Name string `json:"name"`
|
|
Type ApplicationCommandOptionType `json:"type"`
|
|
// NOTE: Contains the value specified by Type.
|
|
Value interface{} `json:"value,omitempty"`
|
|
Options []*ApplicationCommandInteractionDataOption `json:"options,omitempty"`
|
|
|
|
// NOTE: autocomplete interaction only.
|
|
Focused bool `json:"focused,omitempty"`
|
|
}
|
|
|
|
// IntValue is a utility function for casting option value to integer
|
|
func (o ApplicationCommandInteractionDataOption) IntValue() int64 {
|
|
if o.Type != ApplicationCommandOptionInteger {
|
|
panic("IntValue called on data option of type " + o.Type.String())
|
|
}
|
|
return int64(o.Value.(float64))
|
|
}
|
|
|
|
// UintValue is a utility function for casting option value to unsigned integer
|
|
func (o ApplicationCommandInteractionDataOption) UintValue() uint64 {
|
|
if o.Type != ApplicationCommandOptionInteger {
|
|
panic("UintValue called on data option of type " + o.Type.String())
|
|
}
|
|
return uint64(o.Value.(float64))
|
|
}
|
|
|
|
// FloatValue is a utility function for casting option value to float
|
|
func (o ApplicationCommandInteractionDataOption) FloatValue() float64 {
|
|
if o.Type != ApplicationCommandOptionNumber {
|
|
panic("FloatValue called on data option of type " + o.Type.String())
|
|
}
|
|
return o.Value.(float64)
|
|
}
|
|
|
|
// StringValue is a utility function for casting option value to string
|
|
func (o ApplicationCommandInteractionDataOption) StringValue() string {
|
|
if o.Type != ApplicationCommandOptionString {
|
|
panic("StringValue called on data option of type " + o.Type.String())
|
|
}
|
|
return o.Value.(string)
|
|
}
|
|
|
|
// BoolValue is a utility function for casting option value to bool
|
|
func (o ApplicationCommandInteractionDataOption) BoolValue() bool {
|
|
if o.Type != ApplicationCommandOptionBoolean {
|
|
panic("BoolValue called on data option of type " + o.Type.String())
|
|
}
|
|
return o.Value.(bool)
|
|
}
|
|
|
|
// ChannelValue is a utility function for casting option value to channel object.
|
|
// s : Session object, if not nil, function additionally fetches all channel's data
|
|
func (o ApplicationCommandInteractionDataOption) ChannelValue(s *Session) *Channel {
|
|
if o.Type != ApplicationCommandOptionChannel {
|
|
panic("ChannelValue called on data option of type " + o.Type.String())
|
|
}
|
|
chanID := o.Value.(string)
|
|
|
|
if s == nil {
|
|
return &Channel{ID: chanID}
|
|
}
|
|
|
|
ch, err := s.State.Channel(chanID)
|
|
if err != nil {
|
|
ch, err = s.Channel(chanID)
|
|
if err != nil {
|
|
return &Channel{ID: chanID}
|
|
}
|
|
}
|
|
|
|
return ch
|
|
}
|
|
|
|
// RoleValue is a utility function for casting option value to role object.
|
|
// s : Session object, if not nil, function additionally fetches all role's data
|
|
func (o ApplicationCommandInteractionDataOption) RoleValue(s *Session, gID string) *Role {
|
|
if o.Type != ApplicationCommandOptionRole && o.Type != ApplicationCommandOptionMentionable {
|
|
panic("RoleValue called on data option of type " + o.Type.String())
|
|
}
|
|
roleID := o.Value.(string)
|
|
|
|
if s == nil || gID == "" {
|
|
return &Role{ID: roleID}
|
|
}
|
|
|
|
r, err := s.State.Role(roleID, gID)
|
|
if err != nil {
|
|
roles, err := s.GuildRoles(gID)
|
|
if err == nil {
|
|
for _, r = range roles {
|
|
if r.ID == roleID {
|
|
return r
|
|
}
|
|
}
|
|
}
|
|
return &Role{ID: roleID}
|
|
}
|
|
|
|
return r
|
|
}
|
|
|
|
// UserValue is a utility function for casting option value to user object.
|
|
// s : Session object, if not nil, function additionally fetches all user's data
|
|
func (o ApplicationCommandInteractionDataOption) UserValue(s *Session) *User {
|
|
if o.Type != ApplicationCommandOptionUser && o.Type != ApplicationCommandOptionMentionable {
|
|
panic("UserValue called on data option of type " + o.Type.String())
|
|
}
|
|
userID := o.Value.(string)
|
|
|
|
if s == nil {
|
|
return &User{ID: userID}
|
|
}
|
|
|
|
u, err := s.User(userID)
|
|
if err != nil {
|
|
return &User{ID: userID}
|
|
}
|
|
|
|
return u
|
|
}
|
|
|
|
// InteractionResponseType is type of interaction response.
|
|
type InteractionResponseType uint8
|
|
|
|
// Interaction response types.
|
|
const (
|
|
// InteractionResponsePong is for ACK ping event.
|
|
InteractionResponsePong InteractionResponseType = 1
|
|
// InteractionResponseChannelMessageWithSource is for responding with a message, showing the user's input.
|
|
InteractionResponseChannelMessageWithSource InteractionResponseType = 4
|
|
// InteractionResponseDeferredChannelMessageWithSource acknowledges that the event was received, and that a follow-up will come later.
|
|
InteractionResponseDeferredChannelMessageWithSource InteractionResponseType = 5
|
|
// InteractionResponseDeferredMessageUpdate acknowledges that the message component interaction event was received, and message will be updated later.
|
|
InteractionResponseDeferredMessageUpdate InteractionResponseType = 6
|
|
// InteractionResponseUpdateMessage is for updating the message to which message component was attached.
|
|
InteractionResponseUpdateMessage InteractionResponseType = 7
|
|
// InteractionApplicationCommandAutocompleteResult shows autocompletion results. Autocomplete interaction only.
|
|
InteractionApplicationCommandAutocompleteResult InteractionResponseType = 8
|
|
// InteractionResponseModal is for responding to an interaction with a modal window.
|
|
InteractionResponseModal InteractionResponseType = 9
|
|
)
|
|
|
|
// InteractionResponse represents a response for an interaction event.
|
|
type InteractionResponse struct {
|
|
Type InteractionResponseType `json:"type,omitempty"`
|
|
Data *InteractionResponseData `json:"data,omitempty"`
|
|
}
|
|
|
|
// InteractionResponseData is response data for an interaction.
|
|
type InteractionResponseData struct {
|
|
TTS bool `json:"tts"`
|
|
Content string `json:"content"`
|
|
Components []MessageComponent `json:"components"`
|
|
Embeds []*MessageEmbed `json:"embeds"`
|
|
AllowedMentions *MessageAllowedMentions `json:"allowed_mentions,omitempty"`
|
|
Flags uint64 `json:"flags,omitempty"`
|
|
Files []*File `json:"-"`
|
|
|
|
// NOTE: autocomplete interaction only.
|
|
Choices []*ApplicationCommandOptionChoice `json:"choices,omitempty"`
|
|
|
|
// NOTE: modal interaction only.
|
|
|
|
CustomID string `json:"custom_id,omitempty"`
|
|
Title string `json:"title,omitempty"`
|
|
}
|
|
|
|
// VerifyInteraction implements message verification of the discord interactions api
|
|
// signing algorithm, as documented here:
|
|
// https://discord.com/developers/docs/interactions/receiving-and-responding#security-and-authorization
|
|
func VerifyInteraction(r *http.Request, key ed25519.PublicKey) bool {
|
|
var msg bytes.Buffer
|
|
|
|
signature := r.Header.Get("X-Signature-Ed25519")
|
|
if signature == "" {
|
|
return false
|
|
}
|
|
|
|
sig, err := hex.DecodeString(signature)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
if len(sig) != ed25519.SignatureSize {
|
|
return false
|
|
}
|
|
|
|
timestamp := r.Header.Get("X-Signature-Timestamp")
|
|
if timestamp == "" {
|
|
return false
|
|
}
|
|
|
|
msg.WriteString(timestamp)
|
|
|
|
defer r.Body.Close()
|
|
var body bytes.Buffer
|
|
|
|
// at the end of the function, copy the original body back into the request
|
|
defer func() {
|
|
r.Body = ioutil.NopCloser(&body)
|
|
}()
|
|
|
|
// copy body into buffers
|
|
_, err = io.Copy(&msg, io.TeeReader(r.Body, &body))
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
return ed25519.Verify(key, msg.Bytes(), sig)
|
|
}
|