From 4eee5cfd5f10a34dffbbb997ed8bc89ffe9b7c2e Mon Sep 17 00:00:00 2001
From: "Morten@Thinkpad-X220" <unknown@kiln.example.com>
Date: Tue, 9 Oct 2012 08:45:15 -0200
Subject: [PATCH] Adds template repository and an abstract file repository
 U4-967

---
 .../Repositories/FileRepository.cs            | 71 +++++++++++++++++++
 .../Repositories/ITemplateRepository.cs       |  8 +++
 .../Repositories/ScriptRepository.cs          | 53 +++-----------
 .../Repositories/TemplateRepository.cs        | 60 ++++++++++++++++
 src/Umbraco.Core/Umbraco.Core.csproj          |  3 +
 5 files changed, 150 insertions(+), 45 deletions(-)
 create mode 100644 src/Umbraco.Core/Persistence/Repositories/FileRepository.cs
 create mode 100644 src/Umbraco.Core/Persistence/Repositories/ITemplateRepository.cs
 create mode 100644 src/Umbraco.Core/Persistence/Repositories/TemplateRepository.cs

diff --git a/src/Umbraco.Core/Persistence/Repositories/FileRepository.cs b/src/Umbraco.Core/Persistence/Repositories/FileRepository.cs
new file mode 100644
index 0000000000..a0981caa58
--- /dev/null
+++ b/src/Umbraco.Core/Persistence/Repositories/FileRepository.cs
@@ -0,0 +1,71 @@
+using System.Collections.Generic;
+using System.IO;
+using System.Text;
+using Umbraco.Core.IO;
+using Umbraco.Core.Models;
+using Umbraco.Core.Persistence.UnitOfWork;
+
+namespace Umbraco.Core.Persistence.Repositories
+{
+    internal abstract class FileRepository<TId, TEntity> : IRepository<TId, TEntity> 
+        where TEntity : IFile
+    {
+        private IUnitOfWork _work;
+        private readonly IFileSystem _fileSystem;
+
+        protected FileRepository(IUnitOfWork work, IFileSystem fileSystem)
+        {
+            _work = work;
+            _fileSystem = fileSystem;
+        }
+
+        /// <summary>
+        /// Returns the Unit of Work added to the repository
+        /// </summary>
+        protected IUnitOfWork UnitOfWork
+        {
+            get { return _work; }
+        }
+
+        protected IFileSystem FileSystem
+        {
+            get { return _fileSystem; }
+        }
+
+        #region Implementation of IRepository<TId,TEntity>
+
+        public virtual void AddOrUpdate(TEntity entity)
+        {
+            var stream = new MemoryStream(Encoding.UTF8.GetBytes(entity.Content));
+            FileSystem.AddFile(entity.Name, stream, true);
+        }
+
+        public void Delete(TEntity entity)
+        {
+            if (_fileSystem.FileExists(entity.Name))
+            {
+                _fileSystem.DeleteFile(entity.Name);
+            }
+            else if(_fileSystem.FileExists(entity.Path))
+            {
+                _fileSystem.DeleteFile(entity.Path);
+            }
+        }
+
+        public abstract TEntity Get(TId id);
+
+        public abstract IEnumerable<TEntity> GetAll(params TId[] ids);
+
+        public bool Exists(TId id)
+        {
+            return _fileSystem.FileExists(id.ToString());
+        }
+
+        public void SetUnitOfWork(IUnitOfWork work)
+        {
+            _work = work;
+        }
+
+        #endregion
+    }
+}
\ No newline at end of file
diff --git a/src/Umbraco.Core/Persistence/Repositories/ITemplateRepository.cs b/src/Umbraco.Core/Persistence/Repositories/ITemplateRepository.cs
new file mode 100644
index 0000000000..ab3a1def12
--- /dev/null
+++ b/src/Umbraco.Core/Persistence/Repositories/ITemplateRepository.cs
@@ -0,0 +1,8 @@
+using Umbraco.Core.Models;
+
+namespace Umbraco.Core.Persistence.Repositories
+{
+    public interface ITemplateRepository : IRepository<string, Template>
+    {
+    }
+}
\ No newline at end of file
diff --git a/src/Umbraco.Core/Persistence/Repositories/ScriptRepository.cs b/src/Umbraco.Core/Persistence/Repositories/ScriptRepository.cs
index b4a23d9d0f..99b39a6128 100644
--- a/src/Umbraco.Core/Persistence/Repositories/ScriptRepository.cs
+++ b/src/Umbraco.Core/Persistence/Repositories/ScriptRepository.cs
@@ -1,6 +1,5 @@
 using System;
 using System.Collections.Generic;
