import dotenv from 'dotenv';
import { SignerClient, ApiClient, AccountApi } from '../src';

dotenv.config();

const BASE_URL = process.env.BASE_URL || 'https://api-testnet.lighter.xyz';
const PRIVATE_KEY = process.env.API_PRIVATE_KEY || '';
const ACCOUNT_INDEX = parseInt(process.env.ACCOUNT_INDEX || '10');
const API_KEY_INDEX = parseInt(process.env.API_KEY_INDEX || '10');

async function main() {
  console.log('🚀 Starting account tier change example...\n');

  try {
    // Initialize signer client
    const signerClient = new SignerClient({
      url: BASE_URL,
      privateKey: PRIVATE_KEY,
      accountIndex: ACCOUNT_INDEX,
      apiKeyIndex: API_KEY_INDEX,
    });

    // Ensure signerclient initialized
    await signerClient.initialize();
    await signerClient.ensureWasmClient();
    
    console.log(`✅ SignerClient initialized`);
    console.log(`📊 Account Index: ${ACCOUNT_INDEX}`);
    console.log(`🔑 API Key Index: ${API_KEY_INDEX}\n`);
    
    // Initialize API clients
    const apiClient = new ApiClient({ host: BASE_URL });
    const accountApi = new AccountApi(apiClient);

    console.log(`✅ Clients initialized`);
    console.log(`📊 Account Index: ${ACCOUNT_INDEX}`);

    // Create auth token
    const authToken = await signerClient.createAuthTokenWithExpiry();
    console.log(`🔐 Auth token created\n`);

    // Example 1: Upgrade to Premium tier
    console.log('📈 Example 1: Upgrading to PREMIUM tier');
    try {
      const upgradeResult = await accountApi.changeAccountTier(
        ACCOUNT_INDEX,
        'premium',  // New tier
        authToken
      );
      console.log(`✅ Successfully upgraded to PREMIUM tier!`);
      console.log(`📊 Result:`, JSON.stringify(upgradeResult, null, 2));
    } catch (error) {
      console.error(`❌ Error upgrading to premium: ${error}`);
      // This might fail if already premium or not eligible
    }

    // Wait a bit
    await new Promise(resolve => setTimeout(resolve, 2000));

    // Example 2: Revert back to Standard tier
    console.log('\n📉 Example 2: Reverting to STANDARD tier');
    try {
      const revertResult = await accountApi.changeAccountTier(
        ACCOUNT_INDEX,
        'standard',  // New tier
        authToken
      );
      console.log(`✅ Successfully reverted to STANDARD tier!`);
      console.log(`📊 Result:`, JSON.stringify(revertResult, null, 2));
    } catch (error) {
      console.error(`❌ Error reverting to standard: ${error}`);
    }

    console.log('\n✅ Account tier change examples completed!');
    console.log('\n💡 Note: Tier changes may require specific account conditions or fees.');

  } catch (error) {
    console.error('❌ Error:', error);
    process.exit(1);
  }
}

main().catch(console.error);