EmailBill/WebApi/Controllers/Dto/BudgetDto.cs

namespace WebApi.Controllers.Dto;

public class BudgetDto
{
    public long Id { get; set; }
    public string Name { get; set; } = string.Empty;
    public BudgetPeriodType Type { get; set; }
    public decimal Limit { get; set; }
    public decimal Current { get; set; }
    public BudgetCategory Category { get; set; }
    public string[] SelectedCategories { get; set; } = Array.Empty<string>();
    public bool IsStopped { get; set; }
    public string StartDate { get; set; } = string.Empty;
    public string Period { get; set; } = string.Empty;
    public DateTime? PeriodStart { get; set; }
    public DateTime? PeriodEnd { get; set; }

    public static BudgetDto FromEntity(BudgetRecord entity, decimal currentAmount = 0, DateTime? referenceDate = null)
    {
        var date = referenceDate ?? DateTime.Now;
        var (start, end) = BudgetService.GetPeriodRange(entity.StartDate, entity.Type, date);
        
        return new BudgetDto
        {
            Id = entity.Id,
            Name = entity.Name,
            Type = entity.Type,
            Limit = entity.Limit,
            Current = currentAmount,
            Category = entity.Category,
            SelectedCategories = string.IsNullOrEmpty(entity.SelectedCategories) 
                ? Array.Empty<string>() 
                : entity.SelectedCategories.Split(','),
            IsStopped = entity.IsStopped,
            StartDate = entity.StartDate.ToString("yyyy-MM-dd"),
            Period = entity.Type switch
            {
                BudgetPeriodType.Year => $"{start:yy}年",
                BudgetPeriodType.Month => $"{start:yy}年第{start.Month}月",
                _ => $"{start:yyyy-MM-dd} ~ {end:yyyy-MM-dd}"
            },
            PeriodStart = start,
            PeriodEnd = end
        };
    }
}

public class CreateBudgetDto
{
    public string Name { get; set; } = string.Empty;
    public BudgetPeriodType Type { get; set; } = BudgetPeriodType.Month;
    public decimal Limit { get; set; }
    public BudgetCategory Category { get; set; }
    public string[] SelectedCategories { get; set; } = Array.Empty<string>();
    public DateTime? StartDate { get; set; }
}

public class UpdateBudgetDto : CreateBudgetDto
{
    public long Id { get; set; }
    public bool IsStopped { get; set; }
}