Migrated UserData from library sqlite db to jellyfin.db

1 files changed, 89 insertions, 0 deletions

diff --git a/Jellyfin.Server/Migrations/Routines/MigrateUserData.cs b/Jellyfin.Server/Migrations/Routines/MigrateUserData.cs
new file mode 100644
index 000000000..224534d43
--- /dev/null
+++ b/Jellyfin.Server/Migrations/Routines/MigrateUserData.cs
@@ -0,0 +1,89 @@
+using System;
+using System.Collections.Immutable;
+using System.IO;
+using System.Linq;
+using Emby.Server.Implementations.Data;
+using Jellyfin.Data.Entities;
+using Jellyfin.Server.Implementations;
+using MediaBrowser.Controller;
+using Microsoft.Data.Sqlite;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.Extensions.Logging;
+
+namespace Jellyfin.Server.Migrations.Routines;
+
+/// <summary>
+/// The migration routine for migrating the userdata database to EF Core.
+/// </summary>
+public class MigrateUserData : IMigrationRoutine
+{
+    private const string DbFilename = "library.db";
+
+    private readonly ILogger<MigrateUserDb> _logger;
+    private readonly IServerApplicationPaths _paths;
+    private readonly IDbContextFactory<JellyfinDbContext> _provider;
+
+    /// <summary>
+    /// Initializes a new instance of the <see cref="MigrateUserData"/> class.
+    /// </summary>
+    /// <param name="logger">The logger.</param>
+    /// <param name="provider">The database provider.</param>
+    /// <param name="paths">The server application paths.</param>
+    public MigrateUserData(
+        ILogger<MigrateUserDb> logger,
+        IDbContextFactory<JellyfinDbContext> provider,
+        IServerApplicationPaths paths)
+    {
+        _logger = logger;
+        _provider = provider;
+        _paths = paths;
+    }
+
+    /// <inheritdoc/>
+    public Guid Id => Guid.Parse("5bcb4197-e7c0-45aa-9902-963bceab5798");
+
+    /// <inheritdoc/>
+    public string Name => "MigrateUserData";
+
+    /// <inheritdoc/>
+    public bool PerformOnNewInstall => false;
+
+    /// <inheritdoc/>
+    public void Perform()
+    {
+        _logger.LogInformation("Migrating the userdata from library.db may take a while, do not stop Jellyfin.");
+
+        var dataPath = _paths.DataPath;
+        using var connection = new SqliteConnection($"Filename={Path.Combine(dataPath, DbFilename)}");
+
+        connection.Open();
+        using var dbContext = _provider.CreateDbContext();
+
+        var queryResult = connection.Query("SELECT key, userId, rating, played, playCount, isFavorite, playbackPositionTicks, lastPlayedDate, AudioStreamIndex, SubtitleStreamIndex FROM UserDatas");
+
+        dbContext.UserData.ExecuteDelete();
+
+        var users = dbContext.Users.AsNoTracking().ToImmutableArray();
+
+        foreach (SqliteDataReader dto in queryResult)
+        {
+            var entity = new UserData()
+            {
+                Key = dto.GetString(0),
+                UserId = users.ElementAt(dto.GetInt32(1)).Id,
+                Rating = dto.IsDBNull(2) ? null : dto.GetDouble(2),
+                Played = dto.GetBoolean(3),
+                PlayCount = dto.GetInt32(4),
+                IsFavorite = dto.GetBoolean(5),
+                PlaybackPositionTicks = dto.GetInt64(6),
+                LastPlayedDate = dto.IsDBNull(7) ? null : dto.GetDateTime(7),
+                AudioStreamIndex = dto.IsDBNull(8) ? null : dto.GetInt32(8),
+                SubtitleStreamIndex = dto.IsDBNull(9) ? null : dto.GetInt32(9),
+            };
+
+            dbContext.UserData.Add(entity);
+        }
+
+        dbContext.SaveChanges();
+    }
+}