Restore a game

1 files changed, 9 insertions, 2 deletions

diff --git a/pkg/ui/views/game_moves.go b/pkg/ui/views/game_moves.go
index ea1fa02..32f440f 100644
--- a/pkg/ui/views/game_moves.go
+++ b/pkg/ui/views/game_moves.go
@@ -35,10 +35,17 @@ func (m *GameModel) getMoves() tea.Cmd {
 
 	return func() tea.Msg {
 		move := <-m.incomingMoves
-		if multiplayer.MoveType(string(move.Type)) == multiplayer.AbandonGameMessage {
+
+		switch multiplayer.MoveType(string(move.Type)) {
+		case multiplayer.AbandonGameMessage:
 			return EndGameMsg{abandoned: true}
+		case multiplayer.RestoreGameMessage:
+			return SendRestoreMsg{}
+		case multiplayer.RestoreAckGameMessage:
+			return RestoreMoves(string(move.Payload))
+		default:
+			return ChessMoveMsg(string(move.Payload))
 		}
-		return ChessMoveMsg(string(move.Payload))
 	}
 }