@ -1,6 +1,18 @@
using MediaBrowser.Controller.LiveTv ;
using MediaBrowser.Common.Extensions ;
using MediaBrowser.Common.IO ;
using MediaBrowser.Controller ;
using MediaBrowser.Controller.Drawing ;
using MediaBrowser.Controller.LiveTv ;
using MediaBrowser.Controller.Persistence ;
using MediaBrowser.Model.Entities ;
using MediaBrowser.Model.LiveTv ;
using MediaBrowser.Model.Logging ;
using System ;
using System.Collections.Generic ;
using System.IO ;
using System.Linq ;
using System.Threading ;
using System.Threading.Tasks ;
namespace MediaBrowser.Server.Implementations.LiveTv
{
@ -9,7 +21,25 @@ namespace MediaBrowser.Server.Implementations.LiveTv
/// </summary>
public class LiveTvManager : ILiveTvManager
{
private readonly IServerApplicationPaths _appPaths ;
private readonly IFileSystem _fileSystem ;
private readonly ILogger _logger ;
private readonly IItemRepository _itemRepo ;
private readonly IImageProcessor _imageProcessor ;
private List < Channel > _channels = new List < Channel > ( ) ;
private readonly List < ILiveTvService > _services = new List < ILiveTvService > ( ) ;
public LiveTvManager ( IServerApplicationPaths appPaths , IFileSystem fileSystem , ILogger logger , IItemRepository itemRepo , IImageProcessor imageProcessor )
{
_appPaths = appPaths ;
_fileSystem = fileSystem ;
_logger = logger ;
_itemRepo = itemRepo ;
_imageProcessor = imageProcessor ;
}
/// <summary>
/// Gets the services.
/// </summary>
@ -33,16 +63,147 @@ namespace MediaBrowser.Server.Implementations.LiveTv
/// </summary>
/// <param name="info">The info.</param>
/// <returns>ChannelInfoDto.</returns>
public ChannelInfoDto GetChannelInfoDto ( Channel Info info )
public ChannelInfoDto GetChannelInfoDto ( Channel info )
{
return new ChannelInfoDto
{
Name = info . Name ,
ServiceName = info . ServiceName ,
ChannelType = info . ChannelType ,
Id = info . Id ,
Number = info . Number
Id = info . ChannelId ,
Number = info . ChannelNumber ,
PrimaryImageTag = GetLogoImageTag ( info )
} ;
}
private Guid ? GetLogoImageTag ( Channel info )
{
var path = info . PrimaryImagePath ;
if ( string . IsNullOrEmpty ( path ) )
{
return null ;
}
try
{
return _imageProcessor . GetImageCacheTag ( info , ImageType . Primary , path ) ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error getting channel image info for {0}" , ex , info . Name ) ;
}
return null ;
}
public IEnumerable < Channel > GetChannels ( ChannelQuery query )
{
return _channels . OrderBy ( i = >
{
double number = 0 ;
if ( ! string . IsNullOrEmpty ( i . ChannelNumber ) )
{
double . TryParse ( i . ChannelNumber , out number ) ;
}
return number ;
} ) . ThenBy ( i = > i . Name ) ;
}
public Channel GetChannel ( string serviceName , string channelId )
{
return _channels . FirstOrDefault ( i = > string . Equals ( i . ServiceName , serviceName , StringComparison . OrdinalIgnoreCase ) & & string . Equals ( i . ChannelId , channelId , StringComparison . OrdinalIgnoreCase ) ) ;
}
internal async Task RefreshChannels ( IProgress < double > progress , CancellationToken cancellationToken )
{
// Avoid implicitly captured closure
var currentCancellationToken = cancellationToken ;
var tasks = _services . Select ( i = > i . GetChannelsAsync ( currentCancellationToken ) ) ;
var results = await Task . WhenAll ( tasks ) . ConfigureAwait ( false ) ;
var allChannels = results . SelectMany ( i = > i ) ;
var channnelTasks = allChannels . Select ( i = > GetChannel ( i , cancellationToken ) ) ;
var channelEntities = await Task . WhenAll ( channnelTasks ) . ConfigureAwait ( false ) ;
_channels = channelEntities . ToList ( ) ;
}
private async Task < Channel > GetChannel ( ChannelInfo channelInfo , CancellationToken cancellationToken )
{
try
{
return await GetChannelInternal ( channelInfo , cancellationToken ) . ConfigureAwait ( false ) ;
}
catch ( OperationCanceledException )
{
throw ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error getting channel information for {0}" , ex , channelInfo . Name ) ;
return null ;
}
}
private async Task < Channel > GetChannelInternal ( ChannelInfo channelInfo , CancellationToken cancellationToken )
{
var path = Path . Combine ( _appPaths . ItemsByNamePath , "channels" , _fileSystem . GetValidFilename ( channelInfo . ServiceName ) , _fileSystem . GetValidFilename ( channelInfo . Name ) ) ;
var fileInfo = new DirectoryInfo ( path ) ;
var isNew = false ;
if ( ! fileInfo . Exists )
{
Directory . CreateDirectory ( path ) ;
fileInfo = new DirectoryInfo ( path ) ;
if ( ! fileInfo . Exists )
{
throw new IOException ( "Path not created: " + path ) ;
}
isNew = true ;
}
var type = typeof ( Channel ) ;
var id = ( path + channelInfo . Number ) . GetMBId ( type ) ;
var item = _itemRepo . RetrieveItem ( id ) as Channel ;
if ( item = = null )
{
item = new Channel
{
Name = channelInfo . Name ,
Id = id ,
DateCreated = _fileSystem . GetCreationTimeUtc ( fileInfo ) ,
DateModified = _fileSystem . GetLastWriteTimeUtc ( fileInfo ) ,
Path = path ,
ChannelId = channelInfo . Id ,
ChannelNumber = channelInfo . Number ,
ServiceName = channelInfo . ServiceName
} ;
isNew = true ;
}
// Set this now so we don't cause additional file system access during provider executions
item . ResetResolveArgs ( fileInfo ) ;
await item . RefreshMetadata ( cancellationToken , forceSave : isNew , resetResolveArgs : false ) ;
return item ;
}
}
}
Luke Pulverenti
11 years ago