const {
    Connection,
    PublicKey,
    Keypair,
    Transaction,
    sendAndConfirmTransaction,
} = require('@solana/web3.js');
const { getMarket, PlaceOrder, marketAddress } = require('@raydium-io/raydium-sdk');
const fs = require('fs');

// Initialize connection
const connection = new Connection('https://api.mainnet-beta.solana.com', 'confirmed');

// Load wallet keys from a file or secure storage
const wallets = JSON.parse(fs.readFileSync('wallets.json')); // Ensure this contains the keys in the correct format
const SOL_AMOUNT = 100; // Amount of SOL to buy/sell per transaction
const MIN_TOKENS_TO_KEEP = 1; // Minimum tokens to keep in each wallet
const TRANSACTION_COUNT = 300; // Total transactions in 24 hours
const INTERVAL = (24 * 60 * 60 * 1000) / TRANSACTION_COUNT; // Interval between transactions

async function getBalance(wallet) {
    const balance = await connection.getTokenAccountBalance(wallet);
    return balance.value.uiAmount; // Adjust this according to your token's specifics
}

async function tradeOnRaydium(wallet, action) {
    const market = await getMarket(marketAddress); // Replace with the correct market address for your token

    const currentBalance = await getBalance(wallet.publicKey);
    const amountToTrade = action === 'buy' ? SOL_AMOUNT : currentBalance - MIN_TOKENS_TO_KEEP;

    if (amountToTrade <= 0) {
        console.log(`Not enough tokens to ${action}. Wallet: ${wallet.publicKey.toString()}`);
        return; // Skip trading if insufficient tokens
    }

    const transaction = new Transaction();

    // Construct your order based on action (buy or sell)
    const order = new PlaceOrder({
        market,
        side: action === 'buy' ? 'buy' : 'sell',
        price: market.currentPrice, // Set the current market price or desired price
        size: amountToTrade,
        owner: wallet.publicKey,
        // Include other necessary parameters
    });

    transaction.add(order); // Add the order to the transaction

    // Send transaction
    const signature = await sendAndConfirmTransaction(connection, transaction, [wallet]);
    console.log(`${action} transaction signature:`, signature);
}

async function startTrading() {
    for (let i = 0; i < TRANSACTION_COUNT; i++) {
        // Select wallet based on index
        const walletIndex = i % wallets.length;
        const walletKeypair = Keypair.fromSecretKey(Uint8Array.from(wallets[walletIndex]));

        // Alternate between buying and selling
        const action = i % 2 === 0 ? 'buy' : 'sell';
        await tradeOnRaydium(walletKeypair, action);

        // Wait for the specified interval before the next transaction
        await new Promise(resolve => setTimeout(resolve, INTERVAL));
    }
}

startTrading().catch(err => {
    console.error('Error during trading:', err);
});