using MediaBrowser.Common ;
using MediaBrowser.Common.Configuration ;
using MediaBrowser.Common.Net ;
using MediaBrowser.Common.Security ;
using MediaBrowser.Controller.Configuration ;
using MediaBrowser.Controller.Drawing ;
using MediaBrowser.Controller.FileOrganization ;
using MediaBrowser.Controller.Library ;
using MediaBrowser.Controller.LiveTv ;
using MediaBrowser.Controller.MediaEncoding ;
using MediaBrowser.Controller.Providers ;
using MediaBrowser.Model.Dto ;
using MediaBrowser.Model.Entities ;
using MediaBrowser.Model.Events ;
using MediaBrowser.Model.FileOrganization ;
using MediaBrowser.Model.LiveTv ;
using MediaBrowser.Model.Logging ;
using MediaBrowser.Model.Serialization ;
using MediaBrowser.Server.Implementations.FileOrganization ;
using System ;
using System.Collections.Concurrent ;
using System.Collections.Generic ;
using System.Globalization ;
using System.IO ;
using System.Linq ;
using System.Threading ;
using System.Threading.Tasks ;
using CommonIO ;
using MediaBrowser.Common.Events ;
using MediaBrowser.Common.Extensions ;
using MediaBrowser.Controller.Entities ;
using MediaBrowser.Controller.Entities.TV ;
using MediaBrowser.Controller.Power ;
using Microsoft.Win32 ;
namespace MediaBrowser.Server.Implementations.LiveTv.EmbyTV
{
public class EmbyTV : ILiveTvService , IHasRegistrationInfo , IDisposable
{
private readonly IApplicationHost _appHpst ;
private readonly ILogger _logger ;
private readonly IHttpClient _httpClient ;
private readonly IServerConfigurationManager _config ;
private readonly IJsonSerializer _jsonSerializer ;
private readonly ItemDataProvider < SeriesTimerInfo > _seriesTimerProvider ;
private readonly TimerManager _timerProvider ;
private readonly LiveTvManager _liveTvManager ;
private readonly IFileSystem _fileSystem ;
private readonly ISecurityManager _security ;
private readonly ILibraryMonitor _libraryMonitor ;
private readonly ILibraryManager _libraryManager ;
private readonly IProviderManager _providerManager ;
private readonly IFileOrganizationService _organizationService ;
private readonly IMediaEncoder _mediaEncoder ;
public static EmbyTV Current ;
public event EventHandler DataSourceChanged ;
public event EventHandler < RecordingStatusChangedEventArgs > RecordingStatusChanged ;
private readonly ConcurrentDictionary < string , ActiveRecordingInfo > _activeRecordings =
new ConcurrentDictionary < string , ActiveRecordingInfo > ( StringComparer . OrdinalIgnoreCase ) ;
public EmbyTV ( IApplicationHost appHost , ILogger logger , IJsonSerializer jsonSerializer , IHttpClient httpClient , IServerConfigurationManager config , ILiveTvManager liveTvManager , IFileSystem fileSystem , ISecurityManager security , ILibraryManager libraryManager , ILibraryMonitor libraryMonitor , IProviderManager providerManager , IFileOrganizationService organizationService , IMediaEncoder mediaEncoder , IPowerManagement powerManagement )
{
Current = this ;
_appHpst = appHost ;
_logger = logger ;
_httpClient = httpClient ;
_config = config ;
_fileSystem = fileSystem ;
_security = security ;
_libraryManager = libraryManager ;
_libraryMonitor = libraryMonitor ;
_providerManager = providerManager ;
_organizationService = organizationService ;
_mediaEncoder = mediaEncoder ;
_liveTvManager = ( LiveTvManager ) liveTvManager ;
_jsonSerializer = jsonSerializer ;
_seriesTimerProvider = new SeriesTimerManager ( fileSystem , jsonSerializer , _logger , Path . Combine ( DataPath , "seriestimers" ) ) ;
_timerProvider = new TimerManager ( fileSystem , jsonSerializer , _logger , Path . Combine ( DataPath , "timers" ) , powerManagement , _logger ) ;
_timerProvider . TimerFired + = _timerProvider_TimerFired ;
_config . NamedConfigurationUpdated + = _config_NamedConfigurationUpdated ;
}
private void _config_NamedConfigurationUpdated ( object sender , ConfigurationUpdateEventArgs e )
{
if ( string . Equals ( e . Key , "livetv" , StringComparison . OrdinalIgnoreCase ) )
{
OnRecordingFoldersChanged ( ) ;
}
}
public void Start ( )
{
_timerProvider . RestartTimers ( ) ;
SystemEvents . PowerModeChanged + = SystemEvents_PowerModeChanged ;
}
private void OnRecordingFoldersChanged ( )
{
CreateRecordingFolders ( ) ;
}
internal void CreateRecordingFolders ( )
{
try
{
CreateRecordingFoldersInternal ( ) ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error creating recording folders" , ex ) ;
}
}
internal void CreateRecordingFoldersInternal ( )
{
var recordingFolders = GetRecordingFolders ( ) ;
var virtualFolders = _libraryManager . GetVirtualFolders ( )
. ToList ( ) ;
var allExistingPaths = virtualFolders . SelectMany ( i = > i . Locations ) . ToList ( ) ;
var pathsAdded = new List < string > ( ) ;
foreach ( var recordingFolder in recordingFolders )
{
var pathsToCreate = recordingFolder . Locations
. Where ( i = > ! allExistingPaths . Contains ( i , StringComparer . OrdinalIgnoreCase ) )
. ToList ( ) ;
if ( pathsToCreate . Count = = 0 )
{
continue ;
}
try
{
_libraryManager . AddVirtualFolder ( recordingFolder . Name , recordingFolder . CollectionType , pathsToCreate . ToArray ( ) , true ) ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error creating virtual folder" , ex ) ;
}
pathsAdded . AddRange ( pathsToCreate ) ;
}
var config = GetConfiguration ( ) ;
var pathsToRemove = config . MediaLocationsCreated
. Except ( recordingFolders . SelectMany ( i = > i . Locations ) )
. ToList ( ) ;
if ( pathsAdded . Count > 0 | | pathsToRemove . Count > 0 )
{
pathsAdded . InsertRange ( 0 , config . MediaLocationsCreated ) ;
config . MediaLocationsCreated = pathsAdded . Except ( pathsToRemove ) . Distinct ( StringComparer . OrdinalIgnoreCase ) . ToArray ( ) ;
_config . SaveConfiguration ( "livetv" , config ) ;
}
foreach ( var path in pathsToRemove )
{
RemovePathFromLibrary ( path ) ;
}
}
private void RemovePathFromLibrary ( string path )
{
_logger . Debug ( "Removing path from library: {0}" , path ) ;
var requiresRefresh = false ;
var virtualFolders = _libraryManager . GetVirtualFolders ( )
. ToList ( ) ;
foreach ( var virtualFolder in virtualFolders )
{
if ( ! virtualFolder . Locations . Contains ( path , StringComparer . OrdinalIgnoreCase ) )
{
continue ;
}
if ( virtualFolder . Locations . Count = = 1 )
{
// remove entire virtual folder
try
{
_libraryManager . RemoveVirtualFolder ( virtualFolder . Name , true ) ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error removing virtual folder" , ex ) ;
}
}
else
{
try
{
_libraryManager . RemoveMediaPath ( virtualFolder . Name , path ) ;
requiresRefresh = true ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error removing media path" , ex ) ;
}
}
}
if ( requiresRefresh )
{
_libraryManager . ValidateMediaLibrary ( new Progress < Double > ( ) , CancellationToken . None ) ;
}
}
void SystemEvents_PowerModeChanged ( object sender , PowerModeChangedEventArgs e )
{
_logger . Info ( "Power mode changed to {0}" , e . Mode ) ;
if ( e . Mode = = PowerModes . Resume )
{
_timerProvider . RestartTimers ( ) ;
}
}
public string Name
{
get { return "Emby" ; }
}
public string DataPath
{
get { return Path . Combine ( _config . CommonApplicationPaths . DataPath , "livetv" ) ; }
}
private string DefaultRecordingPath
{
get
{
return Path . Combine ( DataPath , "recordings" ) ;
}
}
private string RecordingPath
{
get
{
var path = GetConfiguration ( ) . RecordingPath ;
return string . IsNullOrWhiteSpace ( path )
? DefaultRecordingPath
: path ;
}
}
public string HomePageUrl
{
get { return "http://emby.media" ; }
}
public async Task < LiveTvServiceStatusInfo > GetStatusInfoAsync ( CancellationToken cancellationToken )
{
var status = new LiveTvServiceStatusInfo ( ) ;
var list = new List < LiveTvTunerInfo > ( ) ;
foreach ( var hostInstance in _liveTvManager . TunerHosts )
{
try
{
var tuners = await hostInstance . GetTunerInfos ( cancellationToken ) . ConfigureAwait ( false ) ;
list . AddRange ( tuners ) ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error getting tuners" , ex ) ;
}
}
status . Tuners = list ;
status . Status = LiveTvServiceStatus . Ok ;
status . Version = _appHpst . ApplicationVersion . ToString ( ) ;
status . IsVisible = false ;
return status ;
}
public async Task RefreshSeriesTimers ( CancellationToken cancellationToken , IProgress < double > progress )
{
var seriesTimers = await GetSeriesTimersAsync ( cancellationToken ) . ConfigureAwait ( false ) ;
List < ChannelInfo > channels = null ;
foreach ( var timer in seriesTimers )
{
List < ProgramInfo > epgData ;
if ( timer . RecordAnyChannel )
{
if ( channels = = null )
{
channels = ( await GetChannelsAsync ( true , CancellationToken . None ) . ConfigureAwait ( false ) ) . ToList ( ) ;
}
var channelIds = channels . Select ( i = > i . Id ) . ToList ( ) ;
epgData = GetEpgDataForChannels ( channelIds ) ;
}
else
{
epgData = GetEpgDataForChannel ( timer . ChannelId ) ;
}
await UpdateTimersForSeriesTimer ( epgData , timer , true ) . ConfigureAwait ( false ) ;
}
var timers = await GetTimersAsync ( cancellationToken ) . ConfigureAwait ( false ) ;
foreach ( var timer in timers . ToList ( ) )
{
if ( DateTime . UtcNow > timer . EndDate & & ! _activeRecordings . ContainsKey ( timer . Id ) )
{
_timerProvider . Delete ( timer ) ;
}
}
}
private List < ChannelInfo > _channelCache = null ;
private async Task < IEnumerable < ChannelInfo > > GetChannelsAsync ( bool enableCache , CancellationToken cancellationToken )
{
if ( enableCache & & _channelCache ! = null )
{
return _channelCache . ToList ( ) ;
}
var list = new List < ChannelInfo > ( ) ;
foreach ( var hostInstance in _liveTvManager . TunerHosts )
{
try
{
var channels = await hostInstance . GetChannels ( cancellationToken ) . ConfigureAwait ( false ) ;
list . AddRange ( channels ) ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error getting channels" , ex ) ;
}
}
foreach ( var provider in GetListingProviders ( ) )
{
var enabledChannels = list
. Where ( i = > IsListingProviderEnabledForTuner ( provider . Item2 , i . TunerHostId ) )
. ToList ( ) ;
if ( enabledChannels . Count > 0 )
{
try
{
await provider . Item1 . AddMetadata ( provider . Item2 , enabledChannels , cancellationToken ) . ConfigureAwait ( false ) ;
}
catch ( NotSupportedException )
{
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error adding metadata" , ex ) ;
}
}
}
_channelCache = list . ToList ( ) ;
return list ;
}
public Task < IEnumerable < ChannelInfo > > GetChannelsAsync ( CancellationToken cancellationToken )
{
return GetChannelsAsync ( false , cancellationToken ) ;
}
public Task CancelSeriesTimerAsync ( string timerId , CancellationToken cancellationToken )
{
var timers = _timerProvider
. GetAll ( )
. Where ( i = > string . Equals ( i . SeriesTimerId , timerId , StringComparison . OrdinalIgnoreCase ) )
. ToList ( ) ;
foreach ( var timer in timers )
{
CancelTimerInternal ( timer . Id ) ;
}
var remove = _seriesTimerProvider . GetAll ( ) . FirstOrDefault ( r = > string . Equals ( r . Id , timerId , StringComparison . OrdinalIgnoreCase ) ) ;
if ( remove ! = null )
{
_seriesTimerProvider . Delete ( remove ) ;
}
return Task . FromResult ( true ) ;
}
private void CancelTimerInternal ( string timerId )
{
var remove = _timerProvider . GetAll ( ) . FirstOrDefault ( r = > string . Equals ( r . Id , timerId , StringComparison . OrdinalIgnoreCase ) ) ;
if ( remove ! = null )
{
_timerProvider . Delete ( remove ) ;
}
ActiveRecordingInfo activeRecordingInfo ;
if ( _activeRecordings . TryGetValue ( timerId , out activeRecordingInfo ) )
{
activeRecordingInfo . CancellationTokenSource . Cancel ( ) ;
}
}
public Task CancelTimerAsync ( string timerId , CancellationToken cancellationToken )
{
CancelTimerInternal ( timerId ) ;
return Task . FromResult ( true ) ;
}
public Task DeleteRecordingAsync ( string recordingId , CancellationToken cancellationToken )
{
return Task . FromResult ( true ) ;
}
public Task CreateTimerAsync ( TimerInfo info , CancellationToken cancellationToken )
{
info . Id = Guid . NewGuid ( ) . ToString ( "N" ) ;
_timerProvider . Add ( info ) ;
return Task . FromResult ( 0 ) ;
}
public async Task CreateSeriesTimerAsync ( SeriesTimerInfo info , CancellationToken cancellationToken )
{
info . Id = Guid . NewGuid ( ) . ToString ( "N" ) ;
List < ProgramInfo > epgData ;
if ( info . RecordAnyChannel )
{
var channels = await GetChannelsAsync ( true , CancellationToken . None ) . ConfigureAwait ( false ) ;
var channelIds = channels . Select ( i = > i . Id ) . ToList ( ) ;
epgData = GetEpgDataForChannels ( channelIds ) ;
}
else
{
epgData = GetEpgDataForChannel ( info . ChannelId ) ;
}
// populate info.seriesID
var program = epgData . FirstOrDefault ( i = > string . Equals ( i . Id , info . ProgramId , StringComparison . OrdinalIgnoreCase ) ) ;
if ( program ! = null )
{
info . SeriesId = program . SeriesId ;
}
else
{
throw new InvalidOperationException ( "SeriesId for program not found" ) ;
}
_seriesTimerProvider . Add ( info ) ;
await UpdateTimersForSeriesTimer ( epgData , info , false ) . ConfigureAwait ( false ) ;
}
public async Task UpdateSeriesTimerAsync ( SeriesTimerInfo info , CancellationToken cancellationToken )
{
var instance = _seriesTimerProvider . GetAll ( ) . FirstOrDefault ( i = > string . Equals ( i . Id , info . Id , StringComparison . OrdinalIgnoreCase ) ) ;
if ( instance ! = null )
{
instance . ChannelId = info . ChannelId ;
instance . Days = info . Days ;
instance . EndDate = info . EndDate ;
instance . IsPostPaddingRequired = info . IsPostPaddingRequired ;
instance . IsPrePaddingRequired = info . IsPrePaddingRequired ;
instance . PostPaddingSeconds = info . PostPaddingSeconds ;
instance . PrePaddingSeconds = info . PrePaddingSeconds ;
instance . Priority = info . Priority ;
instance . RecordAnyChannel = info . RecordAnyChannel ;
instance . RecordAnyTime = info . RecordAnyTime ;
instance . RecordNewOnly = info . RecordNewOnly ;
instance . StartDate = info . StartDate ;
_seriesTimerProvider . Update ( instance ) ;
List < ProgramInfo > epgData ;
if ( instance . RecordAnyChannel )
{
var channels = await GetChannelsAsync ( true , CancellationToken . None ) . ConfigureAwait ( false ) ;
var channelIds = channels . Select ( i = > i . Id ) . ToList ( ) ;
epgData = GetEpgDataForChannels ( channelIds ) ;
}
else
{
epgData = GetEpgDataForChannel ( instance . ChannelId ) ;
}
await UpdateTimersForSeriesTimer ( epgData , instance , true ) . ConfigureAwait ( false ) ;
}
}
public Task UpdateTimerAsync ( TimerInfo info , CancellationToken cancellationToken )
{
_timerProvider . Update ( info ) ;
return Task . FromResult ( true ) ;
}
public Task < ImageStream > GetChannelImageAsync ( string channelId , CancellationToken cancellationToken )
{
throw new NotImplementedException ( ) ;
}
public Task < ImageStream > GetRecordingImageAsync ( string recordingId , CancellationToken cancellationToken )
{
throw new NotImplementedException ( ) ;
}
public Task < ImageStream > GetProgramImageAsync ( string programId , string channelId , CancellationToken cancellationToken )
{
throw new NotImplementedException ( ) ;
}
public async Task < IEnumerable < RecordingInfo > > GetRecordingsAsync ( CancellationToken cancellationToken )
{
return new List < RecordingInfo > ( ) ;
}
public Task < IEnumerable < TimerInfo > > GetTimersAsync ( CancellationToken cancellationToken )
{
return Task . FromResult ( ( IEnumerable < TimerInfo > ) _timerProvider . GetAll ( ) ) ;
}
public Task < SeriesTimerInfo > GetNewTimerDefaultsAsync ( CancellationToken cancellationToken , ProgramInfo program = null )
{
var config = GetConfiguration ( ) ;
var defaults = new SeriesTimerInfo ( )
{
PostPaddingSeconds = Math . Max ( config . PostPaddingSeconds , 0 ) ,
PrePaddingSeconds = Math . Max ( config . PrePaddingSeconds , 0 ) ,
RecordAnyChannel = false ,
RecordAnyTime = false ,
RecordNewOnly = false
} ;
if ( program ! = null )
{
defaults . SeriesId = program . SeriesId ;
defaults . ProgramId = program . Id ;
}
return Task . FromResult ( defaults ) ;
}
public Task < IEnumerable < SeriesTimerInfo > > GetSeriesTimersAsync ( CancellationToken cancellationToken )
{
return Task . FromResult ( ( IEnumerable < SeriesTimerInfo > ) _seriesTimerProvider . GetAll ( ) ) ;
}
public async Task < IEnumerable < ProgramInfo > > GetProgramsAsync ( string channelId , DateTime startDateUtc , DateTime endDateUtc , CancellationToken cancellationToken )
{
try
{
return await GetProgramsAsyncInternal ( channelId , startDateUtc , endDateUtc , cancellationToken ) . ConfigureAwait ( false ) ;
}
catch ( OperationCanceledException )
{
throw ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error getting programs" , ex ) ;
return GetEpgDataForChannel ( channelId ) . Where ( i = > i . StartDate < = endDateUtc & & i . EndDate > = startDateUtc ) ;
}
}
private bool IsListingProviderEnabledForTuner ( ListingsProviderInfo info , string tunerHostId )
{
if ( info . EnableAllTuners )
{
return true ;
}
if ( string . IsNullOrWhiteSpace ( tunerHostId ) )
{
throw new ArgumentNullException ( "tunerHostId" ) ;
}
return info . EnabledTuners . Contains ( tunerHostId , StringComparer . OrdinalIgnoreCase ) ;
}
private async Task < IEnumerable < ProgramInfo > > GetProgramsAsyncInternal ( string channelId , DateTime startDateUtc , DateTime endDateUtc , CancellationToken cancellationToken )
{
var channels = await GetChannelsAsync ( true , cancellationToken ) . ConfigureAwait ( false ) ;
var channel = channels . First ( i = > string . Equals ( i . Id , channelId , StringComparison . OrdinalIgnoreCase ) ) ;
foreach ( var provider in GetListingProviders ( ) )
{
if ( ! IsListingProviderEnabledForTuner ( provider . Item2 , channel . TunerHostId ) )
{
_logger . Debug ( "Skipping getting programs for channel {0}-{1} from {2}-{3}, because it's not enabled for this tuner." , channel . Number , channel . Name , provider . Item1 . Name , provider . Item2 . ListingsId ? ? string . Empty ) ;
continue ;
}
_logger . Debug ( "Getting programs for channel {0}-{1} from {2}-{3}" , channel . Number , channel . Name , provider . Item1 . Name , provider . Item2 . ListingsId ? ? string . Empty ) ;
var programs = await provider . Item1 . GetProgramsAsync ( provider . Item2 , channel . Number , channel . Name , startDateUtc , endDateUtc , cancellationToken )
. ConfigureAwait ( false ) ;
var list = programs . ToList ( ) ;
// Replace the value that came from the provider with a normalized value
foreach ( var program in list )
{
program . ChannelId = channelId ;
}
if ( list . Count > 0 )
{
SaveEpgDataForChannel ( channelId , list ) ;
return list ;
}
}
return new List < ProgramInfo > ( ) ;
}
private List < Tuple < IListingsProvider , ListingsProviderInfo > > GetListingProviders ( )
{
return GetConfiguration ( ) . ListingProviders
. Select ( i = >
{
var provider = _liveTvManager . ListingProviders . FirstOrDefault ( l = > string . Equals ( l . Type , i . Type , StringComparison . OrdinalIgnoreCase ) ) ;
return provider = = null ? null : new Tuple < IListingsProvider , ListingsProviderInfo > ( provider , i ) ;
} )
. Where ( i = > i ! = null )
. ToList ( ) ;
}
public Task < MediaSourceInfo > GetRecordingStream ( string recordingId , string streamId , CancellationToken cancellationToken )
{
throw new NotImplementedException ( ) ;
}
public async Task < MediaSourceInfo > GetChannelStream ( string channelId , string streamId , CancellationToken cancellationToken )
{
_logger . Info ( "Streaming Channel " + channelId ) ;
foreach ( var hostInstance in _liveTvManager . TunerHosts )
{
try
{
var result = await hostInstance . GetChannelStream ( channelId , streamId , cancellationToken ) . ConfigureAwait ( false ) ;
result . Item2 . Release ( ) ;
return result . Item1 ;
}
catch ( Exception e )
{
_logger . ErrorException ( "Error getting channel stream" , e ) ;
}
}
throw new ApplicationException ( "Tuner not found." ) ;
}
private async Task < Tuple < MediaSourceInfo , ITunerHost , SemaphoreSlim > > GetChannelStreamInternal ( string channelId , string streamId , CancellationToken cancellationToken )
{
_logger . Info ( "Streaming Channel " + channelId ) ;
foreach ( var hostInstance in _liveTvManager . TunerHosts )
{
try
{
var result = await hostInstance . GetChannelStream ( channelId , streamId , cancellationToken ) . ConfigureAwait ( false ) ;
return new Tuple < MediaSourceInfo , ITunerHost , SemaphoreSlim > ( result . Item1 , hostInstance , result . Item2 ) ;
}
catch ( Exception e )
{
_logger . ErrorException ( "Error getting channel stream" , e ) ;
}
}
throw new ApplicationException ( "Tuner not found." ) ;
}
public async Task < List < MediaSourceInfo > > GetChannelStreamMediaSources ( string channelId , CancellationToken cancellationToken )
{
foreach ( var hostInstance in _liveTvManager . TunerHosts )
{
try
{
var sources = await hostInstance . GetChannelStreamMediaSources ( channelId , cancellationToken ) . ConfigureAwait ( false ) ;
if ( sources . Count > 0 )
{
return sources ;
}
}
catch ( NotImplementedException )
{
}
}
throw new NotImplementedException ( ) ;
}
public Task < List < MediaSourceInfo > > GetRecordingStreamMediaSources ( string recordingId , CancellationToken cancellationToken )
{
throw new NotImplementedException ( ) ;
}
public Task CloseLiveStream ( string id , CancellationToken cancellationToken )
{
return Task . FromResult ( 0 ) ;
}
public Task RecordLiveStream ( string id , CancellationToken cancellationToken )
{
return Task . FromResult ( 0 ) ;
}
public Task ResetTuner ( string id , CancellationToken cancellationToken )
{
return Task . FromResult ( 0 ) ;
}
async void _timerProvider_TimerFired ( object sender , GenericEventArgs < TimerInfo > e )
{
var timer = e . Argument ;
_logger . Info ( "Recording timer fired." ) ;
try
{
var recordingEndDate = timer . EndDate . AddSeconds ( timer . PostPaddingSeconds ) ;
if ( recordingEndDate < = DateTime . UtcNow )
{
_logger . Warn ( "Recording timer fired for timer {0}, Id: {1}, but the program has already ended." , timer . Name , timer . Id ) ;
return ;
}
var activeRecordingInfo = new ActiveRecordingInfo
{
CancellationTokenSource = new CancellationTokenSource ( ) ,
TimerId = timer . Id
} ;
if ( _activeRecordings . TryAdd ( timer . Id , activeRecordingInfo ) )
{
await RecordStream ( timer , recordingEndDate , activeRecordingInfo , activeRecordingInfo . CancellationTokenSource . Token ) . ConfigureAwait ( false ) ;
}
else
{
_logger . Info ( "Skipping RecordStream because it's already in progress." ) ;
}
}
catch ( OperationCanceledException )
{
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error recording stream" , ex ) ;
}
}
private string GetRecordingPath ( TimerInfo timer , ProgramInfo info )
{
var recordPath = RecordingPath ;
var config = GetConfiguration ( ) ;
if ( info . IsMovie )
{
var customRecordingPath = config . MovieRecordingPath ;
var allowSubfolder = true ;
if ( ! string . IsNullOrWhiteSpace ( customRecordingPath ) )
{
allowSubfolder = string . Equals ( customRecordingPath , recordPath , StringComparison . OrdinalIgnoreCase ) ;
recordPath = customRecordingPath ;
}
if ( allowSubfolder & & config . EnableRecordingSubfolders )
{
recordPath = Path . Combine ( recordPath , "Movies" ) ;
}
var folderName = _fileSystem . GetValidFilename ( info . Name ) . Trim ( ) ;
if ( info . ProductionYear . HasValue )
{
folderName + = " (" + info . ProductionYear . Value . ToString ( CultureInfo . InvariantCulture ) + ")" ;
}
recordPath = Path . Combine ( recordPath , folderName ) ;
}
else if ( info . IsSeries )
{
var customRecordingPath = config . SeriesRecordingPath ;
var allowSubfolder = true ;
if ( ! string . IsNullOrWhiteSpace ( customRecordingPath ) )
{
allowSubfolder = string . Equals ( customRecordingPath , recordPath , StringComparison . OrdinalIgnoreCase ) ;
recordPath = customRecordingPath ;
}
if ( allowSubfolder & & config . EnableRecordingSubfolders )
{
recordPath = Path . Combine ( recordPath , "Series" ) ;
}
var folderName = _fileSystem . GetValidFilename ( info . Name ) . Trim ( ) ;
var folderNameWithYear = folderName ;
if ( info . ProductionYear . HasValue )
{
folderNameWithYear + = " (" + info . ProductionYear . Value . ToString ( CultureInfo . InvariantCulture ) + ")" ;
}
if ( Directory . Exists ( Path . Combine ( recordPath , folderName ) ) )
{
recordPath = Path . Combine ( recordPath , folderName ) ;
}
else
{
recordPath = Path . Combine ( recordPath , folderNameWithYear ) ;
}
if ( info . SeasonNumber . HasValue )
{
folderName = string . Format ( "Season {0}" , info . SeasonNumber . Value . ToString ( CultureInfo . InvariantCulture ) ) ;
recordPath = Path . Combine ( recordPath , folderName ) ;
}
}
else if ( info . IsKids )
{
if ( config . EnableRecordingSubfolders )
{
recordPath = Path . Combine ( recordPath , "Kids" ) ;
}
var folderName = _fileSystem . GetValidFilename ( info . Name ) . Trim ( ) ;
if ( info . ProductionYear . HasValue )
{
folderName + = " (" + info . ProductionYear . Value . ToString ( CultureInfo . InvariantCulture ) + ")" ;
}
recordPath = Path . Combine ( recordPath , folderName ) ;
}
else if ( info . IsSports )
{
if ( config . EnableRecordingSubfolders )
{
recordPath = Path . Combine ( recordPath , "Sports" ) ;
}
recordPath = Path . Combine ( recordPath , _fileSystem . GetValidFilename ( info . Name ) . Trim ( ) ) ;
}
else
{
if ( config . EnableRecordingSubfolders )
{
recordPath = Path . Combine ( recordPath , "Other" ) ;
}
recordPath = Path . Combine ( recordPath , _fileSystem . GetValidFilename ( info . Name ) . Trim ( ) ) ;
}
var recordingFileName = _fileSystem . GetValidFilename ( RecordingHelper . GetRecordingName ( timer , info ) ) . Trim ( ) + ".ts" ;
return Path . Combine ( recordPath , recordingFileName ) ;
}
private async Task RecordStream ( TimerInfo timer , DateTime recordingEndDate , ActiveRecordingInfo activeRecordingInfo , CancellationToken cancellationToken )
{
if ( timer = = null )
{
throw new ArgumentNullException ( "timer" ) ;
}
ProgramInfo info = null ;
if ( string . IsNullOrWhiteSpace ( timer . ProgramId ) )
{
_logger . Info ( "Timer {0} has null programId" , timer . Id ) ;
}
else
{
info = GetProgramInfoFromCache ( timer . ChannelId , timer . ProgramId ) ;
}
if ( info = = null )
{
_logger . Info ( "Unable to find program with Id {0}. Will search using start date" , timer . ProgramId ) ;
info = GetProgramInfoFromCache ( timer . ChannelId , timer . StartDate ) ;
}
if ( info = = null )
{
throw new InvalidOperationException ( string . Format ( "Program with Id {0} not found" , timer . ProgramId ) ) ;
}
var recordPath = GetRecordingPath ( timer , info ) ;
var recordingStatus = RecordingStatus . New ;
try
{
var result = await GetChannelStreamInternal ( timer . ChannelId , null , CancellationToken . None ) . ConfigureAwait ( false ) ;
var mediaStreamInfo = result . Item1 ;
var isResourceOpen = true ;
// Unfortunately due to the semaphore we have to have a nested try/finally
try
{
// HDHR doesn't seem to release the tuner right away after first probing with ffmpeg
//await Task.Delay(3000, cancellationToken).ConfigureAwait(false);
var recorder = await GetRecorder ( ) . ConfigureAwait ( false ) ;
recordPath = recorder . GetOutputPath ( mediaStreamInfo , recordPath ) ;
recordPath = EnsureFileUnique ( recordPath , timer . Id ) ;
_fileSystem . CreateDirectory ( Path . GetDirectoryName ( recordPath ) ) ;
activeRecordingInfo . Path = recordPath ;
_libraryMonitor . ReportFileSystemChangeBeginning ( recordPath ) ;
var duration = recordingEndDate - DateTime . UtcNow ;
_logger . Info ( "Beginning recording. Will record for {0} minutes." , duration . TotalMinutes . ToString ( CultureInfo . InvariantCulture ) ) ;
_logger . Info ( "Writing file to path: " + recordPath ) ;
_logger . Info ( "Opening recording stream from tuner provider" ) ;
Action onStarted = ( ) = >
{
result . Item3 . Release ( ) ;
isResourceOpen = false ;
} ;
var pathWithDuration = result . Item2 . ApplyDuration ( mediaStreamInfo . Path , duration ) ;
// If it supports supplying duration via url
if ( ! string . Equals ( pathWithDuration , mediaStreamInfo . Path , StringComparison . OrdinalIgnoreCase ) )
{
mediaStreamInfo . Path = pathWithDuration ;
mediaStreamInfo . RunTimeTicks = duration . Ticks ;
}
await recorder . Record ( mediaStreamInfo , recordPath , duration , onStarted , cancellationToken ) . ConfigureAwait ( false ) ;
recordingStatus = RecordingStatus . Completed ;
_logger . Info ( "Recording completed: {0}" , recordPath ) ;
}
finally
{
if ( isResourceOpen )
{
result . Item3 . Release ( ) ;
}
_libraryMonitor . ReportFileSystemChangeComplete ( recordPath , false ) ;
}
}
catch ( OperationCanceledException )
{
_logger . Info ( "Recording stopped: {0}" , recordPath ) ;
recordingStatus = RecordingStatus . Completed ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error recording to {0}" , ex , recordPath ) ;
recordingStatus = RecordingStatus . Error ;
}
finally
{
ActiveRecordingInfo removed ;
_activeRecordings . TryRemove ( timer . Id , out removed ) ;
}
if ( recordingStatus = = RecordingStatus . Completed )
{
OnSuccessfulRecording ( info . IsSeries , recordPath ) ;
_timerProvider . Delete ( timer ) ;
}
else if ( DateTime . UtcNow < timer . EndDate )
{
const int retryIntervalSeconds = 60 ;
_logger . Info ( "Retrying recording in {0} seconds." , retryIntervalSeconds ) ;
_timerProvider . StartTimer ( timer , TimeSpan . FromSeconds ( retryIntervalSeconds ) ) ;
}
else
{
_timerProvider . Delete ( timer ) ;
}
}
private string EnsureFileUnique ( string path , string timerId )
{
var originalPath = path ;
var index = 1 ;
while ( FileExists ( path , timerId ) )
{
var parent = Path . GetDirectoryName ( originalPath ) ;
var name = Path . GetFileNameWithoutExtension ( originalPath ) ;
name + = "-" + index . ToString ( CultureInfo . InvariantCulture ) ;
path = Path . ChangeExtension ( Path . Combine ( parent , name ) , Path . GetExtension ( originalPath ) ) ;
index + + ;
}
return path ;
}
private bool FileExists ( string path , string timerId )
{
if ( _fileSystem . FileExists ( path ) )
{
return true ;
}
var hasRecordingAtPath = _activeRecordings . Values . ToList ( ) . Any ( i = > string . Equals ( i . Path , path , StringComparison . OrdinalIgnoreCase ) & & ! string . Equals ( i . TimerId , timerId , StringComparison . OrdinalIgnoreCase ) ) ;
if ( hasRecordingAtPath )
{
return true ;
}
return false ;
}
private async Task < IRecorder > GetRecorder ( )
{
var config = GetConfiguration ( ) ;
if ( config . EnableRecordingEncoding )
{
var regInfo = await _security . GetRegistrationStatus ( "embytvrecordingconversion" ) . ConfigureAwait ( false ) ;
if ( regInfo . IsValid )
{
return new EncodedRecorder ( _logger , _fileSystem , _mediaEncoder , _config . ApplicationPaths , _jsonSerializer , config ) ;
}
}
return new DirectRecorder ( _logger , _httpClient , _fileSystem ) ;
}
private async void OnSuccessfulRecording ( bool isSeries , string path )
{
if ( GetConfiguration ( ) . EnableAutoOrganize )
{
if ( isSeries )
{
try
{
// this is to account for the library monitor holding a lock for additional time after the change is complete.
// ideally this shouldn't be hard-coded
await Task . Delay ( 30000 ) . ConfigureAwait ( false ) ;
var organize = new EpisodeFileOrganizer ( _organizationService , _config , _fileSystem , _logger , _libraryManager , _libraryMonitor , _providerManager ) ;
var result = await organize . OrganizeEpisodeFile ( path , CancellationToken . None ) . ConfigureAwait ( false ) ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error processing new recording" , ex ) ;
}
}
}
}
private ProgramInfo GetProgramInfoFromCache ( string channelId , string programId )
{
var epgData = GetEpgDataForChannel ( channelId ) ;
return epgData . FirstOrDefault ( p = > string . Equals ( p . Id , programId , StringComparison . OrdinalIgnoreCase ) ) ;
}
private ProgramInfo GetProgramInfoFromCache ( string channelId , DateTime startDateUtc )
{
var epgData = GetEpgDataForChannel ( channelId ) ;
var startDateTicks = startDateUtc . Ticks ;
// Find the first program that starts within 3 minutes
return epgData . FirstOrDefault ( p = > Math . Abs ( startDateTicks - p . StartDate . Ticks ) < = TimeSpan . FromMinutes ( 3 ) . Ticks ) ;
}
private LiveTvOptions GetConfiguration ( )
{
return _config . GetConfiguration < LiveTvOptions > ( "livetv" ) ;
}
private async Task UpdateTimersForSeriesTimer ( List < ProgramInfo > epgData , SeriesTimerInfo seriesTimer , bool deleteInvalidTimers )
{
var newTimers = GetTimersForSeries ( seriesTimer , epgData , true ) . ToList ( ) ;
var registration = await GetRegistrationInfo ( "seriesrecordings" ) . ConfigureAwait ( false ) ;
if ( registration . IsValid )
{
foreach ( var timer in newTimers )
{
_timerProvider . AddOrUpdate ( timer ) ;
}
}
if ( deleteInvalidTimers )
{
var allTimers = GetTimersForSeries ( seriesTimer , epgData , false )
. Select ( i = > i . Id )
. ToList ( ) ;
var deletes = _timerProvider . GetAll ( )
. Where ( i = > string . Equals ( i . SeriesTimerId , seriesTimer . Id , StringComparison . OrdinalIgnoreCase ) )
. Where ( i = > ! allTimers . Contains ( i . Id , StringComparer . OrdinalIgnoreCase ) & & i . StartDate > DateTime . UtcNow )
. ToList ( ) ;
foreach ( var timer in deletes )
{
await CancelTimerAsync ( timer . Id , CancellationToken . None ) . ConfigureAwait ( false ) ;
}
}
}
private IEnumerable < TimerInfo > GetTimersForSeries ( SeriesTimerInfo seriesTimer ,
IEnumerable < ProgramInfo > allPrograms ,
bool filterByCurrentRecordings )
{
if ( seriesTimer = = null )
{
throw new ArgumentNullException ( "seriesTimer" ) ;
}
if ( allPrograms = = null )
{
throw new ArgumentNullException ( "allPrograms" ) ;
}
// Exclude programs that have already ended
allPrograms = allPrograms . Where ( i = > i . EndDate > DateTime . UtcNow & & i . StartDate > DateTime . UtcNow ) ;
allPrograms = GetProgramsForSeries ( seriesTimer , allPrograms ) ;
if ( filterByCurrentRecordings )
{
allPrograms = allPrograms . Where ( i = > ! IsProgramAlreadyInLibrary ( i ) ) ;
}
return allPrograms . Select ( i = > RecordingHelper . CreateTimer ( i , seriesTimer ) ) ;
}
private bool IsProgramAlreadyInLibrary ( ProgramInfo program )
{
if ( ( program . EpisodeNumber . HasValue & & program . SeasonNumber . HasValue ) | | ! string . IsNullOrWhiteSpace ( program . EpisodeTitle ) )
{
var seriesIds = _libraryManager . GetItemIds ( new InternalItemsQuery
{
IncludeItemTypes = new [ ] { typeof ( Series ) . Name } ,
Name = program . Name
} ) . Select ( i = > i . ToString ( "N" ) ) . ToArray ( ) ;
if ( seriesIds . Length = = 0 )
{
return false ;
}
if ( program . EpisodeNumber . HasValue & & program . SeasonNumber . HasValue )
{
var result = _libraryManager . GetItemsResult ( new InternalItemsQuery
{
IncludeItemTypes = new [ ] { typeof ( Episode ) . Name } ,
ParentIndexNumber = program . SeasonNumber . Value ,
IndexNumber = program . EpisodeNumber . Value ,
AncestorIds = seriesIds ,
ExcludeLocationTypes = new [ ] { LocationType . Virtual }
} ) ;
if ( result . TotalRecordCount > 0 )
{
return true ;
}
}
if ( ! string . IsNullOrWhiteSpace ( program . EpisodeTitle ) )
{
var result = _libraryManager . GetItemsResult ( new InternalItemsQuery
{
IncludeItemTypes = new [ ] { typeof ( Episode ) . Name } ,
Name = program . EpisodeTitle ,
AncestorIds = seriesIds ,
ExcludeLocationTypes = new [ ] { LocationType . Virtual }
} ) ;
if ( result . TotalRecordCount > 0 )
{
return true ;
}
}
}
return false ;
}
private IEnumerable < ProgramInfo > GetProgramsForSeries ( SeriesTimerInfo seriesTimer , IEnumerable < ProgramInfo > allPrograms )
{
if ( ! seriesTimer . RecordAnyTime )
{
allPrograms = allPrograms . Where ( epg = > Math . Abs ( seriesTimer . StartDate . TimeOfDay . Ticks - epg . StartDate . TimeOfDay . Ticks ) < TimeSpan . FromMinutes ( 5 ) . Ticks ) ;
}
if ( seriesTimer . RecordNewOnly )
{
allPrograms = allPrograms . Where ( epg = > ! epg . IsRepeat ) ;
}
if ( ! seriesTimer . RecordAnyChannel )
{
allPrograms = allPrograms . Where ( epg = > string . Equals ( epg . ChannelId , seriesTimer . ChannelId , StringComparison . OrdinalIgnoreCase ) ) ;
}
allPrograms = allPrograms . Where ( i = > seriesTimer . Days . Contains ( i . StartDate . ToLocalTime ( ) . DayOfWeek ) ) ;
if ( string . IsNullOrWhiteSpace ( seriesTimer . SeriesId ) )
{
_logger . Error ( "seriesTimer.SeriesId is null. Cannot find programs for series" ) ;
return new List < ProgramInfo > ( ) ;
}
return allPrograms . Where ( i = > string . Equals ( i . SeriesId , seriesTimer . SeriesId , StringComparison . OrdinalIgnoreCase ) ) ;
}
private string GetChannelEpgCachePath ( string channelId )
{
return Path . Combine ( _config . CommonApplicationPaths . CachePath , "embytvepg" , channelId + ".json" ) ;
}
private readonly object _epgLock = new object ( ) ;
private void SaveEpgDataForChannel ( string channelId , List < ProgramInfo > epgData )
{
var path = GetChannelEpgCachePath ( channelId ) ;
_fileSystem . CreateDirectory ( Path . GetDirectoryName ( path ) ) ;
lock ( _epgLock )
{
_jsonSerializer . SerializeToFile ( epgData , path ) ;
}
}
private List < ProgramInfo > GetEpgDataForChannel ( string channelId )
{
try
{
lock ( _epgLock )
{
return _jsonSerializer . DeserializeFromFile < List < ProgramInfo > > ( GetChannelEpgCachePath ( channelId ) ) ;
}
}
catch
{
return new List < ProgramInfo > ( ) ;
}
}
private List < ProgramInfo > GetEpgDataForChannels ( List < string > channelIds )
{
return channelIds . SelectMany ( GetEpgDataForChannel ) . ToList ( ) ;
}
public void Dispose ( )
{
foreach ( var pair in _activeRecordings . ToList ( ) )
{
pair . Value . CancellationTokenSource . Cancel ( ) ;
}
}
public Task < MBRegistrationRecord > GetRegistrationInfo ( string feature )
{
if ( string . Equals ( feature , "seriesrecordings" , StringComparison . OrdinalIgnoreCase ) )
{
return _security . GetRegistrationStatus ( "embytvseriesrecordings" ) ;
}
return Task . FromResult ( new MBRegistrationRecord
{
IsValid = true ,
IsRegistered = true
} ) ;
}
public List < VirtualFolderInfo > GetRecordingFolders ( )
{
var list = new List < VirtualFolderInfo > ( ) ;
var defaultFolder = RecordingPath ;
var defaultName = "Recordings" ;
if ( Directory . Exists ( defaultFolder ) )
{
list . Add ( new VirtualFolderInfo
{
Locations = new List < string > { defaultFolder } ,
Name = defaultName
} ) ;
}
var customPath = GetConfiguration ( ) . MovieRecordingPath ;
if ( ( ! string . IsNullOrWhiteSpace ( customPath ) & & ! string . Equals ( customPath , defaultFolder , StringComparison . OrdinalIgnoreCase ) ) & & Directory . Exists ( customPath ) )
{
list . Add ( new VirtualFolderInfo
{
Locations = new List < string > { customPath } ,
Name = "Recorded Movies" ,
CollectionType = CollectionType . Movies
} ) ;
}
customPath = GetConfiguration ( ) . SeriesRecordingPath ;
if ( ( ! string . IsNullOrWhiteSpace ( customPath ) & & ! string . Equals ( customPath , defaultFolder , StringComparison . OrdinalIgnoreCase ) ) & & Directory . Exists ( customPath ) )
{
list . Add ( new VirtualFolderInfo
{
Locations = new List < string > { customPath } ,
Name = "Recorded Series" ,
CollectionType = CollectionType . TvShows
} ) ;
}
return list ;
}
class ActiveRecordingInfo
{
public string Path { get ; set ; }
public string TimerId { get ; set ; }
public CancellationTokenSource CancellationTokenSource { get ; set ; }
}
}
}