You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readarr/NzbDrone.Services.Api/SceneMapping/SceneMappingRepository.cs

63 lines
2.0 KiB

using System;
using System.Collections.Generic;
using System.Linq;
using System.Web;
using MongoDB.Bson;
using MongoDB.Driver;
using MongoDB.Driver.Builders;
using NzbDrone.Services.Api.DailySeries;
namespace NzbDrone.Services.Api.SceneMapping
{
public class SceneMappingRepository
{
private readonly MongoDatabase _mongoDb;
public SceneMappingRepository(MongoDatabase mongoDb)
{
_mongoDb = mongoDb;
}
public SceneMappingRepository()
{
}
public List<SceneMappingModel> All()
{
return _mongoDb.GetCollection<SceneMappingModel>(SceneMappingModel.CollectionName).FindAll().ToList();
}
public List<SceneMappingModel> Public()
{
var query = Query<SceneMappingModel>.EQ(d => d.Public, true);
return _mongoDb.GetCollection<SceneMappingModel>(SceneMappingModel.CollectionName).Find(query).ToList();
}
public SceneMappingModel Find(ObjectId mapId)
{
var query = Query<SceneMappingModel>.EQ(d => d.MapId, mapId);
return _mongoDb.GetCollection<SceneMappingModel>(SceneMappingModel.CollectionName).FindOne(query);
}
public ObjectId Insert(SceneMappingModel mapping)
{
mapping.MapId = ObjectId.GenerateNewId();
_mongoDb.GetCollection<SceneMappingModel>(SceneMappingModel.CollectionName).Insert(mapping);
return mapping.MapId;
}
public void Delete(ObjectId mapId)
{
var query = Query<SceneMappingModel>.EQ(d => d.MapId, mapId);
_mongoDb.GetCollection<SceneMappingModel>(SceneMappingModel.CollectionName).Remove(query);
}
public void TogglePublic(ObjectId mapId, bool status)
{
var query = Query<SceneMappingModel>.EQ(d => d.MapId, mapId);
var update = Update<SceneMappingModel>.Set(d => d.Public, status);
_mongoDb.GetCollection<SceneMappingModel>(SceneMappingModel.CollectionName).Update(query, update);
}
}
}