Merge pull request #13589 from JPVenson/feature/DatabaseRefactor

[Feature] Database code refactor

1 files changed, 47 insertions, 0 deletions

diff --git a/src/Jellyfin.Database/Jellyfin.Database.Providers.Sqlite/ModelBuilderExtensions.cs b/src/Jellyfin.Database/Jellyfin.Database.Providers.Sqlite/ModelBuilderExtensions.cs
new file mode 100644
index 000000000..41375874d
--- /dev/null
+++ b/src/Jellyfin.Database/Jellyfin.Database.Providers.Sqlite/ModelBuilderExtensions.cs
@@ -0,0 +1,47 @@
+using System;
+using Jellyfin.Database.Providers.Sqlite.ValueConverters;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Storage.ValueConversion;
+
+namespace Jellyfin.Database.Providers.Sqlite;
+
+/// <summary>
+/// Model builder extensions.
+/// </summary>
+public static class ModelBuilderExtensions
+{
+    /// <summary>
+    /// Specify value converter for the object type.
+    /// </summary>
+    /// <param name="modelBuilder">The model builder.</param>
+    /// <param name="converter">The <see cref="ValueConverter{TModel,TProvider}"/>.</param>
+    /// <typeparam name="T">The type to convert.</typeparam>
+    /// <returns>The modified <see cref="ModelBuilder"/>.</returns>
+    public static ModelBuilder UseValueConverterForType<T>(this ModelBuilder modelBuilder, ValueConverter converter)
+    {
+        var type = typeof(T);
+        foreach (var entityType in modelBuilder.Model.GetEntityTypes())
+        {
+            foreach (var property in entityType.GetProperties())
+            {
+                if (property.ClrType == type)
+                {
+                    property.SetValueConverter(converter);
+                }
+            }
+        }
+
+        return modelBuilder;
+    }
+
+    /// <summary>
+    /// Specify the default <see cref="DateTimeKind"/>.
+    /// </summary>
+    /// <param name="modelBuilder">The model builder to extend.</param>
+    /// <param name="kind">The <see cref="DateTimeKind"/> to specify.</param>
+    public static void SetDefaultDateTimeKind(this ModelBuilder modelBuilder, DateTimeKind kind)
+    {
+        modelBuilder.UseValueConverterForType<DateTime>(new DateTimeKindValueConverter(kind));
+        modelBuilder.UseValueConverterForType<DateTime?>(new DateTimeKindValueConverter(kind));
+    }
+}