You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 lines
3.1 KiB
104 lines
3.1 KiB
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:**\n%s", server.Name, strings.Join(players, "\n")),
|
|
)
|
|
}
|