import { BacktestEngine } from './src/backtest/BacktestEngine';
import { StrategyManager } from './src/strategies/StrategyManager';
import { StorageService } from './src/services/StorageService';
import { ModeManager } from './src/core/ModeManager';
import { MarketDataService } from './src/services/MarketDataService';
import { ExecutionService } from './src/services/ExecutionService';
import { IServiceContainer } from '@stock-bot/di';

async function debugEquityCurve() {
  // Create service container
  const container: IServiceContainer = {
    logger: {
      info: (msg: string, ...args: any[]) => console.log('[INFO]', msg, ...args),
      error: (msg: string, ...args: any[]) => console.error('[ERROR]', msg, ...args),
      warn: (msg: string, ...args: any[]) => console.warn('[WARN]', msg, ...args),
      debug: (msg: string, ...args: any[]) => console.log('[DEBUG]', msg, ...args),
    } as any,
    custom: {}
  };

  // Initialize services
  const storageService = new StorageService();
  const marketDataService = new MarketDataService(container);
  const executionService = new ExecutionService(container);
  const modeManager = new ModeManager(container, marketDataService, executionService, storageService);
  const strategyManager = new StrategyManager(container);

  // Set services in container
  container.custom = {
    MarketDataService: marketDataService,
    ExecutionService: executionService,
    ModeManager: modeManager,
    StorageService: storageService
  };

  // Test with 5000 initial capital
  const config = {
    mode: 'backtest',
    name: 'Debug',
    strategy: 'sma-crossover',
    symbols: ['AAPL'],
    startDate: '2023-01-01T00:00:00Z',
    endDate: '2023-01-05T00:00:00Z', // Just 5 days
    initialCapital: 5000,
    commission: 0.001,
    slippage: 0.0001,
    dataFrequency: '1d',
    speed: 'max'
  };
  
  await modeManager.initializeMode(config);
  const backtestEngine = new BacktestEngine(container, storageService, strategyManager);
  
  console.log('Before runBacktest - checking backtestEngine state...');
  
  const result = await backtestEngine.runBacktest(config);
  
  console.log('\n=== EQUITY CURVE DEBUG ===');
  console.log(`Config Initial Capital: $${config.initialCapital}`);
  console.log(`Number of equity points: ${result.equity.length}`);
  
  // Show all equity points
  result.equity.forEach((point, idx) => {
    console.log(`  ${idx}: ${point.date} -> $${point.value.toFixed(2)}`);
  });
  
  process.exit(0);
}

debugEquityCurve().catch(error => {
  console.error('Test failed:', error);
  process.exit(1);
});