From 34038245eb93d53ccdb42e9c628cb26cfab1709f Mon Sep 17 00:00:00 2001
From: "kay.one" <kay.one@gmail.com>
Date: Sun, 3 Feb 2013 22:27:58 -0800
Subject: [PATCH] added repo base

---
 NzbDrone.Core/Repository/RootDir.cs           | 16 +----
 .../RootFolders/RootFolderRepository.cs       | 58 ++++++++++++++-----
 2 files changed, 45 insertions(+), 29 deletions(-)

diff --git a/NzbDrone.Core/Repository/RootDir.cs b/NzbDrone.Core/Repository/RootDir.cs
index 7cb43d9e4..6e6383374 100644
--- a/NzbDrone.Core/Repository/RootDir.cs
+++ b/NzbDrone.Core/Repository/RootDir.cs
@@ -1,26 +1,14 @@
 using System.Collections.Generic;
+using NzbDrone.Core.RootFolders;
 using PetaPoco;
 
 namespace NzbDrone.Core.Repository
 {
-    public interface IRootDir
-    {
-        int Id { get; set; }
-        string Path { get; set; }
-
-        [ResultColumn]
-        ulong FreeSpace { get; set; }
-
-        [Ignore]
-        List<string> UnmappedFolders { get; set; }
-    }
 
     [TableName("RootDirs")]
     [PrimaryKey("Id", autoIncrement = true)]
-    public class RootDir : IRootDir
+    public class RootDir : BaseModel
     {
-        public virtual int Id { get; set; }
-
         public string Path { get; set; }
 
         [ResultColumn]
diff --git a/NzbDrone.Core/RootFolders/RootFolderRepository.cs b/NzbDrone.Core/RootFolders/RootFolderRepository.cs
index 3492c6a15..ff3f7dd7c 100644
--- a/NzbDrone.Core/RootFolders/RootFolderRepository.cs
+++ b/NzbDrone.Core/RootFolders/RootFolderRepository.cs
@@ -1,45 +1,73 @@
 using System.Collections.Generic;
+using Eloquera.Client;
 using NzbDrone.Core.Datastore;
 using NzbDrone.Core.Repository;
 using System.Linq;
 
 namespace NzbDrone.Core.RootFolders
 {
-    public interface IRootFolderRepository
+
+    public abstract class BaseModel
     {
-        List<RootDir> All();
-        RootDir Get(int rootFolderId);
-        RootDir Add(RootDir rootFolder);
+        [ID]
+        public int Id;
+    }
+
+    public interface IBasicRepository<TModel>
+    {
+        List<TModel> All();
+        TModel Get(int rootFolderId);
+        TModel Add(TModel rootFolder);
         void Delete(int rootFolderId);
     }
 
-    public class RootFolderRepository : IRootFolderRepository
+
+    public abstract class BasicRepository<TModel> : IBasicRepository<TModel> where TModel : BaseModel, new()
     {
-        private readonly EloqueraDb _db;
 
-        public RootFolderRepository(EloqueraDb db)
+
+        public BasicRepository(EloqueraDb eloqueraDb)
         {
-            _db = db;
+            EloqueraDb = eloqueraDb;
         }
 
-        public List<RootDir> All()
+        protected EloqueraDb EloqueraDb { get; private set; }
+
+
+        public List<TModel> All()
         {
-            return _db.AsQueryable<RootDir>().ToList();
+            return EloqueraDb.AsQueryable<TModel>().ToList();
         }
 
-        public RootDir Get(int rootFolderId)
+        public TModel Get(int rootFolderId)
         {
-            return _db.AsQueryable<RootDir>().Single(c => c.Id == rootFolderId);
+            return EloqueraDb.AsQueryable<TModel>().Single(c => c.Id == rootFolderId);
         }
 
-        public RootDir Add(RootDir rootFolder)
+        public TModel Add(TModel rootFolder)
         {
-            return _db.Insert(rootFolder);
+            return EloqueraDb.Insert(rootFolder);
         }
 
         public void Delete(int rootFolderId)
         {
-            _db.Delete(Get(rootFolderId));
+            var itemToDelete = Get(rootFolderId);
+            EloqueraDb.Delete(itemToDelete);
+        }
+    }
+
+    public interface IRootFolderRepository : IBasicRepository<RootDir>
+    {
+
+    }
+
+
+    //This way we only need to implement none_custom methods for repos, like custom queries etc... rest is done automagically.
+    public class RootFolderRepository : BasicRepository<RootDir>, IRootFolderRepository
+    {
+        public RootFolderRepository(EloqueraDb eloqueraDb)
+            : base(eloqueraDb)
+        {
         }
     }