From 79e5ba799b104c8bc5e0738b231e88ee602190d0 Mon Sep 17 00:00:00 2001
From: SowinskiBraeden <sowinskibraeden@gmail.com>
Date: Sun, 17 Nov 2024 19:24:26 -0800
Subject: [PATCH] fix everything

---
 cogs/event.py           |   5 ++
 handlers/logs.py        | 162 +++++++++++++++++++++++-----------------
 src/ReforgerStatsBot.py |  20 ++---
 3 files changed, 110 insertions(+), 77 deletions(-)

diff --git a/cogs/event.py b/cogs/event.py
index cd23474..0b7aee7 100644
--- a/cogs/event.py
+++ b/cogs/event.py
@@ -37,6 +37,11 @@ class Event(app_commands.Group):
     self.bot.gamertags  = []
     self.bot.message_id = ""
     
+    self.sftp.close()
+    self.ssh.close()
+    self.sftp = None
+    self.ssh  = None
+    
     await interaction.response.send_message("Stopping event leaderboard...", ephemeral=True)
 
 async def setup(bot: commands.Bot) -> None:
diff --git a/handlers/logs.py b/handlers/logs.py
index 36cb37c..8f25f17 100644
--- a/handlers/logs.py
+++ b/handlers/logs.py
@@ -3,6 +3,7 @@ from re import match
 from paramiko import SFTPClient
 from datetime import datetime
 from typing import Tuple
+from discord.ext import commands
 
 # Regex templates
 class Templates:
@@ -13,74 +14,23 @@ class Templates:
   Admin:       str = r'(?P<time>.*)   NETWORK      : Player \'(?P<gamertag>.*)\' signed in as server admin.'
   Kill:        str = r'(?P<time>.*)   SCRIPT       : ServerAdminTools \| Event serveradmintools_player_killed \| player: (?P<victim_gamertag>.*), instigator: (?P<killer_gamertag>.*), friendly: (?P<friendly>.*)'
 
-# Function to process new logs in an array from a given starting index
-def readLogFromIndex(
-  startingIndex: int,
-  logs:          list[str],
-  identities: list[dict],
-  gamertags: list[str]
-) -> Tuple[list[dict], list[str]]: # Return updated identities
-    
-  # Check each line for data to extract and load into json
-  for i in range(startingIndex, len(logs)):
-    gt_data       = match(Templates.Gamertag, logs[i])
-    identity_data = match(Templates.PlayerID, logs[i])
-    admin_data    = match(Templates.Admin, logs[i])
-    kill_data     = match(Templates.Kill, logs[i])
-
-    # Gather in game player GUID    
-    if identity_data is not None:
-      for identity in identities:
-        if identity["gamertag"] == identity_data.group("gamertag"):
-          identities[identities.index(identity)]["player_guid"] = identity_data.group("player_guid")
-      continue
-
-    # Has player logged into admin/gamemaster
-    if admin_data is not None:
-      for identity in identities:
-        if identity["gamertag"] == admin_data.group("gamertag"):
-          identities[identities.index(identity)]["admin"] = True  
-      continue
-
-    # Contribute to playerts KDR
-    if kill_data is not None:
-      if kill_data.group("friendly") == "true": continue # Ignore friendly kills
-      if kill_data.group("killer_gamertag") == kill_data.group("victim_gamertag"): continue # Ignore suicide
-      for identity in identities:
-        if identity["gamertag"] == kill_data.group("killer_gamertag"):
-          index = identities.index(identity)
-          
-          identities[index]["kills"]          += 1
-          identities[index]["killstreak"]     += 1
-          identities[index]["deathstreak"]    = 0
-          identities[index]["bestKillstreak"] = identities[index]["killstreak"] if identities[index]["killstreak"] > identities[index]["bestKillstreak"] else identities[index]["bestKillstreak"]
-          identities[index]["KDR"]            = round(identities[index]["kills"] / (1 if identities[index]["deaths"] == 0 else identities[index]["deaths"]), 2)
-          pass
-        if identity["gamertag"] == kill_data.group("victim_gamertag"):
-          index = identities.index(identity)
-
-          identities[index]["deaths"] += 1
-          identities[index]["deathstreak"] += 1
-          identities[index]["killstreak"] = 0
-          identities[index]["worstDeathstreak"] = identities[index]["deathstreak"] if identities[index]["deathstreak"] > identities[index]["worstDeathstreak"] else identities[index]["worstDeathstreak"]
-          identities[index]["KDR"] = round(identities[index]["kills"] / (1 if identities[index]["deaths"] == 0 else identities[index]["deaths"]), 2)
-          pass
-
-      continue
-
+# Get initial players data
+def getPlayers(bot: commands.Bot, logs: list[str]) -> None:
+  gt_data = match(Templates.Gamertag, logs[i])
+  for i in range(0, len(logs)):
     # Generate new player data or update connections/IP info
     if gt_data is not None:
       ip_data = match(Templates.IpAddress, logs[i+1])
