MigrateUserDb.cs 11 KB

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