from enum import Enum

import json

from message.request_message import send_request

webhook_url = 'https://qyapi.weixin.qq.com/cgi-bin/webhook/send'
webhook_key_estun = 'f09eb098-f709-4fc6-83b5-4a08a8431b8f'
webhook_key_sweet_hut = '5ccae111-43a1-4672-a94f-0d34713259c0'


class MessageEnum(Enum):
    TEXT = 'text'
    MARKDOWN = 'markdown'
    MARKDOWN2 = 'markdown2'


def send_wechat_message(message_type: MessageEnum, message: str, has_sweet_hut=False):
    wechat_message = {}
    match message_type:
        case MessageEnum.TEXT:
            wechat_message = {"msgtype": "text", "text": {"content": message}}
        case MessageEnum.MARKDOWN:
            wechat_message = {"msgtype": "markdown", "markdown": {"content": message}}
        case MessageEnum.MARKDOWN2:
            wechat_message = {"msgtype": "markdown_v2", "markdown_v2": {"content": message}}

    send_request(
        url=webhook_url,
        method="POST",
        headers={"Content-Type": "application/json"},
        params={'key': webhook_key_estun},
        data=json.dumps(wechat_message, ensure_ascii=False).encode('utf-8'),
        timeout=5
    )

    if has_sweet_hut:
        send_request(
            url=webhook_url,
            method="POST",
            headers={"Content-Type": "application/json"},
            params={'key': webhook_key_sweet_hut},
            data=json.dumps(wechat_message, ensure_ascii=False).encode('utf-8'),
            timeout=5
        )