jellyfin/MediaBrowser.Server.Implementations/Playlists/PlaylistManager.cs

239 lines
8.0 KiB
C#
Raw Normal View History

2014-08-02 02:34:45 +00:00
using MediaBrowser.Common.IO;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Audio;
2014-08-02 02:34:45 +00:00
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Playlists;
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
2014-08-02 02:34:45 +00:00
using MediaBrowser.Model.Logging;
2014-08-21 15:55:35 +00:00
using MediaBrowser.Model.Playlists;
2014-08-02 02:34:45 +00:00
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
namespace MediaBrowser.Server.Implementations.Playlists
{
public class PlaylistManager : IPlaylistManager
{
private readonly ILibraryManager _libraryManager;
private readonly IFileSystem _fileSystem;
private readonly ILibraryMonitor _iLibraryMonitor;
private readonly ILogger _logger;
private readonly IUserManager _userManager;
2015-07-24 15:20:11 +00:00
private readonly IProviderManager _providerManager;
2014-08-02 02:34:45 +00:00
2015-07-24 15:20:11 +00:00
public PlaylistManager(ILibraryManager libraryManager, IFileSystem fileSystem, ILibraryMonitor iLibraryMonitor, ILogger logger, IUserManager userManager, IProviderManager providerManager)
2014-08-02 02:34:45 +00:00
{
_libraryManager = libraryManager;
_fileSystem = fileSystem;
_iLibraryMonitor = iLibraryMonitor;
_logger = logger;
_userManager = userManager;
2015-07-24 15:20:11 +00:00
_providerManager = providerManager;
2014-08-02 02:34:45 +00:00
}
public IEnumerable<Playlist> GetPlaylists(string userId)
{
2014-09-14 15:10:51 +00:00
var user = _userManager.GetUserById(userId);
2014-08-02 02:34:45 +00:00
return GetPlaylistsFolder(userId).GetChildren(user, true).OfType<Playlist>();
}
2014-08-21 15:55:35 +00:00
public async Task<PlaylistCreationResult> CreatePlaylist(PlaylistCreationRequest options)
2014-08-02 02:34:45 +00:00
{
var name = options.Name;
var folderName = _fileSystem.GetValidFilename(name) + " [playlist]";
var parentFolder = GetPlaylistsFolder(null);
if (parentFolder == null)
{
throw new ArgumentException();
}
if (string.IsNullOrWhiteSpace(options.MediaType))
{
foreach (var itemId in options.ItemIdList)
{
var item = _libraryManager.GetItemById(itemId);
if (item == null)
{
throw new ArgumentException("No item exists with the supplied Id");
}
if (!string.IsNullOrWhiteSpace(item.MediaType))
{
options.MediaType = item.MediaType;
}
else if (item is MusicArtist || item is MusicAlbum || item is MusicGenre)
{
options.MediaType = MediaType.Audio;
}
else if (item is Genre)
{
options.MediaType = MediaType.Video;
}
else
{
var folder = item as Folder;
if (folder != null)
{
2015-01-25 06:34:50 +00:00
options.MediaType = folder.GetRecursiveChildren(i => !i.IsFolder && i.SupportsAddingToPlaylist)
.Select(i => i.MediaType)
.FirstOrDefault(i => !string.IsNullOrWhiteSpace(i));
}
}
if (!string.IsNullOrWhiteSpace(options.MediaType))
{
break;
}
}
}
if (string.IsNullOrWhiteSpace(options.MediaType))
{
throw new ArgumentException("A playlist media type is required.");
}
2014-09-14 15:10:51 +00:00
var user = _userManager.GetUserById(options.UserId);
2014-08-22 02:24:38 +00:00
2014-08-02 02:34:45 +00:00
var path = Path.Combine(parentFolder.Path, folderName);
path = GetTargetPath(path);
2014-08-02 02:34:45 +00:00
_iLibraryMonitor.ReportFileSystemChangeBeginning(path);
try
{
Directory.CreateDirectory(path);
var playlist = new Playlist
2014-08-02 02:34:45 +00:00
{
Name = name,
2014-12-13 03:56:30 +00:00
Path = path
2014-08-02 02:34:45 +00:00
};
2014-12-13 03:56:30 +00:00
playlist.Shares.Add(new Share
{
UserId = options.UserId,
CanEdit = true
});
playlist.SetMediaType(options.MediaType);
await parentFolder.AddChild(playlist, CancellationToken.None).ConfigureAwait(false);
2014-08-02 02:34:45 +00:00
await playlist.RefreshMetadata(new MetadataRefreshOptions { ForceSave = true }, CancellationToken.None)
2014-08-02 02:34:45 +00:00
.ConfigureAwait(false);
if (options.ItemIdList.Count > 0)
{
2014-08-22 02:24:38 +00:00
await AddToPlaylistInternal(playlist.Id.ToString("N"), options.ItemIdList, user);
2014-08-02 02:34:45 +00:00
}
2014-08-21 15:55:35 +00:00
return new PlaylistCreationResult
{
Id = playlist.Id.ToString("N")
};
2014-08-02 02:34:45 +00:00
}
finally
{
// Refresh handled internally
_iLibraryMonitor.ReportFileSystemChangeComplete(path, false);
}
}
private string GetTargetPath(string path)
{
while (Directory.Exists(path))
{
path += "1";
}
return path;
}
2014-08-22 02:24:38 +00:00
private IEnumerable<BaseItem> GetPlaylistItems(IEnumerable<string> itemIds, string playlistMediaType, User user)
{
var items = itemIds.Select(i => _libraryManager.GetItemById(i)).Where(i => i != null);
2014-08-22 02:24:38 +00:00
return Playlist.GetPlaylistItems(playlistMediaType, items, user);
}
public Task AddToPlaylist(string playlistId, IEnumerable<string> itemIds, string userId)
{
2014-09-14 15:10:51 +00:00
var user = string.IsNullOrWhiteSpace(userId) ? null : _userManager.GetUserById(userId);
2014-08-22 02:24:38 +00:00
return AddToPlaylistInternal(playlistId, itemIds, user);
}
2014-08-22 02:24:38 +00:00
private async Task AddToPlaylistInternal(string playlistId, IEnumerable<string> itemIds, User user)
2014-08-02 02:34:45 +00:00
{
var playlist = _libraryManager.GetItemById(playlistId) as Playlist;
2014-08-02 02:34:45 +00:00
if (playlist == null)
2014-08-02 02:34:45 +00:00
{
throw new ArgumentException("No Playlist exists with the supplied Id");
}
var list = new List<LinkedChild>();
2014-08-22 02:24:38 +00:00
var items = GetPlaylistItems(itemIds, playlist.MediaType, user)
.Where(i => i.SupportsAddingToPlaylist)
.ToList();
2014-08-02 02:34:45 +00:00
2014-08-07 02:51:09 +00:00
foreach (var item in items)
{
list.Add(LinkedChild.Create(item));
2014-08-02 02:34:45 +00:00
}
playlist.LinkedChildren.AddRange(list);
await playlist.UpdateToRepository(ItemUpdateType.MetadataEdit, CancellationToken.None).ConfigureAwait(false);
2015-07-24 15:20:11 +00:00
_providerManager.QueueRefresh(playlist.Id, new MetadataRefreshOptions
2014-08-21 15:55:35 +00:00
{
ForceSave = true
2015-07-24 15:20:11 +00:00
});
2014-08-02 02:34:45 +00:00
}
2014-08-06 04:18:13 +00:00
public async Task RemoveFromPlaylist(string playlistId, IEnumerable<string> entryIds)
2014-08-02 02:34:45 +00:00
{
2014-08-06 04:18:13 +00:00
var playlist = _libraryManager.GetItemById(playlistId) as Playlist;
if (playlist == null)
{
throw new ArgumentException("No Playlist exists with the supplied Id");
}
2014-08-11 23:41:11 +00:00
var children = playlist.GetManageableItems().ToList();
2014-08-06 04:18:13 +00:00
var idList = entryIds.ToList();
2014-08-11 23:41:11 +00:00
var removals = children.Where(i => idList.Contains(i.Item1.Id));
2014-08-06 04:18:13 +00:00
playlist.LinkedChildren = children.Except(removals)
2014-08-11 23:41:11 +00:00
.Select(i => i.Item1)
2014-08-06 04:18:13 +00:00
.ToList();
await playlist.UpdateToRepository(ItemUpdateType.MetadataEdit, CancellationToken.None).ConfigureAwait(false);
2015-07-24 15:20:11 +00:00
_providerManager.QueueRefresh(playlist.Id, new MetadataRefreshOptions
2014-08-06 04:18:13 +00:00
{
ForceSave = true
2015-07-24 15:20:11 +00:00
});
2014-08-02 02:34:45 +00:00
}
public Folder GetPlaylistsFolder(string userId)
{
return _libraryManager.RootFolder.Children.OfType<PlaylistsFolder>()
.FirstOrDefault();
}
}
}