-      if gt_data.group("gamertag") in gamertags:
-        for identity in identities:
+      if gt_data.group("gamertag") in bot.gamertags:
+        for identity in bot.players:
           if identity["gamertag"] == gt_data.group("gamertag"):
-            index = identities.index(identity)
-            identities[index]["connections"] += 1 
-            identities[index]["ip"]          = ip_data.group("ip_addr")
-            identities[index]["port"]        = ip_data.group("port")
+            index = bot.players.index(identity)
+            bot.players[index]["connections"] += 1 
+            bot.players[index]["ip"]          = ip_data.group("ip_addr")
+            bot.players[index]["port"]        = ip_data.group("port")
       else:
         be_data = match(Templates.BattlEyeGUI, logs[i+3])
-        identities.append({
+        bot.players.append({
           "gamertag":         gt_data.group("gamertag"),
           "ip":               ip_data.group("ip_addr"),
           "port":             ip_data.group("port"),
@@ -96,10 +46,91 @@ def readLogFromIndex(
           "worstDeathstreak": 0,
           "admin":            False
         })
-        gamertags.append(gt_data.group("gamertag"))
+        bot.gamertags.append(gt_data.group("gamertag"))
+
+# Function to process new logs in an array from a given starting index
+def readLogFromIndex(
+  bot: commands.Bot,
+  logs: list[str]
+) -> None:
+
+  # Check each line for data to extract and load into json
+  for i in range(bot.log_index, len(logs)):
+    gt_data       = match(Templates.Gamertag, logs[i])
+    identity_data = match(Templates.PlayerID, logs[i])
+    admin_data    = match(Templates.Admin, logs[i])
+    kill_data     = match(Templates.Kill, logs[i])
+
+    # Gather in game player GUID    
+    if identity_data is not None:
+      for identity in bot.players:
+        if identity["gamertag"] == identity_data.group("gamertag"):
+          bot.players[bot.players.index(identity)]["player_guid"] = identity_data.group("player_guid")
       continue
 
-  return identities, gamertags
+    # Has player logged into admin/gamemaster
+    if admin_data is not None:
+      for identity in bot.players:
+        if identity["gamertag"] == admin_data.group("gamertag"):
+          bot.players[bot.players.index(identity)]["admin"] = True  
+      continue
+
+    # Contribute to playerts KDR
+    if kill_data is not None:
+      if kill_data.group("friendly") == "true": continue # Ignore friendly kills
+      if kill_data.group("killer_gamertag") == kill_data.group("victim_gamertag"): continue # Ignore suicide
+      for identity in bot.players:
+        if identity["gamertag"] == kill_data.group("killer_gamertag"):
+          index = bot.players.index(identity)          
+          bot.players[index]["kills"]          += 1
+          bot.players[index]["killstreak"]     += 1
+          bot.players[index]["deathstreak"]    = 0
+          bot.players[index]["bestKillstreak"] = bot.players[index]["killstreak"] if bot.players[index]["killstreak"] > bot.players[index]["bestKillstreak"] else bot.players[index]["bestKillstreak"]
+          bot.players[index]["KDR"]            = round(bot.players[index]["kills"] / (1 if bot.players[index]["deaths"] == 0 else bot.players[index]["deaths"]), 2)
+
+        if identity["gamertag"] == kill_data.group("victim_gamertag"):
+          index = bot.players.index(identity)
+          bot.players[index]["deaths"] += 1
+          bot.players[index]["deathstreak"] += 1
+          bot.players[index]["killstreak"] = 0
+          bot.players[index]["worstDeathstreak"] = bot.players[index]["deathstreak"] if bot.players[index]["deathstreak"] > bot.players[index]["worstDeathstreak"] else bot.players[index]["worstDeathstreak"]
+          bot.players[index]["KDR"] = round(bot.players[index]["kills"] / (1 if bot.players[index]["deaths"] == 0 else bot.players[index]["deaths"]), 2)
+
+      continue
+
+    # Generate new player data or update connections/IP info
+    if gt_data is not None:
+      ip_data = match(Templates.IpAddress, logs[i+1])
+      if gt_data.group("gamertag") in bot.gamertags:
+        for identity in bot.players:
+          if identity["gamertag"] == gt_data.group("gamertag"):
+            index = bot.players.index(identity)
+            bot.players[index]["connections"] += 1 
+            bot.players[index]["ip"]          = ip_data.group("ip_addr")
+            bot.players[index]["port"]        = ip_data.group("port")
+      else:
+        be_data = match(Templates.BattlEyeGUI, logs[i+3])
+        bot.players.append({
+          "gamertag":         gt_data.group("gamertag"),
+          "ip":               ip_data.group("ip_addr"),
+          "port":             ip_data.group("port"),
+          "player_guid":      None,
+          "battleye_guid":    be_data.group("battleye_guid"),
+          "connections":      1,
+          "kills":            0,
+          "deaths":           0,
+          "KDR":              0,
+          "killstreak":       0,
+          "bestKillstreak":   0,
+          "deathstreak":      0,
+          "worstDeathstreak": 0,
+          "admin":            False
+        })
+        bot.gamertags.append(gt_data.group("gamertag"))
+      continue
+
+  print("done reading logs...")
+  return
 
 def getLatestDir(sftp: SFTPClient, remote_path: str) -> str:
   log_directory:    str       = ""
@@ -117,6 +148,3 @@ def getLatestDir(sftp: SFTPClient, remote_path: str) -> str:
       log_directory = directory
 
   return log_directory
-
-def scrape(bot) -> None:
-  pass
diff --git a/src/ReforgerStatsBot.py b/src/ReforgerStatsBot.py
index a96e170..4daa15b 100644
--- a/src/ReforgerStatsBot.py
+++ b/src/ReforgerStatsBot.py
@@ -8,7 +8,7 @@ from os import listdir, getcwd
 import requests
 
 from handlers.sftp import createSSHClient
-from handlers.logs import getLatestDir, readLogFromIndex
+from handlers.logs import getLatestDir, readLogFromIndex, getPlayers
 from config import Config
 
 class ReforgerStats(commands.Bot):
@@ -27,7 +27,6 @@ class ReforgerStats(commands.Bot):
   async def on_ready(self: commands.Bot) -> None:
     print(f"Logged in as {self.user}")
     await self.load_cogs()
-    # self.scrape_file.start()
 
   async def load_cogs(self: commands.Bot):
     for filename in listdir(f"{getcwd()}/cogs"):
@@ -87,16 +86,17 @@ class ReforgerStats(commands.Bot):
     # Read data
     file:  SFTPFile  = self.sftp.open(f"{remote_path}/{self.log_dir}/console.log")
     lines: list[str] = file.readlines()
-    if self.log_index == -1: self.log_index = len(lines) - 1
-    self.players, self.gamertags = readLogFromIndex(self.log_index, lines, self.players, self.gamertags)
-    self.log_index = len(lines) - 1
     file.close()
+    
+    if self.log_index == -1:
+      self.log_index = len(lines) - 1
+      getPlayers(self, lines) # Get all players from this log file to add to self.players / self.gamertags
+      return
+
+    readLogFromIndex(self, lines)
+    self.log_index = len(lines) - 1
 
     # Send leaderboard embeds via webhook
     content: dict = await self.createLeaderboardEmbed(self.players)
-    if self.message_id == "": await self.webhookSend(content)
+    if self.message_id == "": self.message_id = await self.webhookSend(content)
     else: await self.webhookUpdateMessage(content)
-
-    # Close SSH && SFTP sessions and file
-    self.sftp.close()
-    self.ssh.close()