astro-domu/internal/app/ui.go

package app

import (
	"fmt"

	"github.com/gdamore/tcell/v2"
	"github.com/jordanorelli/astro-domu/internal/exit"
	"github.com/jordanorelli/astro-domu/internal/math"
	"github.com/jordanorelli/astro-domu/internal/sim"
	"github.com/jordanorelli/astro-domu/internal/wire"
	"github.com/jordanorelli/blammo"
)

type UI struct {
	*blammo.Log
	PlayerName string
	screen     tcell.Screen
	room       *wire.Room
	client     *wire.Client

	statusBar *statusBar
	gameView  *gameView
	testList  *menuList
	chatView  *chatView
	focussed  view
}

func (ui *UI) Run() {
	ui.setupTerminal()
	defer ui.clearTerminal()

	ui.room = new(wire.Room)

	if err := ui.connect(); err != nil {
		return
	}

	res, err := ui.client.Send(wire.Login{Name: ui.PlayerName})
	if err != nil {
		ui.Error("login error: %v", err)
		return
	}

	welcome, ok := res.Body.(*wire.Welcome)
	if !ok {
		ui.Error("unexpected initial message of type %t", res.Body)
		return
	}
	ui.Info("welcome: %v", welcome)
	meta := welcome.Players[ui.PlayerName]
	room := welcome.Rooms[meta.Room]
	ui.room = &room
	ui.gameView = &gameView{
		Log:  ui.Child("game-view"),
		room: &room,
		me: &wire.Entity{
			ID:       meta.Avatar,
			Glyph:    room.Entities[meta.Avatar].Glyph,
			Position: room.Entities[meta.Avatar].Position,
		},
	}
	ui.chatView = &chatView{
		Log:     ui.Child("chat-view"),
		history: make([]sim.ChatMessage, 0, 32),
	}
	ui.statusBar = &statusBar{}
	ui.testList = &menuList{
		Log:     ui.Child("menu-list"),
		choices: []string{"apple", "banana", "orange"},
	}
	ui.focussed = ui.gameView

	ui.Info("running ui")
	if ui.handleUserInput() {
		ui.Info("user requested close")
		ui.Info("closing client")
		ui.client.Close()
		ui.Info("client closed")
		ui.Info("finalizing screen")
	}
	ui.Info("run loop done, shutting down")
}

func (ui *UI) connect() error {
	ui.client = &wire.Client{
		Log:  ui.Child("client"),
		Host: "cdm.jordanorelli.com",
		Port: 12805,
	}

	c, err := ui.client.Dial()
	if err != nil {
		return fmt.Errorf("unable to dial server: %v", err)
	}
	go ui.handleNotifications(c)
	return nil
}

func (ui *UI) setupTerminal() {
	screen, err := tcell.NewScreen()
	if err != nil {
		exit.WithMessage(1, "unable to create a screen: %v", err)
	}
	ui.screen = screen
	ui.Debug("sceen created")

	if err := screen.Init(); err != nil {
		exit.WithMessage(1, "unable to initialize screen: %v", err)
	}
	ui.Debug("screen initialized")
	width, height := screen.Size()
	ui.Debug("screen width: %d", width)
	ui.Debug("screen height: %d", height)
	ui.Debug("screen colors: %v", screen.Colors())
	ui.Debug("screen has mouse: %v", screen.HasMouse())
}

func (ui *UI) clearTerminal() {
	ui.statusBar.clearCount++
	ui.screen.Clear()
	ui.screen.Fini()
}

func (ui *UI) handleNotifications(c <-chan wire.Response) {
	for n := range c {
		ui.statusBar.msgCount++
		if ui.handleNotification(n.Body) {
			if ui.gameView != nil {
				ui.render()
			}
		}
	}
	ui.Info("notifications channel is closed so we must be done")
	ui.Info("clearing and finalizing screen from notifications goroutine")
	ui.statusBar.clearCount++
	ui.screen.Clear()
	ui.screen.Fini()
	ui.Info("screen finalized")
}

func (ui *UI) handleNotification(v wire.Value) bool {
	switch n := v.(type) {

	case *wire.Entity:
		ui.room.Entities[n.ID] = *n
		return true

	case *wire.Frame:
		if ui.room == nil {
			ui.room = new(wire.Room)
		}

		ui.room.Name = n.RoomName
		ui.room.Rect = n.RoomSize
		ui.room.Entities = n.Entities
		return true

	case *wire.Delta:
		if n.RoomSize != nil {
			ui.room.Rect = *n.RoomSize
		}

		if len(n.Entities) > 0 {
			for id, e := range n.Entities {
				if e != nil {
					ui.room.Entities[id] = *e
				} else {
					delete(ui.room.Entities, id)
				}
			}
		}
		return true

	case *sim.ChatMessage:
		ui.chatView.history = append(ui.chatView.history, *n)
		return true

	default:
		ui.Info("ignoring notification: %v", n)
		return false
	}
}

// writeString writes a string in the given style from left to right beginning
// at the location (x, y). Writing of the screen just fails silently so don't
// do that.
func (ui *UI) writeString(x, y int, s string, style tcell.Style) {
	width, height := ui.screen.Size()
	if y > height {
		return
	}
	for i, r := range []rune(s) {
		x := x + i
		if x > width {
			return
		}
		ui.screen.SetContent(x, y, r, nil, style)
	}
}

func (ui *UI) handleUserInput() bool {
	ui.statusBar.clearCount++
	ui.screen.Clear()
	ui.render()

	for {
		e := ui.screen.PollEvent()
		if e == nil {
			ui.Info("run loop sees nil event, breaking out")
			// someone else shut us down, so return false
			return false
		}

		switch v := e.(type) {
		case *tcell.EventKey:
			key := v.Key()
			if key == tcell.KeyCtrlC {
				ui.Info("saw ctrl+c keyboard input, shutting down")
				// we want to shut things down
				return true
			}
			if key == tcell.KeyTab {
				ui.Info("saw tab from keyboard input, switching focussed view")
				ui.focussed.setFocus(false)
				switch ui.focussed {
				case ui.gameView:
					ui.focussed = ui.chatView
				case ui.chatView:
					ui.focussed = ui.testList
				case ui.testList:
					ui.focussed = ui.gameView
				}
				ui.focussed.setFocus(true)
				goto HANDLED
			}
		}

		ui.focussed.handleEvent(ui, e)
		ui.statusBar.clearCount++
		ui.screen.Clear()
		ui.render()
		ui.statusBar.showCount++
		ui.screen.Show()
	HANDLED:
	}
}

func (ui *UI) render() {
	width, height := ui.screen.Size()

	{
		b := newBuffer(width, 1)
		ui.statusBar.draw(b)
		b.blit(ui.screen, math.Vec{0, 0})
	}

	gameViewHeight := math.Max((height-1)/2, 8)
	{
		b := newBuffer(width/2, gameViewHeight)
		ui.gameView.draw(b)
		b.blit(ui.screen, math.Vec{0, 1})
	}
	{
		b := newBuffer(width/2, gameViewHeight)
		ui.testList.draw(b)
		b.blit(ui.screen, math.Vec{width / 2, 1})
	}
	{
		b := newBuffer(width, height-gameViewHeight-1)
		ui.chatView.draw(b)
		b.blit(ui.screen, math.Vec{0, gameViewHeight + 1})
	}

	ui.statusBar.showCount++
	ui.screen.Show()
}