from flask import Flask, render_template, request, jsonify
from solana.rpc.api import Client
from dexscreener import DexscreenerClient
import requests
import os

app = Flask(__name__)
solana_client = Client("https://api.mainnet-beta.solana.com")
dexscreener_client = DexscreenerClient()

# Replace with the wallet address you want to follow
FOLLOWED_WALLET = "REPLACE_WITH_WALLET_ADDRESS"

# Replace with your wallet address
YOUR_WALLET = "REPLACE_WITH_YOUR_WALLET_ADDRESS"

# Simulated wallet balances (replace with actual balance fetching logic)
wallet_balances = {
    FOLLOWED_WALLET: {"SOL": 100, "USDC": 1000},
    YOUR_WALLET: {"SOL": 10, "USDC": 100}
}

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/tokens', methods=['GET'])
def get_tokens():
    # Fetch tokens from your wallet
    return jsonify(list(wallet_balances[YOUR_WALLET].keys()))

@app.route('/swap', methods=['POST'])
def swap_tokens():
    data = request.json
    token_name = data['token_name']
    amount = data['amount']
    # Here you would add logic to perform the token swap
    return jsonify({'status': 'success', 'message': f'Swapped {amount} of {token_name}'})
    
@app.route('/balances', methods=['GET'])
def get_balances():
    return jsonify(wallet_balances[YOUR_WALLET])

@app.route('/followed_wallet_moves', methods=['GET'])
def get_followed_wallet_moves():
    # In a real-world scenario, you'd use a blockchain explorer API to get recent transactions
    # For this example, we'll simulate a move
    simulated_move = {
        "token": "SOL",
        "action": "swap",
        "amount": 5,
        "to_token": "USDC"
    }
    return jsonify(simulated_move)

@app.route('/follow_move', methods=['POST'])
def follow_move():
    move = request.json
    followed_balance = wallet_balances[FOLLOWED_WALLET][move['token']]
    your_balance = wallet_balances[YOUR_WALLET][move['token']]
    
    proportion = your_balance / followed_balance
    amount_to_swap = move['amount'] * proportion

    # Perform the swap (simulated)
    if wallet_balances[YOUR_WALLET][move['token']] >= amount_to_swap:
        wallet_balances[YOUR_WALLET][move['token']] -= amount_to_swap
        
        # Get the current price of the token pair
        pair = dexscreener_client.get_token_pair("solana", move['token'])
        price = float(pair['priceUsd'])
        
        received_amount = amount_to_swap * price
        wallet_balances[YOUR_WALLET][move['to_token']] += received_amount

        return jsonify({
            'status': 'success',
            'message': f"Swapped {amount_to_swap} {move['token']} for {received_amount} {move['to_token']}"
        })
    else:
        return jsonify({
            'status': 'error',
            'message': f"Insufficient balance to swap {amount_to_swap} {move['token']}"
        })

if __name__ == '__main__':
    app.run(debug=True)