bot

package
v0.0.0-...-644a4e2 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Mar 14, 2024 License: GPL-3.0 Imports: 14 Imported by: 0

Documentation

Index

Constants

View Source 
const (

	// Chat commands
	Help                              = "help"
	Stats                             = "stats"
	Settings                          = "settings"
	GetUserInfoFromChatCommandContext = "query"
	AddCustomSlashCommand             = "addcommand"
	RemoveCustomSlashCommand          = "deletecommand"

	// User commands
	GetUserInfoFromUserContext      = "Check info"
	DocumentBehaviorFromUserContext = "Save evidence"

	// Message commands
	GetUserInfoFromMessageContext      = "Get user info"
	DocumentBehaviorFromMessageContext = "Save as evidence"

	// Premade Option IDs (semi-reusable)
	// TODO: actions that delete messages, ban users etc
	UserOption    = "user"
	ChannelOption = "channel"
	MessageOption = "message"
	ReasonOption  = "reason"

	// Buttons
	// Settings (the names affect the column names in the DB)
	EvidenceChannelSettingID = "Evidence channel"
	ModeratorRoleSettingID   = "Moderator role"

	// Moderation buttons
	IncreaseUserReputation      = "⬆️ Reputation"
	DecreaseUserReputation      = "⬇️ Reputation"
	ShowEvidenceCollectionModal = "Add notes"
	SubmitReport                = "Submit report"

	// Modals
	SaveEvidenceNotes        = "Save evidence notes"
	SaveCustomSlashCommand   = "Save custom slash command"
	DeleteCustomSlashCommand = "Remove custom slash command"

	// Modal options
	EvidenceNotes                 = "Evidence notes"
	CustomSlashName               = "Name for this custom slash command"
	CustomSlashCommandDescription = "Description for this custom slash command"
	CustomSlashCommandContent     = "Message to paste if this command is used"
	CustomCommandIdentifier       = "Custom command: "

	// Colors
	FrenchGray = 13424349
	Purple     = 7283691
	DarkRed    = 9109504
	Green      = 4306266

	// Constants
	MaxCommandContentLength     = 32   // https://discord.com/developers/docs/interactions/application-commands#create-global-application-command
	MaxMessageContentLength     = 2000 // https://discord.com/developers/docs/resources/channel#create-message
	MaxDescriptionContentLength = 100  // https://discord.com/developers/docs/interactions/application-commands#application-command-object

	// Text fragments
	CurrentReputation      = "Current reputation"
	PreviousReputation     = "Previous reputation"
	ModerationSuccessful   = "Moderation action saved"
	ModerationUnSuccessful = "There was a problem saving moderation action"
	OriginalMessageContent = "Content of original message"
	Attachments            = "Attachments"
	Notes                  = "Notes"
	// If images have this in front of their name, they're spoilered
	Spoiler = "SPOILER_"

	// Errors
	SettingFailedResponseMessage = "Error changing setting"
	UnexpectedRowsAffected       = "unexpected number of rows affected getting user reputation: %v"

	// URLs
	MessageURLTemplate = "https://discord.com/channels/%s/%s/%s"

	// BotHelpText Shown to the user when `/help` is called
	BotHelpText = `**Usage**
	In the Discord app, right click (Desktop) or long press (mobile) a message or user to see the available options.

Configure the bot (Highly recommended to do so only people in a specific role can use the moderation commands):

` + "`/settings`" + `

Look up a user (more coming soon):

` + "`/query`" + `

Add a custom command (simply posts your desired block of text, Markdown formatting enabled)

` + "`/addcommand`" + `

Remove a custom command

` + "`/deletecommand`" + `

Get this help message:

` + "`/help`"
)

Variables

