from __future__ import annotations

from io import BytesIO
from typing import List, Union

from PIL import Image
from httpx import AsyncClient

from .models import Maps, Point, XYPoint

CLIENT = AsyncClient()


async def get_img(url: str) -> Image.Image:
    resp = await CLIENT.get(url)
    resp.raise_for_status()
    return Image.open(BytesIO(resp.read()))


async def make_map(map: Maps) -> Image.Image:
    """
    获取所有地图并拼接

    警告：可能导致内存溢出

    在测试中，获取并合成「提瓦特」地图时占用了约 1.4 GiB

    建议使用 `genshinmap.utils.get_map_by_pos` 获取地图单片

    参数：
        map: `Maps`
            地图数据，可通过 `get_maps` 获取

    返回：
        `PIL.Image.Image` 对象

    另见：
        `get_map_by_pos`
    """
    img = Image.new("RGBA", tuple(map.total_size))
    y = 0
    for y_index in map.slices:
        x = 0
        for url in y_index:
            m = await get_img(url)
            img.paste(m, (x, y))
            x += 4096
        y += 4096
    return img


async def get_map_by_pos(map: Maps, x: Union[int, float]) -> Image.Image:
    """
    根据横坐标获取地图单片

    参数：
        map: `Maps`
            地图数据，可通过 `get_maps` 获取

        x: `int | float`
            横坐标

    返回：
        `PIL.Image.Image` 对象
    """
    return await get_img(map.slices[int(x // 4096)])


def get_points_by_id(id_: int, points: List[Point]) -> List[XYPoint]:
    """
    根据 Label ID 获取坐标点

    参数：
        id_: `int`
            Label ID

        points: `list[Point]`
            米游社坐标点列表，可通过 `get_points` 获取

    返回：
        `list[XYPoint]`
    """
    return [
        XYPoint(point.x_pos, point.y_pos)
        for point in points
        if point.label_id == id_
    ]


def convert_pos(points: List[XYPoint], origin: List[int]) -> List[XYPoint]:
    """
    将米游社资源坐标转换为以左上角为原点的坐标系的坐标

    参数：
        points: `list[XYPoint]`
            米游社资源坐标

        origin: `list[Point]`
            米游社地图 Origin，可通过 `get_maps` 获取

    返回：
        `list[XYPoint]`

    示例：
        >>> from genshinmap.models import XYPoint
        >>> points = [XYPoint(1200, 5000), XYPoint(-4200, 1800)]
        >>> origin = [4844,4335]
        >>> convert_pos(points, origin)
        [XYPoint(x=6044, y=9335), XYPoint(x=644, y=6135)]
    """
    return [XYPoint(x + origin[0], y + origin[1]) for x, y in points]