using CharacterApi.Models;
using MongoDB.Driver;

namespace CharacterApi.Services;

public class CharacterStore
{
    private readonly IMongoCollection<Character> _col;
    private readonly IMongoCollection<VisibleLocation> _locations;

    public CharacterStore(IConfiguration cfg)
    {
        var cs = cfg["MongoDB:ConnectionString"] ?? "mongodb://127.0.0.1:27017";
        var dbName = cfg["MongoDB:DatabaseName"] ?? "GameDb";
        var client = new MongoClient(cs);
        var db = client.GetDatabase(dbName);
        _col = db.GetCollection<Character>("Characters");
        _locations = db.GetCollection<VisibleLocation>("Locations");

        var ownerIndex = Builders<Character>.IndexKeys.Ascending(c => c.OwnerUserId);
        _col.Indexes.CreateOne(new CreateIndexModel<Character>(ownerIndex));
    }

    public Task CreateAsync(Character character) => _col.InsertOneAsync(character);

    public Task<List<Character>> GetForOwnerAsync(string ownerUserId) =>
        _col.Find(c => c.OwnerUserId == ownerUserId).ToListAsync();

    public async Task<Character?> GetForOwnerByIdAsync(string id, string ownerUserId, bool allowAnyOwner)
    {
        var filter = Builders<Character>.Filter.Eq(c => c.Id, id);
        if (!allowAnyOwner)
        {
            filter = Builders<Character>.Filter.And(
                filter,
                Builders<Character>.Filter.Eq(c => c.OwnerUserId, ownerUserId)
            );
        }

        return await _col.Find(filter).FirstOrDefaultAsync();
    }

    public Task<List<VisibleLocation>> GetVisibleLocationsAsync(Character character)
    {
        var radius = character.VisionRadius > 0 ? character.VisionRadius : 3;
        var minX = character.Coord.X - radius;
        var maxX = character.Coord.X + radius;
        var minY = character.Coord.Y - radius;
        var maxY = character.Coord.Y + radius;

        var filter = Builders<VisibleLocation>.Filter.And(
            Builders<VisibleLocation>.Filter.Gte(l => l.Coord.X, minX),
            Builders<VisibleLocation>.Filter.Lte(l => l.Coord.X, maxX),
            Builders<VisibleLocation>.Filter.Gte(l => l.Coord.Y, minY),
            Builders<VisibleLocation>.Filter.Lte(l => l.Coord.Y, maxY)
        );

        return _locations.Find(filter).ToListAsync();
    }

    public async Task<bool> DeleteForOwnerAsync(string id, string ownerUserId, bool allowAnyOwner)
    {
        var filter = Builders<Character>.Filter.Eq(c => c.Id, id);
        if (!allowAnyOwner)
        {
            filter = Builders<Character>.Filter.And(
                filter,
                Builders<Character>.Filter.Eq(c => c.OwnerUserId, ownerUserId)
            );
        }

        var result = await _col.DeleteOneAsync(filter);
        return result.DeletedCount > 0;
    }
}