mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 22:07:13 -04:00
129 lines
5.0 KiB
C#
129 lines
5.0 KiB
C#
|
using System;
|
|||
|
using System.Collections.Generic;
|
|||
|
using System.Linq;
|
|||
|
using Jellyfin.Api.Extensions;
|
|||
|
using Jellyfin.Api.Helpers;
|
|||
|
using MediaBrowser.Controller.Dto;
|
|||
|
using MediaBrowser.Controller.Entities;
|
|||
|
using MediaBrowser.Controller.Entities.Audio;
|
|||
|
using MediaBrowser.Controller.Library;
|
|||
|
using MediaBrowser.Model.Dto;
|
|||
|
using MediaBrowser.Model.Querying;
|
|||
|
using Microsoft.AspNetCore.Mvc;
|
|||
|
|
|||
|
namespace Jellyfin.Api.Controllers
|
|||
|
{
|
|||
|
/// <summary>
|
|||
|
/// The albums controller.
|
|||
|
/// </summary>
|
|||
|
public class AlbumsController : BaseJellyfinApiController
|
|||
|
{
|
|||
|
private readonly IUserManager _userManager;
|
|||
|
private readonly ILibraryManager _libraryManager;
|
|||
|
private readonly IDtoService _dtoService;
|
|||
|
|
|||
|
/// <summary>
|
|||
|
/// Initializes a new instance of the <see cref="AlbumsController"/> class.
|
|||
|
/// </summary>
|
|||
|
/// <param name="userManager">Instance of the <see cref="IUserManager"/> interface.</param>
|
|||
|
/// <param name="libraryManager">Instance of the <see cref="ILibraryManager"/> interface.</param>
|
|||
|
/// <param name="dtoService">Instance of the <see cref="IDtoService"/> interface.</param>
|
|||
|
public AlbumsController(
|
|||
|
IUserManager userManager,
|
|||
|
ILibraryManager libraryManager,
|
|||
|
IDtoService dtoService)
|
|||
|
{
|
|||
|
_userManager = userManager;
|
|||
|
_libraryManager = libraryManager;
|
|||
|
_dtoService = dtoService;
|
|||
|
}
|
|||
|
|
|||
|
/// <summary>
|
|||
|
/// Finds albums similar to a given album.
|
|||
|
/// </summary>
|
|||
|
/// <param name="albumId">The album id.</param>
|
|||
|
/// <param name="userId">Optional. Filter by user id, and attach user data.</param>
|
|||
|
/// <param name="excludeArtistIds">Optional. Ids of artists to exclude.</param>
|
|||
|
/// <param name="limit">Optional. The maximum number of records to return.</param>
|
|||
|
/// <returns>A <see cref="QueryResult{BaseItemDto}"/> with similar albums.</returns>
|
|||
|
[HttpGet("/Albums/{albumId}/Similar")]
|
|||
|
public ActionResult<QueryResult<BaseItemDto>> GetSimilarAlbums(
|
|||
|
[FromRoute] string albumId,
|
|||
|
[FromQuery] Guid userId,
|
|||
|
[FromQuery] string excludeArtistIds,
|
|||
|
[FromQuery] int? limit)
|
|||
|
{
|
|||
|
var dtoOptions = new DtoOptions().AddClientFields(Request);
|
|||
|
|
|||
|
return SimilarItemsHelper.GetSimilarItemsResult(
|
|||
|
dtoOptions,
|
|||
|
_userManager,
|
|||
|
_libraryManager,
|
|||
|
_dtoService,
|
|||
|
userId,
|
|||
|
albumId,
|
|||
|
excludeArtistIds,
|
|||
|
limit,
|
|||
|
new[] { typeof(MusicAlbum) },
|
|||
|
GetAlbumSimilarityScore);
|
|||
|
}
|
|||
|
|
|||
|
/// <summary>
|
|||
|
/// Finds artists similar to a given artist.
|
|||
|
/// </summary>
|
|||
|
/// <param name="artistId">The artist id.</param>
|
|||
|
/// <param name="userId">Optional. Filter by user id, and attach user data.</param>
|
|||
|
/// <param name="excludeArtistIds">Optional. Ids of artists to exclude.</param>
|
|||
|
/// <param name="limit">Optional. The maximum number of records to return.</param>
|
|||
|
/// <returns>A <see cref="QueryResult{BaseItemDto}"/> with similar artists.</returns>
|
|||
|
[HttpGet("/Artists/{artistId}/Similar")]
|
|||
|
public ActionResult<QueryResult<BaseItemDto>> GetSimilarArtists(
|
|||
|
[FromRoute] string artistId,
|
|||
|
[FromQuery] Guid userId,
|
|||
|
[FromQuery] string excludeArtistIds,
|
|||
|
[FromQuery] int? limit)
|
|||
|
{
|
|||
|
var dtoOptions = new DtoOptions().AddClientFields(Request);
|
|||
|
|
|||
|
return SimilarItemsHelper.GetSimilarItemsResult(
|
|||
|
dtoOptions,
|
|||
|
_userManager,
|
|||
|
_libraryManager,
|
|||
|
_dtoService,
|
|||
|
userId,
|
|||
|
artistId,
|
|||
|
excludeArtistIds,
|
|||
|
limit,
|
|||
|
new[] { typeof(MusicArtist) },
|
|||
|
SimilarItemsHelper.GetSimiliarityScore);
|
|||
|
}
|
|||
|
|
|||
|
/// <summary>
|
|||
|
/// Gets a similairty score of two albums.
|
|||
|
/// </summary>
|
|||
|
/// <param name="item1">The first item.</param>
|
|||
|
/// <param name="item1People">The item1 people.</param>
|
|||
|
/// <param name="allPeople">All people.</param>
|
|||
|
/// <param name="item2">The second item.</param>
|
|||
|
/// <returns>System.Int32.</returns>
|
|||
|
private int GetAlbumSimilarityScore(BaseItem item1, List<PersonInfo> item1People, List<PersonInfo> allPeople, BaseItem item2)
|
|||
|
{
|
|||
|
var points = SimilarItemsHelper.GetSimiliarityScore(item1, item1People, allPeople, item2);
|
|||
|
|
|||
|
var album1 = (MusicAlbum)item1;
|
|||
|
var album2 = (MusicAlbum)item2;
|
|||
|
|
|||
|
var artists1 = album1
|
|||
|
.GetAllArtists()
|
|||
|
.DistinctNames()
|
|||
|
.ToList();
|
|||
|
|
|||
|
var artists2 = new HashSet<string>(
|
|||
|
album2.GetAllArtists().DistinctNames(),
|
|||
|
StringComparer.OrdinalIgnoreCase);
|
|||
|
|
|||
|
return points + artists1.Where(artists2.Contains).Sum(i => 5);
|
|||
|
}
|
|||
|
}
|
|||
|
}
|