@ -6,21 +6,22 @@ using Ninject;
using NLog ;
using NzbDrone.Core.Providers.Core ;
using NzbDrone.Core.Repository ;
using PetaPoco ;
using SubSonic.Repository ;
namespace NzbDrone.Core.Providers
{
public class RootDirProvider
{
private readonly I Repository _repository ;
private readonly I Database _database ;
private static readonly Logger Logger = LogManager . GetCurrentClassLogger ( ) ;
private readonly DiskProvider _diskProvider ;
private readonly SeriesProvider _seriesProvider ;
[Inject]
public RootDirProvider ( I Repository repository , SeriesProvider seriesProvider , DiskProvider diskProvider )
public RootDirProvider ( I Database database , SeriesProvider seriesProvider , DiskProvider diskProvider )
{
_ repository = repository ;
_ database = database ;
_diskProvider = diskProvider ;
_seriesProvider = seriesProvider ;
}
@ -29,26 +30,25 @@ namespace NzbDrone.Core.Providers
public virtual List < RootDir > GetAll ( )
{
return _ repository. All < RootDir > ( ) . ToList ( ) ;
return _ database. Fetch < RootDir > ( ) ;
}
public virtual int Add ( RootDir rootDir )
{
ValidatePath ( rootDir ) ;
return ( int ) _repository . Add ( rootDir ) ;
return Convert . ToInt32 ( _database . Insert ( rootDir ) ) ;
}
public virtual void Remove ( int rootDirId )
{
_ repository . Delete < RootDir > ( rootDirId ) ;
_ database . Delete < RootDir > ( rootDirId ) ;
}
public virtual void Update ( RootDir rootDir )
{
ValidatePath ( rootDir ) ;
_ repository . Update ( rootDir ) ;
_ database . Update ( rootDir ) ;
}
private static void ValidatePath ( RootDir rootDir )
@ -61,7 +61,7 @@ namespace NzbDrone.Core.Providers
public virtual RootDir GetRootDir ( int rootDirId )
{
return _ repository. Single < RootDir > ( rootDirId ) ;
return _ database. SingleOrDefault < RootDir > ( rootDirId ) ;
}
public List < String > GetUnmappedFolders ( string path )
@ -78,7 +78,6 @@ namespace NzbDrone.Core.Providers
return results ;
}
foreach ( string seriesFolder in _diskProvider . GetDirectories ( path ) )
{
var cleanPath = Parser . NormalizePath ( new DirectoryInfo ( seriesFolder ) . FullName ) ;