Geeksbot/internal/exts/utils/utils.go

package utils

import (
	"fmt"
	"sort"
	"strconv"
	"strings"
	"time"

	"github.com/bwmarrin/discordgo"
	"github.com/dustinpianalto/disgoman"
	"github.com/dustinpianalto/geeksbot"
	"github.com/dustinpianalto/geeksbot/internal/discord_utils"
	"github.com/dustinpianalto/geeksbot/pkg/services"
)

var PingCommand = &disgoman.Command{
	Name:                "ping",
	Aliases:             []string{" "},
	Description:         "Check the bot's ping",
	OwnerOnly:           false,
	Hidden:              false,
	RequiredPermissions: 0,
	Invoke:              pingCommandFunc,
}

func pingCommandFunc(ctx disgoman.Context, _ []string) {
	timeBefore := time.Now()
	msg, _ := ctx.Send("Pong!")
	took := time.Now().Sub(timeBefore)
	_, err := ctx.Session.ChannelMessageEdit(ctx.Message.ChannelID, msg.ID, fmt.Sprintf("Pong!\nPing Took **%s**", took.String()))
	if err != nil {
		ctx.CommandManager.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Ping Failed",
			Error:   err,
		}
	}
}

var InviteCommand = &disgoman.Command{
	Name:                "invite",
	Aliases:             nil,
	Description:         "Get the invite link for this bot or others",
	OwnerOnly:           false,
	Hidden:              false,
	RequiredPermissions: 0,
	Invoke:              inviteCommandFunc,
}

func inviteCommandFunc(ctx disgoman.Context, args []string) {
	var ids []string
	if len(args) == 0 && len(ctx.Message.Mentions) == 0 {
		ids = []string{ctx.Session.State.User.ID}
	} else {
		if len(ctx.Message.Mentions) > 0 {
			for _, user := range ctx.Message.Mentions {
				member, err := ctx.Session.GuildMember(ctx.Guild.ID, user.ID)
				if err != nil {
					ctx.CommandManager.ErrorChannel <- disgoman.CommandError{
						Context: ctx,
						Message: "Could not find member " + user.Username,
						Error:   err,
					}
					continue
				}
				ids = append(ids, member.User.ID)
			}
		}
		if len(args) > 0 {
			for _, id := range args {
				member, err := ctx.Session.GuildMember(ctx.Guild.ID, id)
				if err != nil {
					ctx.CommandManager.ErrorChannel <- disgoman.CommandError{
						Context: ctx,
						Message: "Could not find member " + id,
						Error:   err,
					}
					continue
				}
				ids = append(ids, member.User.ID)
			}
		}
	}
	if len(ids) == 0 {
		return
	}
	for _, id := range ids {
		url := fmt.Sprintf("<https://discordapp.com/oauth2/authorize?client_id=%v&scope=bot>", id)
		_, err := ctx.Send(url)
		if err != nil {
			ctx.CommandManager.ErrorChannel <- disgoman.CommandError{
				Context: ctx,
				Message: "Couldn't send the invite link.",
				Error:   err,
			}
		}
	}
}

var GitCommand = &disgoman.Command{
	Name:                "git",
	Aliases:             nil,
	Description:         "Show my github link",
	OwnerOnly:           false,
	Hidden:              false,
	RequiredPermissions: 0,
	Invoke:              gitCommandFunc,
}

func gitCommandFunc(ctx disgoman.Context, _ []string) {
	embed := &discordgo.MessageEmbed{
		Title: "Hi there, My code is on Github",
		Color: 0,
		URL:   "https://github.com/dustinpianalto/Geeksbot",
	}
	_, err := ctx.Session.ChannelMessageSendEmbed(ctx.Channel.ID, embed)
	if err != nil {
		ctx.CommandManager.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Git failed",
			Error:   err,
		}
	}
}

var SayCommand = &disgoman.Command{
	Name:                "say",
	Aliases:             nil,
	Description:         "Repeat a message",
	OwnerOnly:           false,
	Hidden:              false,
	RequiredPermissions: 0,
	SanitizeEveryone:    true,
	Invoke:              sayCommandFunc,
}

func sayCommandFunc(ctx disgoman.Context, args []string) {
	resp := strings.Join(args, " ")
	resp = strings.ReplaceAll(resp, "@everyone", "@\ufff0everyone")
	resp = strings.ReplaceAll(resp, "@here", "@\ufff0here")
	_, err := ctx.Session.ChannelMessageSend(ctx.Message.ChannelID, resp)
	if err != nil {
		ctx.CommandManager.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Say Failed",
			Error:   err,
		}
	}
}

