#! /usr/bin/env python3
# -*- coding: utf-8 -*-

import signal
from argparse import ArgumentParser

from configuration import Config
from constants import ChatType
from robot import Robot, __version__
from wcferry import Wcf

def main(chat_type: int):
    config = Config()
    wcf = Wcf(debug=True)

    def handler(sig, frame):
        wcf.cleanup()  # 退出前清理环境
        exit(0)

    signal.signal(signal.SIGINT, handler)

    robot = Robot(config, wcf, chat_type)
    robot.LOG.info(f"WeChatRobot【{__version__}】成功启动···")

    # 机器人启动发送测试消息
    robot.sendTextMsg("机器人启动成功！", "filehelper")

    # 接收消息
    # robot.enableRecvMsg()     # 可能会丢消息？
    robot.enableReceivingMsg()  # 加队列

    # 每天 8:30 发送新闻
    robot.onEveryTime("08:30", robot.newsBaiduReportAuto)

    # epic
    robot.onEveryTime("10:30", robot.sendEpicFreeGames)

    # message report 1:数据自动从redis 转到sqllite
    robot.onEveryTime("00:30", robot.messageCountToDB)
    # 从db中提取并发送给相关群
    robot.onEveryTime("09:30", robot.generateAndSendRanking)

    # sehuatang
    robot.onEveryTime("15:00", robot.generateSehuatangPdf)

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


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