AssetManager.API/AssetManager.Services/StrategyService.cs

using AssetManager.Data;
using AssetManager.Models.DTOs;
using SqlSugar;

namespace AssetManager.Services;

public class StrategyService : IStrategyService
{
    private readonly ISqlSugarClient _db;

    public StrategyService(ISqlSugarClient db)
    {
        _db = db;
    }

    public Strategy CreateStrategy(CreateStrategyRequest request, string userId)
    {
        var strategy = new Strategy
        {
            Id = Guid.NewGuid().ToString(),
            UserId = userId,
            Alias = request.name,
            Type = request.type,
            Description = request.description,
            Tags = System.Text.Json.JsonSerializer.Serialize(request.tags),
            RiskLevel = request.riskLevel,
            Config = System.Text.Json.JsonSerializer.Serialize(request.parameters),
            CreatedAt = DateTime.Now,
            UpdatedAt = DateTime.Now
        };

        _db.Insertable(strategy).ExecuteCommand();
        return strategy;
    }

    public List<Strategy> GetStrategies(string userId)
    {
        return _db.Queryable<Strategy>()
            .Where(it => it.UserId == userId)
            .ToList();
    }

    public Strategy GetStrategyById(string id, string userId)
    {
        var strategy = _db.Queryable<Strategy>()
            .Where(it => it.Id == id && it.UserId == userId)
            .First();

        if (strategy == null)
        {
            throw new Exception("Strategy not found or access denied");
        }

        return strategy;
    }

    public Strategy UpdateStrategy(string id, UpdateStrategyRequest request, string userId)
    {
        var strategy = GetStrategyById(id, userId);

        strategy.Alias = request.name;
        strategy.Type = request.type;
        strategy.Description = request.description;
        strategy.Tags = System.Text.Json.JsonSerializer.Serialize(request.tags);
        strategy.RiskLevel = request.riskLevel;
        strategy.Config = System.Text.Json.JsonSerializer.Serialize(request.parameters);
        strategy.UpdatedAt = DateTime.Now;

        _db.Updateable(strategy).ExecuteCommand();
        return strategy;
    }

    public bool DeleteStrategy(string id, string userId)
    {
        var strategy = GetStrategyById(id, userId);
        return _db.Deleteable(strategy).ExecuteCommand() > 0;
    }
}