# Copyright (c) 2026 Emanuele Bellocchia
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.

"""Module containing utility classes for Ton keys derivation (ton-crypto style)."""

# Imports
from __future__ import annotations

from typing import Union

from bip_utils.ecc.common.ikeys import IPrivateKey
from bip_utils.ton.addr import (
    TonAddrVersions,
    TonV3R1AddrEncoder,
    TonV3R2AddrEncoder,
    TonV4AddrEncoder,
    TonV5R1AddrEncoder,
)
from bip_utils.ton.ton_keys import TonPrivateKey, TonPublicKey


class Ton:
    """
    TON class.
    It allows to generate keys and addresses for TON wallets like ton-crypto.
    """

    m_priv_key: TonPrivateKey
    m_pub_key: TonPublicKey

    @classmethod
    def FromSeed(cls,
                 seed_bytes: bytes) -> Ton:
        """
        Construct class from seed bytes.

        Args:
            seed_bytes (bytes): Seed bytes

        Returns:
            Ton object: Ton object

        Raises:
            TonKeyError: If the key constructed from the bytes is not valid
        """
        return cls(seed_bytes[:32])

    def __init__(self,
                 priv_key: Union[bytes, IPrivateKey]) -> None:
        """
        Construct class.

        Args:
            priv_key (Union[bytes, IPrivateKey]): Private key bytes or object

        Raises:
            TonKeyError: If the key constructed from the bytes is not valid
        """
        self.m_priv_key = TonPrivateKey.FromBytesOrKeyObject(priv_key)
        self.m_pub_key = self.m_priv_key.PublicKey()

    def PublicKey(self) -> TonPublicKey:
        """
        Return public key object.

        Returns:
            TonPublicKey object: TonPublicKey object
        """
        return self.m_pub_key

    def PrivateKey(self) -> TonPrivateKey:
        """
        Return private key object.

        Returns:
            TonPrivateKey object: TonPrivateKey object
        """
        return self.m_priv_key

    def GetAddress(self,
                   version: TonAddrVersions = TonAddrVersions.V5R1,
                   is_bounceable: bool = False) -> str:
        """
        Get address from public key.

        Args:
            version (TonAddrVersions, optional): Address version (default: v5r1)
            is_bounceable (bool, optional)     : Whether the address is bounceable (default: False)

        Returns:
            str: Address string
        """
        if version == TonAddrVersions.V5R1:
            return TonV5R1AddrEncoder(self.m_pub_key).Encode(is_bounceable)
        elif version == TonAddrVersions.V4:
            return TonV4AddrEncoder(self.m_pub_key).Encode(is_bounceable)
        elif version == TonAddrVersions.V3R2:
            return TonV3R2AddrEncoder(self.m_pub_key).Encode(is_bounceable)
        else:
            return TonV3R1AddrEncoder(self.m_pub_key).Encode(is_bounceable)