Test

main.py

@@ -233,6 +233,112 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
 #################### 文件操作 #############################
 
 
+async def tg_show_files(update: Update, context: CallbackContext):
+    if "file_id" not in context.user_data:
+        context.user_data["file_id"] = ""
+    files = await THUNDERX_CLIENT.file_list(
+        100, context.user_data["file_id"], "", {}, {}
+    )
+    keyboard = []
+
+    if files["files"] is None:
+        await update.message.reply_text("❌未找到文件!!")
+    else:
+        # 为每个文件创建按钮和操作选项
+        for file in files["files"]:
+            if file["kind"].lower() == "drive#folder":
+                keyboard.append(
+                    [
+                        InlineKeyboardButton(
+                            f"查看📁: {file['name']}",
+                            callback_data=f"list_file:{file['id']}",
+                        ),
+                        InlineKeyboardButton(
+                            f"删除",
+                            callback_data=f"delete_file:{file['id']}",
+                        ),
+                    ]
+                )
+            else:
+                keyboard.append(
+                    [
+                        InlineKeyboardButton(
+                            f"下载📄: {file['name']}",
+                            callback_data=f"download_file:{file['id']}",
+                        ),
+                        InlineKeyboardButton(
+                            f"删除",
+                            callback_data=f"delete_file:{file['id']}",
+                        ),
+                    ]
+                )
+
+        reply_markup = InlineKeyboardMarkup(keyboard)
+        await update.message.reply_text(f"📋文件列表:", reply_markup=reply_markup)
+
+
+async def handle_file_confirmation(update: Update, context: CallbackContext):
+    query = update.callback_query
+    await query.answer()
+
+    # 获取确认操作的类型和文件 ID
+    action, file_id = query.data.split(":")[0], query.data.split(":")[1]
+
+    if action == "confirm_file_delete_file":
+        await THUNDERX_CLIENT.delete_forever([file_id])
+        await query.edit_message_text(f"✅文件 {file_id} 已删除。")
+
+
+async def handle_file_cancel(update: Update, context: CallbackContext):
+    query = update.callback_query
+    await query.answer()
+    # 获取取消操作的类型和文件 ID
+    action, file_id = query.data.split(":")[0], query.data.split(":")[1]
+    # 返回文件夹列表
+    await query.edit_message_text(f"操作已取消")
+
+
+# 处理任务操作的回调
+async def handle_file_operation(update: Update, context: CallbackContext):
+    query = update.callback_query
+    await query.answer()
+
+    # 获取操作类型和文件 ID
+    action, file_id = query.data.split(":")
+
+    # 需要确认的操作
+    if action in ["delete_file"]:
+        # 生成确认消息
+        keyboard = [
+            [
+                InlineKeyboardButton(
+                    "确认", callback_data=f"confirm_file_{action}:{file_id}"
+                )
+            ],
+            [
+                InlineKeyboardButton(
+                    "取消", callback_data=f"cancel_file_{action}:{file_id}"
+                )
+            ],
+        ]
+        reply_markup = InlineKeyboardMarkup(keyboard)
+        await query.edit_message_text(
+            f"你确定要{action}文件 {file_id} 吗？", reply_markup=reply_markup
+        )
+    else:
+        # 不需要确认的操作，直接处理
+        await perform_file_action(update, context, action, file_id)
+
+
+async def perform_file_action(
+    update: Update, context: CallbackContext, action: str, file_id: str
+):
+    if action == "list_file":
+        context.user_data["file_id"] = file_id
+        await tg_show_files(update, context)
+        # await update.callback_query.edit_message_text(f"你选择了取消任务：{file_id}")
+
+
 #################### 离线任务处理 ##########################
 # 确认操作的回调
 async def handle_task_confirmation(update: Update, context: CallbackContext):
@@ -427,11 +533,27 @@ async def init_client():
         TG_BOT_APPLICATION.add_handler(
             CallbackQueryHandler(handle_task_confirmation, pattern="^confirm_task")
         )
+
+        ########## 文件操作 ###############
+
+        TG_BOT_APPLICATION.add_handler(
+            CallbackQueryHandler(handle_file_operation, pattern="^delete_file:")
+        )
+        # 处理取消任务操作
+        TG_BOT_APPLICATION.add_handler(
+            CallbackQueryHandler(handle_file_cancel, pattern="^cancel_file")
+        )
+        # 处理确认操作（确认删除、复制等）
+        TG_BOT_APPLICATION.add_handler(
+            CallbackQueryHandler(handle_file_confirmation, pattern="^confirm_file")
+        )
+
         TG_BOT_APPLICATION.add_handler(CommandHandler("start", start))
         TG_BOT_APPLICATION.add_handler(CommandHandler("help", help))
         TG_BOT_APPLICATION.add_handler(CommandHandler("quota", quota))
         TG_BOT_APPLICATION.add_handler(CommandHandler("emptytrash", tg_emptytrash))
         TG_BOT_APPLICATION.add_handler(CommandHandler("tasks", tg_show_task))
+        TG_BOT_APPLICATION.add_handler(CommandHandler("files", tg_show_files))
         # Message 消息处理相关的命令！
         TG_BOT_APPLICATION.add_handler(MessageHandler(filters.TEXT, handle_message))