Add broadcast command

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/go-bindata/go-bindata v3.1.2+incompatible // indirect
 	github.com/golang-migrate/migrate v3.5.4+incompatible // indirect
 	github.com/golang-migrate/migrate/v4 v4.14.1
+	github.com/gorcon/rcon v1.3.1
 	github.com/hashicorp/errwrap v1.1.0 // indirect
 	github.com/lib/pq v1.8.0
 )

go.sum

@@ -152,6 +152,8 @@ github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+
 github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+vpHVxEJEs9eg=
 github.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk=
+github.com/gorcon/rcon v1.3.1 h1:z6a5iOlojfdkvA1qaKEng7QfCJuCzYlC9BUDs6/M+74=
+github.com/gorcon/rcon v1.3.1/go.mod h1:2gztBPSV2WxkPkqV4jiJkdHs+NT46mNSGb8JxbPesx4=
 github.com/gorilla/context v1.1.1/go.mod h1:kBGZzfjB9CEq2AlWe17Uuf7NDRt0dE0s8S51q0aT7Yg=
 github.com/gorilla/handlers v1.4.2/go.mod h1:Qkdc/uu4tH4g6mTK6auzZ766c4CA0Ng8+o/OAirnOIQ=
 github.com/gorilla/mux v1.6.2/go.mod h1:1lud6UwP+6orDFRuTfBEV8e9/aOM/c4fVVCaMa2zaAs=

internal/exts/arcon/arcon.go

@@ -0,0 +1,182 @@
+package arcon
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/dustinpianalto/disgoman"
+	"github.com/dustinpianalto/geeksbot"
+	"github.com/dustinpianalto/geeksbot/internal/discord_utils"
+	"github.com/dustinpianalto/geeksbot/pkg/services"
+	"github.com/gorcon/rcon"
+)
+
+var ListplayersCommand = &disgoman.Command{
+	Name:                "listplayers",
+	Aliases:             nil,
+	Description:         "List the players currently connected to a ARK server.",
+	OwnerOnly:           false,
+	Hidden:              false,
+	RequiredPermissions: 0,
+	Invoke:              listplayersCommandFunc,
+}
+
+func listplayersCommandFunc(ctx disgoman.Context, args []string) {
+	guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Error getting Guild from the database", err)
+		return
+	}
+	author, err := services.UserService.GetOrCreateUser(ctx.Message.Author.ID)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Sorry, there was a problem getting your user.", err)
+		return
+	}
+	if !discord_utils.IsGuildAdmin(ctx, author) && !discord_utils.IsGuildMod(ctx, author) {
+		return
+	}
+	if len(args) == 0 {
+		servers, err := services.ServerService.GuildServers(guild)
+		if err != nil {
+			discord_utils.SendErrorMessage(ctx, "Could not find any servers for this guild", err)
+			return
+		}
+		for _, server := range servers {
+			go listplayers(ctx, server)
+		}
+		return
+	}
+	serverName := strings.Join(args, " ")
+	server, err := services.ServerService.ServerByName(serverName, guild)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx,
+			fmt.Sprintf("Could not find **%s** in this guild.", serverName),
+			err,
+		)
+		return
+	}
+	listplayers(ctx, server)
+}
+
+func listplayers(ctx disgoman.Context, server geeksbot.Server) {
+	msg, err := ctx.Send(fmt.Sprintf("**Getting data for %s**", server.Name))
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "There was an error getting the player list", err)
+		return
+	}
+	conn, err := rcon.Dial(fmt.Sprintf("%s:%d", server.IPAddr, server.Port), server.Password)
+	if err != nil {
+		_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
+			fmt.Sprintf("**Could not open connection to %s**", server.Name),
+		)
+		return
+	}
+	defer conn.Close()
+	response, err := conn.Execute("listplayers")
+	if err != nil {
+		_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
+			fmt.Sprintf("**There was a problem getting a response from %s**", server.Name),
+		)
+		return
+	}
+	if strings.HasPrefix(response, "No Players") {
+		_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
+			fmt.Sprintf("**%s: %s**", server.Name, response),
+		)
+		return
+	}
+	players := strings.Split(response, "\n")
+	for i, player := range players {
+		parts := strings.Split(player, ", ")
+		steamID := parts[len(parts)-1]
+		user, err := services.UserService.GetBySteamID(steamID)
+		if err == nil {
+			duser, err := ctx.Session.GuildMember(ctx.Guild.ID, user.ID)
+			if err == nil {
+				players[i] = fmt.Sprintf("%s (%s)", player, duser.Mention())
+			}
+		}
+	}
+	_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
+		fmt.Sprintf("**%s:**%s", server.Name, strings.Join(players, "\n")),
+	)
+}
+
+var BroadcastCommand = &disgoman.Command{
+	Name:                "broadcast",
+	Aliases:             nil,
+	Description:         "Broadcast a message to ARK servers.",
+	OwnerOnly:           false,
+	Hidden:              false,
+	RequiredPermissions: 0,
+	Invoke:              broadcastCommandFunc,
+}
+
+func broadcastCommandFunc(ctx disgoman.Context, args []string) {
+	guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Error getting Guild from the database", err)
+		return
+	}
+	author, err := services.UserService.GetOrCreateUser(ctx.Message.Author.ID)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Sorry, there was a problem getting your user.", err)
+		return
+	}
+	if !discord_utils.IsGuildAdmin(ctx, author) && !discord_utils.IsGuildMod(ctx, author) {
+		return
+	}
+	message := strings.Join(args[1:len(args)], " ")
+	if strings.ToLower(args[0]) == "all" {
+		servers, err := services.ServerService.GuildServers(guild)
+		if err != nil {
+			discord_utils.SendErrorMessage(ctx, "Could not find any servers for this guild", err)
+			return
+		}
+		for _, server := range servers {
+			go broadcast(ctx, server, message)
+		}
+		return
+	} else {
+		serverName := strings.Title(strings.ReplaceAll(args[0], "_", " "))
+		server, err := services.ServerService.ServerByName(serverName, guild)
+		if err != nil {
+			discord_utils.SendErrorMessage(ctx, fmt.Sprintf("Could not find %s", serverName), err)
+			return
+		}
+		broadcast(ctx, server, message)
+	}
+}
+
+func broadcast(ctx disgoman.Context, server geeksbot.Server, message string) {
+	msg, err := ctx.Send(fmt.Sprintf("**Broadcasting to: %s**", server.Name))
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "There was an error getting the player list", err)
+		return
+	}
+	conn, err := rcon.Dial(fmt.Sprintf("%s:%d", server.IPAddr, server.Port), server.Password)
+	if err != nil {
+		_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
+			fmt.Sprintf("**Could not open connection to %s**", server.Name),
+		)
+		return
+	}
+	defer conn.Close()
+	userName := discord_utils.GetDisplayName(ctx, ctx.Message.Author.ID)
+	response, err := conn.Execute(fmt.Sprintf("broadcast %s: %s", userName, message))
+	if err != nil {
+		_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
+			fmt.Sprintf("**There was a problem getting a response from %s**", server.Name),
+		)
+		return
+	}
+	if response == "Server recieved, But no response!!" {
+		_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
+			fmt.Sprintf("**%s Broadcast Successful**", server.Name),
+		)
+		return
+	}
+	_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
+		fmt.Sprintf("**Broadcasting to %s Failed!**", server.Name),
+	)
+}

