Goff/exts/roles.go

package exts

import (
	"fmt"
	"strconv"
	"strings"

	"github.com/bwmarrin/discordgo"
	"github.com/dustinpianalto/disgoman"
	"github.com/dustinpianalto/goff/utils"
)

func makeRoleSelfAssignable(ctx disgoman.Context, args []string) {
	var roleString string
	var roleID string
	if len(args) > 0 {
		roleString = strings.Join(args, " ")
		if strings.HasPrefix(roleString, "<@&") && strings.HasSuffix(roleString, ">") {
			roleID = roleString[3 : len(roleString)-1]
		} else if _, err := strconv.Atoi(roleString); err == nil {
			roleID = roleString
		} else {
			for _, role := range ctx.Guild.Roles {
				if roleString == role.Name {
					roleID = role.ID
				}
			}
		}
	}
	fmt.Println(roleID)
	var role *discordgo.Role
	var err error
	if role, err = ctx.Session.State.Role(ctx.Guild.ID, roleID); err != nil {
		ctx.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Can't find that Role.",
			Error:   err,
		}
		return
	}
	_, err = utils.Database.Exec("INSERT INTO roles (id, guild_id, self_assignable) VALUES ($1, $2, true) ON CONFLICT (id) DO UPDATE SET self_assignable=true", role.ID, ctx.Guild.ID)
	if err != nil {
		ctx.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Error Updating Database",
			Error:   err,
		}
		return
	}
	_, _ = ctx.Send(fmt.Sprintf("%s is now self assignable", role.Name))
}

func removeSelfAssignableRole(ctx disgoman.Context, args []string) {
	var roleString string
	var roleID string
	if len(args) > 0 {
		roleString = strings.Join(args, " ")
		if strings.HasPrefix(roleString, "<@&") && strings.HasSuffix(roleString, ">") {
			roleID = roleString[3 : len(roleString)-1]
		} else if _, err := strconv.Atoi(roleString); err == nil {
			roleID = roleString
		} else {
			for _, role := range ctx.Guild.Roles {
				if roleString == role.Name {
					roleID = role.ID
				}
			}
		}
	}
	fmt.Println(roleID)
	var err error
	var role *discordgo.Role
	if role, err = ctx.Session.State.Role(ctx.Guild.ID, roleID); err != nil {
		ctx.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Can't find that Role.",
			Error:   err,
		}
		return
	}
	_, err = utils.Database.Exec("INSERT INTO roles (id, guild_id, self_assignable) VALUES ($1, $2, false) ON CONFLICT (id) DO UPDATE SET self_assignable=false", role.ID, ctx.Guild.ID)
	if err != nil {
		ctx.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Error Updating Database",
			Error:   err,
		}
		return
	}
	_, _ = ctx.Send(fmt.Sprintf("%s's self assignability has been removed.", role.Name))
}

func selfAssignRole(ctx disgoman.Context, args []string) {
	var roleString string
	var roleID string
	if len(args) > 0 {
		roleString = strings.Join(args, " ")
		if strings.HasPrefix(roleString, "<@&") && strings.HasSuffix(roleString, ">") {
			roleID = roleString[3 : len(roleString)-1]
		} else if _, err := strconv.Atoi(roleString); err == nil {
			roleID = roleString
		} else {
			for _, role := range ctx.Guild.Roles {
				if roleString == role.Name {
					roleID = role.ID
				}
			}
		}
	}
	fmt.Println(roleID)
	var role *discordgo.Role
	var err error
	if role, err = ctx.Session.State.Role(ctx.Guild.ID, roleID); err != nil {
		ctx.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Can't find that Role.",
			Error:   err,
		}
		return
	}
	if memberHasRole(ctx.Member, role.ID) {
		_, _ = ctx.Send(fmt.Sprintf("You already have the %s role silly...", role.Name))
		return
	}
	var selfAssignable bool
	err = utils.Database.QueryRow("SELECT self_assignable FROM roles where id=$1", role.ID).Scan(&selfAssignable)
	if err != nil {
		ctx.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Error Updating Database",
			Error:   err,
		}
		return
	}
	if !selfAssignable {
		_, _ = ctx.Send(fmt.Sprintf("You aren't allowed to assign yourself the %s role", role.Name))
		return
	}
	err = ctx.Session.GuildMemberRoleAdd(ctx.Guild.ID, ctx.User.ID, role.ID)
	if err != nil {
		ctx.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "There was a problem adding that role to you.",
			Error:   err,
		}
		return
	}
	_, _ = ctx.Send(fmt.Sprintf("Congratulations! The %s role has been added to your... Ummm... Thing.", role.Name))
}

func unAssignRole(ctx disgoman.Context, args []string) {
	var roleString string
	var roleID string
	if len(args) > 0 {
		roleString = strings.Join(args, " ")
		if strings.HasPrefix(roleString, "<@&") && strings.HasSuffix(roleString, ">") {
			roleID = roleString[3 : len(roleString)-1]
		} else if _, err := strconv.Atoi(roleString); err == nil {
			roleID = roleString
		} else {
			for _, role := range ctx.Guild.Roles {
				if roleString == role.Name {
					roleID = role.ID
				}
			}
		}
	}
	fmt.Println(roleID)
	var role *discordgo.Role
	var err error
	if role, err = ctx.Session.State.Role(ctx.Guild.ID, roleID); err != nil {
		ctx.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Can't find that Role.",
			Error:   err,
		}
		return
	}
	if !memberHasRole(ctx.Member, role.ID) {
		_, _ = ctx.Send(fmt.Sprintf("I can't remove the %s role from you because you don't have it...", role.Name))
		return
	}
	var selfAssignable bool
	err = utils.Database.QueryRow("SELECT self_assignable FROM roles where id=$1", role.ID).Scan(&selfAssignable)
	if err != nil {
		ctx.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "Error Updating Database",
			Error:   err,
		}
		return
	}
	if !selfAssignable {
		_, _ = ctx.Send(fmt.Sprintf("You aren't allowed to remove the %s role from yourself", role.Name))
		return
	}
	err = ctx.Session.GuildMemberRoleRemove(ctx.Guild.ID, ctx.User.ID, role.ID)
	if err != nil {
		ctx.ErrorChannel <- disgoman.CommandError{
			Context: ctx,
			Message: "There was a problem removing that role from you.",
			Error:   err,
		}
		return
	}
	_, _ = ctx.Send(fmt.Sprintf("Sad to see you go... but the %s role has been removed.", role.Name))
}

func memberHasRole(m *discordgo.Member, id string) bool {
	for _, r := range m.Roles {
		if r == id {
			return true
		}
	}
	return false
}