From 7e2c841b59e7a3893e7850bca349b8a34906e32d Mon Sep 17 00:00:00 2001
From: Dan Barber <github@danbarber.me>
Date: Mon, 12 Jul 2021 12:29:41 -0500
Subject: [PATCH] Correctly choose pieces

---
 lib/chess_web/templates/game/show.html.eex |  1 +
 lib/chess_web/views/live/board_live.ex     | 17 +++++++++++++----
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/lib/chess_web/templates/game/show.html.eex b/lib/chess_web/templates/game/show.html.eex
index 2a9b0b4..560399e 100644
--- a/lib/chess_web/templates/game/show.html.eex
+++ b/lib/chess_web/templates/game/show.html.eex
@@ -12,6 +12,7 @@
         <img class="game-info__opponent-status" src="/images/eye-closed.svg" alt="offline">
       </p>
     </div>
+
     <div class="move-list">
       <table class="table table--condensed">
         <thead>
diff --git a/lib/chess_web/views/live/board_live.ex b/lib/chess_web/views/live/board_live.ex
index 5636895..ff7612d 100644
--- a/lib/chess_web/views/live/board_live.ex
+++ b/lib/chess_web/views/live/board_live.ex
@@ -6,12 +6,15 @@ defmodule ChessWeb.BoardLive do
   alias Chess.Repo
   alias Chess.Board
 
+  import Chess.Auth, only: [get_user!: 1]
+
   def render(assigns) do
     Phoenix.View.render(ChessWeb.GameView, "board.html", assigns)
   end
 
   def mount(_params, %{"user_id" => user_id, "game_id" => game_id}, socket) do
     user = Repo.get!(User, user_id)
+
     game =
       Game.for_user(user)
       |> Repo.get!(game_id)
@@ -24,19 +27,25 @@ defmodule ChessWeb.BoardLive do
   end
 
   defp handle_click(socket, file, rank) do
-    board = socket.assigns[:game].board
+    game = socket.assigns[:game]
+    board = game.board
+    user = socket.assigns[:user]
+
+    colour = ChessWeb.GameView.player_colour(user, game)
 
     assigns =
       case socket.assigns do
         %{:selected => nil} ->
           case Board.piece(board, {file, rank}) do
-            %{"colour" => "white"} ->
+            %{"colour" => ^colour} ->
               [{:selected, selected(file, rank)}]
+
             _ ->
               []
           end
+
         _ ->
-          []
+          [{:selected, nil}]
       end
 
     assign(socket, assigns)
@@ -45,7 +54,7 @@ defmodule ChessWeb.BoardLive do
   defp selected(file, rank) do
     {
       String.to_integer(file),
-      String.to_integer(rank),
+      String.to_integer(rank)
     }
   end
 end