mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-07 06:17:14 -04:00
150 lines
5.5 KiB
C#
150 lines
5.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using Emby.Server.Implementations.Data;
|
|
using Jellyfin.Data.Entities.Security;
|
|
using Jellyfin.Server.Implementations;
|
|
using MediaBrowser.Controller;
|
|
using MediaBrowser.Controller.Library;
|
|
using Microsoft.Data.Sqlite;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace Jellyfin.Server.Migrations.Routines
|
|
{
|
|
/// <summary>
|
|
/// A migration that moves data from the authentication database into the new schema.
|
|
/// </summary>
|
|
public class MigrateAuthenticationDb : IMigrationRoutine
|
|
{
|
|
private const string DbFilename = "authentication.db";
|
|
|
|
private readonly ILogger<MigrateAuthenticationDb> _logger;
|
|
private readonly IDbContextFactory<JellyfinDbContext> _dbProvider;
|
|
private readonly IServerApplicationPaths _appPaths;
|
|
private readonly IUserManager _userManager;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="MigrateAuthenticationDb"/> class.
|
|
/// </summary>
|
|
/// <param name="logger">The logger.</param>
|
|
/// <param name="dbProvider">The database provider.</param>
|
|
/// <param name="appPaths">The server application paths.</param>
|
|
/// <param name="userManager">The user manager.</param>
|
|
public MigrateAuthenticationDb(
|
|
ILogger<MigrateAuthenticationDb> logger,
|
|
IDbContextFactory<JellyfinDbContext> dbProvider,
|
|
IServerApplicationPaths appPaths,
|
|
IUserManager userManager)
|
|
{
|
|
_logger = logger;
|
|
_dbProvider = dbProvider;
|
|
_appPaths = appPaths;
|
|
_userManager = userManager;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public Guid Id => Guid.Parse("5BD72F41-E6F3-4F60-90AA-09869ABE0E22");
|
|
|
|
/// <inheritdoc />
|
|
public string Name => "MigrateAuthenticationDatabase";
|
|
|
|
/// <inheritdoc />
|
|
public bool PerformOnNewInstall => false;
|
|
|
|
/// <inheritdoc />
|
|
public void Perform()
|
|
{
|
|
var dataPath = _appPaths.DataPath;
|
|
using (var connection = new SqliteConnection($"Filename={Path.Combine(dataPath, DbFilename)}"))
|
|
{
|
|
connection.Open();
|
|
using var dbContext = _dbProvider.CreateDbContext();
|
|
|
|
var authenticatedDevices = connection.Query("SELECT * FROM Tokens");
|
|
|
|
foreach (var row in authenticatedDevices)
|
|
{
|
|
var dateCreatedStr = row.GetString(9);
|
|
_ = DateTime.TryParse(dateCreatedStr, out var dateCreated);
|
|
var dateLastActivityStr = row.GetString(10);
|
|
_ = DateTime.TryParse(dateLastActivityStr, out var dateLastActivity);
|
|
|
|
if (row.IsDBNull(6))
|
|
{
|
|
dbContext.ApiKeys.Add(new ApiKey(row.GetString(3))
|
|
{
|
|
AccessToken = row.GetString(1),
|
|
DateCreated = dateCreated,
|
|
DateLastActivity = dateLastActivity
|
|
});
|
|
}
|
|
else
|
|
{
|
|
var userId = row.GetGuid(6);
|
|
var user = _userManager.GetUserById(userId);
|
|
if (user is null)
|
|
{
|
|
// User doesn't exist, don't bring over the device.
|
|
continue;
|
|
}
|
|
|
|
dbContext.Devices.Add(new Device(
|
|
userId,
|
|
row.GetString(3),
|
|
row.GetString(4),
|
|
row.GetString(5),
|
|
row.GetString(2))
|
|
{
|
|
AccessToken = row.GetString(1),
|
|
IsActive = row.GetBoolean(8),
|
|
DateCreated = dateCreated,
|
|
DateLastActivity = dateLastActivity
|
|
});
|
|
}
|
|
}
|
|
|
|
var deviceOptions = connection.Query("SELECT * FROM Devices");
|
|
var deviceIds = new HashSet<string>();
|
|
foreach (var row in deviceOptions)
|
|
{
|
|
if (row.IsDBNull(2))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
var deviceId = row.GetString(2);
|
|
if (deviceIds.Contains(deviceId))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
deviceIds.Add(deviceId);
|
|
|
|
dbContext.DeviceOptions.Add(new DeviceOptions(deviceId)
|
|
{
|
|
CustomName = row.IsDBNull(1) ? null : row.GetString(1)
|
|
});
|
|
}
|
|
|
|
dbContext.SaveChanges();
|
|
}
|
|
|
|
try
|
|
{
|
|
File.Move(Path.Combine(dataPath, DbFilename), Path.Combine(dataPath, DbFilename + ".old"));
|
|
|
|
var journalPath = Path.Combine(dataPath, DbFilename + "-journal");
|
|
if (File.Exists(journalPath))
|
|
{
|
|
File.Move(journalPath, Path.Combine(dataPath, DbFilename + ".old-journal"));
|
|
}
|
|
}
|
|
catch (IOException e)
|
|
{
|
|
_logger.LogError(e, "Error renaming legacy activity log database to 'authentication.db.old'");
|
|
}
|
|
}
|
|
}
|
|
}
|