-using System.IO;
 using System.Linq;
 using System.Text;
 using Umbraco.Core.IO;
@@ -12,61 +11,35 @@ namespace Umbraco.Core.Persistence.Repositories
     /// <summary>
     /// Represents the Script Repository
     /// </summary>
-    internal class ScriptRepository : IScriptRepository
+    internal class ScriptRepository : FileRepository<string, Script>, IScriptRepository
     {
-        private IUnitOfWork _work;
-        private readonly IFileSystem _fileSystem;
-
         public ScriptRepository(IUnitOfWork work)
+            : base(work, FileSystemProviderManager.Current.GetFileSystemProvider("script"))
         {
-            _work = work;
-            _fileSystem = FileSystemProviderManager.Current.GetFileSystemProvider("script");
-        }
-
-        /// <summary>
-        /// Returns the Unit of Work added to the repository
-        /// </summary>
-        protected IUnitOfWork UnitOfWork
-        {
-            get { return _work; }
         }
 
         #region Implementation of IRepository<string,Script>
 
-        public void AddOrUpdate(Script entity)
+        public override Script Get(string id)
         {
-            var stream = new MemoryStream(Encoding.UTF8.GetBytes(entity.Content));
-            _fileSystem.AddFile(entity.Name, stream, true);
-        }
-
-        public void Delete(Script entity)
-        {
-            if (_fileSystem.FileExists(entity.Name))
-            {
-                _fileSystem.DeleteFile(entity.Name);
-            }
-        }
-
-        public Script Get(string id)
-        {
-            if(!_fileSystem.FileExists(id))
+            if(!FileSystem.FileExists(id))
             {
                 throw new Exception(string.Format("The file {0} was not found", id));
             }
 
-            var stream = _fileSystem.OpenFile(id);
+            var stream = FileSystem.OpenFile(id);
             byte[] bytes = new byte[stream.Length];
             stream.Position = 0;
             stream.Read(bytes, 0, (int)stream.Length);
             var content = Encoding.UTF8.GetString(bytes);
 
-            var path = _fileSystem.GetRelativePath(id);
+            var path = FileSystem.GetRelativePath(id);
 
             var script = new Script(path) {Content = content};
             return script;
         }
 
-        public IEnumerable<Script> GetAll(params string[] ids)
+        public override IEnumerable<Script> GetAll(params string[] ids)
         {
             if (ids.Any())
             {
@@ -77,7 +50,7 @@ namespace Umbraco.Core.Persistence.Repositories
             }
             else
             {
-                var files = _fileSystem.GetFiles("", "*");
+                var files = FileSystem.GetFiles("", "*");
                 foreach (var file in files)
                 {
                     yield return Get(file);
@@ -85,16 +58,6 @@ namespace Umbraco.Core.Persistence.Repositories
             }
         }
 
-        public bool Exists(string id)
-        {
-            return _fileSystem.FileExists(id);
-        }
-
-        public void SetUnitOfWork(IUnitOfWork work)
-        {
-            _work = work;
-        }
-
         #endregion
     }
 }
