MigrateUserDb.cs 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233
  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. var userDbPath = Path.Combine(dataPath, DbFilename);
  56. if (!File.Exists(userDbPath))
  57. {
  58. _logger.LogWarning("{UserDbPath} doesn't exist, nothing to migrate", userDbPath);
  59. return;
  60. }
  61. _logger.LogInformation("Migrating the user database may take a while, do not stop Jellyfin.");
  62. using (var connection = new SqliteConnection($"Filename={userDbPath}"))
  63. {
  64. connection.Open();
  65. var tableQuery = connection.Query("SELECT count(*) FROM sqlite_master WHERE type='table' AND name='LocalUsersv2';");
  66. foreach (var row in tableQuery)
  67. {
  68. if (row.GetInt32(0) == 0)
  69. {
  70. _logger.LogWarning("Table 'LocalUsersv2' doesn't exist in {UserDbPath}, nothing to migrate", userDbPath);
  71. break;
  72. }
  73. }
  74. using var dbContext = _provider.CreateDbContext();
  75. var queryResult = connection.Query("SELECT * FROM LocalUsersv2");
  76. dbContext.RemoveRange(dbContext.Users);
  77. dbContext.SaveChanges();
  78. foreach (var entry in queryResult)
  79. {
  80. UserMockup? mockup = JsonSerializer.Deserialize<UserMockup>(entry.GetStream(2), JsonDefaults.Options);
  81. if (mockup is null)
  82. {
  83. continue;
  84. }
  85. var userDataDir = Path.Combine(_paths.UserConfigurationDirectoryPath, mockup.Name);
  86. var configPath = Path.Combine(userDataDir, "config.xml");
  87. var config = File.Exists(configPath)
  88. ? (UserConfiguration?)_xmlSerializer.DeserializeFromFile(typeof(UserConfiguration), configPath) ?? new UserConfiguration()
  89. : new UserConfiguration();
  90. var policyPath = Path.Combine(userDataDir, "policy.xml");
  91. var policy = File.Exists(policyPath)
  92. ? (UserPolicy?)_xmlSerializer.DeserializeFromFile(typeof(UserPolicy), policyPath) ?? new UserPolicy()
  93. : new UserPolicy();
  94. policy.AuthenticationProviderId = policy.AuthenticationProviderId?.Replace(
  95. "Emby.Server.Implementations.Library",
  96. "Jellyfin.Server.Implementations.Users",
  97. StringComparison.Ordinal)
  98. ?? typeof(DefaultAuthenticationProvider).FullName;
  99. policy.PasswordResetProviderId = typeof(DefaultPasswordResetProvider).FullName;
  100. int? maxLoginAttempts = policy.LoginAttemptsBeforeLockout switch
  101. {
  102. -1 => null,
  103. 0 => 3,
  104. _ => policy.LoginAttemptsBeforeLockout
  105. };
  106. var user = new User(mockup.Name, policy.AuthenticationProviderId!, policy.PasswordResetProviderId!)
  107. {
  108. Id = entry.GetGuid(1),
  109. InternalId = entry.GetInt64(0),
  110. MaxParentalRatingScore = policy.MaxParentalRating,
  111. MaxParentalRatingSubScore = null,
  112. EnableUserPreferenceAccess = policy.EnableUserPreferenceAccess,
  113. RemoteClientBitrateLimit = policy.RemoteClientBitrateLimit,
  114. InvalidLoginAttemptCount = policy.InvalidLoginAttemptCount,
  115. LoginAttemptsBeforeLockout = maxLoginAttempts,
  116. SubtitleMode = config.SubtitleMode,
  117. HidePlayedInLatest = config.HidePlayedInLatest,
  118. EnableLocalPassword = config.EnableLocalPassword,
  119. PlayDefaultAudioTrack = config.PlayDefaultAudioTrack,
  120. DisplayCollectionsView = config.DisplayCollectionsView,
  121. DisplayMissingEpisodes = config.DisplayMissingEpisodes,
  122. AudioLanguagePreference = config.AudioLanguagePreference,
  123. RememberAudioSelections = config.RememberAudioSelections,
  124. EnableNextEpisodeAutoPlay = config.EnableNextEpisodeAutoPlay,
  125. RememberSubtitleSelections = config.RememberSubtitleSelections,
  126. SubtitleLanguagePreference = config.SubtitleLanguagePreference,
  127. Password = mockup.Password,
  128. LastLoginDate = mockup.LastLoginDate,
  129. LastActivityDate = mockup.LastActivityDate
  130. };
  131. if (mockup.ImageInfos.Length > 0)
  132. {
  133. ItemImageInfo info = mockup.ImageInfos[0];
  134. user.ProfileImage = new ImageInfo(info.Path)
  135. {
  136. LastModified = info.DateModified
  137. };
  138. }
  139. user.SetPermission(PermissionKind.IsAdministrator, policy.IsAdministrator);
  140. user.SetPermission(PermissionKind.IsHidden, policy.IsHidden);
  141. user.SetPermission(PermissionKind.IsDisabled, policy.IsDisabled);
  142. user.SetPermission(PermissionKind.EnableSharedDeviceControl, policy.EnableSharedDeviceControl);
  143. user.SetPermission(PermissionKind.EnableRemoteAccess, policy.EnableRemoteAccess);
  144. user.SetPermission(PermissionKind.EnableLiveTvManagement, policy.EnableLiveTvManagement);
  145. user.SetPermission(PermissionKind.EnableLiveTvAccess, policy.EnableLiveTvAccess);
  146. user.SetPermission(PermissionKind.EnableMediaPlayback, policy.EnableMediaPlayback);
  147. user.SetPermission(PermissionKind.EnableAudioPlaybackTranscoding, policy.EnableAudioPlaybackTranscoding);
  148. user.SetPermission(PermissionKind.EnableVideoPlaybackTranscoding, policy.EnableVideoPlaybackTranscoding);
  149. user.SetPermission(PermissionKind.EnableContentDeletion, policy.EnableContentDeletion);
  150. user.SetPermission(PermissionKind.EnableContentDownloading, policy.EnableContentDownloading);
  151. user.SetPermission(PermissionKind.EnableSyncTranscoding, policy.EnableSyncTranscoding);
  152. user.SetPermission(PermissionKind.EnableMediaConversion, policy.EnableMediaConversion);
  153. user.SetPermission(PermissionKind.EnableAllChannels, policy.EnableAllChannels);
  154. user.SetPermission(PermissionKind.EnableAllDevices, policy.EnableAllDevices);
  155. user.SetPermission(PermissionKind.EnableAllFolders, policy.EnableAllFolders);
  156. user.SetPermission(PermissionKind.EnableRemoteControlOfOtherUsers, policy.EnableRemoteControlOfOtherUsers);
  157. user.SetPermission(PermissionKind.EnablePlaybackRemuxing, policy.EnablePlaybackRemuxing);
  158. user.SetPermission(PermissionKind.ForceRemoteSourceTranscoding, policy.ForceRemoteSourceTranscoding);
  159. user.SetPermission(PermissionKind.EnablePublicSharing, policy.EnablePublicSharing);
  160. user.SetPermission(PermissionKind.EnableCollectionManagement, policy.EnableCollectionManagement);
  161. foreach (var policyAccessSchedule in policy.AccessSchedules)
  162. {
  163. user.AccessSchedules.Add(policyAccessSchedule);
  164. }
  165. user.SetPreference(PreferenceKind.BlockedTags, policy.BlockedTags);
  166. user.SetPreference(PreferenceKind.EnabledChannels, policy.EnabledChannels);
  167. user.SetPreference(PreferenceKind.EnabledDevices, policy.EnabledDevices);
  168. user.SetPreference(PreferenceKind.EnabledFolders, policy.EnabledFolders);
  169. user.SetPreference(PreferenceKind.EnableContentDeletionFromFolders, policy.EnableContentDeletionFromFolders);
  170. user.SetPreference(PreferenceKind.OrderedViews, config.OrderedViews);
  171. user.SetPreference(PreferenceKind.GroupedFolders, config.GroupedFolders);
  172. user.SetPreference(PreferenceKind.MyMediaExcludes, config.MyMediaExcludes);
  173. user.SetPreference(PreferenceKind.LatestItemExcludes, config.LatestItemsExcludes);
  174. dbContext.Users.Add(user);
  175. }
  176. dbContext.SaveChanges();
  177. }
  178. try
  179. {
  180. File.Move(Path.Combine(dataPath, DbFilename), Path.Combine(dataPath, DbFilename + ".old"));
  181. var journalPath = Path.Combine(dataPath, DbFilename + "-journal");
  182. if (File.Exists(journalPath))
  183. {
  184. File.Move(journalPath, Path.Combine(dataPath, DbFilename + ".old-journal"));
  185. }
  186. }
  187. catch (IOException e)
  188. {
  189. _logger.LogError(e, "Error renaming legacy user database to 'users.db.old'");
  190. }
  191. }
  192. #nullable disable
  193. internal class UserMockup
  194. {
  195. public string Password { get; set; }
  196. public string EasyPassword { get; set; }
  197. public DateTime? LastLoginDate { get; set; }
  198. public DateTime? LastActivityDate { get; set; }
  199. public string Name { get; set; }
  200. public ItemImageInfo[] ImageInfos { get; set; }
  201. }
  202. }