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

import re

from func_chengyu import cy
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)
        self.config = config

    def toChengyu(self, msg):
        texts = re.findall(r"^([#|?|？])(.*)$", msg.content)
        # [('#', '天天向上')]
        if texts:
            flag = texts[0][0]
            text = texts[0][1]
            if flag == "#":  # 接龙
                if cy.isChengyu(text):
                    rsp = cy.getNext(text)
                    if rsp:
                        self.sendTextMsg(msg.roomId, rsp)
            elif flag in ["?", "？"]:  # 查词
                if cy.isChengyu(text):
                    rsp = cy.getMeaning(text)
                    if rsp:
                        self.sendTextMsg(msg.roomId, rsp)

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

        self.printRawMsg(msg)  # 打印信息

        # 群聊消息
        if self.isGroupChat(msg):
            # 如果在群里被 @，回复发信人：“收到你的消息了！” 并 @他
            if msg.roomId not in self.config.GROUPS:  # 不在配置的响应的群列表里，忽略
                return

            if self.isAt(msg):   # 被@
                rsp = "你@我干嘛？"
                self.sendTextMsg(msg.roomId, rsp, msg.wxId)

            else:
                self.toChengyu(msg)

        # 非群聊信息
        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()