From 9fb130c944d20e74bf3281601bc2b1a9cfb5432f Mon Sep 17 00:00:00 2001
From: Tissevert <tissevert+devel@marvid.fr>
Date: Tue, 27 Aug 2019 21:57:26 +0200
Subject: [PATCH] Fix bug in end of game handling

---
 src/Automaton.hs | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/Automaton.hs b/src/Automaton.hs
index 3b71a01..721cb8d 100644
--- a/src/Automaton.hs
+++ b/src/Automaton.hs
@@ -78,12 +78,12 @@ answer (Message.Relay {Message.message = Message.Invitation {}}) state = do
   return state
 
 answer (Message.Game {Message.game, Message.logs}) state@(Playing {key, name}) = do
-  if playing game == key
-  then do
-    case step game of
-      Over -> send Message.Quit >> return (LoggedIn {key, name})
-      _ -> send (Message.Play {Message.move = AI.move key game}) >> return state
-  else return state
+  case step game of
+    Over -> send Message.Quit >> return (LoggedIn {key, name})
+    _ -> do
+      if playing game == key
+      then send (Message.Play {Message.move = AI.move key game}) >> return state
+      else return state
 
 answer (Message.Relay {Message.from, Message.message = Message.LogOut}) (Playing {key, name, against})
   | from == against = send Message.Quit >> return (LoggedIn {key, name})