#! /usr/bin/env python3
# -*- coding: utf-8 -*-
import threading
from argparse import ArgumentParser
from configuration import Config
from constants import ChatType
from robot import Robot

from loguru import logger

# INFO 日志（包含 INFO、DEBUG，但不包含 WARNING、ERROR）
logger.add(
    f"wx_info.log",
    level="INFO",
    filter=lambda record: record["level"].name in ["INFO", "DEBUG"],
    rotation="10 MB",
    retention="7 days",
    encoding="utf-8"
)

# ERROR 日志（仅 ERROR 及以上）
logger.add(
    f"wx_error.log",
    level="ERROR",
    rotation="10 MB",
    retention="7 days",
    encoding="utf-8"
)
# ERROR 日志（仅 ERROR 及以上）
logger.add(
    f"wx_debug.log",
    level="DEBUG",
    rotation="10 MB",
    retention="7 days",
    encoding="utf-8"
)


def main(chat_type: int):
    config = Config()

    # 创建机器人实例
    robot = Robot(config)
    robot.LOG.info(f"WeChatRobot 正在启动...")

    # 初始化并启动wechat_ipad客户端
    if robot.init_wechat_ipad():
        robot.LOG.info("wechat_ipad客户端启动成功")
    else:
        robot.LOG.error("wechat_ipad客户端启动失败")
    # # 每天 8:30 发送新闻
    robot.onEveryTime("08:30", robot.news_baidu_report_auto)
    #
    # # epic
    robot.onEveryTime("10:30", robot.send_epic_free_games)
    #
    # # message report 1:数据自动从redis 转到sqllite
    robot.onEveryTime("02:30", robot.message_count_to_db)
    # # 从db中提取并发送给相关群
    robot.onEveryTime("09:30", robot.generate_and_send_ranking)
    #
    # # sehuatang
    # robot.onEveryTime("15:30", robot.generate_sehuatang_pdf)
    #
    # # 秀人网每天自动下载帖子
    robot.onEveryTime("01:30", robot.xiu_ren_download_task)
    #
    # # 秀人网每天自动发pdf
    # robot.onEveryTime("17:30", robot.xiu_ren_pdf_send)

    # 每天进行二次登录检查
    robot.onEveryTime("00:01", robot.login_twice_auto_auth)
    robot.onEveryTime("12:01", robot.login_twice_auto_auth)

    # 启动Dashboard服务器
    dashboard_server = None
    try:
        # 创建Dashboard服务器实例，共享robot对象
        from admin.dashboard.server import DashboardServer
        dashboard_server = DashboardServer(robot_instance=robot)

        # 在单独的线程中启动Dashboard服务器
        dashboard_thread = threading.Thread(target=dashboard_server.run, daemon=True)
        dashboard_thread.start()
        robot.LOG.info(f"Dashboard服务器已在 http://{dashboard_server.host}:{dashboard_server.port} 启动")
    except Exception as e:
        robot.LOG.error(f"Dashboard服务器启动失败: {e}")

    # 让机器人一直跑
    robot.keep_running_and_block_process()


if __name__ == "__main__":
    parser = ArgumentParser()
    parser.add_argument('-c', type=int, default=0, help=f'选择模型参数序号: {ChatType.help_hint()}')
    args = parser.parse_args()
    main(args.c)