From f22f8f772a4c91337caef195a76026e9a78eb344 Mon Sep 17 00:00:00 2001
From: snowy <snowykami@outlook.com>
Date: Sat, 11 May 2024 00:34:11 +0800
Subject: [PATCH] :sparkles: Add sign status for Lagrange.Core

---
 liteyuki/plugins/sign_status.py                 | 17 ++++++++++++++---
 .../lagrange_sign/templates/js/sign_status.js   | 16 ++++++++++++++--
 2 files changed, 28 insertions(+), 5 deletions(-)

diff --git a/liteyuki/plugins/sign_status.py b/liteyuki/plugins/sign_status.py
index 8082447..8f77e7b 100644
--- a/liteyuki/plugins/sign_status.py
+++ b/liteyuki/plugins/sign_status.py
@@ -47,7 +47,7 @@ sign_status = on_alconna(Alconna(
     "sign",
     Subcommand(
         "chart",
-        Args["limit", int, 60]
+        Args["limit", int, 10000]
     ),
     Subcommand(
         "count"
@@ -57,6 +57,8 @@ sign_status = on_alconna(Alconna(
     )
 ))
 
+cache_img: bytes = None
+
 
 @sign_status.assign("count")
 async def _():
@@ -85,19 +87,26 @@ async def _():
 
 @sign_status.assign("chart")
 async def _(arp: CommandResult = AlconnaResult()):
-    limit = arp.result.main_args.get("limit", 60)
+    limit = arp.result.subcommands.get("chart").args.get("limit")
+    if limit == 10000:
+        if cache_img:
+            await sign_status.send(UniMessage.image(raw=cache_img))
+            return
     img = await generate_chart(limit)
     await sign_status.send(UniMessage.image(raw=img))
 
 
 @scheduler.scheduled_job("interval", seconds=SIGN_COUNT_DURATION, next_run_time=datetime.datetime.now())
 async def update_sign_count():
+    global cache_img
     if not SIGN_COUNT_URLS:
         return
     data = await get_now_sign()
     for name, count in data.items():
         await save_sign_count(count[0], count[1], SIGN_COUNT_URLS[name])
 
+    cache_img = await generate_chart(10000)
+
 
 async def get_now_sign() -> dict[str, tuple[float, int]]:
     """
@@ -129,7 +138,8 @@ async def save_sign_count(timestamp: float, count: int, sid: str):
 async def generate_chart(limit):
     data = []
     for name, url in SIGN_COUNT_URLS.items():
-        count_rows = sign_db.all(SignCount(), "sid = ? LIMIT ?", url, limit)
+        count_rows = sign_db.all(SignCount(), "sid = ? ORDER BY id DESC LIMIT ?", url, limit)
+        count_rows.reverse()
         data.append(
             {
                     "name"  : name,
@@ -138,6 +148,7 @@ async def generate_chart(limit):
                     "counts": [row.count for row in count_rows]
             }
         )
+        print(len(count_rows))
 
     img = await template2image(
         template=get_path("templates/sign_status.html", debug=True),
diff --git a/liteyuki/resources/lagrange_sign/templates/js/sign_status.js b/liteyuki/resources/lagrange_sign/templates/js/sign_status.js
index ae1bbd1..2452ce2 100644
--- a/liteyuki/resources/lagrange_sign/templates/js/sign_status.js
+++ b/liteyuki/resources/lagrange_sign/templates/js/sign_status.js
@@ -24,7 +24,7 @@ data.forEach((item) => {
             },
             xAxis: {
                 type: 'category',
-                data: item["times"],
+                data: item["times"].map(timestampToTime),
             },
             yAxis: {
                 type: 'value',
@@ -38,4 +38,16 @@ data.forEach((item) => {
             ]
         }
     )
-})
\ No newline at end of file
+})
+
+
+function timestampToTime(timestamp) {
+    let date = new Date(timestamp * 1000)
+    let Y = date.getFullYear() + '-'
+    let M = (date.getMonth() + 1 < 10 ? '0' + (date.getMonth() + 1) : date.getMonth() + 1) + '-'
+    let D = date.getDate() + ' '
+    let h = date.getHours() + ':'
+    let m = date.getMinutes() + ':'
+    let s = date.getSeconds()
+    return M + D + h + m + s
+}
\ No newline at end of file