From 791e2350b64993ab499958c03753656a185a7b1d Mon Sep 17 00:00:00 2001
From: Ari Archer <ari.web.xyz@gmail.com>
Date: Sun, 17 Jul 2022 16:42:52 +0300
Subject: [PATCH] Add chat log dumping and autodumping

Signed-off-by: Ari Archer <ari.web.xyz@gmail.com>
---
 .gitignore  |  1 +
 src/main.py | 73 +++++++++++++++++++++++++++++++++++++++++++++--------
 2 files changed, 63 insertions(+), 11 deletions(-)

diff --git a/.gitignore b/.gitignore
index 5eaac78..30cdb84 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
 /config.json
 /venv
 .mypy_cache/
+/logs
diff --git a/src/main.py b/src/main.py
index b12fe8e..8c4b9a9 100644
--- a/src/main.py
+++ b/src/main.py
@@ -6,10 +6,11 @@ import asyncio
 import json
 import os
 import sys
-from datetime import datetime
+import time
+from datetime import datetime, timezone
 from html import unescape as html_unescape
 from secrets import SystemRandom
-from time import sleep
+from string import punctuation as special_characters
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 from uuid import uuid4
 from warnings import filterwarnings as filter_warnings
@@ -61,6 +62,8 @@ CONFIG: Dict[str, Any] = {
     "report-webhook-url": "",
     "authkey-webhook-url": "",
     "chatlog-limit": 500,
+    "logs-dir": "logs",
+    "autodump-chatlogs": True,
 }
 RANDOM: SystemRandom = SystemRandom()
 
@@ -223,6 +226,51 @@ def unguac_msg(msg: str) -> Optional[List[str]]:
     return result
 
 
+def dump_log(time: str) -> str:
+    if not os.path.exists(CONFIG["logs-dir"]):
+        log(f"Making {CONFIG['logs-dir']!r} directory")
+        os.mkdir(CONFIG["logs-dir"])
+
+    _log_file: str = os.path.join(
+        CONFIG["logs-dir"],
+        f"{''.join(c.replace(' ', '-') for c in time if c not in special_characters)}.log",
+    )
+
+    log(f"Dumping chatlog to {_log_file!r}")
+
+    with open(
+        _log_file,
+        "w",
+    ) as chatlog:
+        chatlog.write("\n".join(STATE["chatlog"]))
+
+    return chatlog.name
+
+
+def generate_time_str() -> str:
+    _utc: datetime = datetime.now(tz=timezone.utc)
+
+    return datetime.strftime(
+        _utc, f"{_utc.timestamp()} UNIX / %Y-%m-%d %H:%M:%S (%f microseconds) UTC"
+    )
+
+
+def chatlog_entry(message: str, user: str, header: Optional[str] = None) -> None:
+    _time: str = generate_time_str()
+
+    if len(STATE["chatlog"]) > CONFIG["chatlog-limit"]:
+        if CONFIG["autodump-chatlog"]:
+            dump_log(_time)
+
+        STATE["chatlog"].clear()
+
+    STATE["chatlog"].append(
+        f"\n{(str(header) + ' ') if header is not None else ''}\
+{user!r} @ {_time}: \
+{message}"
+    )
+
+
 class CommandParser:
     @staticmethod
     def cmd_hi(user: str, args: List[str]) -> Tuple[str]:
@@ -657,16 +705,19 @@ class CommandParser:
             ),
         )
 
+    @staticmethod
+    def cmd_dumplog(user: str, args: List[str]) -> Tuple[str]:
+        """Auth command, dumps current chatlog
+        Syntax: dumplog"""
 
-def chatlog_entry(message: str, user: str, header: Optional[str] = None) -> None:
-    if len(STATE["chatlog"]) > CONFIG["chatlog-limit"]:
-        STATE["chatlog"].clear()
+        _dumplog_filename: str = dump_log(generate_time_str())
 
-    STATE["chatlog"].append(
-        f"\n{(str(header) + ' ') if header is not None else ''}\
-{user!r} @ {datetime.strftime(datetime.utcnow(), '%Y-%m-%d %H:%M:%S (%f microseconds)')} UTC: \
-{message}"
-    )
+        return (
+            guac_msg(
+                "chat",
+                f"@{user} Dumped to {_dumplog_filename}",
+            ),
+        )
 
 
 class MessageParser:
@@ -925,6 +976,6 @@ if __name__ == "__main__":
 
         if STATE["run"]:
             log("Reconnecting after 30s")
-            sleep(30)
+            time.sleep(30)
 
     sys.exit(ret)