mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 05:47:14 -04:00
Add Argument*Exceptions now use proper nameof operators. Added exception messages to quite a few Argument*Exceptions. Fixed rethorwing to be proper syntax. Added a ton of null checkes. (This is only a start, there are about 500 places that need proper null handling) Added some TODOs to log certain exceptions. Fix sln again. Fixed all AssemblyInfo's and added proper copyright (where I could find them) We live in *current year*. Fixed the use of braces. Fixed a ton of properties, and made a fair amount of functions static that should be and can be static. Made more Methods that should be static static. You can now use static to find bad functions! Removed unused variable. And added one more proper XML comment.
230 lines
7.2 KiB
C#
230 lines
7.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Threading;
|
|
using MediaBrowser.Controller;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Persistence;
|
|
using MediaBrowser.Model.IO;
|
|
using Microsoft.Extensions.Logging;
|
|
using MediaBrowser.Model.Serialization;
|
|
using SQLitePCL.pretty;
|
|
|
|
namespace Emby.Server.Implementations.Data
|
|
{
|
|
/// <summary>
|
|
/// Class SQLiteUserRepository
|
|
/// </summary>
|
|
public class SqliteUserRepository : BaseSqliteRepository, IUserRepository
|
|
{
|
|
private readonly IJsonSerializer _jsonSerializer;
|
|
|
|
public SqliteUserRepository(ILogger logger, IServerApplicationPaths appPaths, IJsonSerializer jsonSerializer)
|
|
: base(logger)
|
|
{
|
|
_jsonSerializer = jsonSerializer;
|
|
|
|
DbFilePath = Path.Combine(appPaths.DataPath, "users.db");
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the name of the repository
|
|
/// </summary>
|
|
/// <value>The name.</value>
|
|
public string Name => "SQLite";
|
|
|
|
/// <summary>
|
|
/// Opens the connection to the database
|
|
/// </summary>
|
|
/// <returns>Task.</returns>
|
|
public void Initialize()
|
|
{
|
|
using (var connection = CreateConnection())
|
|
{
|
|
RunDefaultInitialization(connection);
|
|
|
|
var localUsersTableExists = TableExists(connection, "LocalUsersv2");
|
|
|
|
connection.RunQueries(new[] {
|
|
"create table if not exists LocalUsersv2 (Id INTEGER PRIMARY KEY, guid GUID NOT NULL, data BLOB NOT NULL)",
|
|
"drop index if exists idx_users"
|
|
});
|
|
|
|
if (!localUsersTableExists && TableExists(connection, "Users"))
|
|
{
|
|
TryMigrateToLocalUsersTable(connection);
|
|
}
|
|
}
|
|
}
|
|
|
|
private void TryMigrateToLocalUsersTable(ManagedConnection connection)
|
|
{
|
|
try
|
|
{
|
|
connection.RunQueries(new[]
|
|
{
|
|
"INSERT INTO LocalUsersv2 (guid, data) SELECT guid,data from users"
|
|
});
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Logger.LogError(ex, "Error migrating users database");
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Save a user in the repo
|
|
/// </summary>
|
|
public void CreateUser(User user)
|
|
{
|
|
if (user == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(user));
|
|
}
|
|
|
|
var serialized = _jsonSerializer.SerializeToBytes(user);
|
|
|
|
using (WriteLock.Write())
|
|
{
|
|
using (var connection = CreateConnection())
|
|
{
|
|
connection.RunInTransaction(db =>
|
|
{
|
|
using (var statement = db.PrepareStatement("insert into LocalUsersv2 (guid, data) values (@guid, @data)"))
|
|
{
|
|
statement.TryBind("@guid", user.Id.ToGuidBlob());
|
|
statement.TryBind("@data", serialized);
|
|
|
|
statement.MoveNext();
|
|
}
|
|
|
|
var createdUser = GetUser(user.Id, false);
|
|
|
|
if (createdUser == null)
|
|
{
|
|
throw new ApplicationException("created user should never be null");
|
|
}
|
|
|
|
user.InternalId = createdUser.InternalId;
|
|
|
|
}, TransactionMode);
|
|
}
|
|
}
|
|
}
|
|
|
|
public void UpdateUser(User user)
|
|
{
|
|
if (user == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(user));
|
|
}
|
|
|
|
var serialized = _jsonSerializer.SerializeToBytes(user);
|
|
|
|
using (WriteLock.Write())
|
|
{
|
|
using (var connection = CreateConnection())
|
|
{
|
|
connection.RunInTransaction(db =>
|
|
{
|
|
using (var statement = db.PrepareStatement("update LocalUsersv2 set data=@data where Id=@InternalId"))
|
|
{
|
|
statement.TryBind("@InternalId", user.InternalId);
|
|
statement.TryBind("@data", serialized);
|
|
statement.MoveNext();
|
|
}
|
|
|
|
}, TransactionMode);
|
|
}
|
|
}
|
|
}
|
|
|
|
private User GetUser(Guid guid, bool openLock)
|
|
{
|
|
using (openLock ? WriteLock.Read() : null)
|
|
{
|
|
using (var connection = CreateConnection(true))
|
|
{
|
|
using (var statement = connection.PrepareStatement("select id,guid,data from LocalUsersv2 where guid=@guid"))
|
|
{
|
|
statement.TryBind("@guid", guid);
|
|
|
|
foreach (var row in statement.ExecuteQuery())
|
|
{
|
|
return GetUser(row);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
private User GetUser(IReadOnlyList<IResultSetValue> row)
|
|
{
|
|
var id = row[0].ToInt64();
|
|
var guid = row[1].ReadGuidFromBlob();
|
|
|
|
using (var stream = new MemoryStream(row[2].ToBlob()))
|
|
{
|
|
stream.Position = 0;
|
|
var user = _jsonSerializer.DeserializeFromStream<User>(stream);
|
|
user.InternalId = id;
|
|
user.Id = guid;
|
|
return user;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Retrieve all users from the database
|
|
/// </summary>
|
|
/// <returns>IEnumerable{User}.</returns>
|
|
public List<User> RetrieveAllUsers()
|
|
{
|
|
var list = new List<User>();
|
|
|
|
using (WriteLock.Read())
|
|
{
|
|
using (var connection = CreateConnection(true))
|
|
{
|
|
foreach (var row in connection.Query("select id,guid,data from LocalUsersv2"))
|
|
{
|
|
list.Add(GetUser(row));
|
|
}
|
|
}
|
|
}
|
|
|
|
return list;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Deletes the user.
|
|
/// </summary>
|
|
/// <param name="user">The user.</param>
|
|
/// <returns>Task.</returns>
|
|
/// <exception cref="System.ArgumentNullException">user</exception>
|
|
public void DeleteUser(User user)
|
|
{
|
|
if (user == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(user));
|
|
}
|
|
|
|
using (WriteLock.Write())
|
|
{
|
|
using (var connection = CreateConnection())
|
|
{
|
|
connection.RunInTransaction(db =>
|
|
{
|
|
using (var statement = db.PrepareStatement("delete from LocalUsersv2 where Id=@id"))
|
|
{
|
|
statement.TryBind("@id", user.InternalId);
|
|
statement.MoveNext();
|
|
}
|
|
}, TransactionMode);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|