game.go

package main

import (
	"fmt"
	"os"
	"time"

	pubnub "github.com/pubnub/go"

	"github.com/gosuri/uiprogress"
	term "github.com/nsf/termbox-go"
)

func countdown(hostName string, guestName string) {
	fmt.Println("")
	fmt.Println("The game is about to start between " + hostName + " and " + guestName + "!")
	fmt.Println("Alternate pressing SPACE and the RIGHT ARROW KEY to race!")
	fmt.Println("")
	time.Sleep(750 * time.Millisecond)
	fmt.Print("3... ")
	time.Sleep(1 * time.Second)
	fmt.Print("2... ")
	time.Sleep(1 * time.Second)
	fmt.Print("1...  ")
	time.Sleep(1 * time.Second)
	fmt.Print("Race!!")
	fmt.Println("")
	time.Sleep(500 * time.Millisecond)
}

func startGame(isHost bool, lobby string, hostName string, guestName string, pn *pubnub.PubNub) {
	var (
		spaced   bool
		progress int
		winner   string
	)
	data := make(map[string]interface{})

	countdown(hostName, guestName) // Countdown before starting game.

	uiprogress.Start()
	hostBar := uiprogress.AddBar(100).AppendCompleted()
	hostBar.AppendFunc(func(b *uiprogress.Bar) string {
		if isHost {
			return hostName + " (you)"
		}
		return hostName + " (host)"
	})
	guestBar := uiprogress.AddBar(100).AppendCompleted()
	guestBar.AppendFunc(func(b *uiprogress.Bar) string {
		if !isHost {
			return guestName + " (you)"
		}
		return guestName + " (guest)"
	})

	gamelistener := pubnub.NewListener()
	go func() {
		for {
			select {
			case message := <-gamelistener.Message:
				if msg, ok := message.Message.(map[string]interface{}); ok {
					if val, ok := msg["guestProgress"]; ok {
						guestBar.Set(int(val.(float64)))
					}
					if val, ok := msg["hostProgress"]; ok {
						hostBar.Set(int(val.(float64)))
					}
				}
			}
		}
	}()
	pn.AddListener(gamelistener)
	pn.Subscribe().
		Channels([]string{lobby}).
		Execute()

	err := term.Init()
	if err != nil {
		panic(err)
	}
	defer term.Close()
keyPressListenerLoop:
	for {
		if hostBar.Current() == hostBar.Total { // Check for winner.
			winner = hostName
			break keyPressListenerLoop
		} else if guestBar.Current() == guestBar.Total {
			winner = guestName
			break keyPressListenerLoop
		}
		event := term.PollEvent()
		switch {
		case event.Key == term.KeyEsc:
			break keyPressListenerLoop
		case event.Key == term.KeySpace:
			if !spaced {
				progress = progress + 1
				if isHost {
					if hostBar.Current() < hostBar.Total {
						data["hostProgress"] = progress
						pn.Publish().
							Channel(lobby).
							Message(data).
							Execute()
					}
				} else {
					if guestBar.Current() < guestBar.Total {
						data["guestProgress"] = progress
						pn.Publish().
							Channel(lobby).
							Message(data).
							Execute()
					}
				}
				spaced = true
			}
		case event.Key == term.KeyArrowRight: // Prevents the player from using keyrepeat to cheat.
			spaced = false
		}
	}

	uiprogress.Stop()
	term.Close()
	pn.Unsubscribe().
		Channels([]string{lobby}).
		Execute()
	fmt.Println("")
	if winner != "" {
		fmt.Println(winner + " won the game!")
	} else {
		fmt.Println("You left the game.") // The other player wins if the player leaves the game.
		if isHost {
			data["guestProgress"] = guestBar.Total
		} else {
			data["hostProgress"] = hostBar.Total
		}
		pn.Publish().
			Channel(lobby).
			Message(data).
			Execute()
	}
	fmt.Println("")
	fmt.Println("Thanks for playing!")
	os.Exit(0)
}