file4.go

package main

import (
	"fmt"
	"log"
	"os"
	"strings"

	"github.com/ThomasK33/UpstartGriefBot/src"
	"github.com/gempir/go-twitch-irc/v2"
	_ "github.com/joho/godotenv/autoload"
	"github.com/reiver/go-telnet"
)

var (
	// Version - Version number
	Version string
	// Build - Build number
	Build string
)

// Command - Twitch Command structure
type Command struct {
	requiredVotes int
	currentVotes  int
	handler       func(string)
}

func main() {
	log.Println("Starting " + Version + " (" + Build + ")")

	channelName, botName, oauthToken := "channel1", "botxx", "m9cnIDlo5Dk07yPwes3Lk0TWmdDkZe3aN3QjewdGm59p4eWjvNS8M8uAh03I6JjcUMJIwIF58"

	caller := &src.TelnetCaller{}
	gc := &src.GameCommands{Caller: caller}

	commands := map[string]*Command{
		"!disconnect":  {3, 0, gc.Disconnect},
		"!quit":        {3, 0, gc.Quit},
		"!buy":         {3, 0, gc.BuySlot},
		"!sell":        {3, 0, gc.SellUnit},
		"!spray":       {3, 0, gc.BoardSpray},
		"!benchUnit":   {3, 0, gc.BenchUnit},
		"!fakeDown":    {3, 0, gc.FakeGCDown},
		"!levelup":     {3, 0, gc.Levelup},
		"!lock":        {3, 0, gc.Lock},
		"!reroll":      {3, 0, gc.Reroll},
		"!toggle":      {1, 0, gc.Toggle},
		"!cameraDown":  {3, 0, gc.CameraDown},
		"!cameraUp":    {3, 0, gc.CameraUp},
		"!enemiesDown": {3, 0, gc.EnemiesDown},
		"!enemiesUp":   {3, 0, gc.EnemiesUp},
		"!away":        {3, 0, gc.Away},
		"!home":        {3, 0, gc.Home},
		"!opponent":    {3, 0, gc.Opponent},
		"!dps":         {3, 0, gc.Dps},
		"!sharecode":   {3, 0, gc.Sharecode},
	}

	// TODO: Refactor this into person specific counter
	// counter := map[string]map[string]time.Time{}
	// counter["greycodes"] = map[string]time.Time{}
	// counter["greycodes"]["!buy"] = time.Now()

	go func() {
		client := twitch.NewClient(botName, oauthToken)

		client.OnPrivateMessage(func(message twitch.PrivateMessage) {
			log.Println(message.Message)

			for key, value := range commands {
				if strings.HasPrefix(message.Message, key) {
					value.currentVotes++

					if value.currentVotes >= value.requiredVotes {
						argString := strings.TrimSpace(strings.TrimPrefix(message.Message, key))

						value.handler(argString)
						value.currentVotes = 0

						client.Say(channelName, "Executed "+key)
					} else {
						client.Say(channelName, "Need "+fmt.Sprint(value.requiredVotes-value.currentVotes)+" more votes to execute "+key)
					}
				}
			}

			if strings.HasPrefix(message.Message, "!help") {
				var builder strings.Builder

				builder.WriteString("!help")
				for key := range commands {
					builder.WriteString(", " + key)
				}

				client.Say(channelName, "Following commands are available: "+builder.String())
			}
		})

		client.OnConnect(func() {
			log.Println("Connected to Twitch")
			client.Say(channelName, "Bring it on!")
		})

		client.Join(channelName)

		twitchErr := client.Connect()
		if twitchErr != nil {
			panic(twitchErr)
		}
	}()

	err := telnet.DialToAndCall("localhost:27015", caller)

	if err != nil {
		log.Fatal(err)
	}
}