from web3 import Web3
from dataclasses import dataclass
import base64
import json

# Connect to Ethereum node
web3 = Web3(Web3.HTTPProvider('https://mainnet.infura.io/v3/YOUR_INFURA_PROJECT_ID'))

# Address and ABI of deployed contract
contract_address = "YOUR_CONTRACT_ADDRESS"
contract_abi = 'YOUR_CONTRACT_ABI_JSON_HERE'

# Initialize contract
contract = web3.eth.contract(address=contract_address, abi=contract_abi)

@dataclass
class LynxAttributes:
    name: str
    description: str
    metadata_uri: str
    is_blessed: bool
    is_enchanted: bool
    is_protected_by_brinks: bool

class QuantumLynx:
    def __init__(self, base_uri):
        self.base_uri = base_uri
        self.token_id_counter = 0
        self.lynx_attributes = {}

    def mint_lynx(self, to, name, description, metadata_uri, is_blessed, is_enchanted, is_protected_by_brinks):
        self.token_id_counter += 1
        token_id = self.token_id_counter
        self.lynx_attributes[token_id] = LynxAttributes(
            name, description, metadata_uri, is_blessed, is_enchanted, is_protected_by_brinks
        )
        tx_hash = contract.functions.mintLynx(
            to, name, description, metadata_uri, is_blessed, is_enchanted, is_protected_by_brinks
        ).transact({'from': web3.eth.defaultAccount})
        web3.eth.waitForTransactionReceipt(tx_hash)
        print(f"Lynx NFT minted: {tx_hash.hex()}")

    def token_uri(self, token_id):
        assert token_id in self.lynx_attributes, "Token ID does not exist"
        attributes = self.lynx_attributes[token_id]
        attributes_dict = {
            "name": attributes.name,
            "description": attributes.description,
            "image": f"{self.base_uri}{token_id}",
            "attributes": [
                {"trait_type": "Blessed by Archangel Michael", "value": "Yes" if attributes.is_blessed else "No"},
                {"trait_type": "Enchanted", "value": "Yes" if attributes.is_enchanted else "No"},
                {"trait_type": "Protected by BRINKS", "value": "Yes" if attributes.is_protected_by_brinks else "No"}
            ]
        }
        attributes_json = json.dumps(attributes_dict)
        attributes_base64 = base64.b64encode(attributes_json.encode()).decode()
        return f"data:application/json;base64,{attributes_base64}"

    def set_base_uri(self, base_uri):
        self.base_uri = base_uri

# Example Usage
quantum_lynx = QuantumLynx("https://baseuri.com/")
quantum_lynx.mint_lynx(
    "recipient_address",
    "Quantum Lynx #1",
    "An enchanted NFT blessed by Archangel Michael",
    "https://baseuri.com/metadata/1",
    True,
    True,
    True
)
print(quantum_lynx.token_uri(1))