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

import re

import robot.sdk.wcferry as WxSDK
from robot.base_robot import BaseRobot
from robot.configuration import Config


class Robot(BaseRobot):
    """个性化自己的机器人
    """

    def __init__(self, sdk: WxSDK, config: Config) -> None:
        super().__init__(sdk, config)

    def processMsg(self, msg) -> None:
        """当接收到消息的时候，会调用本方法。如果不实现本方法，则打印原始消息。
        """

        self.printRawMsg(msg)  # 打印信息

        # 如果在群里被 @，回复发信人：“收到你的消息了！” 并 @他
        if self.isGroupChat(msg):  # 是群消息
            if self.isAt(msg):   # 被@
                if msg.roomId in self.config.GROUPS:  # 在配置的响应的群列表里
                    self.sendTextMsg(msg.roomId, "收到你的消息了！", msg.wxId)

        # 非群聊信息
        elif msg.type == 37:     # 好友请求
            self.autoAcceptFriendRequest(msg)

        elif msg.type == 10000:  # 系统信息
            nickName = re.findall(r"你已添加了(.*)，现在可以开始聊天了。", msg.content)
            if nickName:
                # 添加了好友，更新好友列表
                self.allContacts[msg.wxId] = nickName


def weather_report(robot: Robot):
    """模拟发送天气预报
    """
    # 获取接收人
    receivers = ["filehelper"]

    # 获取天气，需要自己实现，可以参考 https://gitee.com/lch0821/WeatherScrapy 获取天气。
    report = "这就是获取到的天气情况了"

    for r in receivers:
        robot.sendTextMsg(r, report)


def main():
    robot = Robot(WxSDK, Config())

    # 初始化机器人
    robot.initSDK()

    # 接收消息
    robot.enableRecvMsg()

    # 每天7点发送天气预报
    robot.onEveryTime("07:00", weather_report, robot=robot)

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


if __name__ == "__main__":
    main()