From 49e9491d3b103618757d65e0d32a26904f350047 Mon Sep 17 00:00:00 2001
From: enisn <info@enisnecipoglu.com>
Date: Tue, 5 Apr 2022 10:51:31 +0300
Subject: [PATCH] Add BlogFeatureDataSeedContributor

---
 .../Blogs/BlogFeatureDataSeedContributor.cs   | 29 +++++++++++++++++++
 .../Volo/CmsKit/Blogs/BlogFeatureManager.cs   | 20 +++++++++++++
 2 files changed, 49 insertions(+)
 create mode 100644 modules/cms-kit/src/Volo.CmsKit.Domain/Volo/CmsKit/Blogs/BlogFeatureDataSeedContributor.cs

diff --git a/modules/cms-kit/src/Volo.CmsKit.Domain/Volo/CmsKit/Blogs/BlogFeatureDataSeedContributor.cs b/modules/cms-kit/src/Volo.CmsKit.Domain/Volo/CmsKit/Blogs/BlogFeatureDataSeedContributor.cs
new file mode 100644
index 0000000000..b212e5e21a
--- /dev/null
+++ b/modules/cms-kit/src/Volo.CmsKit.Domain/Volo/CmsKit/Blogs/BlogFeatureDataSeedContributor.cs
@@ -0,0 +1,29 @@
+using System.Threading.Tasks;
+using Volo.Abp.Data;
+using Volo.Abp.DependencyInjection;
+
+namespace Volo.CmsKit.Blogs;
+
+public class BlogFeatureDataSeedContributor : IDataSeedContributor, ITransientDependency
+{
+    private readonly BlogFeatureManager _blogFeatureManager;
+    private readonly IBlogRepository _blogRepository;
+
+    public BlogFeatureDataSeedContributor(
+        BlogFeatureManager blogFeatureManager,
+        IBlogRepository blogRepository)
+    {
+        _blogFeatureManager = blogFeatureManager;
+        _blogRepository = blogRepository;
+    }
+
+    public async Task SeedAsync(DataSeedContext context)
+    {
+        var blogs = await _blogRepository.GetListAsync();
+
+        foreach (var blog in blogs)
+        {
+            await _blogFeatureManager.SetDefaultsIfNotSetAsync(blog.Id);
+        }
+    }
+}
diff --git a/modules/cms-kit/src/Volo.CmsKit.Domain/Volo/CmsKit/Blogs/BlogFeatureManager.cs b/modules/cms-kit/src/Volo.CmsKit.Domain/Volo/CmsKit/Blogs/BlogFeatureManager.cs
index 77a519daf3..f9dd101ef9 100644
--- a/modules/cms-kit/src/Volo.CmsKit.Domain/Volo/CmsKit/Blogs/BlogFeatureManager.cs
+++ b/modules/cms-kit/src/Volo.CmsKit.Domain/Volo/CmsKit/Blogs/BlogFeatureManager.cs
@@ -44,4 +44,24 @@ public class BlogFeatureManager : DomainService
             await SetAsync(blogId, feature.FeatureName, isEnabled: true);
         }
     }
+
+    public async Task SetIfNotSetAsync(Guid blogId, string featureName, bool isEnabled)
+    {
+        var blogFeature = await BlogFeatureRepository.FindAsync(blogId, featureName);
+        if (blogFeature == null)
+        {
+            var newBlogFeature = new BlogFeature(blogId, featureName, isEnabled);
+            await BlogFeatureRepository.InsertAsync(newBlogFeature);
+        }
+    }
+
+    public async Task SetDefaultsIfNotSetAsync(Guid blogId)
+    {
+        var defaultFeatures = await DefaultBlogFeatureProvider.GetDefaultFeaturesAsync(blogId);
+
+        foreach (var feature in defaultFeatures)
+        {
+            await SetIfNotSetAsync(blogId, feature.FeatureName, isEnabled: true);
+        }
+    }
 }