添加查看存储使用信息

main.py

@@ -94,6 +94,24 @@ async def verify_token(
         )
 
 
+def format_bytes(size: int) -> str:
+    # 预设单位
+    units = ["B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB"]
+
+    # 确保字节数是正数
+    if size < 0:
+        raise ValueError("字节大小不能为负数")
+
+    # 选择合适的单位
+    unit_index = 0
+    while size >= 1024 and unit_index < len(units) - 1:
+        size /= 1024.0
+        unit_index += 1
+
+    # 格式化输出，保留两位小数
+    return f"{size:.2f} {units[unit_index]}"
+
+
 app = FastAPI()
 
 
@@ -126,13 +144,55 @@ TG_BASE_URL = "https://tg.alist.dpdns.org/bot"
 # 定义命令处理函数
 async def start(update: Update, context):
     commands = (
-        "欢迎使用我的机器人！\n\n" "以下是您可以使用的命令：\n" "/help - 获取帮助信息\n"
+        "欢迎使用我的机器人！\n\n"
+        "以下是您可以使用的命令：\n"
+        "/tasks - 查看下载任务\n"
+        "/files - 查看文件列表\n"
+        "/quota - 查看存储空间\n"
+        "/emptytrash - 清空回收站\n"
+        "/help - 获取帮助信息\n"
     )
     await update.message.reply_text(commands)
 
 
 async def help(update: Update, context):
-    await update.message.reply_text("这是一个帮助信息！")
+    commands = (
+        "欢迎使用我的机器人！\n\n"
+        "以下是您可以使用的命令：\n"
+        "/tasks - 查看下载任务\n"
+        "/files - 查看文件列表\n"
+        "/quota - 查看存储空间\n"
+        "/emptytrash - 清空回收站\n"
+        "/help - 获取帮助信息\n"
+    )
+    await update.message.reply_text(commands)
+
+
+async def quota(update: Update, context):
+    """
+    返回信息
+    {
+      "kind": "drive#about",
+      "quota": {
+        "kind": "drive#quota",
+        "limit": "72057604737418240",
+        "usage": "18700975438",
+        "usage_in_trash": "0",
+        "play_times_limit": "2",
+        "play_times_usage": "0",
+        "is_unlimited": true
+      },
+      "expires_at": "2026-04-08T21:47:59.000+08:00",
+      "quotas": {}
+    }
+    """
+    quota_info = await THUNDERX_CLIENT.get_quota_info()
+    if quota_info["usage"] is None:
+        await update.message.reply_text("未找到使用信息，请稍后再试！")
+    else:
+        await update.message.reply_text(
+            f"使用信息:\n{format_bytes(quota_info['limit'])}/{format_bytes(quota_info['usage'])}\n到期时间:\n{quota_info['expires_at']}"
+        )
 
 
 @app.on_event("startup")
@@ -177,6 +237,7 @@ async def init_client():
         # 将命令处理函数添加到 dispatcher
         TG_BOT_APPLICATION.add_handler(CommandHandler("start", start))
         TG_BOT_APPLICATION.add_handler(CommandHandler("help", help))
+        TG_BOT_APPLICATION.add_handler(CommandHandler("quota", quota))
 
 
 # FastAPI 路由：接收来自 Telegram 的 Webhook 回调