Add sar commands

guild.go

@@ -25,4 +25,5 @@ type GuildService interface {
 	UpdateRole(r Role) (Role, error)
 	DeleteRole(r Role) error
 	GetOrCreateGuild(id string) (Guild, error)
+	CreateOrUpdateRole(r Role) (Role, error)
 }

internal/database/channel.go

@@ -69,12 +69,9 @@ func (s channelService) GuildChannels(g geeksbot.Guild) ([]geeksbot.Channel, err
 
 func (s channelService) GetOrCreateChannel(id string, guild_id string) (geeksbot.Channel, error) {
 	channel, err := s.Channel(id)
-	log.Println(err)
+	if err == sql.ErrNoRows {
-	log.Println(1)
+		var guild geeksbot.Guild
-	if err.Error() == sql.ErrNoRows.Error() {
+		guild, err = GuildService.GetOrCreateGuild(guild_id)
-		guild, err := GuildService.GetOrCreateGuild(guild_id)
-		log.Println(err)
-		log.Println(2)
 		if err != nil {
 			return geeksbot.Channel{}, err
 		}
@@ -86,7 +83,5 @@ func (s channelService) GetOrCreateChannel(id string, guild_id string) (geeksbot
 			NewPatron: false,
 		})
 	}
-	log.Println(err)
-	log.Println(3)
 	return channel, err
 }

internal/database/guild.go

@@ -111,3 +111,11 @@ func (s guildService) GetOrCreateGuild(id string) (geeksbot.Guild, error) {
 	}
 	return guild, err
 }
+
+func (s guildService) CreateOrUpdateRole(r geeksbot.Role) (geeksbot.Role, error) {
+	role, err := s.CreateRole(r)
+	if err != nil && err.Error() == `pq: duplicate key value violates unique constraint "roles_pkey"` {
+		role, err = s.UpdateRole(r)
+	}
+	return role, err
+}

internal/database/migrations/000004_add_sar_role_type.down.sql

@@ -0,0 +1,14 @@
+BEGIN;
+    CREATE TYPE role_type_new AS ENUM (
+        'normal',
+        'moderator',
+        'admin',
+        'patreon',
+    );
+    UPDATE roles SET role_type = 'normal' WHERE role_type = 'sar';
+    ALTER TABLE roles 
+        ALTER COLUMN roles TYPE role_type_new;
+            USING (roles::text::role_type_new)
+    DROP TYPE role_type;
+    ALTER TYPE role_type_new RENAME TO role_type;
+COMMIT;

internal/database/migrations/000004_add_sar_role_type.up.sql

@@ -0,0 +1 @@
+ALTER TYPE role_type ADD VALUE 'sar';

internal/database/request.go

