managing-apps/src/Managing.Application/Scenarios/ScenarioService.cs

using Managing.Application.Abstractions;
using Managing.Application.Abstractions.Services;
using Managing.Domain.Scenarios;
using Managing.Domain.Strategies;
using Managing.Domain.Users;
using Microsoft.Extensions.Logging;
using static Managing.Common.Enums;

namespace Managing.Application.Scenarios
{
    public class ScenarioService : IScenarioService
    {
        private readonly ILogger<ScenarioService> _logger;
        private readonly ITradingService _tradingService;

        public ScenarioService(ILogger<ScenarioService> logger, ITradingService tradingService)
        {
            _logger = logger;
            _tradingService = tradingService;
        }

        public async Task<Scenario> CreateScenario(string name, List<string> strategies, int? loopbackPeriod = 1)
        {
            var scenario = new Scenario(name, loopbackPeriod);

            foreach (var strategy in strategies)
            {
                scenario.AddIndicator(await _tradingService.GetStrategyByNameAsync(strategy));
            }

            try
            {
                await _tradingService.InsertScenarioAsync(scenario);
            }
            catch (Exception ex)
            {
                _logger.LogError(ex.Message);
                throw new Exception("Cannot create scenario");
            }

            return scenario;
        }

        public async Task<Indicator> CreateStrategy(
            IndicatorType type,
            string name,
            int? period = null,
            int? fastPeriods = null,
            int? slowPeriods = null,
            int? signalPeriods = null,
            double? multiplier = null,
            int? stochPeriods = null,
            int? smoothPeriods = null,
            int? cyclePeriods = null)
        {
            var strategy = ScenarioHelpers.BuildIndicator(
                type,
                name,
                period,
                fastPeriods,
                slowPeriods,
                signalPeriods,
                multiplier,
                stochPeriods,
                smoothPeriods,
                cyclePeriods);
            await _tradingService.InsertStrategyAsync(strategy);
            return strategy;
        }

        public async Task<IEnumerable<Scenario>> GetScenariosAsync()
        {
            return await _tradingService.GetScenariosAsync();
        }

        public async Task<IEnumerable<Indicator>> GetIndicatorsAsync()
        {
            return await _tradingService.GetStrategiesAsync();
        }

        public async Task<bool> DeleteScenarioAsync(string name)
        {
            try
            {
                await _tradingService.DeleteScenarioAsync(name);
                return true;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex.Message);
                return false;
            }
        }

        public async Task<bool> UpdateScenario(string name, List<string> strategies, int? loopbackPeriod)
        {
            try
            {
                var scenario = await _tradingService.GetScenarioByNameAsync(name);
                scenario.Indicators.Clear();
                foreach (var strategy in strategies)
                {
                    scenario.AddIndicator(await _tradingService.GetStrategyByNameAsync(strategy));
                }

                scenario.LoopbackPeriod = loopbackPeriod ?? 1;
                await _tradingService.UpdateScenarioAsync(scenario);
                return true;
            }
            catch (Exception e)
            {
                Console.WriteLine(e);
                return false;
            }
        }

        public async Task<bool> UpdateStrategy(IndicatorType indicatorType, string name, int? period, int? fastPeriods,
            int? slowPeriods,
            int? signalPeriods, double? multiplier, int? stochPeriods, int? smoothPeriods, int? cyclePeriods)
        {
            try
            {
                var strategy = await _tradingService.GetStrategyByNameAsync(name);
                strategy.Type = indicatorType;
                strategy.Period = period;
                strategy.FastPeriods = fastPeriods;
                strategy.SlowPeriods = slowPeriods;
                strategy.SignalPeriods = signalPeriods;
                strategy.Multiplier = multiplier;
                strategy.StochPeriods = stochPeriods;
                strategy.SmoothPeriods = smoothPeriods;
                strategy.CyclePeriods = cyclePeriods;
                await _tradingService.UpdateStrategyAsync(strategy);
                return true;
            }
            catch (Exception e)
            {
                Console.WriteLine(e);
                return false;
            }
        }

        public async Task<IEnumerable<Scenario>> GetScenariosByUserAsync(User user)
        {
            var scenarios = await _tradingService.GetScenariosAsync();
            return scenarios.Where(s => s.User?.Name == user.Name);
        }

        public async Task<Scenario> CreateScenarioForUser(User user, string name, List<string> strategies, int? loopbackPeriod = 1)
        {
            var scenario = new Scenario(name, loopbackPeriod ?? 1)
            {
                User = user
            };

            foreach (var strategyName in strategies)
            {
                var strategy = await _tradingService.GetStrategyByNameAsync(strategyName);
                if (strategy != null && strategy.User?.Name == user.Name)
                {
                    scenario.AddIndicator(strategy);
                }
            }

            await _tradingService.InsertScenarioAsync(scenario);
            return scenario;
        }

