using System;
using System.Linq;
using cAlgo.API;
using cAlgo.API.Indicators;
using cAlgo.API.Internals;
using cAlgo.Indicators;

namespace cAlgo.Robots {
  // Attribute to set the robot's time zone and access rights
  [Robot(TimeZone = TimeZones.UTC, AccessRights = AccessRights.None)]
  public class SampleMartingaleRobot: Robot {
    // Parameters for initial volume, stop loss, take profit, trailing stop loss, and spread limit
    [Parameter("Initial Volume", DefaultValue = 1000, MinValue = 0)]
    public int InitialVolume {
      get;
      set;
    }

    [Parameter("Stop Loss", DefaultValue = 15)]
    public int StopLoss {
      get;
      set;
    }

    [Parameter("Take Profit", DefaultValue = 24)]
    public int TakeProfit {
      get;
      set;
    }

    [Parameter("Trailing Stop Loss", DefaultValue = 21)]
    public int TrailingStopLoss {
      get;
      set;
    }

    [Parameter("Spread Limit", DefaultValue = 0.5)]
    public double SpreadLimit {
      get;
      set;
    }

    // Random number generator
    private Random random = new Random();

    // Function that is called when the robot starts
    protected override void OnStart() {
      // Event handler for when positions are closed
      Positions.Closed += OnPositionsClosed;

      // Check if spread is within the limit and execute an initial order
      if (Symbol.Spread <= SpreadLimit) {
        ExecuteOrder(InitialVolume, GetRandomTradeType());
      }
    }

    // Function to execute an order
    private void ExecuteOrder(long volume, TradeType tradeType) {
      // Check if spread is within the limit
      if (Symbol.Spread <= SpreadLimit) {
        var result = ExecuteMarketOrder(tradeType, SymbolName, volume, "Martingale", StopLoss, TakeProfit);
        // Modify stop loss if order execution is successful
        if (result.IsSuccessful) {
          result.Position.ModifyStopLossPips(TrailingStopLoss);
        }

        // If error is not enough money, stop the robot
        if (result.Error == ErrorCode.NoMoney)
          Stop();
      }
    }

    // Event handler function for when positions are closed
    private void OnPositionsClosed(PositionClosedEventArgs args) {
      Print("Closed");
      var position = args.Position;

      // Check if the closed position has the correct label and symbol
      if (position.Label != "Martingale" || position.SymbolName != SymbolName)
        return;

      // If the position has a positive gross profit, execute another order with the initial volume
      if (position.GrossProfit > 0) {
        if (Symbol.Spread <= SpreadLimit) {
          ExecuteOrder(InitialVolume, GetRandomTradeType());
        }
      }
      // If the position has a negative gross profit, execute another order with double the previous volume
      else {
        if (Symbol.Spread <= SpreadLimit) {
          ExecuteOrder((int) position.VolumeInUnits * 2, position.TradeType);
        }
      }
    }

    // Function to generate a random trade type (buy or sell)
    private TradeType GetRandomTradeType() {
      return random.Next(2) == 0 ? TradeType.Buy : TradeType.Sell;
    }
  }
}