@@ -124,6 +124,14 @@ func (s requestService) CreateRequest(r geeksbot.Request) (geeksbot.Request, err
 							completed, completed_at, completed_by, message_id, completed_message)
 						VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10) RETURNING id`
 	var id int64
+	var completedByID sql.NullString
+	if r.CompletedBy == nil {
+		completedByID.String = ""
+		completedByID.Valid = false
+	} else {
+		completedByID.String = r.CompletedBy.ID
+		completedByID.Valid = true
+	}
 	err := s.db.QueryRow(queryString,
 		r.Author.ID,
 		r.Channel.ID,
@@ -132,7 +140,7 @@ func (s requestService) CreateRequest(r geeksbot.Request) (geeksbot.Request, err
 		r.RequestedAt,
 		r.Completed,
 		r.CompletedAt,
-		r.CompletedBy.ID,
+		completedByID,
 		r.Message.ID,
 		r.CompletedMessage).Scan(&id)
 	if err != nil {

internal/exts/guild/roles.go

@@ -0,0 +1,404 @@
+package guild
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+
+	"github.com/bwmarrin/discordgo"
+	"github.com/dustinpianalto/disgoman"
+	"github.com/dustinpianalto/geeksbot"
+	"github.com/dustinpianalto/geeksbot/internal/discord_utils"
+	"github.com/dustinpianalto/geeksbot/internal/services"
+	"github.com/dustinpianalto/geeksbot/internal/utils"
+)
+
+var AddModeratorRoleCommand = &disgoman.Command{
+	Name:                "addMod",
+	Aliases:             []string{"addModerator", "addModRole"},
+	Description:         "Add a role which is allowed to run moderator commands",
+	OwnerOnly:           false,
+	Hidden:              false,
+	RequiredPermissions: disgoman.PermissionManageServer,
+	Invoke:              addModeratorRoleCommandFunc,
+}
+
+func addModeratorRoleCommandFunc(ctx disgoman.Context, args []string) {
+	var count int
+	added := make(map[string]bool)
+	guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Something went wrong getting the guild", err)
+		return
+	}
+	roles := append(args, ctx.Message.MentionRoles...)
+	if len(roles) > 0 {
+		for _, id := range roles {
+			if strings.HasPrefix(id, "<@&") && strings.HasSuffix(id, ">") {
+				continue
+			}
+			if _, ok := added[id]; ok {
+				continue
+			}
+			if _, err = ctx.Session.State.Role(ctx.Guild.ID, id); err != nil {
+				_, _ = ctx.Send(fmt.Sprintf("%s does not reference a valid role for this guild.", id))
+				continue
+			}
+			_, err := services.GuildService.CreateOrUpdateRole(geeksbot.Role{
+				ID:       id,
+				RoleType: "moderator",
+				Guild:    guild,
+			})
+			if err != nil {
+				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("There was a problem adding <@&%s>", id), err)
+				continue
+			}
+			added[id] = true
+			count++
+			_, _ = ctx.Send(fmt.Sprintf("Added <@&%s> as a moderator role.", id))
+		}
+		_, _ = ctx.Send(fmt.Sprintf("Added %d moderator %s.", count, utils.PluralizeString("role", count)))
+	} else {
+		_, _ = ctx.Send("Please include at least one role to make a moderator role.")
+	}
+}
+
+var AddAdminRoleCommand = &disgoman.Command{
+	Name:                "addAdmin",
+	Aliases:             []string{"addAdminRole"},
+	Description:         "Add a role which is allowed to run admin commands",
+	OwnerOnly:           false,
+	Hidden:              false,
+	RequiredPermissions: disgoman.PermissionManageServer,
+	Invoke:              addAdminRoleCommandFunc,
+}
+
+func addAdminRoleCommandFunc(ctx disgoman.Context, args []string) {
+	var count int
+	added := make(map[string]bool)
+	guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Something went wrong getting the guild", err)
+		return
+	}
+	roles := append(args, ctx.Message.MentionRoles...)
+	if len(roles) > 0 {
+		for _, id := range roles {
+			if strings.HasPrefix(id, "<@&") && strings.HasSuffix(id, ">") {
+				continue
+			}
+			if _, ok := added[id]; ok {
+				continue
+			}
+			if _, err = ctx.Session.State.Role(ctx.Guild.ID, id); err != nil {
+				_, _ = ctx.Send(fmt.Sprintf("%s does not reference a valid role for this guild.", id))
+				continue
+			}
+			_, err := services.GuildService.CreateOrUpdateRole(geeksbot.Role{
+				ID:       id,
+				RoleType: "admin",
+				Guild:    guild,
+			})
+			if err != nil {
+				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("There was a problem adding <@&%s>", id), err)
+				continue
+			}
+			added[id] = true
+			count++
+			_, _ = ctx.Send(fmt.Sprintf("Added <@&%s> as an admin role.", id))
+		}
+		_, _ = ctx.Send(fmt.Sprintf("Added %d admin %s.", count, utils.PluralizeString("role", count)))
+	} else {
+		_, _ = ctx.Send("Please include at least one role to make an admin role.")
+	}
+}
+
+var RemoveModRoleCommand = &disgoman.Command{
+	Name:                "removeMod",
+	Aliases:             []string{"removeModeratorRole", "removeModRole", "removeAdmin", "removeAdminRole"},
+	Description:         "Remove a role or several roles from the moderator or admin list",
+	OwnerOnly:           false,
+	Hidden:              false,
+	RequiredPermissions: disgoman.PermissionManageServer,
+	Invoke:              removeModRoleCommandFunc,
+}
+
+func removeModRoleCommandFunc(ctx disgoman.Context, args []string) {
+	var count int
+	added := make(map[string]bool)
+	guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Something went wrong getting the guild", err)
+		return
+	}
+	roles := append(args, ctx.Message.MentionRoles...)
+	if len(roles) > 0 {
+		for _, id := range roles {
+			if strings.HasPrefix(id, "<@&") && strings.HasSuffix(id, ">") {
+				continue
+			}
+			if _, ok := added[id]; ok {
+				continue
+			}
+			if _, err = ctx.Session.State.Role(ctx.Guild.ID, id); err != nil {
+				if r, err := services.GuildService.Role(id); err != nil {
+					_, _ = ctx.Send(fmt.Sprintf("%s does not reference a valid role for this guild.", id))
+					continue
+				} else {
+					err = services.GuildService.DeleteRole(r)
+					if err != nil {
+						discord_utils.SendErrorMessage(ctx, "Something went wrong deleting the role", err)
+					}
+					_, _ = ctx.Send(fmt.Sprintf("Deleted <@&%s> as a no longer a valid role.", id))
+					continue
+				}
+			}
+			_, err := services.GuildService.CreateOrUpdateRole(geeksbot.Role{
+				ID:       id,
+				RoleType: "normal",
+				Guild:    guild,
+			})
+			if err != nil {
+				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("There was a problem updating <@&%s>", id), err)
+				continue
+			}
+			added[id] = true
+			count++
+			_, _ = ctx.Send(fmt.Sprintf("Set <@&%s> as a normal role.", id))
+		}
+		_, _ = ctx.Send(fmt.Sprintf("Set %d %s to normal.", count, utils.PluralizeString("role", count)))
+	} else {
+		_, _ = ctx.Send("Please include at least one role to remove from the moderator or admin lists.")
+	}
+}
+
+var MakeRoleSelfAssignableCommand = &disgoman.Command{
+	Name:                "make-role-self-assignable",
+	Aliases:             []string{"makesar", "addsar"},
+	Description:         "Makes the passed in role self assignable by anyone",
+	OwnerOnly:           false,
+	Hidden:              false,
+	RequiredPermissions: disgoman.PermissionManageServer,
+	Invoke:              makeRoleSelfAssignableCommandFunc,
+}
+
+func makeRoleSelfAssignableCommandFunc(ctx disgoman.Context, args []string) {
+	added := make(map[string]bool)
+	guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Something went wrong getting the guild", err)
+		return
+	}
+
+	roles := append(args, ctx.Message.MentionRoles...)
+	if len(roles) > 0 {
+		for _, id := range roles {
+			if strings.HasPrefix(id, "<@&") && strings.HasSuffix(id, ">") {
+				continue
+			}
+			if _, ok := added[id]; ok {
+				continue
+			}
+
+			var role *discordgo.Role
+			var err error
+			if role, err = ctx.Session.State.Role(ctx.Guild.ID, id); err != nil {
+				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("%s does not reference a valid role for this guild", id), err)
+				return
+			}
+
+			_, err = services.GuildService.CreateOrUpdateRole(geeksbot.Role{
+				ID:       role.ID,
+				RoleType: "sar",
+				Guild:    guild,
+			})
+			if err != nil {
+				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("There was a problem updating <@&%s>", role.ID), err)
+				return
+			}
+			_, _ = ctx.Send(fmt.Sprintf("%s is now self assignable", role.Name))
+
+		}
+	} else {
+		_, _ = ctx.Send("Please include at least one role to make self assignable")
+	}
+
+}
+
+var RemoveSelfAssignableCommand = &disgoman.Command{
+	Name:                "remove-self-assignable-role",
+	Aliases:             []string{"removesar"},
+	Description:         "Makes a role that was previously self assignable not so",
+	OwnerOnly:           false,
+	Hidden:              false,
+	RequiredPermissions: disgoman.PermissionManageServer,
+	Invoke:              removeSelfAssignableRoleCommandFunc,
+}
+
+func removeSelfAssignableRoleCommandFunc(ctx disgoman.Context, args []string) {
+	removed := make(map[string]bool)
+	guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Something went wrong getting the guild", err)
+		return
+	}
+
+	roles := append(args, ctx.Message.MentionRoles...)
+	if len(roles) > 0 {
+		for _, id := range roles {
+			if strings.HasPrefix(id, "<@&") && strings.HasSuffix(id, ">") {
+				continue
+			}
+			if _, ok := removed[id]; ok {
+				continue
+			}
+
+			var err error
+			var role *discordgo.Role
+			if role, err = ctx.Session.State.Role(ctx.Guild.ID, id); err != nil {
+				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("%s does not reference a valid role for this guild", id), err)
+				return
+			}
+			_, err = services.GuildService.CreateOrUpdateRole(geeksbot.Role{
+				ID:       role.ID,
+				RoleType: "normal",
+				Guild:    guild,
+			})
+			if err != nil {
+				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("There was a problem updating <@&%s>", role.ID), err)
+				return
+			}
+			_, _ = ctx.Send(fmt.Sprintf("%s's self assignability has been removed.", role.Name))
+		}
+	} else {
+		_, _ = ctx.Send("Please include at least one role to make self assignable")
+	}
+
+}
+
+var SelfAssignRoleCommand = &disgoman.Command{
+	Name:                "giverole",
+	Aliases:             []string{"iwant", "givetome"},
+	Description:         "Assigns a person the passed in role if it is self assignable",
+	OwnerOnly:           false,
+	Hidden:              false,
+	RequiredPermissions: 0,
+	Invoke:              selfAssignRoleCommandFunc,
+}
+
+func selfAssignRoleCommandFunc(ctx disgoman.Context, args []string) {
+	added := make(map[string]bool)
+	roles := append(args, ctx.Message.MentionRoles...)
+	if len(roles) > 0 {
+		for _, id := range roles {
+			var roleID string
+			if strings.HasPrefix(id, "<@&") && strings.HasSuffix(id, ">") {
+				continue
+			} else if _, err := strconv.Atoi(id); err == nil {
+				roleID = id
+			} else {
+				for _, role := range ctx.Guild.Roles {
+					if id == role.Name {
+						roleID = role.ID
+					}
+				}
+			}
+
+			if _, ok := added[id]; ok {
+				continue
+			}
+			var role *discordgo.Role
+			var err error
+			if role, err = ctx.Session.State.Role(ctx.Guild.ID, roleID); err != nil {
+				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("%s does not reference a valid role for this guild", roleID), err)
+				return
+			}
+			if memberHasRole(ctx.Member, role.ID) {
+				_, _ = ctx.Send(fmt.Sprintf("You already have the %s role silly...", role.Name))
+				return
+			}
+			r, err := services.GuildService.Role(role.ID)
+			if err != nil || r.RoleType != "sar" {
+				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("You aren't allowed to assign yourself the %s role", role.Name), err)
+				return
+			}
+			err = ctx.Session.GuildMemberRoleAdd(ctx.Guild.ID, ctx.User.ID, role.ID)
+			if err != nil {
+				discord_utils.SendErrorMessage(ctx, "There was a problem adding that role to you.", err)
+				return
+			}
+			_, _ = ctx.Send(fmt.Sprintf("Congratulations! The %s role has been added to your... Ummm... Thing.", role.Name))
+		}
+	} else {
+		_, _ = ctx.Send("Please include at least one role to make self assignable")
+	}
+
+}
+
+var UnAssignRoleCommand = &disgoman.Command{
+	Name:                "removerole",
+	Aliases:             []string{"idon'twant"},
+	Description:         "Removes a role from a person if the role is self assignable",
+	OwnerOnly:           false,
+	Hidden:              false,
+	RequiredPermissions: 0,
+	Invoke:              unAssignRoleCommandFunc,
+}
+
+func unAssignRoleCommandFunc(ctx disgoman.Context, args []string) {
+	removed := make(map[string]bool)
+	roles := append(args, ctx.Message.MentionRoles...)
+	if len(roles) > 0 {
+		for _, id := range roles {
+			var roleID string
+			if strings.HasPrefix(id, "<@&") && strings.HasSuffix(id, ">") {
+				continue
+			} else if _, err := strconv.Atoi(id); err == nil {
+				roleID = id
+			} else {
+				for _, role := range ctx.Guild.Roles {
+					if id == role.Name {
+						roleID = role.ID
+					}
+				}
+			}
+			if _, ok := removed[id]; ok {
+				continue
+			}
+
+			var role *discordgo.Role
+			var err error
+			if role, err = ctx.Session.State.Role(ctx.Guild.ID, roleID); err != nil {
+				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("%s does not reference a valid role for this guild", roleID), 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
+			}
+			r, err := services.GuildService.Role(role.ID)
+			if err != nil || r.RoleType != "sar" {
+				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("You aren't allowed to assign yourself the %s role", role.Name), err)
+				return
+			}
+			err = ctx.Session.GuildMemberRoleRemove(ctx.Guild.ID, ctx.User.ID, role.ID)
+			if err != nil {
+				discord_utils.SendErrorMessage(ctx, "There was a problem removing that role from your account", err)
+				return
+			}
+			_, _ = ctx.Send(fmt.Sprintf("Sad to see you go... but the %s role has been removed.", role.Name))
+		}
+	} else {
+		_, _ = ctx.Send("Please include at least one role to make self assignable")
+	}
+
+}
+
+func memberHasRole(m *discordgo.Member, id string) bool {
+	for _, r := range m.Roles {
+		if r == id {
+			return true
+		}
+	}
+	return false
+}

internal/exts/init.go

@@ -24,5 +24,12 @@ func AddCommandHandlers(g *disgoman.CommandManager) {
 	_ = g.AddCommand(utils.PingCommand)
 	_ = g.AddCommand(guild.AddPrefixCommand)
 	_ = g.AddCommand(guild.RemovePrefixCommand)
+	_ = g.AddCommand(guild.AddModeratorRoleCommand)
+	_ = g.AddCommand(guild.AddAdminRoleCommand)
+	_ = g.AddCommand(guild.RemoveModRoleCommand)
+	_ = g.AddCommand(guild.MakeRoleSelfAssignableCommand)
+	_ = g.AddCommand(guild.RemoveSelfAssignableCommand)
+	_ = g.AddCommand(guild.SelfAssignRoleCommand)
+	_ = g.AddCommand(guild.UnAssignRoleCommand)
 	_ = g.AddCommand(requests.RequestCommand)
 }

internal/exts/requests/requests.go

@@ -84,7 +84,7 @@ func requestCommandFunc(ctx disgoman.Context, args []string) {
 						"In: %s",
 						mentionRolesString,
 						request.ID,
-						ctx.Message.Author.Mention,
+						ctx.Message.Author.Mention(),
 						request.Content,
 						request.RequestedAt.UTC().Format(time.UnixDate),
 						ctx.Channel.Mention(),
@@ -95,7 +95,7 @@ func requestCommandFunc(ctx disgoman.Context, args []string) {
 	}
 	_, err = ctx.Send(fmt.Sprintf("%s The admin have recieved your request.\n "+
 		"If you would like to close or add a comment to this request please reference ID `%v`",
-		ctx.Message.Author.Mention, request.ID,
+		ctx.Message.Author.Mention(), request.ID,
 	))
 	if err != nil {
 		discord_utils.SendErrorMessage(ctx, "There was an error sending the message. The request was created.", err)

internal/utils/utils.go

@@ -12,3 +12,10 @@ func RemoveDuplicateStrings(s []string) []string {
 	}
 	return o
 }
+
+func PluralizeString(s string, i int) string {
+	if i == 1 {
+		return s
+	}
+	return s + "s"
+}