from uniswap import Uniswap
from web3 import Web3
import config

def create_uniswap_instance():
    return Uniswap(address=config.WALLET, private_key=config.PRIV, provider=config.PROVIDER, version=2)

def buy_token(token_address, eth_amount):
    eth = "0x0000000000000000000000000000000000000000"
    uniswap = create_uniswap_instance()
    
    eth_balance_eth = Web3.from_wei(uniswap.get_eth_balance(), 'ether')
    print(f"ETH Balance: {eth_balance_eth} ETH")
    
    eth_amount_wei = Web3.to_wei(eth_amount, 'ether')
    
    tx = uniswap.make_trade(eth, token_address, eth_amount_wei)
    print(f"Transaction hash: {tx.hex()}")

def sell_token(token_address):
    eth = "0x0000000000000000000000000000000000000000"
    uniswap = create_uniswap_instance()
    
    token_balance_wei = uniswap.get_token_balance(token_address)
    print(f"Token Balance: {token_balance_wei} ")
    
    tx = uniswap.make_trade(token_address, eth, token_balance_wei)
    print(f"Transaction hash: {tx.hex()}")