View Source 
var (
	ChatCommands = []*discordgo.ApplicationCommand{
		{
			Name:        Help,
			Description: "How to use this bot",
		},
		{
			Name:        GetUserInfoFromChatCommandContext,
			Description: "Check a user's reputation information",
			Type:        discordgo.ChatApplicationCommand,
			Options: []*discordgo.ApplicationCommandOption{
				{
					Name:        UserOption,
					Description: "User to look up",
					Type:        discordgo.ApplicationCommandOptionString,
					Required:    true,
				},
			},
		},
		{
			Name:        Settings,
			Description: "Change settings",
		},
		{
			Name:        AddCustomSlashCommand,
			Description: "Create custom slash command",
		},
		{
			Name:        RemoveCustomSlashCommand,
			Description: "Remove custom slash command",
		},
	}
	UserCommands = []*discordgo.ApplicationCommand{
		{
			Name: GetUserInfoFromUserContext,
			Type: discordgo.UserApplicationCommand,
		},
		{
			Name: DocumentBehaviorFromUserContext,
			Type: discordgo.UserApplicationCommand,
		},
	}
	MessageCommands = []*discordgo.ApplicationCommand{
		{
			Name: GetUserInfoFromMessageContext,
			Type: discordgo.MessageApplicationCommand,
		},
		{
			Name: DocumentBehaviorFromMessageContext,
			Type: discordgo.MessageApplicationCommand,
		},
	}

	RegisteredCommands = []*discordgo.ApplicationCommand{}

	// ConfiguredCommands This object is used to register chat commands, so if it's
	// not in here, it won't get registered properly. This is updated
	// during runtime.
	ConfiguredCommands = append(append(ChatCommands, UserCommands...), MessageCommands...)
)
View Source 
var (
	// Enabled takes a boolean and returns "enabled" or "disabled" as a string
	Enabled = map[bool]string{
			// contains filtered or unexported fields
	}
	// ActiveButton Takes a boolean and returns a colorized button if true
	ActiveButton = map[bool]discordgo.ButtonStyle{
					// contains filtered or unexported fields
	}
)

Functions

This section is empty.

Types

type CustomCommand 

type CustomCommand struct {
	ID            string
	GuildConfigID string `gorm:"type:varchar(191)"`
	Name          string
	Description   string
	Content       string
}

CustomCommand Custom commands registered with a specific server ID is registered with the Discord API GuildConfigID is the server ID where the command is registered

type GuildConfig 

type GuildConfig struct {
	ID                       string          `pretty:"Server ID" gorm:"type:varchar(191)"`
	Name                     string          `pretty:"Server Name" gorm:"default:default"`
	Active                   *bool           `pretty:"Bot is active in the server" gorm:"default:true"`
	EvidenceChannelSettingID string          `pretty:"Channel to document evidence in"`
	ModeratorRoleSettingID   string          `pretty:"Role for moderators"`
	CustomCommands           []CustomCommand `pretty:"Custom commands"`
}

GuildConfig Servers are called Guilds in the Discord API

type ModeratedUser 

type ModeratedUser struct {
	ID               string
	GuildId          string
	GuildName        string
	UserID           string
	UserName         string
	ModerationEvents []ModerationEvent
	Reputation       *int64 `gorm:"default:1"`
}

A ModeratedUser represents a specific user/server combination that serves to record events and a "Reputation" which is only visible to people who are in the moderator's configurable role.

type ModerationEvent 

type ModerationEvent struct {
	ID                 uint `gorm:"primaryKey"`
	ModeratedUserID    string
	GuildId            string
	GuildName          string
	UserID             string
	UserName           string
	Notes              string
	PreviousReputation *int64
	CurrentReputation  *int64
	ModeratorID        string
	ModeratorName      string
	ReportURL          string
}

ModerationEvent This is the representation of a moderation action

type ModeratorBot 

type ModeratorBot struct {
	DB     *gorm.DB
	DG     *discordgo.Session
	Config ModeratorBotConfig
}

ModeratorBot Handlers This is the main type passed around throughout the code It has many functions for overall bot management

func (*ModeratorBot) BotReadyHandler 

func (bot *ModeratorBot) BotReadyHandler(s *discordgo.Session, r *discordgo.Ready)

BotReadyHandler is called when the bot is considered ready to use the Discord session

func (*ModeratorBot) ChangeUserReputation 

func (bot *ModeratorBot) ChangeUserReputation(i *discordgo.InteractionCreate, difference int) (err error)

ChangeUserReputation Updates a user reputation, given the source interaction and value to add

