ctucx.git: ctucxbot

reset daily stats on daychange

diff --git a/src/cmd/stats.nim b/src/cmd/stats.nim
@@ -1,4 +1,4 @@
-import asyncdispatch, telebot, strutils, options, os, json, algorithm
+import asyncdispatch, telebot, strutils, options, os, json, algorithm, times
 
 type User = tuple
   username: string

@@ -46,6 +46,7 @@ proc statsHandler(message: Message) {.async.} =
   let filePath = getCurrentDir() & "/data/stats/" & $(message.chat.id) & ".json"
 
   var data = %* {
+    "lastReset": getDateStr(now()),
     "messagesCount":{
       "today": 0,
       "total": 0

@@ -56,6 +57,11 @@ proc statsHandler(message: Message) {.async.} =
   if fileExists(filePath):
     data = parseFile(filePath)
 
+
+  if not data.hasKey("lastReset"):
+    data.add("lastReset", %getDateStr(now()))
+
+
   if not data["users"].hasKey($(user.id)):
     data["users"].add($(user.id), %* {
       "username": user.username,

@@ -64,11 +70,19 @@ proc statsHandler(message: Message) {.async.} =
     })
 
   if not user.username.isSome:
-    data["users"][$(user.id)]["username"] = newJString(user.firstName)
+    data["users"][$(user.id)]["username"] = %user.firstName
+  else:
+    data["users"][$(user.id)]["username"] = %user.username
 
   data["users"][$(user.id)]["todayMessagesCount"] = newJInt(data["users"][$(user.id)]["todayMessagesCount"].getInt()+1)
   data["users"][$(user.id)]["totalMessagesCount"] = newJInt(data["users"][$(user.id)]["totalMessagesCount"].getInt()+1)
   data["messagesCount"]["today"] = newJInt(data["messagesCount"]["today"].getInt()+1)
   data["messagesCount"]["total"] = newJInt(data["messagesCount"]["total"].getInt()+1)
 
+  if getDateStr(now()) > data["lastReset"].getStr:
+    data["messagesCount"]["today"] = %0
+    
+    for id in keys(data["users"]):
+      data["users"][id]["todayMessagesCount"] = %0
+
   writeFile(filePath, $data)