import os
import sys
from requests import post, get
from pytz import timezone
from datetime import datetime
from pagermaid.listener import listener
from pagermaid.utils import alias_command


@listener(is_plugin=True, outgoing=True, command=alias_command("epic"),
          description="获取 Epic 喜加一信息")
async def epic(context):
    await context.edit("获取中 . . .")
    epic_url = "https://www.epicgames.com/store/backend/graphql-proxy"
    headers = {
        "Referer": "https://www.epicgames.com/store/zh-CN/",
        "Content-Type": "application/json; charset=utf-8",
    }
    data = {
        "query":
            "query searchStoreQuery($allowCountries: String, $category: String, $count: Int, $country: String!, "
            "$keywords: String, $locale: String, $namespace: String, $sortBy: String, $sortDir: String, $start: Int, "
            "$tag: String, $withPrice: Boolean = false, $withPromotions: Boolean = false) {\n Catalog {\n "
            "searchStore(allowCountries: $allowCountries, category: $category, count: $count, country: $country, "
            "keywords: $keywords, locale: $locale, namespace: $namespace, sortBy: $sortBy, sortDir: $sortDir, "
            "start: $start, tag: $tag) {\n elements {\n title\n id\n namespace\n description\n effectiveDate\n "
            "keyImages {\n type\n url\n }\n seller {\n id\n name\n }\n productSlug\n urlSlug\n url\n items {\n id\n "
            "namespace\n }\n customAttributes {\n key\n value\n }\n categories {\n path\n }\n price(country: "
            "$country) @include(if: $withPrice) {\n totalPrice {\n discountPrice\n originalPrice\n voucherDiscount\n "
            "discount\n currencyCode\n currencyInfo {\n decimals\n }\n fmtPrice(locale: $locale) {\n originalPrice\n "
            "discountPrice\n intermediatePrice\n }\n }\n lineOffers {\n appliedRules {\n id\n endDate\n "
            "discountSetting {\n discountType\n }\n }\n }\n }\n promotions(category: $category) @include(if: "
            "$withPromotions) {\n promotionalOffers {\n promotionalOffers {\n startDate\n endDate\n discountSetting {"
            "\n discountType\n discountPercentage\n }\n }\n }\n upcomingPromotionalOffers {\n promotionalOffers {\n "
            "startDate\n endDate\n discountSetting {\n discountType\n discountPercentage\n }\n }\n }\n }\n }\n paging "
            "{\n count\n total\n }\n }\n }\n}\n",
        "variables": {
            "allowCountries": "CN",
            "category": "freegames",
            "count": 1000,
            "country": "CN",
            "locale": "zh-CN",
            "sortBy": "effectiveDate",
            "sortDir": "asc",
            "withPrice": True,
            "withPromotions": True
        }
    }
    try:
        res = post(epic_url, headers=headers, json=data, timeout=10.0)
        resJson = res.json()
        games = resJson["data"]["Catalog"]["searchStore"]["elements"]
    except Exception as e:
        return await context.edit("请求 Epic Store API 错误：" + str(sys.exc_info()[0]) + "\n" + str(e))
    if not games:
        return await context.edit("Epic 可能又抽风啦，请稍后再试（")
    else:
        for game in games:
            try:
                game_name = game["title"]
                game_corp = game["seller"]["name"]
                game_price = game["price"]["totalPrice"]["fmtPrice"]["originalPrice"]
                game_promotions = game["promotions"]["promotionalOffers"]
                upcoming_promotions = game["promotions"]["upcomingPromotionalOffers"]
                if not game_promotions and upcoming_promotions:
                    continue  # 促销即将上线，跳过
                elif game["price"]["totalPrice"]["discountPrice"] != 0:
                    continue  # 非免费游戏，跳过
                else:
                    game_thumbnail, game_dev, game_pub = None, None, None
                    for image in game["keyImages"]:
                        game_thumbnail = image["url"] if image["type"] == "Thumbnail" else None
                    for pair in game["customAttributes"]:
                        game_dev = pair["value"] if pair["key"] == "developerName" else game_corp
                        game_pub = pair["value"] if pair["key"] == "publisherName" else game_corp
                    game_desp = game["description"]
                    end_date_iso = game["promotions"]["promotionalOffers"][0]["promotionalOffers"][0]["endDate"][:-1]
                    end_date = datetime.fromisoformat(end_date_iso).replace(
                        tzinfo=timezone('UTC')).astimezone(timezone('Asia/Chongqing')).strftime("%Y-%m-%d %H:%M:%S")
                    # API 返回不包含游戏商店 URL，此处自行拼接，可能出现少数游戏 404
                    game_url = f"https://www.epicgames.com/store/zh-CN/p/{game['productSlug'].replace('/home', '')}"
                    msg = f"**FREE now :: {game_name} ({game_price})**\n\n{game_desp}\n\n"
                    msg += f"游戏由 {game_pub} 发售，" if game_dev == game_pub else f"游戏由 {game_dev} 开发、{game_pub} 出版，"
                    msg += f"将在 **{end_date}** 结束免费游玩，戳下面的链接领取吧~\n{game_url}"

                    if game_thumbnail:
                        r = get(game_thumbnail, timeout=10.0)
                        with open("epic.jpg", "wb") as code:
                            code.write(r.content)
                        await context.client.send_file(context.chat_id, "epic.jpg", caption=msg, force_document=False)
                    else:
                        await context.respond(msg)
            except (TypeError, IndexError):
                pass
            except Exception as e:
                await context.respond("获取 Epic 信息错误：" + str(sys.exc_info()[0]) + "\n" + str(e))
        try:
            os.remove("epic.jpg")
        except FileNotFoundError:
            pass
        await context.delete()