using Microsoft.EntityFrameworkCore;

namespace SVSim.Database.Repositories.Guild;

public sealed class GuildRepository : IGuildRepository
{
    private readonly SVSimDbContext _db;

    public GuildRepository(SVSimDbContext db) { _db = db; }

    public Task<Entities.Guild.Guild?> GetByIdAsync(int guildId, CancellationToken ct = default)
        => _db.Guilds.FirstOrDefaultAsync(g => g.GuildId == guildId, ct);

    public Task<Entities.Guild.Guild?> GetActiveByIdAsync(int guildId, CancellationToken ct = default)
        => _db.Guilds.FirstOrDefaultAsync(g => g.GuildId == guildId && g.BreakupAt == null, ct);

    public Task<Entities.Guild.Guild?> GetWithMembersAsync(int guildId, CancellationToken ct = default)
        => _db.Guilds
            .Include(g => g.Members)
            .AsSplitQuery()
            .FirstOrDefaultAsync(g => g.GuildId == guildId, ct);

    public Task<bool> NameExistsAsync(string name, CancellationToken ct = default)
        => _db.Guilds.AnyAsync(g => g.Name == name && g.BreakupAt == null, ct);

    public async Task<Entities.Guild.Guild> AddAsync(Entities.Guild.Guild g, CancellationToken ct = default)
    {
        _db.Guilds.Add(g);
        await _db.SaveChangesAsync(ct);
        return g;
    }

    public async Task MarkBrokenUpAsync(int guildId, DateTime brokenUpAt, CancellationToken ct = default)
    {
        var guild = await _db.Guilds.FirstOrDefaultAsync(g => g.GuildId == guildId, ct);
        if (guild is null) return;
        guild.BreakupAt = brokenUpAt;
        await _db.SaveChangesAsync(ct);
    }

    public async Task<IReadOnlyList<Entities.Guild.Guild>> SearchAsync(
        string name, int activity, int joinCondition, int memberBucket,
        int maxMemberCap, int resultCap, CancellationToken ct = default)
    {
        IQueryable<Entities.Guild.Guild> q = _db.Guilds.Where(g => g.BreakupAt == null);

        if (!string.IsNullOrEmpty(name))
            q = q.Where(g => g.Name.Contains(name));

        if (activity != 0)
            q = q.Where(g => (int)g.Activity == activity);

        if (joinCondition != 0)
            q = q.Where(g => (int)g.JoinCondition == joinCondition);

        if (memberBucket != 0)
        {
            q = memberBucket switch
            {
                1 => q.Where(g => _db.GuildMembers.Count(m => m.GuildId == g.GuildId) >= 1
                                  && _db.GuildMembers.Count(m => m.GuildId == g.GuildId) <= 10),
                2 => q.Where(g => _db.GuildMembers.Count(m => m.GuildId == g.GuildId) >= 11
                                  && _db.GuildMembers.Count(m => m.GuildId == g.GuildId) <= 25),
                3 => q.Where(g => _db.GuildMembers.Count(m => m.GuildId == g.GuildId) >= 26
                                  && _db.GuildMembers.Count(m => m.GuildId == g.GuildId) <= maxMemberCap),
                _ => q,
            };
        }

        return await q.Take(resultCap).ToListAsync(ct);
    }
}