Geeksbot/internal/exts/arcon/arcon.go

package arcon

import (
	"fmt"
	"log"
	"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
	}
	log.Printf("%#v", response)
	if strings.Contains(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),
	)
}