From 4ca3b9d56cbfb42acbae85e959a521edb04566f9 Mon Sep 17 00:00:00 2001
From: HF <hf@pixelplanet.fun>
Date: Thu, 15 Sep 2022 00:58:26 +0200
Subject: [PATCH] make RpgEvent work on shards

---
 src/core/RpgEvent.js        | 65 ++++++++++++++++++++++++++-----------
 src/core/draw.js            |  9 ++---
 src/server.js               |  4 +--
 src/socket/MessageBroker.js |  8 +++++
 src/socket/SockEvents.js    |  9 +++++
 5 files changed, 69 insertions(+), 26 deletions(-)

diff --git a/src/core/RpgEvent.js b/src/core/RpgEvent.js
index d0d1c97..7d1c78a 100644
--- a/src/core/RpgEvent.js
+++ b/src/core/RpgEvent.js
@@ -14,11 +14,11 @@ import {
   clearOldEvent,
   CANVAS_ID,
 } from '../data/redis/Event';
-import { setCoolDownFactor } from './draw';
 import Void from './Void';
 import { protectCanvasArea } from './Image';
 import { setPixelByOffset } from './setPixel';
 import { TILE_SIZE, EVENT_USER_NAME } from './constants';
+import socketEvents from '../socket/socketEvents';
 import chatProvider from './ChatProvider';
 import canvases from './canvases';
 
@@ -77,15 +77,16 @@ class RpgEvent {
   chatTimeout; // number
 
   constructor() {
-    this.eventState = -1;
-    this.eventCenterC = null;
-    this.void = null;
-    this.chatTimeout = 0;
     this.runEventLoop = this.runEventLoop.bind(this);
   }
 
-  setSuccess(success, save = true) {
+  setSuccess(success) {
     this.success = success;
+    setSuccess(success);
+    RpgEvent.setCoolDownFactorFromSuccess(success);
+  }
+
+  static setCoolDownFactorFromSuccess(success) {
     let fac = 1;
     switch (success) {
       case 1:
@@ -97,24 +98,31 @@ class RpgEvent {
       default:
         // nothing
     }
-    if (save) {
-      setSuccess(success);
-    }
-    setCoolDownFactor(fac);
+    socketEvents.setCoolDownFactor(fac);
   }
 
   async initialize() {
+    this.eventState = -1;
+    this.eventCenterC = null;
+    this.void = null;
+    this.chatTimeout = 0;
     const success = await getSuccess();
-    this.setSuccess(success, false);
-    let eventTimestamp = await nextEvent();
-    if (!eventTimestamp) {
-      eventTimestamp = await RpgEvent.setNextEvent();
+    this.success = success;
+    RpgEvent.setCoolDownFactorFromSuccess(success);
+    if (socketEvents.amIImportant()) {
+      let eventTimestamp = await nextEvent();
+      if (!eventTimestamp) {
+        eventTimestamp = await RpgEvent.setNextEvent();
+        await this.calcEventCenter();
+        const [x, y, w, h] = this.eventArea;
+        await protectCanvasArea(CANVAS_ID, x, y, w, h, true);
+      }
+      this.eventTimestamp = eventTimestamp;
       await this.calcEventCenter();
-      const [x, y, w, h] = this.eventArea;
-      await protectCanvasArea(CANVAS_ID, x, y, w, h, true);
+      logger.info('initialized Event');
+    } else {
+      logger.info('Loaded Event cooldown factor, but i am not running it');
     }
-    this.eventTimestamp = eventTimestamp;
-    await this.calcEventCenter();
     this.runEventLoop();
   }
 
@@ -181,6 +189,25 @@ class RpgEvent {
   }
 
   async runEventLoop() {
+    /*
+     * if we aren't the main shard, we just wait and regularly check,
+     * re-intilializing if we become it
+     */
+    if (!socketEvents.amIImportant()) {
+      this.iAmNotImportant = true;
+      if (this.void) {
+        this.void.cancel();
+        this.void = null;
+      }
+      setTimeout(this.runEventLoop, 180000);
+      return;
+    }
+    if (this.iAmNotImportant) {
+      this.iAmNotImportant = false;
+      this.initialize();
+      return;
+    }
+
     const {
       eventState,
     } = this;
@@ -338,7 +365,7 @@ class RpgEvent {
         // 32min after last Event
         // clear old event area
         // reset success state
-        logger.info('Restoring old event area');
+        logger.info('Restoring old Event area');
         await clearOldEvent();
         if (this.success === 1) {
           RpgEvent.broadcastChatMessage(
diff --git a/src/core/draw.js b/src/core/draw.js
index d0aa44b..b7ff29a 100644
--- a/src/core/draw.js
+++ b/src/core/draw.js
@@ -1,5 +1,5 @@
 /*
- * draw pixel on canvas
+ * draw pixel on canvas by user
  */
 
 import {
@@ -8,6 +8,7 @@ import {
 import logger, { pixelLogger } from './logger';
 import RedisCanvas from '../data/redis/RedisCanvas';
 import allowPlace from '../data/redis/cooldown';
+import socketEvents from '../socket/socketEvents';
 import {
   setPixelByOffset,
   setPixelByCoords,
@@ -18,9 +19,9 @@ import canvases from './canvases';
 import { THREE_CANVAS_HEIGHT, THREE_TILE_SIZE, TILE_SIZE } from './constants';
 
 let coolDownFactor = 1;
-export function setCoolDownFactor(fac) {
-  coolDownFactor = fac;
-}
+socketEvents.on('setCoolDownFactor', (newFac) => {
+  coolDownFactor = newFac;
+});
 
 /*
  * IPs who are currently requesting pixels
diff --git a/src/server.js b/src/server.js
index 1600945..24e8172 100644
--- a/src/server.js
+++ b/src/server.js
@@ -119,9 +119,7 @@ sequelize.sync({ alter: { drop: false } })
       logger.info('I am the main shard');
     }
     rankings.initialize();
-    if (HOURLY_EVENT && !SHARD_NAME) {
-      // TODO make it wok in a cluster
-      logger.info('Initializing RpgEvent');
+    if (HOURLY_EVENT) {
       rpgEvent.initialize();
     }
   });
diff --git a/src/socket/MessageBroker.js b/src/socket/MessageBroker.js
index bdddb9f..00e8aa3 100644
--- a/src/socket/MessageBroker.js
+++ b/src/socket/MessageBroker.js
@@ -215,6 +215,14 @@ class MessageBroker extends SocketEvents {
     super.emit('chunkUpdate', canvasId, [i, j]);
   }
 
+  setCoolDownFactor(fac) {
+    if (this.amIImportant()) {
+      this.emit('setCoolDownFactor', fac);
+    } else {
+      super.emit('setCoolDownFactor', fac);
+    }
+  }
+
   broadcastChunkUpdate(
     canvasId,
     chunk,
diff --git a/src/socket/SockEvents.js b/src/socket/SockEvents.js
index 69ffc40..0c3ba13 100644
--- a/src/socket/SockEvents.js
+++ b/src/socket/SockEvents.js
@@ -103,6 +103,15 @@ class SocketEvents extends EventEmitter {
     this.emit('recvChatMessage', user, message, channelId);
   }
 
+  /*
+   * set cooldownfactor
+   * (used by RpgEvent)
+   * @param fac factor by which cooldown changes globally
+   */
+  setCoolDownFactor(fac) {
+    this.emit('setCoolDownFactor', fac);
+  }
+
   /*
    * broadcast chat message to all users in channel
    * @param name chatname