        public async Task<IEnumerable<Indicator>> GetIndicatorsByUserAsync(User user)
        {
            var indicators = await GetIndicatorsAsync();
            return indicators.Where(s => s.User?.Name == user.Name);
        }

        public async Task<bool> DeleteIndicatorByUser(User user, string name)
        {
            var strategy = await _tradingService.GetStrategyByNameAsync(name);
            if (strategy != null && strategy.User?.Name == user.Name)
            {
                await _tradingService.DeleteStrategyAsync(strategy.Name);
                return true;
            }

            return false;
        }

        public async Task<bool> DeleteScenarioByUser(User user, string name)
        {
            try
            {
                var scenarios = await GetScenariosByUserAsync(user);
                foreach (var scenario in scenarios.Where(s => s.Name == name))
                {
                    await _tradingService.DeleteScenarioAsync(scenario.Name);
                }
                return true;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex.Message);
                return false;
            }
        }

        public async Task<bool> DeleteScenariosByUser(User user)
        {
            try
            {
                var scenarios = await GetScenariosByUserAsync(user);
                foreach (var scenario in scenarios)
                {
                    await _tradingService.DeleteScenarioAsync(scenario.Name);
                }
                return true;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex.Message);
                return false;
            }
        }

        public async Task<Scenario> GetScenarioByUser(User user, string name)
        {
            var scenario = await _tradingService.GetScenarioByNameAsync(name);
            return scenario != null && scenario.User?.Name == user.Name ? scenario : null;
        }

        public async Task<Indicator> CreateIndicatorForUser(User user, IndicatorType type, string name, int? period = null,
            int? fastPeriods = null, int? slowPeriods = null, int? signalPeriods = null,
            double? multiplier = null, int? stochPeriods = null, int? smoothPeriods = null,
            int? cyclePeriods = null)
        {
            // Create a new strategy using the existing implementation
            var strategy = await CreateStrategy(type, name, period, fastPeriods, slowPeriods, signalPeriods,
                multiplier, stochPeriods, smoothPeriods, cyclePeriods);

            // Set the user
            strategy.User = user;

            // Update the strategy to save the user property
            await _tradingService.UpdateStrategyAsync(strategy);

            return strategy;
        }

        public async Task<bool> DeleteStrategiesByUser(User user)
        {
            try
            {
                var strategies = await GetIndicatorsByUserAsync(user);
                foreach (var strategy in strategies)
                {
                    await _tradingService.DeleteStrategyAsync(strategy.Name);
                }
                return true;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex.Message);
                return false;
            }
        }

        public async Task<bool> UpdateScenarioByUser(User user, string name, List<string> strategies, int? loopbackPeriod)
        {
            var scenario = await _tradingService.GetScenarioByNameAsync(name);
            if (scenario == null || scenario.User?.Name != user.Name)
            {
                return false;
            }

            scenario.Indicators.Clear();
            scenario.LoopbackPeriod = loopbackPeriod ?? 1;

            foreach (var strategyName in strategies)
            {
                var strategy = await _tradingService.GetStrategyByNameAsync(strategyName);
                if (strategy != null && strategy.User?.Name == user.Name)
                {
                    scenario.AddIndicator(strategy);
                }
            }

            await _tradingService.UpdateScenarioAsync(scenario);
            return true;
        }

        public async Task<bool> UpdateIndicatorByUser(User user, IndicatorType indicatorType, string name, int? period,
            int? fastPeriods, int? slowPeriods, int? signalPeriods, double? multiplier,
            int? stochPeriods, int? smoothPeriods, int? cyclePeriods)
        {
            var strategy = await _tradingService.GetStrategyByNameAsync(name);
            if (strategy == null || strategy.User?.Name != user.Name)
            {
                return false;
            }

            // Use the existing update strategy logic
            var result = await UpdateStrategy(indicatorType, name, period, fastPeriods, slowPeriods,
                signalPeriods, multiplier, stochPeriods, smoothPeriods, cyclePeriods);

            return result;
        }

        public async Task<Scenario> GetScenarioByNameAndUserAsync(string scenarioName, User user)
        {
            var scenario = await _tradingService.GetScenarioByNameAsync(scenarioName);
            if (scenario == null)
            {
                throw new InvalidOperationException($"Scenario {scenarioName} not found for user {user.Name}");
            }

            return scenario;
        }
    }
}