MigrateUserDb.cs 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216
  1. using System;
  2. using System.IO;
  3. using Emby.Server.Implementations.Data;
  4. using Jellyfin.Data;
  5. using Jellyfin.Database.Implementations;
  6. using Jellyfin.Database.Implementations.Entities;
  7. using Jellyfin.Database.Implementations.Enums;
  8. using Jellyfin.Extensions.Json;
  9. using Jellyfin.Server.Implementations.Users;
  10. using MediaBrowser.Controller;
  11. using MediaBrowser.Controller.Entities;
  12. using MediaBrowser.Model.Configuration;
  13. using MediaBrowser.Model.Serialization;
  14. using MediaBrowser.Model.Users;
  15. using Microsoft.Data.Sqlite;
  16. using Microsoft.EntityFrameworkCore;
  17. using Microsoft.Extensions.Logging;
  18. using JsonSerializer = System.Text.Json.JsonSerializer;
  19. namespace Jellyfin.Server.Migrations.Routines;
  20. /// <summary>
  21. /// The migration routine for migrating the user database to EF Core.
  22. /// </summary>
  23. #pragma warning disable CS0618 // Type or member is obsolete
  24. [JellyfinMigration("2025-04-20T10:00:00", nameof(MigrateUserDb), "5C4B82A2-F053-4009-BD05-B6FCAD82F14C")]
  25. public class MigrateUserDb : IMigrationRoutine
  26. #pragma warning restore CS0618 // Type or member is obsolete
  27. {
  28. private const string DbFilename = "users.db";
  29. private readonly ILogger<MigrateUserDb> _logger;
  30. private readonly IServerApplicationPaths _paths;
  31. private readonly IDbContextFactory<JellyfinDbContext> _provider;
  32. private readonly IXmlSerializer _xmlSerializer;
  33. /// <summary>
  34. /// Initializes a new instance of the <see cref="MigrateUserDb"/> class.
  35. /// </summary>
  36. /// <param name="logger">The logger.</param>
  37. /// <param name="paths">The server application paths.</param>
  38. /// <param name="provider">The database provider.</param>
  39. /// <param name="xmlSerializer">The xml serializer.</param>
  40. public MigrateUserDb(
  41. ILogger<MigrateUserDb> logger,
  42. IServerApplicationPaths paths,
  43. IDbContextFactory<JellyfinDbContext> provider,
  44. IXmlSerializer xmlSerializer)
  45. {
  46. _logger = logger;
  47. _paths = paths;
  48. _provider = provider;
  49. _xmlSerializer = xmlSerializer;
  50. }
  51. /// <inheritdoc/>
  52. public void Perform()
  53. {
  54. var dataPath = _paths.DataPath;
  55. _logger.LogInformation("Migrating the user database may take a while, do not stop Jellyfin.");
  56. using (var connection = new SqliteConnection($"Filename={Path.Combine(dataPath, DbFilename)}"))
  57. {
  58. connection.Open();
  59. using var dbContext = _provider.CreateDbContext();
  60. var queryResult = connection.Query("SELECT * FROM LocalUsersv2");
  61. dbContext.RemoveRange(dbContext.Users);
  62. dbContext.SaveChanges();
  63. foreach (var entry in queryResult)
  64. {
  65. UserMockup? mockup = JsonSerializer.Deserialize<UserMockup>(entry.GetStream(2), JsonDefaults.Options);
  66. if (mockup is null)
  67. {
  68. continue;
  69. }
  70. var userDataDir = Path.Combine(_paths.UserConfigurationDirectoryPath, mockup.Name);
  71. var configPath = Path.Combine(userDataDir, "config.xml");
  72. var config = File.Exists(configPath)
  73. ? (UserConfiguration?)_xmlSerializer.DeserializeFromFile(typeof(UserConfiguration), configPath) ?? new UserConfiguration()
  74. : new UserConfiguration();
  75. var policyPath = Path.Combine(userDataDir, "policy.xml");
  76. var policy = File.Exists(policyPath)
  77. ? (UserPolicy?)_xmlSerializer.DeserializeFromFile(typeof(UserPolicy), policyPath) ?? new UserPolicy()
  78. : new UserPolicy();
  79. policy.AuthenticationProviderId = policy.AuthenticationProviderId?.Replace(
  80. "Emby.Server.Implementations.Library",
  81. "Jellyfin.Server.Implementations.Users",
  82. StringComparison.Ordinal)
  83. ?? typeof(DefaultAuthenticationProvider).FullName;
  84. policy.PasswordResetProviderId = typeof(DefaultPasswordResetProvider).FullName;
  85. int? maxLoginAttempts = policy.LoginAttemptsBeforeLockout switch
  86. {
  87. -1 => null,
  88. 0 => 3,
  89. _ => policy.LoginAttemptsBeforeLockout
  90. };
  91. var user = new User(mockup.Name, policy.AuthenticationProviderId!, policy.PasswordResetProviderId!)
  92. {
  93. Id = entry.GetGuid(1),
  94. InternalId = entry.GetInt64(0),
  95. MaxParentalRatingScore = policy.MaxParentalRating,
  96. MaxParentalRatingSubScore = null,
  97. EnableUserPreferenceAccess = policy.EnableUserPreferenceAccess,
  98. RemoteClientBitrateLimit = policy.RemoteClientBitrateLimit,
  99. InvalidLoginAttemptCount = policy.InvalidLoginAttemptCount,
  100. LoginAttemptsBeforeLockout = maxLoginAttempts,
  101. SubtitleMode = config.SubtitleMode,
  102. HidePlayedInLatest = config.HidePlayedInLatest,
  103. EnableLocalPassword = config.EnableLocalPassword,
  104. PlayDefaultAudioTrack = config.PlayDefaultAudioTrack,
  105. DisplayCollectionsView = config.DisplayCollectionsView,
  106. DisplayMissingEpisodes = config.DisplayMissingEpisodes,
  107. AudioLanguagePreference = config.AudioLanguagePreference,
  108. RememberAudioSelections = config.RememberAudioSelections,
  109. EnableNextEpisodeAutoPlay = config.EnableNextEpisodeAutoPlay,
  110. RememberSubtitleSelections = config.RememberSubtitleSelections,
  111. SubtitleLanguagePreference = config.SubtitleLanguagePreference,
  112. Password = mockup.Password,
  113. LastLoginDate = mockup.LastLoginDate,
  114. LastActivityDate = mockup.LastActivityDate
  115. };
  116. if (mockup.ImageInfos.Length > 0)
  117. {
  118. ItemImageInfo info = mockup.ImageInfos[0];
  119. user.ProfileImage = new ImageInfo(info.Path)
  120. {
  121. LastModified = info.DateModified
  122. };
  123. }
  124. user.SetPermission(PermissionKind.IsAdministrator, policy.IsAdministrator);
  125. user.SetPermission(PermissionKind.IsHidden, policy.IsHidden);
  126. user.SetPermission(PermissionKind.IsDisabled, policy.IsDisabled);
  127. user.SetPermission(PermissionKind.EnableSharedDeviceControl, policy.EnableSharedDeviceControl);
  128. user.SetPermission(PermissionKind.EnableRemoteAccess, policy.EnableRemoteAccess);
  129. user.SetPermission(PermissionKind.EnableLiveTvManagement, policy.EnableLiveTvManagement);
  130. user.SetPermission(PermissionKind.EnableLiveTvAccess, policy.EnableLiveTvAccess);
  131. user.SetPermission(PermissionKind.EnableMediaPlayback, policy.EnableMediaPlayback);
  132. user.SetPermission(PermissionKind.EnableAudioPlaybackTranscoding, policy.EnableAudioPlaybackTranscoding);
  133. user.SetPermission(PermissionKind.EnableVideoPlaybackTranscoding, policy.EnableVideoPlaybackTranscoding);
  134. user.SetPermission(PermissionKind.EnableContentDeletion, policy.EnableContentDeletion);
  135. user.SetPermission(PermissionKind.EnableContentDownloading, policy.EnableContentDownloading);
  136. user.SetPermission(PermissionKind.EnableSyncTranscoding, policy.EnableSyncTranscoding);
  137. user.SetPermission(PermissionKind.EnableMediaConversion, policy.EnableMediaConversion);
  138. user.SetPermission(PermissionKind.EnableAllChannels, policy.EnableAllChannels);
  139. user.SetPermission(PermissionKind.EnableAllDevices, policy.EnableAllDevices);
  140. user.SetPermission(PermissionKind.EnableAllFolders, policy.EnableAllFolders);
  141. user.SetPermission(PermissionKind.EnableRemoteControlOfOtherUsers, policy.EnableRemoteControlOfOtherUsers);
  142. user.SetPermission(PermissionKind.EnablePlaybackRemuxing, policy.EnablePlaybackRemuxing);
  143. user.SetPermission(PermissionKind.ForceRemoteSourceTranscoding, policy.ForceRemoteSourceTranscoding);
  144. user.SetPermission(PermissionKind.EnablePublicSharing, policy.EnablePublicSharing);
  145. user.SetPermission(PermissionKind.EnableCollectionManagement, policy.EnableCollectionManagement);
  146. foreach (var policyAccessSchedule in policy.AccessSchedules)
  147. {
  148. user.AccessSchedules.Add(policyAccessSchedule);
  149. }
  150. user.SetPreference(PreferenceKind.BlockedTags, policy.BlockedTags);
  151. user.SetPreference(PreferenceKind.EnabledChannels, policy.EnabledChannels);
  152. user.SetPreference(PreferenceKind.EnabledDevices, policy.EnabledDevices);
  153. user.SetPreference(PreferenceKind.EnabledFolders, policy.EnabledFolders);
  154. user.SetPreference(PreferenceKind.EnableContentDeletionFromFolders, policy.EnableContentDeletionFromFolders);
  155. user.SetPreference(PreferenceKind.OrderedViews, config.OrderedViews);
  156. user.SetPreference(PreferenceKind.GroupedFolders, config.GroupedFolders);
  157. user.SetPreference(PreferenceKind.MyMediaExcludes, config.MyMediaExcludes);
  158. user.SetPreference(PreferenceKind.LatestItemExcludes, config.LatestItemsExcludes);
  159. dbContext.Users.Add(user);
  160. }
  161. dbContext.SaveChanges();
  162. }
  163. try
  164. {
  165. File.Move(Path.Combine(dataPath, DbFilename), Path.Combine(dataPath, DbFilename + ".old"));
  166. var journalPath = Path.Combine(dataPath, DbFilename + "-journal");
  167. if (File.Exists(journalPath))
  168. {
  169. File.Move(journalPath, Path.Combine(dataPath, DbFilename + ".old-journal"));
  170. }
  171. }
  172. catch (IOException e)
  173. {
  174. _logger.LogError(e, "Error renaming legacy user database to 'users.db.old'");
  175. }
  176. }
  177. #nullable disable
  178. internal class UserMockup
  179. {
  180. public string Password { get; set; }
  181. public string EasyPassword { get; set; }
  182. public DateTime? LastLoginDate { get; set; }
  183. public DateTime? LastActivityDate { get; set; }
  184. public string Name { get; set; }
  185. public ItemImageInfo[] ImageInfos { get; set; }
  186. }
  187. }