internal/exts/init.go

@@ -2,6 +2,7 @@ package exts
 
 import (
 	"github.com/dustinpianalto/disgoman"
+	"github.com/dustinpianalto/geeksbot/internal/exts/arcon"
 	"github.com/dustinpianalto/geeksbot/internal/exts/guild"
 	"github.com/dustinpianalto/geeksbot/internal/exts/requests"
 	"github.com/dustinpianalto/geeksbot/internal/exts/utils"
@@ -36,4 +37,6 @@ func AddCommandHandlers(g *disgoman.CommandManager) {
 	_ = g.AddCommand(requests.ListCommand)
 	_ = g.AddCommand(requests.ViewCommand)
 	_ = g.AddCommand(requests.CommentCommand)
+	_ = g.AddCommand(arcon.ListplayersCommand)
+	_ = g.AddCommand(arcon.BroadcastCommand)
 }

internal/exts/requests/requests.go

@@ -79,7 +79,7 @@ func requestCommandFunc(ctx disgoman.Context, args []string) {
 			if c.Admin {
 				_, _ = ctx.Session.ChannelMessageSend(c.ID,
 					fmt.Sprintf("%s\n"+
-						"New Request ID %d "+
+						"New Request ID %d\n"+
 						"%s has requested assistance: \n"+
 						"```\n%s\n```\n"+
 						"Requested At: %s\n"+
@@ -154,7 +154,7 @@ func closeCommandFunc(ctx disgoman.Context, args []string) {
 			continue
 		}
 		if request.Author != closer && !closer.IsStaff && !closer.IsAdmin {
-			if !discord_utils.IsGuildMod(ctx, closer) || !discord_utils.IsGuildAdmin(ctx, closer) {
+			if !discord_utils.IsGuildMod(ctx, closer) && !discord_utils.IsGuildAdmin(ctx, closer) {
 				discord_utils.SendErrorMessage(ctx, fmt.Sprintf("You are not authorized to close %d", id), nil)
 				continue
 			}

pkg/database/server.go

@@ -74,7 +74,7 @@ func (s serverService) ServerByID(id int) (geeksbot.Server, error) {
 
 func (s serverService) ServerByName(name string, guild geeksbot.Guild) (geeksbot.Server, error) {
 	var id int
-	queryString := "SELECT id FROM servers WHERE name = $1 AND guild_id = $2"
+	queryString := "SELECT id FROM servers WHERE LOWER(name) = LOWER($1) AND guild_id = $2"
 	row := s.db.QueryRow(queryString, name, guild.ID)
 	err := row.Scan(&id)
 	if err != nil {

pkg/database/user.go

@@ -49,3 +49,14 @@ func (s userService) GetOrCreateUser(id string) (geeksbot.User, error) {
 	}
 	return user, err
 }
+
+func (s userService) GetBySteamID(steamID string) (geeksbot.User, error) {
+	var id string
+	queryString := "SELECT id FROM users WHERE steam_id = $1"
+	err := s.db.QueryRow(queryString, steamID).Scan(&id)
+	if err != nil {
+		return geeksbot.User{}, err
+	}
+	user, err := s.User(id)
+	return user, err
+}

user.go

@@ -16,4 +16,5 @@ type UserService interface {
 	DeleteUser(u User) error
 	UpdateUser(u User) (User, error)
 	GetOrCreateUser(id string) (User, error)
+	GetBySteamID(steamID string) (User, error)
 }