SceneMapping added to Services API

2026-04-20 21:55:03 -04:00 · 2013-01-31 20:18:11 -08:00
parent 4176e7e9e4
commit 27afb2402a
10 changed files with 171 additions and 15 deletions
@@ -0,0 +1,63 @@
+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);
+        }
+    }
+}