commands_roles.go

/*
DiscordConsole is a software aiming to give you full control over accounts, bots and webhooks!
Copyright (C) 2020 Mnpn

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/
package main

import (
	"io"
	"sort"
	"strconv"
	"strings"

	"github.com/bwmarrin/discordgo"
	"github.com/jD91mZM2/gtable"
	"github.com/jD91mZM2/stdutil"
)

func commandsRoles(session *discordgo.Session, cmd string, args []string, nargs int, w io.Writer) (returnVal string) {
	switch cmd {
	case "role":
		if nargs < 1 {
			stdutil.PrintErr("role list/add/rem/create/edit/delete", nil)
			return
		}
		switch strings.ToLower(args[0]) {
		case "list":
			if loc.guild == nil {
				stdutil.PrintErr(tl("invalid.guild"), nil)
				return
			}

			roles, err := session.GuildRoles(loc.guild.ID)
			if err != nil {
				stdutil.PrintErr(tl("failed.roles"), err)
				return
			}
			sort.Slice(roles, func(i, j int) bool {
				return roles[i].Position > roles[j].Position
			})

			table := gtable.NewStringTable()
			table.AddStrings("ID", "Name", "Permissions", "Color")

			for _, role := range roles {
				table.AddRow()
				table.AddStrings(role.ID, role.Name, strconv.Itoa(role.Permissions), strconv.Itoa(role.Color))
			}

			writeln(w, table.String())
		case "add":
			fallthrough
		case "rem":
			if nargs < 3 {
				stdutil.PrintErr("role add/rem <user id> <role id>", nil)
				return
			}
			if loc.guild == nil {
				stdutil.PrintErr(tl("invalid.guild"), nil)
				return
			}

			var err error
			if args[0] == "add" {
				err = session.GuildMemberRoleAdd(loc.guild.ID, args[1], args[2])
			} else {
				err = session.GuildMemberRoleRemove(loc.guild.ID, args[1], args[2])
			}

			if err != nil {
				stdutil.PrintErr(tl("failed.role.change"), err)
			}
		case "create":
			if loc.guild == nil {
				stdutil.PrintErr(tl("invalid.guild"), nil)
				return
			}

			role, err := session.GuildRoleCreate(loc.guild.ID)
			if err != nil {
				stdutil.PrintErr(tl("failed.role.create"), err)
				return
			}
			writeln(w, "Created role with ID "+role.ID)
			lastUsedRole = role.ID
			returnVal = role.ID
		case "edit":
			if nargs < 4 {
				stdutil.PrintErr("role edit <role id> <flag> <value>", nil)
				return
			}
			if loc.guild == nil {
				stdutil.PrintErr(tl("invalid.guild"), nil)
				return
			}

			value := strings.Join(args[3:], " ")

			roles, err := session.GuildRoles(loc.guild.ID)
			if err != nil {
				stdutil.PrintErr(tl("failed.roles"), err)
				return
			}

			var role *discordgo.Role
			for _, r := range roles {
				if r.ID == args[1] {
					role = r
					break
				}
			}
			if role == nil {
				stdutil.PrintErr(tl("invalid.role"), nil)
				return
			}

			name := role.Name
			color := int64(role.Color)
			hoist := role.Hoist
			perms := role.Permissions
			mention := role.Mentionable

			switch strings.ToLower(args[2]) {
			case "name":
				name = value
			case "color":
				value = strings.TrimPrefix(value, "#")
				color, err = strconv.ParseInt(value, 16, 0)
				if err != nil {
					stdutil.PrintErr(tl("invalid.number"), nil)
					return
				}
			case "separate":
				hoist, err = parseBool(value)
				if err != nil {
					stdutil.PrintErr(err.Error(), nil)
					return
				}
			case "perms":
				perms, err = strconv.Atoi(value)
				if err != nil {
					stdutil.PrintErr(tl("invalid.number"), nil)
					return
				}
			case "mention":
				mention, err = parseBool(value)
				if err != nil {
					stdutil.PrintErr(err.Error(), nil)
					return
				}
			default:
				stdutil.PrintErr(tl("invalid.value")+": "+args[2], nil)
				return
			}

			role, err = session.GuildRoleEdit(loc.guild.ID, args[1], name, int(color), hoist, perms, mention)
			if err != nil {
				stdutil.PrintErr(tl("failed.role.edit"), err)
				return
			}
			lastUsedRole = role.ID
			writeln(w, "Edited role "+role.ID)
		case "delete":
			if nargs < 2 {
				stdutil.PrintErr("role delete <role id>", nil)
				return
			}
			if loc.guild == nil {
				stdutil.PrintErr(tl("invalid.guild"), nil)
				return
			}

			err := session.GuildRoleDelete(loc.guild.ID, args[1])
			if err != nil {
				stdutil.PrintErr(tl("failed.role.delete"), err)
			}
		default:
			stdutil.PrintErr("role list/add/rem/create/edit/delete", nil)
			return
		}
	}
	return
}