import json
from pprint import pprint
from typing import Dict
from urllib.parse import urljoin

import requests


class NotifyBot(object):

    def __init__(self, content, title="什么值得买签到", **kwargs: Dict) -> None:
        self.content = content
        self.title = title
        self.kwargs = kwargs

        self.push_plus()
        self.server_chain()
        self.tg_bot()

    def push_plus(self, template='html'):
        try:
            if self.kwargs.get("PUSH_PLUS_TOKEN", None):
                PUSH_PLUS_TOKEN = self.kwargs.get("PUSH_PLUS_TOKEN")
            else:
                pprint("⚠️ PUSH_PLUS_TOKEN not set, skip PushPlus nofitication")
                return

            url = 'https://www.pushplus.plus/send'
            body = {
                'token': PUSH_PLUS_TOKEN,
                'title': self.title,
                'content': self.content,
                'template': template
            }
            data = json.dumps(body).encode(encoding='utf-8')
            headers = {'Content-Type': 'application/json'}
            resp = requests.post(url, data=data, headers=headers)
            if resp.status_code == 200:
                pprint("✅ Push Plus notified")
            return resp.json()
        except Exception as e:
            pprint(e)

    def server_chain(self):
        try:
            if self.kwargs.get("SC_KEY", None):
                SC_KEY = self.kwargs.get("SC_KEY")
            else:
                pprint("⚠️ SC_KEY not set, skip ServerChain notification")
                return

            url = f'http://sc.ftqq.com/{SC_KEY}.send'

            data = {'text': self.title, 'desp': self.content}
            resp = requests.post(url, data=data)
            if resp.status_code == 200:
                pprint("✅ Server Chain notified")
            return resp.json()
        except Exception as e:
            pprint(e)

    def tg_bot(self):
        try:
            if self.kwargs.get("TG_BOT_TOKEN", None) and self.kwargs.get("TG_USER_ID", None):
                TG_BOT_TOKEN = self.kwargs.get("TG_BOT_TOKEN")
                TG_USER_ID = self.kwargs.get("TG_USER_ID")
            else:
                pprint(
                    "⚠️ TG_BOT_TOKEN & TG_USER_ID not set, skip TelegramBot notification")
                return

            TG_BOT_API = self.kwargs.get(
                "TG_BOT_API", "https://api.telegram.org")
            url = urljoin(TG_BOT_API, f"/bot{TG_BOT_TOKEN}/sendMessage")
            headers = {'Content-Type': 'application/x-www-form-urlencoded'}
            payload = {'chat_id': str(TG_USER_ID),
                       'text': f'{self.title}\n\n{self.content}',
                       'disable_web_page_preview': 'true'}
            resp = requests.post(url=url, headers=headers, params=payload)
            if resp.status_code == 200:
                pprint("✅ Telegram Bot notified")
            return resp.json()
        except Exception as e:
            pprint(e)