mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 11:34:57 +02:00
449 lines
13 KiB
C#
449 lines
13 KiB
C#
#nullable disable
|
|
#pragma warning disable CS1591
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Globalization;
|
|
using SQLitePCL.pretty;
|
|
|
|
namespace Emby.Server.Implementations.Data
|
|
{
|
|
public static class SqliteExtensions
|
|
{
|
|
private const string DatetimeFormatUtc = "yyyy-MM-dd HH:mm:ss.FFFFFFFK";
|
|
private const string DatetimeFormatLocal = "yyyy-MM-dd HH:mm:ss.FFFFFFF";
|
|
|
|
/// <summary>
|
|
/// An array of ISO-8601 DateTime formats that we support parsing.
|
|
/// </summary>
|
|
private static readonly string[] _datetimeFormats = new string[]
|
|
{
|
|
"THHmmssK",
|
|
"THHmmK",
|
|
"HH:mm:ss.FFFFFFFK",
|
|
"HH:mm:ssK",
|
|
"HH:mmK",
|
|
DatetimeFormatUtc,
|
|
"yyyy-MM-dd HH:mm:ssK",
|
|
"yyyy-MM-dd HH:mmK",
|
|
"yyyy-MM-ddTHH:mm:ss.FFFFFFFK",
|
|
"yyyy-MM-ddTHH:mmK",
|
|
"yyyy-MM-ddTHH:mm:ssK",
|
|
"yyyyMMddHHmmssK",
|
|
"yyyyMMddHHmmK",
|
|
"yyyyMMddTHHmmssFFFFFFFK",
|
|
"THHmmss",
|
|
"THHmm",
|
|
"HH:mm:ss.FFFFFFF",
|
|
"HH:mm:ss",
|
|
"HH:mm",
|
|
DatetimeFormatLocal,
|
|
"yyyy-MM-dd HH:mm:ss",
|
|
"yyyy-MM-dd HH:mm",
|
|
"yyyy-MM-ddTHH:mm:ss.FFFFFFF",
|
|
"yyyy-MM-ddTHH:mm",
|
|
"yyyy-MM-ddTHH:mm:ss",
|
|
"yyyyMMddHHmmss",
|
|
"yyyyMMddHHmm",
|
|
"yyyyMMddTHHmmssFFFFFFF",
|
|
"yyyy-MM-dd",
|
|
"yyyyMMdd",
|
|
"yy-MM-dd"
|
|
};
|
|
|
|
public static void RunQueries(this SQLiteDatabaseConnection connection, string[] queries)
|
|
{
|
|
ArgumentNullException.ThrowIfNull(queries);
|
|
|
|
connection.RunInTransaction(conn =>
|
|
{
|
|
conn.ExecuteAll(string.Join(';', queries));
|
|
});
|
|
}
|
|
|
|
public static Guid ReadGuidFromBlob(this ResultSetValue result)
|
|
{
|
|
return new Guid(result.ToBlob());
|
|
}
|
|
|
|
public static string ToDateTimeParamValue(this DateTime dateValue)
|
|
{
|
|
var kind = DateTimeKind.Utc;
|
|
|
|
return (dateValue.Kind == DateTimeKind.Unspecified)
|
|
? DateTime.SpecifyKind(dateValue, kind).ToString(
|
|
GetDateTimeKindFormat(kind),
|
|
CultureInfo.InvariantCulture)
|
|
: dateValue.ToString(
|
|
GetDateTimeKindFormat(dateValue.Kind),
|
|
CultureInfo.InvariantCulture);
|
|
}
|
|
|
|
private static string GetDateTimeKindFormat(DateTimeKind kind)
|
|
=> (kind == DateTimeKind.Utc) ? DatetimeFormatUtc : DatetimeFormatLocal;
|
|
|
|
public static DateTime ReadDateTime(this ResultSetValue result)
|
|
{
|
|
var dateText = result.ToString();
|
|
|
|
return DateTime.ParseExact(
|
|
dateText,
|
|
_datetimeFormats,
|
|
DateTimeFormatInfo.InvariantInfo,
|
|
DateTimeStyles.AdjustToUniversal);
|
|
}
|
|
|
|
public static bool TryReadDateTime(this IReadOnlyList<ResultSetValue> reader, int index, out DateTime result)
|
|
{
|
|
var item = reader[index];
|
|
if (item.IsDbNull())
|
|
{
|
|
result = default;
|
|
return false;
|
|
}
|
|
|
|
var dateText = item.ToString();
|
|
|
|
if (DateTime.TryParseExact(dateText, _datetimeFormats, DateTimeFormatInfo.InvariantInfo, DateTimeStyles.AdjustToUniversal, out var dateTimeResult))
|
|
{
|
|
result = dateTimeResult;
|
|
return true;
|
|
}
|
|
|
|
result = default;
|
|
return false;
|
|
}
|
|
|
|
public static bool TryGetGuid(this IReadOnlyList<ResultSetValue> reader, int index, out Guid result)
|
|
{
|
|
var item = reader[index];
|
|
if (item.IsDbNull())
|
|
{
|
|
result = default;
|
|
return false;
|
|
}
|
|
|
|
result = item.ReadGuidFromBlob();
|
|
return true;
|
|
}
|
|
|
|
public static bool IsDbNull(this ResultSetValue result)
|
|
{
|
|
return result.SQLiteType == SQLiteType.Null;
|
|
}
|
|
|
|
public static string GetString(this IReadOnlyList<ResultSetValue> result, int index)
|
|
{
|
|
return result[index].ToString();
|
|
}
|
|
|
|
public static bool TryGetString(this IReadOnlyList<ResultSetValue> reader, int index, out string result)
|
|
{
|
|
result = null;
|
|
var item = reader[index];
|
|
if (item.IsDbNull())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
result = item.ToString();
|
|
return true;
|
|
}
|
|
|
|
public static bool GetBoolean(this IReadOnlyList<ResultSetValue> result, int index)
|
|
{
|
|
return result[index].ToBool();
|
|
}
|
|
|
|
public static bool TryGetBoolean(this IReadOnlyList<ResultSetValue> reader, int index, out bool result)
|
|
{
|
|
var item = reader[index];
|
|
if (item.IsDbNull())
|
|
{
|
|
result = default;
|
|
return false;
|
|
}
|
|
|
|
result = item.ToBool();
|
|
return true;
|
|
}
|
|
|
|
public static bool TryGetInt32(this IReadOnlyList<ResultSetValue> reader, int index, out int result)
|
|
{
|
|
var item = reader[index];
|
|
if (item.IsDbNull())
|
|
{
|
|
result = default;
|
|
return false;
|
|
}
|
|
|
|
result = item.ToInt();
|
|
return true;
|
|
}
|
|
|
|
public static long GetInt64(this IReadOnlyList<ResultSetValue> result, int index)
|
|
{
|
|
return result[index].ToInt64();
|
|
}
|
|
|
|
public static bool TryGetInt64(this IReadOnlyList<ResultSetValue> reader, int index, out long result)
|
|
{
|
|
var item = reader[index];
|
|
if (item.IsDbNull())
|
|
{
|
|
result = default;
|
|
return false;
|
|
}
|
|
|
|
result = item.ToInt64();
|
|
return true;
|
|
}
|
|
|
|
public static bool TryGetSingle(this IReadOnlyList<ResultSetValue> reader, int index, out float result)
|
|
{
|
|
var item = reader[index];
|
|
if (item.IsDbNull())
|
|
{
|
|
result = default;
|
|
return false;
|
|
}
|
|
|
|
result = item.ToFloat();
|
|
return true;
|
|
}
|
|
|
|
public static bool TryGetDouble(this IReadOnlyList<ResultSetValue> reader, int index, out double result)
|
|
{
|
|
var item = reader[index];
|
|
if (item.IsDbNull())
|
|
{
|
|
result = default;
|
|
return false;
|
|
}
|
|
|
|
result = item.ToDouble();
|
|
return true;
|
|
}
|
|
|
|
public static Guid GetGuid(this IReadOnlyList<ResultSetValue> result, int index)
|
|
{
|
|
return result[index].ReadGuidFromBlob();
|
|
}
|
|
|
|
[Conditional("DEBUG")]
|
|
private static void CheckName(string name)
|
|
{
|
|
throw new ArgumentException("Invalid param name: " + name, nameof(name));
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, double value)
|
|
{
|
|
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
|
|
{
|
|
bindParam.Bind(value);
|
|
}
|
|
else
|
|
{
|
|
CheckName(name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, string value)
|
|
{
|
|
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
|
|
{
|
|
if (value is null)
|
|
{
|
|
bindParam.BindNull();
|
|
}
|
|
else
|
|
{
|
|
bindParam.Bind(value);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
CheckName(name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, bool value)
|
|
{
|
|
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
|
|
{
|
|
bindParam.Bind(value);
|
|
}
|
|
else
|
|
{
|
|
CheckName(name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, float value)
|
|
{
|
|
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
|
|
{
|
|
bindParam.Bind(value);
|
|
}
|
|
else
|
|
{
|
|
CheckName(name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, int value)
|
|
{
|
|
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
|
|
{
|
|
bindParam.Bind(value);
|
|
}
|
|
else
|
|
{
|
|
CheckName(name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, Guid value)
|
|
{
|
|
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
|
|
{
|
|
Span<byte> byteValue = stackalloc byte[16];
|
|
value.TryWriteBytes(byteValue);
|
|
bindParam.Bind(byteValue);
|
|
}
|
|
else
|
|
{
|
|
CheckName(name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, DateTime value)
|
|
{
|
|
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
|
|
{
|
|
bindParam.Bind(value.ToDateTimeParamValue());
|
|
}
|
|
else
|
|
{
|
|
CheckName(name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, long value)
|
|
{
|
|
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
|
|
{
|
|
bindParam.Bind(value);
|
|
}
|
|
else
|
|
{
|
|
CheckName(name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, ReadOnlySpan<byte> value)
|
|
{
|
|
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
|
|
{
|
|
bindParam.Bind(value);
|
|
}
|
|
else
|
|
{
|
|
CheckName(name);
|
|
}
|
|
}
|
|
|
|
public static void TryBindNull(this IStatement statement, string name)
|
|
{
|
|
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
|
|
{
|
|
bindParam.BindNull();
|
|
}
|
|
else
|
|
{
|
|
CheckName(name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, DateTime? value)
|
|
{
|
|
if (value.HasValue)
|
|
{
|
|
TryBind(statement, name, value.Value);
|
|
}
|
|
else
|
|
{
|
|
TryBindNull(statement, name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, Guid? value)
|
|
{
|
|
if (value.HasValue)
|
|
{
|
|
TryBind(statement, name, value.Value);
|
|
}
|
|
else
|
|
{
|
|
TryBindNull(statement, name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, double? value)
|
|
{
|
|
if (value.HasValue)
|
|
{
|
|
TryBind(statement, name, value.Value);
|
|
}
|
|
else
|
|
{
|
|
TryBindNull(statement, name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, int? value)
|
|
{
|
|
if (value.HasValue)
|
|
{
|
|
TryBind(statement, name, value.Value);
|
|
}
|
|
else
|
|
{
|
|
TryBindNull(statement, name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, float? value)
|
|
{
|
|
if (value.HasValue)
|
|
{
|
|
TryBind(statement, name, value.Value);
|
|
}
|
|
else
|
|
{
|
|
TryBindNull(statement, name);
|
|
}
|
|
}
|
|
|
|
public static void TryBind(this IStatement statement, string name, bool? value)
|
|
{
|
|
if (value.HasValue)
|
|
{
|
|
TryBind(statement, name, value.Value);
|
|
}
|
|
else
|
|
{
|
|
TryBindNull(statement, name);
|
|
}
|
|
}
|
|
|
|
public static IEnumerable<IReadOnlyList<ResultSetValue>> ExecuteQuery(this IStatement statement)
|
|
{
|
|
while (statement.MoveNext())
|
|
{
|
|
yield return statement.Current;
|
|
}
|
|
}
|
|
}
|
|
}
|