From 22efa391c3f49ae7b91cd86ccf11da3af74523fe Mon Sep 17 00:00:00 2001
From: Benjamin Auder <benjamin.auder@somewhere>
Date: Mon, 2 Dec 2019 16:15:07 +0100
Subject: [PATCH] Fix clocks update

---
 client/src/utils/gameStorage.js | 17 +++++++----------
 client/src/views/Game.vue       | 25 +++++++++++++++++--------
 2 files changed, 24 insertions(+), 18 deletions(-)

diff --git a/client/src/utils/gameStorage.js b/client/src/utils/gameStorage.js
index a8aa6edf..2ccb494f 100644
--- a/client/src/utils/gameStorage.js
+++ b/client/src/utils/gameStorage.js
@@ -68,21 +68,18 @@ export const GameStorage =
 
   // TODO: also option to takeback a move ?
   // NOTE: for live games only (all on server for corr)
-  update: function(gameId, obj) //colorIdx, nextIdx, move, fen, addTime, score
+  update: function(gameId, obj) //move, fen, clocks, score, initime, ...
   {
     dbOperation((db) => {
       let objectStore = db.transaction("games", "readwrite").objectStore("games");
       objectStore.get(gameId).onsuccess = function(event) {
         const game = event.target.result;
-        if (!!obj.move)
-        {
-          game.moves.push(obj.move);
-          game.fen = obj.fen;
-          game.clocks[obj.colorIdx] += obj.addTime;
-          game.initime[obj.nextIdx] = Date.now();
-        }
-        if (!!obj.score)
-          game.score = obj.score;
+        Object.keys(obj).forEach(k => {
+          if (k == "move")
+            game.moves.push(obj[k]);
+          else
+            game[k] = obj[k];
+        });
         objectStore.put(game); //save updated data
       }
     });
diff --git a/client/src/views/Game.vue b/client/src/views/Game.vue
index 52e97c53..2fdd2238 100644
--- a/client/src/views/Game.vue
+++ b/client/src/views/Game.vue
@@ -310,17 +310,26 @@ export default {
           //also using move.played fields
           game.clocks = [-1, -1];
           game.initime = [0, 0];
+          // TODO: compute clocks + initime
         }
         const tc = extractTime(game.timeControl);
+        const myIdx = game.players.findIndex(p => p.sid == this.st.user.sid);
         if (game.clocks[0] < 0) //game unstarted
         {
           game.clocks = [tc.mainTime, tc.mainTime];
           game.initime[0] = Date.now();
+          if (myIdx >= 0) //I play in this game
+          {
+            GameStorage.update(game.gameId,
+            {
+              clocks: game.clocks,
+              initime: game.initime,
+            });
+          }
         }
         const vModule = await import("@/variants/" + vname + ".js");
         window.V = vModule.VariantRules;
         this.vr = new V(game.fen);
-        const myIdx = game.players.findIndex(p => p.sid == this.st.user.sid);
         this.game = Object.assign({},
           game,
           // NOTE: assign mycolor here, since BaseGame could also bs VS computer
@@ -388,11 +397,14 @@ export default {
       const nextIdx = ["w","b"].indexOf(this.vr.turn);
       GameStorage.update(this.gameRef.id,
       {
-        colorIdx: colorIdx,
-        nextIdx: nextIdx,
         move: filtered_move,
         fen: move.fen,
-        addTime: addTime,
+        clocks: this.game.clocks.map((t,i) => i==colorIdx
+          ? this.game.clocks[i] + addTime
+          : this.game.clocks[i]),
+        initime: this.game.initime.map((t,i) => i==nextIdx
+          ? Date.now()
+          : this.game.initime[i]),
       });
       // Also update current game object:
       this.game.moves.push(move);
@@ -404,10 +416,7 @@ export default {
     // TODO: this update function should also work for corr games
     gameOver: function(score) {
       this.game.mode = "analyze";
-      GameStorage.update(this.gameRef.id,
-      {
-        score: score,
-      });
+      GameStorage.update(this.gameRef.id, { score: score });
     },
   },
 };
-- 
2.44.0