import json
from enum import Enum

import requests

from config.setting import settings


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


def send_wechat_message(message_type: MessageEnum, message: str):
    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}}

    requests.post(
        settings.wechat_webhook_url,
        headers={"Content-Type": "application/json"},
        params={'key': settings.wechat_webhook_key},
        data=json.dumps(wechat_message, ensure_ascii=False).encode('utf-8')
    )

    return True