import re

with open('/home/crogers2287/projects/dope_wars_2026/lib/services/game_service.dart', 'r') as f:
    content = f.read()

new_methods = '''
  /// Player initiates combat against rival territory
  ActionResult initiateCombat(GameState state) {
    final player = state.player;
    if (player.heat > 85) {
      return ActionResult(
        success: false,
        message: '🚔 Too hot to start a fight right now. Cops everywhere.',
      );
    }
    final combatEvent = _eventService.initiateRivalCombat(player.currentCityId);
    return ActionResult(
      success: true,
      message: 'You made your move...',
      newState: state.copyWith(pendingCombat: combatEvent),
    );
  }

  /// Scaled price helper — prices increase with player wealth
  double scaledPrice(double basePrice, double netWorth, double scaleDivisor) {
    final multiplier = 1.0 + (netWorth / scaleDivisor).clamp(0, 10.0);
    return basePrice * multiplier;
  }

  /// Apply a severe bust event: lose ALL product + 50% cash
  GameState applyBust(GameState state) {
    final player = state.player;
    final hasSafeHouse = player.upgradeLevel('safe_house') > 0;
    final inventoryLoss = hasSafeHouse ? 0.5 : 1.0;

    var newInv = Map<String, int>.from(player.inventory);
    for (final key in newInv.keys.toList()) {
      final qty = newInv[key]!;
      newInv[key] = (qty * (1 - inventoryLoss)).floor();
      if (newInv[key]! == 0) newInv.remove(key);
    }

    final newPlayer = player.copyWith(
      inventory: newInv,
      money: (player.money * 0.5).clamp(0, double.infinity),
      heat: 90.0,
      bustCount: player.bustCount + 1,
      comboCount: 0,
    );

    final newLog = List<String>.from(state.eventLog)
      ..insert(0, '🚔 BUSTED! ${hasSafeHouse ? "50%" : "ALL"} product seized + 50% cash gone! 30 days heat maxed.');

    return state.copyWith(
      player: newPlayer,
      eventLog: newLog,
      statusMessage: '🚔 BUSTED — You need to lay low.',
    );
  }

'''

# Insert before the last closing brace of GameService
# Find "}\n\n// Upgrade definitions"
content = content.replace('  }\n}\n\n// Upgrade definitions', new_methods + '  }\n}\n\n// Upgrade definitions', 1)

with open('/home/crogers2287/projects/dope_wars_2026/lib/services/game_service.dart', 'w') as f:
    f.write(content)

print("Done")
