# encoding=utf-8
"""
cron: 0
new ENV('hostloc新帖推送机');
cmd: nohup python3 -u /ql/scripts/hostloc2tg_api.py >> /ql/log/hostloc.txt 2>&1 &
------------
env
HOST_GROUP_ID: 用户、频道、群组 ID
HOST_BOT_TOKEN: 机器人 TOKEN
"""

import os
import time

import requests


def post_tg(url, count):
    try:
        r = requests.get(url)
        if '"ok":true,' in r.text:
            print("发送成功！")
        else:
            count = count + 1
            if count <= 5:
                time.sleep(3)
                print("发送失败，正在重试")
                post_tg(url, count)
    except Exception:
        time.sleep(3)
        print("发送失败，正在重试")
        post_tg(url, count)


hostloc_list = [
    "1",
    "2",
    "3",
    "4",
    "5",
    "6",
    "7",
    "8",
    "9",
    "10",
    "11",
    "12",
    "13",
    "14",
    "15",
    "16",
    "17",
    "18",
    "19",
    "20",
]
hostloc_title = [
    "1",
    "2",
    "3",
    "4",
    "5",
    "6",
    "7",
    "8",
    "9",
    "10",
    "11",
    "12",
    "13",
    "14",
    "15",
    "16",
    "17",
    "18",
    "19",
    "20",
]
url_1 = "https://www.hostloc.com/"
while True:
    try:
        with requests.get("https://hostloc.cherbim.ml/", stream=True, timeout=5) as r:
            print(time.strftime("%m-%d %H:%M:%S", time.localtime()))
            for i in r.json()["new_data"][0][15:]:
                if (
                    i["主题ID"] not in hostloc_list
                    and i["主题"] not in hostloc_title
                ):
                    hostloc_list = hostloc_list[1::]
                    hostloc_list.append(i["主题ID"])
                    hostloc_title = hostloc_title[1::]
                    hostloc_title.append(i["主题"])
                    a = "https://www.hostloc.com/thread-{0}-1-1.html".format(i["主题ID"])
                    time_1 = time.strftime("%Y-%m-%d  %H:%M:%S", time.localtime())
                    a = a if "论坛bug，此贴内容无法查看~" not in i["主题内容"][:100] else f"<s>{a}</s>"
                    text = (
                        f'主        题：<b>{i["主题"].replace("&", "%26").replace("<", "%26lt%3b").replace(">", "%26gt%3b").replace("#", " ")}</b>'
                        + "\n"
                        + "发  布  者："
                        + """<a href="{0}">{1}</a>""".format(
                            i["发布者链接"], i["发布者"]
                        )
                        + "\n"
                        + "时        间："
                        + time_1
                        + "\n"
                        + "内容预览："
                        + """<b>{0}</b>""".format(
                            i["主题内容"][:100]
                            .replace("&", "%26")
                            .replace("<", "%26lt%3b")
                            .replace(">", "%26gt%3b")
                            .replace("#", " ")
                        )
                        + "\n"
                        + "直达链接： "
                        + a
                    )

                    print(text)
                    # 修改为你自己的bot api token和chat_id(可以是用户也可以是频道）
                    chat_id = os.environ.get("HOST_GROUP_ID")
                    bot_api_token = os.environ.get("HOST_BOT_TOKEN")
                    tg_url = f"https://api.telegram.org/bot{bot_api_token}/sendMessage?parse_mode=HTML&chat_id={chat_id}&text={text}"

                    b = 0
                    post_tg(tg_url, b)
            time.sleep(2)
    except Exception:
        print("网络错误，请稍后重试")
        time.sleep(5)