from pathlib import Path
from typing import Any

import aiofiles

from ..misc import supports_argument
from ..syncs import asyncable

try:
    from ujson import dumps, loads
except ImportError:
    from json import dumps, loads


@asyncable
def json_read(path: str | Path) -> Any:
    """Read contents of a JSON file and return it.

    Args:
        path (str | Path): Path-like object or path to the file as a string.

    Returns:
        Any: File contents.
    """
    with open(str(path), mode="r", encoding="utf-8") as f:
        data = f.read()

    return loads(data)


@json_read.asynchronous
async def json_read(path: str | Path) -> Any:
    """Read contents of a JSON file and return it.

    Args:
        path (str | Path): Path-like object or path to the file as a string.

    Returns:
        Any: File contents.
    """
    async with aiofiles.open(str(path), mode="r", encoding="utf-8") as f:
        data = await f.read()

    return loads(data)


@asyncable
def json_write(data: Any, path: str | Path) -> None:
    """Write contents to a JSON file.

    Args:
        data (Any): Contents to write. Must be a JSON-serializable object.
        path (str | Path): Path-like object or path to the file as a string.
    """
    with open(str(path), mode="w", encoding="utf-8") as f:
        f.write(
            dumps(data, ensure_ascii=False, escape_forward_slashes=False, indent=4)
            if supports_argument(dumps, "escape_forward_slashes")
            else dumps(data, ensure_ascii=False, indent=4)
        )


@json_write.asynchronous
async def json_write(data: Any, path: str | Path) -> None:
    """Write contents to a JSON file.

    Args:
        data (Any): Contents to write. Must be a JSON-serializable object.
        path (str | Path): Path-like object or path to the file as a string.
    """
    async with aiofiles.open(str(path), mode="w", encoding="utf-8") as f:
        await f.write(
            dumps(data, ensure_ascii=False, escape_forward_slashes=False, indent=4)
            if supports_argument(dumps, "escape_forward_slashes")
            else dumps(data, ensure_ascii=False, indent=4)
        )