var UserCommand = &disgoman.Command{
	Name:                "user",
	Aliases:             nil,
	Description:         "Get user info",
	OwnerOnly:           false,
	Hidden:              false,
	RequiredPermissions: 0,
	Invoke:              userCommandFunc,
}

func userCommandFunc(ctx disgoman.Context, args []string) {
	var member *discordgo.Member
	if len(args) == 0 {
		member, _ = ctx.Session.GuildMember(ctx.Guild.ID, ctx.Message.Author.ID)
	} else {
		var err error
		if len(ctx.Message.Mentions) > 0 {
			member, err = ctx.Session.GuildMember(ctx.Guild.ID, ctx.Message.Mentions[0].ID)
		} else {
			member, err = ctx.Session.GuildMember(ctx.Guild.ID, args[0])
		}
		if err != nil {
			ctx.CommandManager.ErrorChannel <- disgoman.CommandError{
				Context: ctx,
				Message: "Couldn't get that member",
				Error:   err,
			}
			return
		}
	}
	thumb := &discordgo.MessageEmbedThumbnail{
		URL: member.User.AvatarURL(""),
	}

	var botString string
	if member.User.Bot {
		botString = "BOT"
	} else {
		botString = ""
	}

	var roles []*discordgo.Role
	for _, roleID := range member.Roles {
		role, _ := ctx.Session.State.Role(ctx.Guild.ID, roleID)
		roles = append(roles, role)
	}
	sort.Slice(roles, func(i, j int) bool { return roles[i].Position > roles[j].Position })
	var roleMentions []string
	for _, role := range roles {
		roleMentions = append(roleMentions, role.Mention())
	}
	var rolesString string
	if len(roleMentions) > 0 {
		rolesString = strings.Join(roleMentions, " ")
	} else {
		rolesString = "None"
	}

	rolesField := &discordgo.MessageEmbedField{
		Name:   "Roles:",
		Value:  rolesString,
		Inline: false,
	}

	guildJoinTime, _ := member.JoinedAt.Parse()
	guildJoinedField := &discordgo.MessageEmbedField{
		Name:   "Joined Guild:",
		Value:  discord_utils.ParseDateString(guildJoinTime),
		Inline: false,
	}

	int64ID, _ := strconv.ParseInt(member.User.ID, 10, 64)
	s := discord_utils.ParseSnowflake(int64ID)
	discordJoinedField := &discordgo.MessageEmbedField{
		Name:   "Joined Discord:",
		Value:  discord_utils.ParseDateString(s.CreationTime),
		Inline: false,
	}

	embed := &discordgo.MessageEmbed{
		Title:       fmt.Sprintf("%v#%v  %v", member.User.Username, member.User.Discriminator, botString),
		Description: fmt.Sprintf("**%v** (%v)", member.Nick, member.User.ID),
		Color:       ctx.Session.State.UserColor(member.User.ID, ctx.Channel.ID),
		Thumbnail:   thumb,
		Fields: []*discordgo.MessageEmbedField{
			guildJoinedField,
			discordJoinedField,
			rolesField,
		},
	}
	_, err := ctx.Session.ChannelMessageSendEmbed(ctx.Channel.ID, embed)
	if err != nil {
		ctx.CommandManager.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Couldn't send the user embed",
			Error:   err,
		}
	}
}

var AddUserCommand = &disgoman.Command{
	Name:                "adduser",
	Aliases:             nil,
	Description:         "Get user info",
	OwnerOnly:           false,
	Hidden:              false,
	RequiredPermissions: 0,
	Invoke:              addUserCommandFunc,
}

func addUserCommandFunc(ctx disgoman.Context, args []string) {
	if ctx.Message.Author.ID == ctx.CommandManager.Owners[0] {
		user := geeksbot.User{
			ID:       ctx.Message.Author.ID,
			IsActive: true,
			IsStaff:  true,
			IsAdmin:  true,
		}
		user, err := services.UserService.CreateUser(user)
		if err != nil {
			ctx.CommandManager.ErrorChannel <- disgoman.CommandError{
				Context: ctx,
				Message: "Error with adding user",
				Error:   err,
			}
			return
		}
		ctx.Session.MessageReactionAdd(ctx.Channel.ID, ctx.Message.ID, "✅")
	}
}