mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-07 11:57:55 +02:00
160 lines
5.8 KiB
C#
160 lines
5.8 KiB
C#
using System;
|
|
using System.Linq;
|
|
using Jellyfin.Api.Helpers;
|
|
using MediaBrowser.Controller.Dto;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.Querying;
|
|
using Microsoft.AspNetCore.Http;
|
|
|
|
namespace Jellyfin.Api.Extensions
|
|
{
|
|
/// <summary>
|
|
/// Dto Extensions.
|
|
/// </summary>
|
|
public static class DtoExtensions
|
|
{
|
|
/// <summary>
|
|
/// Add Dto Item fields.
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// Converted from IHasItemFields.
|
|
/// Legacy order: 1.
|
|
/// </remarks>
|
|
/// <param name="dtoOptions">DtoOptions object.</param>
|
|
/// <param name="fields">Comma delimited string of fields.</param>
|
|
/// <returns>Modified DtoOptions object.</returns>
|
|
internal static DtoOptions AddItemFields(this DtoOptions dtoOptions, string? fields)
|
|
{
|
|
if (string.IsNullOrEmpty(fields))
|
|
{
|
|
dtoOptions.Fields = Array.Empty<ItemFields>();
|
|
}
|
|
else
|
|
{
|
|
dtoOptions.Fields = fields.Split(',')
|
|
.Select(v =>
|
|
{
|
|
if (Enum.TryParse(v, true, out ItemFields value))
|
|
{
|
|
return (ItemFields?)value;
|
|
}
|
|
|
|
return null;
|
|
})
|
|
.Where(i => i.HasValue)
|
|
.Select(i => i!.Value)
|
|
.ToArray();
|
|
}
|
|
|
|
return dtoOptions;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Add additional fields depending on client.
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// Use in place of GetDtoOptions.
|
|
/// Legacy order: 2.
|
|
/// </remarks>
|
|
/// <param name="dtoOptions">DtoOptions object.</param>
|
|
/// <param name="request">Current request.</param>
|
|
/// <returns>Modified DtoOptions object.</returns>
|
|
internal static DtoOptions AddClientFields(
|
|
this DtoOptions dtoOptions, HttpRequest request)
|
|
{
|
|
dtoOptions.Fields ??= Array.Empty<ItemFields>();
|
|
|
|
string? client = ClaimHelpers.GetClient(request.HttpContext.User);
|
|
|
|
// No client in claim
|
|
if (string.IsNullOrEmpty(client))
|
|
{
|
|
return dtoOptions;
|
|
}
|
|
|
|
if (!dtoOptions.ContainsField(ItemFields.RecursiveItemCount))
|
|
{
|
|
if (client.IndexOf("kodi", StringComparison.OrdinalIgnoreCase) != -1 ||
|
|
client.IndexOf("wmc", StringComparison.OrdinalIgnoreCase) != -1 ||
|
|
client.IndexOf("media center", StringComparison.OrdinalIgnoreCase) != -1 ||
|
|
client.IndexOf("classic", StringComparison.OrdinalIgnoreCase) != -1)
|
|
{
|
|
int oldLen = dtoOptions.Fields.Length;
|
|
var arr = new ItemFields[oldLen + 1];
|
|
dtoOptions.Fields.CopyTo(arr, 0);
|
|
arr[oldLen] = ItemFields.RecursiveItemCount;
|
|
dtoOptions.Fields = arr;
|
|
}
|
|
}
|
|
|
|
if (!dtoOptions.ContainsField(ItemFields.ChildCount))
|
|
{
|
|
if (client.IndexOf("kodi", StringComparison.OrdinalIgnoreCase) != -1 ||
|
|
client.IndexOf("wmc", StringComparison.OrdinalIgnoreCase) != -1 ||
|
|
client.IndexOf("media center", StringComparison.OrdinalIgnoreCase) != -1 ||
|
|
client.IndexOf("classic", StringComparison.OrdinalIgnoreCase) != -1 ||
|
|
client.IndexOf("roku", StringComparison.OrdinalIgnoreCase) != -1 ||
|
|
client.IndexOf("samsung", StringComparison.OrdinalIgnoreCase) != -1 ||
|
|
client.IndexOf("androidtv", StringComparison.OrdinalIgnoreCase) != -1)
|
|
{
|
|
int oldLen = dtoOptions.Fields.Length;
|
|
var arr = new ItemFields[oldLen + 1];
|
|
dtoOptions.Fields.CopyTo(arr, 0);
|
|
arr[oldLen] = ItemFields.ChildCount;
|
|
dtoOptions.Fields = arr;
|
|
}
|
|
}
|
|
|
|
return dtoOptions;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Add additional DtoOptions.
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// Converted from IHasDtoOptions.
|
|
/// Legacy order: 3.
|
|
/// </remarks>
|
|
/// <param name="dtoOptions">DtoOptions object.</param>
|
|
/// <param name="enableImages">Enable images.</param>
|
|
/// <param name="enableUserData">Enable user data.</param>
|
|
/// <param name="imageTypeLimit">Image type limit.</param>
|
|
/// <param name="enableImageTypes">Enable image types.</param>
|
|
/// <returns>Modified DtoOptions object.</returns>
|
|
internal static DtoOptions AddAdditionalDtoOptions(
|
|
this DtoOptions dtoOptions,
|
|
bool? enableImages,
|
|
bool? enableUserData,
|
|
int? imageTypeLimit,
|
|
ImageType[] enableImageTypes)
|
|
{
|
|
dtoOptions.EnableImages = enableImages ?? true;
|
|
|
|
if (imageTypeLimit.HasValue)
|
|
{
|
|
dtoOptions.ImageTypeLimit = imageTypeLimit.Value;
|
|
}
|
|
|
|
if (enableUserData.HasValue)
|
|
{
|
|
dtoOptions.EnableUserData = enableUserData.Value;
|
|
}
|
|
|
|
if (enableImageTypes.Length != 0)
|
|
{
|
|
dtoOptions.ImageTypes = enableImageTypes;
|
|
}
|
|
|
|
return dtoOptions;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Check if DtoOptions contains field.
|
|
/// </summary>
|
|
/// <param name="dtoOptions">DtoOptions object.</param>
|
|
/// <param name="field">Field to check.</param>
|
|
/// <returns>Field existence.</returns>
|
|
internal static bool ContainsField(this DtoOptions dtoOptions, ItemFields field)
|
|
=> dtoOptions.Fields != null && dtoOptions.Fields.Contains(field);
|
|
}
|
|
}
|