// SPDX-License-Identifier: MIT
pragma solidity ^0.8.24;

import { ITokenMetadata, Attribute, StdTokenMetadata } from "../interfaces/metadata/ITokenMetadata.sol";
import { JsonUtil } from "../utils/JsonUtil.sol";
import { JsonStore } from "../utils/JsonStore.sol";
import { Strings } from "../utils/Strings.sol";

abstract contract TokenMetadata is ITokenMetadata {
    /// @dev Indicates whether any token exist with a given id, or not.
    function exists(uint256 _tokenId) external view virtual returns (bool) {
        return _exists(_tokenId);
    }

    /// @dev Returns the URI of the token with the given id.
    function uri(uint256 _tokenId) external view virtual returns (string memory) {
        return _uri(_tokenId);
    }

    /// @dev Returns the URI of the token with the given id.
    function tokenURI(uint256 _tokenId) external view virtual returns (string memory) {
        return _uri(_tokenId);
    }

    function getTokenMetadata(uint256 _tokenId) external view virtual returns (string memory) {
        return _getTokenMetadata(_tokenId);
    }

    function _uri(uint256 _tokenId) internal view virtual returns (string memory) {
        return JsonStore.uri(_getTokenMetadataKey(_tokenId));
    }

    function _exists(uint256 _tokenId) internal view virtual returns (bool) {
        return JsonStore.exists(_getTokenMetadataKey(_tokenId));
    }

    function _exists(uint256 _tokenId, string memory _path) internal view virtual returns (bool) {
        string memory metadata = _getTokenMetadata(_tokenId);
        return JsonUtil.exists(metadata, _path);
    }

    function _getTokenMetadata(uint256 _tokenId) internal view virtual returns (string memory) {
        return JsonStore.get(_getTokenMetadataKey(_tokenId));
    }

    function _getTokenMetadata(uint256 _tokenId, string memory _path) internal view returns (string memory) {
        string memory metadata = _getTokenMetadata(_tokenId);
        return JsonUtil.get(metadata, _path);
    }

    function _getTokenMetadataInt(uint256 _tokenId, string memory _path) internal view returns (int256) {
        string memory metadata = _getTokenMetadata(_tokenId);
        return JsonUtil.getInt(metadata, _path);
    }

    function _getTokenMetadataUint(uint256 _tokenId, string memory _path) internal view returns (uint256) {
        string memory metadata = _getTokenMetadata(_tokenId);
        return JsonUtil.getUint(metadata, _path);
    }

    function _getTokenMetadataBool(uint256 _tokenId, string memory _path) internal view returns (bool) {
        string memory metadata = _getTokenMetadata(_tokenId);
        return JsonUtil.getBool(metadata, _path);
    }

    /// @dev Returns the attribute of the token with the given id and trait type as a string.
    function _getTokenAttribute(uint256 _tokenId, string memory _traitType) internal view returns (string memory) {
        return _getTokenMetadata(_tokenId, _getTokenAttributeValuePath(_traitType));
    }

    /// @dev Returns the attribute of the token with the given id and trait type as `int256`.
    function _getTokenAttributeInt(uint256 _tokenId, string memory _traitType) internal view returns (int256) {
        return _getTokenMetadataInt(_tokenId, _getTokenAttributeValuePath(_traitType));
    }

    /// @dev Returns the attribute of the token with the given id and trait type as `uint256`.
    function _getTokenAttributeUint(uint256 _tokenId, string memory _traitType) internal view returns (uint256) {
        return _getTokenMetadataUint(_tokenId, _getTokenAttributeValuePath(_traitType));
    }

    /// @dev Returns the attribute of the token with the given id and trait type as `bool`.
    function _getTokenAttributeBool(uint256 _tokenId, string memory _traitType) internal view returns (bool) {
        return _getTokenMetadataBool(_tokenId, _getTokenAttributeValuePath(_traitType));
    }

    function _hasTokenAttribute(uint256 _tokenId, string memory _traitType) internal view returns (bool) {
        return _exists(_tokenId, _getTokenAttributePath(_traitType));
    }

    function _getTokenAttributePath(string memory _traitType) internal pure returns (string memory) {
        return Strings.replace('attributes.#(trait_type==":trait_type:")', ":trait_type:", _traitType, 1);
    }

    function _getTokenAttributeValuePath(string memory _traitType) internal pure returns (string memory) {
        return Strings.replace('attributes.#(trait_type==":trait_type:").value', ":trait_type:", _traitType, 1);
    }

    function _tokenMetadataToJson(StdTokenMetadata memory _data) internal pure returns (string memory) {
        string memory metadata = '{"attributes":[]}';

        string[] memory paths = new string[](5);
        paths[0] = "name";
        paths[1] = "description";
        paths[2] = "image";
        paths[3] = "external_url";
        paths[4] = "animation_url";
        string[] memory values = new string[](5);
        values[0] = _data.name;
        values[1] = _data.description;
        values[2] = _data.image;
        values[3] = _data.externalURL;
        values[4] = _data.animationURL;
        metadata = JsonUtil.set(metadata, paths, values);

        for (uint8 i = 0; i < _data.attributes.length; i++) {
            metadata = JsonUtil.setRaw(metadata, "attributes.-1", _tokenAttributeToJson(_data.attributes[i]));
        }

        return metadata;
    }

    function _tokenAttributeToJson(Attribute memory _attribute) internal pure returns (string memory) {
        string memory attribute = "{}";
        string[] memory paths = new string[](2);
        paths[0] = "trait_type";
        paths[1] = "value";
        string[] memory values = new string[](2);
        values[0] = _attribute.traitType;
        values[1] = _attribute.value;
        attribute = JsonUtil.set(attribute, paths, values);
        if (bytes(_attribute.displayType).length > 0) {
            attribute = JsonUtil.set(attribute, "display_type", _attribute.displayType);
        }
        return attribute;
    }

    function _setTokenMetadata(uint256 _tokenId, string memory _metadata) internal virtual {
        if (_exists(_tokenId)) {
            revert ITokenMetadata.TokenMetadataImmutable(_tokenId);
        }
        _setTokenMetadataForced(_tokenId, _metadata);
    }

    function _setTokenMetadata(uint256 _tokenId, StdTokenMetadata memory _data) internal virtual {
        _setTokenMetadata(_tokenId, _tokenMetadataToJson(_data));
    }

    function _setTokenMetadataForced(uint256 _tokenId, string memory _metadata) internal virtual {
        _setTokenMetadataForced(_getTokenMetadataKey(_tokenId), _metadata);
        emit ITokenMetadata.MetadataUpdate(_tokenId);
    }

    function _setTokenMetadataForced(bytes32 _key, string memory _metadata) internal virtual {
        JsonStore.set(_key, _metadata);
    }

    function _getTokenMetadataKey(uint256 _tokenId) internal view virtual returns (bytes32) {
        return bytes32(_tokenId);
    }
}