import hashlib
from eth_utils import keccak
import base58
import ecdsa
from tronpy.keys import PrivateKey


def test1():
    # 示例私鑰，可替換為你自己的私鑰
    private_key_hex = "your_private_key_hex_string"

    # 將十六進(jìn)制私鑰轉(zhuǎn)換為字節(jié)形式
    private_key_bytes = bytes.fromhex(private_key_hex)

    # 使用 SECP256k1 曲線生成私鑰對(duì)象
    sk = ecdsa.SigningKey.from_string(private_key_bytes, curve=ecdsa.SECP256k1)

    # 從私鑰派生公鑰
    vk = sk.get_verifying_key()
    public_key_bytes = vk.to_string()
    public_key_point = vk.pubkey.point

    # Hash the public key using Keccak-256
    public_key_hash = keccak(public_key_bytes)

    # Take the last 20 bytes of the hash and prepend 0x41
    # 添加 TRON 地址前綴 (0x41)
    tron_prefix = b'\x41'
    prefixed_hash160 = tron_prefix + public_key_hash[-20:]

    # 對(duì)添加前綴后的結(jié)果進(jìn)行兩次 SHA-256 哈希以獲取校驗(yàn)和
    first_sha256 = hashlib.sha256(prefixed_hash160).digest()
    second_sha256 = hashlib.sha256(first_sha256).digest()

    # 取前 4 字節(jié)作為校驗(yàn)和
    checksum = second_sha256[:4]

    # 將前綴、哈希結(jié)果和校驗(yàn)和拼接
    full_payload = prefixed_hash160 + checksum

    # Convert to hexadecimal string
    trc_address_hex = prefixed_hash160.hex()

    # 使用 Base58 編碼生成最終的 TRC 地址
    trc_address = base58.b58encode(full_payload).decode()

    # 打印私鑰（以十六進(jìn)制字符串形式）
    print("私鑰 (十六進(jìn)制):", sk.to_string().hex())
    # 打印公鑰（以十六進(jìn)制字符串形式）
    print("公鑰 (十六進(jìn)制):", vk.to_string().hex())
    # 打印公鑰對(duì)應(yīng)的橢圓曲線上的點(diǎn)
    print("公鑰對(duì)應(yīng)的橢圓曲線上的點(diǎn):")
    print("  x 坐標(biāo) (十六進(jìn)制):", hex(public_key_point.x()))
    print("  y 坐標(biāo) (十六進(jìn)制):", hex(public_key_point.y()))
    print("Keccak-256哈希:", public_key_hash.hex())
    print("校驗(yàn)和:", checksum.hex())
    print("完整地址數(shù)據(jù):", full_payload.hex())
    print("TRC 地址數(shù)據(jù):", trc_address_hex)
    print("TRC 地址:", trc_address)
    print('\n')


def test2():
    # 指定私鑰（請(qǐng)?zhí)鎿Q為你實(shí)際的私鑰）
    private_key_hex = "your_private_key_hex_string"

    # 將十六進(jìn)制私鑰轉(zhuǎn)換為PrivateKey對(duì)象
    private_key = PrivateKey(bytes.fromhex(private_key_hex))

    # 從私鑰派生公鑰
    public_key = private_key.public_key

    # 從公鑰生成TRC地址
    address = public_key.to_base58check_address()

    print("指定的私鑰 (十六進(jìn)制):", private_key_hex)
    print("公鑰 (十六進(jìn)制):", public_key.hex())
    print("TRC地址:", address)
    print('\n')


test1()
test2()