76 lines
2.7 KiB
C#
76 lines
2.7 KiB
C#
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;
|
|
}
|
|
}
|