From 688641d74975ef0f0cf02cad737ae8b8447de7c4 Mon Sep 17 00:00:00 2001
From: Stefan Rupp <struppi@struppi.name>
Date: Thu, 21 May 2020 18:49:52 +0200
Subject: [PATCH] implement undo function

---
 templates/home/control.html |  2 ++
 views/control.py            | 31 +++++++++++++++++++++++++++++++
 wiz_game.py                 |  4 ++++
 3 files changed, 37 insertions(+)

diff --git a/templates/home/control.html b/templates/home/control.html
index f1d56b3..78c9cbf 100644
--- a/templates/home/control.html
+++ b/templates/home/control.html
@@ -26,6 +26,8 @@
                 Winner: {{trick_winner}} with Card {{highest_card}}
                 <a href="/control/next_hand/">start next hand</a>
                 {% endif %}
+                <br>
+                <a href="/control/undo_last/">undo last card</a>
 <!--
             <ul>
                 <li>
diff --git a/views/control.py b/views/control.py
index 10ef8db..7bd7ae1 100644
--- a/views/control.py
+++ b/views/control.py
@@ -51,3 +51,34 @@ async def start_next_trick(req, resp):
             pass
     api.redirect(resp, '/control/', status_code=303)
 
+
+@api.route("/control/undo_last/")
+async def undo_last(req, resp):
+    hand_finished = the_game.is_hand_finished()
+    if the_game.played_cards:
+        if not hand_finished:
+            p, c = the_game.played_cards.pop()
+            the_game.active_player -= 1
+            p.add_card(c)
+            for ws in the_game.websockets:
+                try:
+                    await ws.send_text("reload")
+                except Exception as e:
+                    pass
+    else:
+        if the_game.num_tricks_played > 0:
+            the_game.num_tricks_played -= 1;
+            the_game.played_cards = the_game.last_trick;
+            the_game.active_player = len(the_game.players) - 1
+            player, _ = the_game.get_trick_winner()
+            player.tricks_taken -= 1
+            the_game.players_ordered = [p for p, _ in the_game.played_cards]
+            p, c = the_game.played_cards.pop()
+            p.add_card(c)
+            for ws in the_game.websockets:
+                try:
+                    await ws.send_text("reload")
+                except Exception as e:
+                    pass
+
+    api.redirect(resp, '/control/', status_code=303)
diff --git a/wiz_game.py b/wiz_game.py
index 8b83e05..91411e3 100644
--- a/wiz_game.py
+++ b/wiz_game.py
@@ -73,6 +73,10 @@ class Player:
     def take_trick(self):
         self.tricks_taken += 1
 
+    def add_card(self, card):
+        self.cards.append(card)
+        self.cards = sorted(self.cards, key=attrgetter('color', 'value'))
+
 
 class WizGame:
     def __init__(self, player_names: list):