\ No newline at end of file
diff --git a/src/Umbraco.Core/Persistence/Repositories/TemplateRepository.cs b/src/Umbraco.Core/Persistence/Repositories/TemplateRepository.cs
new file mode 100644
index 0000000000..c00e94a2e1
--- /dev/null
+++ b/src/Umbraco.Core/Persistence/Repositories/TemplateRepository.cs
@@ -0,0 +1,60 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using Umbraco.Core.IO;
+using Umbraco.Core.Models;
+using Umbraco.Core.Persistence.UnitOfWork;
+
+namespace Umbraco.Core.Persistence.Repositories
+{
+    internal class TemplateRepository : FileRepository<string, Template>, ITemplateRepository
+    {
+        public TemplateRepository(IUnitOfWork work)
+            : base(work, FileSystemProviderManager.Current.GetFileSystemProvider("template"))
+        {
+        }
+
+        #region Overrides of FileRepository<string,Template>
+
+        public override Template Get(string id)
+        {
+            if (!FileSystem.FileExists(id))
+            {
+                throw new Exception(string.Format("The file {0} was not found", id));
+            }
+
+            var stream = FileSystem.OpenFile(id);
+            byte[] bytes = new byte[stream.Length];
+            stream.Position = 0;
+            stream.Read(bytes, 0, (int)stream.Length);
+            var content = Encoding.UTF8.GetString(bytes);
+
+            var path = FileSystem.GetRelativePath(id);
+
+            var template = new Template(path) { Content = content };
+            return template;
+        }
+
+        public override IEnumerable<Template> GetAll(params string[] ids)
+        {
+            if (ids.Any())
+            {
+                foreach (var id in ids)
+                {
+                    yield return Get(id);
+                }
+            }
+            else
+            {
+                var files = FileSystem.GetFiles("", "*");
+                foreach (var file in files)
+                {
+                    yield return Get(file);
+                }
+            }
+        }
+
+        #endregion
+    }
+}
\ No newline at end of file
diff --git a/src/Umbraco.Core/Umbraco.Core.csproj b/src/Umbraco.Core/Umbraco.Core.csproj
index 27f1a1a460..f364eb3b1c 100644
--- a/src/Umbraco.Core/Umbraco.Core.csproj
+++ b/src/Umbraco.Core/Umbraco.Core.csproj
@@ -121,6 +121,7 @@
     <Compile Include="Persistence\Repositories\ContentTypeRepository.cs" />
     <Compile Include="Persistence\Repositories\DataTypeDefinitionRepository.cs" />
     <Compile Include="Persistence\Repositories\DictionaryRepository.cs" />
+    <Compile Include="Persistence\Repositories\FileRepository.cs" />
     <Compile Include="Persistence\Repositories\IContentRepository.cs" />
     <Compile Include="Persistence\Repositories\IContentTypeRepository.cs" />
     <Compile Include="Persistence\Repositories\IDataTypeDefinitionRepository.cs" />
@@ -133,6 +134,7 @@
     <Compile Include="Persistence\Relators\TabPropertyTypeRelator.cs" />
     <Compile Include="Persistence\Repositories\IScriptRepository.cs" />
     <Compile Include="Persistence\Repositories\IRepository.cs" />
+    <Compile Include="Persistence\Repositories\ITemplateRepository.cs" />
     <Compile Include="Persistence\Repositories\LanguageRepository.cs" />
     <Compile Include="Persistence\Repositories\MacroRepository.cs" />
     <Compile Include="Persistence\Repositories\PetaPocoRepositoryBase.cs" />
@@ -140,6 +142,7 @@
     <Compile Include="Persistence\Repositories\RelationTypeRepository.cs" />
     <Compile Include="Persistence\Repositories\RepositoryBase.cs" />
     <Compile Include="Persistence\Repositories\ScriptRepository.cs" />
+    <Compile Include="Persistence\Repositories\TemplateRepository.cs" />
     <Compile Include="Persistence\RepositoryResolver.cs" />
     <Compile Include="Persistence\TransactionType.cs" />
     <Compile Include="Persistence\UnitOfWork\FileUnitOfWork.cs" />