dcbot-old/discord/coords_translation.go

package discord

import (
	"context"
	"github.com/tribalwarshelp/shared/tw/twmodel"
	"github.com/tribalwarshelp/shared/tw/twurlbuilder"
	"regexp"

	"github.com/bwmarrin/discordgo"
	"github.com/nicksnyder/go-i18n/v2/i18n"
	"github.com/tribalwarshelp/golang-sdk/sdk"

	"github.com/tribalwarshelp/dcbot/message"
	"github.com/tribalwarshelp/dcbot/util/twutil"
)

const (
	coordsLimit                         = 20
	cmdCoordsTranslation        command = "coordstranslation"
	cmdDisableCoordsTranslation command = "disablecoordstranslation"
)

var coordsRegex = regexp.MustCompile(`(\d+)\|(\d+)`)

type hndlrCoordsTranslation struct {
	*Session
}

var _ commandHandler = &hndlrCoordsTranslation{}

func (hndlr *hndlrCoordsTranslation) cmd() command {
	return cmdCoordsTranslation
}

func (hndlr *hndlrCoordsTranslation) requireAdmPermissions() bool {
	return true
}

func (hndlr *hndlrCoordsTranslation) execute(ctx *commandCtx, m *discordgo.MessageCreate, args ...string) {
	argsLength := len(args)
	if argsLength != 1 {
		hndlr.SendMessage(
			m.ChannelID,
			m.Author.Mention()+" "+ctx.localizer.MustLocalize(&i18n.LocalizeConfig{
				MessageID: message.HelpCoordsTranslation,
				TemplateData: map[string]interface{}{
					"Command": hndlr.cmd().WithPrefix(hndlr.cfg.CommandPrefix),
				},
			}),
		)
		return
	}

	serverKey := args[0]
	server, err := hndlr.cfg.API.Server.Read(serverKey, nil)
	if err != nil || server == nil {
		hndlr.SendMessage(
			m.ChannelID,
			ctx.localizer.MustLocalize(&i18n.LocalizeConfig{
				MessageID: message.CoordsTranslationServerNotFound,
				TemplateData: map[string]interface{}{
					"Mention": m.Author.Mention(),
				},
			}),
		)
		return
	}

	ctx.server.CoordsTranslation = serverKey
	go hndlr.cfg.ServerRepository.Update(context.Background(), ctx.server)

	hndlr.SendMessage(m.ChannelID,
		ctx.localizer.MustLocalize(&i18n.LocalizeConfig{
			MessageID: message.CoordsTranslationSuccess,
			TemplateData: map[string]interface{}{
				"Mention": m.Author.Mention(),
			},
		}))
}

type hndlrDisableCoordsTranslation struct {
	*Session
}

var _ commandHandler = &hndlrDisableCoordsTranslation{}

func (hndlr *hndlrDisableCoordsTranslation) cmd() command {
	return cmdDisableCoordsTranslation
}

func (hndlr *hndlrDisableCoordsTranslation) requireAdmPermissions() bool {
	return true
}

func (hndlr *hndlrDisableCoordsTranslation) execute(ctx *commandCtx, m *discordgo.MessageCreate, args ...string) {
	ctx.server.CoordsTranslation = ""
	go hndlr.cfg.ServerRepository.Update(context.Background(), ctx.server)

	hndlr.SendMessage(m.ChannelID,
		ctx.localizer.MustLocalize(&i18n.LocalizeConfig{
			MessageID: message.DisableCoordsTranslationSuccess,
			TemplateData: map[string]interface{}{
				"Mention": m.Author.Mention(),
			},
		}))
}

type procTranslateCoords struct {
	*Session
}

var _ messageProcessor = &procTranslateCoords{}

func (p *procTranslateCoords) process(ctx *commandCtx, m *discordgo.MessageCreate) {
	if ctx.server.CoordsTranslation == "" {
		return
	}

	coords := coordsRegex.FindAllString(m.Content, -1)
	coordsLen := len(coords)
	if coordsLen <= 0 {
		return
	}

	version, err := p.cfg.API.Version.Read(twmodel.VersionCodeFromServerKey(ctx.server.CoordsTranslation))
	if err != nil || version == nil {
		return
	}
	if coordsLen > coordsLimit {
		coords = coords[0:coordsLimit]
	}
	list, err := p.cfg.API.Village.Browse(ctx.server.CoordsTranslation,
		0,
		0,
		[]string{},
		&twmodel.VillageFilter{
			XY: coords,
		},
		&sdk.VillageInclude{
			Player: true,
			PlayerInclude: sdk.PlayerInclude{
				Tribe: true,
			},
		},
	)
	if err != nil || list == nil || len(list.Items) <= 0 {
		return
	}

	bldr := &MessageEmbedFieldBuilder{}
	for _, village := range list.Items {
		villageURL := twurlbuilder.BuildVillageURL(ctx.server.CoordsTranslation, version.Host, village.ID)
		playerName := "-"
		playerURL := ""
		if !twutil.IsPlayerNil(village.Player) {
			playerName = village.Player.Name
			playerURL = twurlbuilder.BuildPlayerURL(ctx.server.CoordsTranslation, version.Host, village.Player.ID)
		}
		tribeName := "-"
		tribeURL := ""
		if !twutil.IsPlayerTribeNil(village.Player) {
			tribeName = village.Player.Tribe.Name
			tribeURL = twurlbuilder.BuildTribeURL(ctx.server.CoordsTranslation, version.Host, village.Player.Tribe.ID)
		}

		bldr.Append(ctx.localizer.MustLocalize(&i18n.LocalizeConfig{
			MessageID: message.CoordsTranslationMessage,
			TemplateData: map[string]interface{}{
				"Village": BuildLink(village.FullName(), villageURL),
				"Player":  BuildLink(playerName, playerURL),
				"Tribe":   BuildLink(tribeName, tribeURL),
			},
		}) + "\n")
	}

	title := ctx.localizer.MustLocalize(&i18n.LocalizeConfig{
		MessageID: message.CoordsTranslationTitle,
	})
	bldr.SetName(title)
	p.SendEmbed(m.ChannelID, NewEmbed().
		SetTitle(title).
		SetFields(bldr.ToMessageEmbedFields()))
}