func (*ModeratorBot) CreateCustomSlashCommandFromChatCommandContext 

func (bot *ModeratorBot) CreateCustomSlashCommandFromChatCommandContext(i *discordgo.InteractionCreate)

CreateCustomSlashCommandFromChatCommandContext Creates a new slash command

func (*ModeratorBot) DeleteCustomSlashCommandFromButtonContext 

func (bot *ModeratorBot) DeleteCustomSlashCommandFromButtonContext(i *discordgo.InteractionCreate, commandID string)

DeleteCustomSlashCommandFromButtonContext Deletes a custom command

func (*ModeratorBot) DeleteCustomSlashCommandFromChatCommandContext 

func (bot *ModeratorBot) DeleteCustomSlashCommandFromChatCommandContext(i *discordgo.InteractionCreate)

func (*ModeratorBot) DocumentBehaviorFromButtonContext 

func (bot *ModeratorBot) DocumentBehaviorFromButtonContext(i *discordgo.InteractionCreate)

DocumentBehaviorFromButtonContext Called from the App menu, this displays an embed for the moderator to choose to change the reputation of the posting user and (PLANNED) produces output in the evidence channel with information about the message, user and moderation actions taken

func (*ModeratorBot) DocumentBehaviorFromMessageContext 

func (bot *ModeratorBot) DocumentBehaviorFromMessageContext(i *discordgo.InteractionCreate)

DocumentBehaviorFromMessageContext Called from the App menu, this displays an embed for the moderator to choose to change the reputation of the posting user and (PLANNED) produces output in the evidence channel with information about the message, user and moderation actions taken

func (*ModeratorBot) DocumentBehaviorFromUserContext 

func (bot *ModeratorBot) DocumentBehaviorFromUserContext(i *discordgo.InteractionCreate)

DocumentBehaviorFromUserContext Very similar to GenerateEvidenceReportFromMessageContext, but this is called when the target is a user and not a message, therefore this will be implicitly without any message reference

func (*ModeratorBot) GenerateEvidenceReportFromMessage 

func (bot *ModeratorBot) GenerateEvidenceReportFromMessage(i *discordgo.InteractionCreate, message *discordgo.Message) (resp *discordgo.InteractionResponse)

Returns a discordgo.InteractionResponse with an evidence report based on a message provided

func (*ModeratorBot) GetCustomCommandHandlers 

func (bot *ModeratorBot) GetCustomCommandHandlers() (cmds map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate))

GetCustomCommandHandlers returns a map[string]func of command handlers for every ServerConfig

func (*ModeratorBot) GetHelpFromChatCommandContext 

func (bot *ModeratorBot) GetHelpFromChatCommandContext(i *discordgo.InteractionCreate)

GetHelpFromChatCommandContext This produces the help text seen from the chat command `/help`

func (*ModeratorBot) GetModeratedUser 

func (bot *ModeratorBot) GetModeratedUser(GuildId string, userID string) (moderatedUser ModeratedUser)

GetModeratedUser Returns a ModeratedUser record from the DB using server and user ID (a user can be in multiple servers)

func (*ModeratorBot) GetUserInfoFromChatCommandContext 

func (bot *ModeratorBot) GetUserInfoFromChatCommandContext(i *discordgo.InteractionCreate)

GetUserInfoFromChatCommandContext Gets user info from the `/query` command

func (*ModeratorBot) GetUserInfoFromMessageContext 

func (bot *ModeratorBot) GetUserInfoFromMessageContext(i *discordgo.InteractionCreate)

Produces user info such as reputation and (PLANNED) stats

func (*ModeratorBot) GetUserInfoFromUserContext 

func (bot *ModeratorBot) GetUserInfoFromUserContext(i *discordgo.InteractionCreate)

GetUserInfoFromUserContext Produces user info such as reputation and (PLANNED) stats

func (*ModeratorBot) GuildCreateHandler 

func (bot *ModeratorBot) GuildCreateHandler(s *discordgo.Session, gc *discordgo.GuildCreate)

GuildCreateHandler is called whenever the bot joins a new guild.

func (*ModeratorBot) GuildDeleteHandler 

func (bot *ModeratorBot) GuildDeleteHandler(s *discordgo.Session, gd *discordgo.GuildDelete)

GuildDeleteHandler is called whenever the bot leaves a guild.

func (*ModeratorBot) InteractionHandler 

func (bot *ModeratorBot) InteractionHandler(s *discordgo.Session, i *discordgo.InteractionCreate)

InteractionHandler configures all interactive commands

func (*ModeratorBot) RegisterCustomCommandHandler 

func (bot *ModeratorBot) RegisterCustomCommandHandler(cmds []CustomCommand)

RegisterCustomCommandHandler registers the provided commands

func (*ModeratorBot) RespondToSettingsChoice 

func (bot *ModeratorBot) RespondToSettingsChoice(i *discordgo.InteractionCreate,
	setting string, value string)

RespondToSettingsChoice Updates a server setting according to the column name (setting) and the value

func (*ModeratorBot) SaveCustomSlashCommand 

func (bot *ModeratorBot) SaveCustomSlashCommand(i *discordgo.InteractionCreate)

SaveCustomSlashCommand Takes user-submitted command and adds it to the server config and registers it in the guild

func (*ModeratorBot) SaveEvidenceNotes 

func (bot *ModeratorBot) SaveEvidenceNotes(i *discordgo.InteractionCreate)

SaveEvidenceNotes Takes user-submitted notes and adds it to an in-progress report

func (*ModeratorBot) SetSettingsFromChatCommandContext 

func (bot *ModeratorBot) SetSettingsFromChatCommandContext(i *discordgo.InteractionCreate)

SetSettingsFromChatCommandContext Sets setting choices from the `/settings` command

func (*ModeratorBot) SettingsIntegrationResponse 

func (bot *ModeratorBot) SettingsIntegrationResponse(cfg GuildConfig) *discordgo.InteractionResponseData

SettingsIntegrationResponse returns server settings in a *discordgo.InteractionResponseData

func (*ModeratorBot) ShowEvidenceCollectionModal 

func (bot *ModeratorBot) ShowEvidenceCollectionModal(i *discordgo.InteractionCreate)

ShowEvidenceCollectionModal shows the user a modal to fill in information about selected evidence

func (*ModeratorBot) StartHealthAPI 

func (b *ModeratorBot) StartHealthAPI()

StartHealthAPI Starts a simple health API that pings the DB

func (*ModeratorBot) SubmitReport 

func (bot *ModeratorBot) SubmitReport(i *discordgo.InteractionCreate)

SubmitReport Submits evidence to the configured channel (including notes)

func (*ModeratorBot) UpdateCommands 

func (bot *ModeratorBot) UpdateCommands() (err error)

UpdateCommands iterates through all configured commands and ensures they are registered, updated or removed

func (*ModeratorBot) UpdateInactiveRegistrations 

func (bot *ModeratorBot) UpdateInactiveRegistrations(activeGuilds []*discordgo.Guild)

UpdateInactiveRegistrations goes through every server registration and updates the DB as to whether or not it's active

func (*ModeratorBot) UseCustomSlashCommandFromChatCommandContext 

func (bot *ModeratorBot) UseCustomSlashCommandFromChatCommandContext(i *discordgo.InteractionCreate, content string)

UseCustomSlashCommandFromChatCommandContext Creates a new slash command

type ModeratorBotConfig 

type ModeratorBotConfig struct {
	DBType     string `env:"DB_TYPE"`
	DBHost     string `env:"DB_HOST"`
	DBPort     string `env:"DB_PORT"`
	DBName     string `env:"DB_NAME"`
	DBPassword string `env:"DB_PASSWORD"`
	DBUser     string `env:"DB_USER"`
	LogLevel   string `env:"LOG_LEVEL"`
	Token      string `env:"TOKEN"`
}

ModeratorBotConfig is attached to ModeratorBot so config settings can be accessed easily

type RSSFeeds 

type RSSFeeds struct {
	ID        uint `gorm:"primaryKey"`
	GuildId   string
	GuildName string
	URL       string
}

RSSFeeds This represents the stored feed URLs to monitor

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.