2013-10-16 04:11:45 +00:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
2017-08-09 19:38:18 +00:00
|
|
|
|
using System.IO;
|
2013-10-16 04:11:45 +00:00
|
|
|
|
using System.Linq;
|
2017-08-09 19:38:18 +00:00
|
|
|
|
using System.Text.RegularExpressions;
|
2013-10-16 04:11:45 +00:00
|
|
|
|
using NLog;
|
2014-01-06 06:20:08 +00:00
|
|
|
|
using NzbDrone.Common.Disk;
|
2013-10-16 04:11:45 +00:00
|
|
|
|
using NzbDrone.Core.Tv;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.DiskSpace
|
|
|
|
|
{
|
|
|
|
|
public interface IDiskSpaceService
|
|
|
|
|
{
|
|
|
|
|
List<DiskSpace> GetFreeSpace();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class DiskSpaceService : IDiskSpaceService
|
|
|
|
|
{
|
|
|
|
|
private readonly ISeriesService _seriesService;
|
|
|
|
|
private readonly IDiskProvider _diskProvider;
|
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
2018-07-07 09:09:04 +00:00
|
|
|
|
private static readonly Regex _regexSpecialDrive = new Regex("^/var/lib/(docker|rancher|kubelet)(/|$)|^/(boot|etc)(/|$)|/docker(/var)?/aufs(/|$)", RegexOptions.Compiled);
|
2017-08-09 19:38:18 +00:00
|
|
|
|
|
2017-02-11 06:46:39 +00:00
|
|
|
|
public DiskSpaceService(ISeriesService seriesService, IDiskProvider diskProvider, Logger logger)
|
2013-10-16 04:11:45 +00:00
|
|
|
|
{
|
|
|
|
|
_seriesService = seriesService;
|
|
|
|
|
_diskProvider = diskProvider;
|
|
|
|
|
_logger = logger;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public List<DiskSpace> GetFreeSpace()
|
|
|
|
|
{
|
2017-02-11 06:46:39 +00:00
|
|
|
|
var importantRootFolders = GetSeriesRootPaths().Distinct().ToList();
|
2013-10-16 04:11:45 +00:00
|
|
|
|
|
2017-08-10 17:01:22 +00:00
|
|
|
|
var optionalRootFolders = GetFixedDisksRootPaths().Except(importantRootFolders).Distinct().ToList();
|
|
|
|
|
|
|
|
|
|
var diskSpace = GetDiskSpace(importantRootFolders).Concat(GetDiskSpace(optionalRootFolders, true)).ToList();
|
|
|
|
|
|
|
|
|
|
return diskSpace;
|
2013-10-16 04:11:45 +00:00
|
|
|
|
}
|
|
|
|
|
|
2017-08-09 19:38:18 +00:00
|
|
|
|
private IEnumerable<string> GetSeriesRootPaths()
|
2013-10-16 04:11:45 +00:00
|
|
|
|
{
|
2020-10-18 21:59:26 +00:00
|
|
|
|
return _seriesService.GetAllSeriesPaths()
|
|
|
|
|
.Where(s => _diskProvider.FolderExists(s))
|
|
|
|
|
.Select(s => _diskProvider.GetPathRoot(s))
|
2017-02-07 22:06:14 +00:00
|
|
|
|
.Distinct();
|
2013-10-16 04:11:45 +00:00
|
|
|
|
}
|
|
|
|
|
|
2017-08-09 19:38:18 +00:00
|
|
|
|
private IEnumerable<string> GetFixedDisksRootPaths()
|
2013-10-16 04:11:45 +00:00
|
|
|
|
{
|
2017-08-09 19:38:18 +00:00
|
|
|
|
return _diskProvider.GetMounts()
|
|
|
|
|
.Where(d => d.DriveType == DriveType.Fixed)
|
|
|
|
|
.Where(d => !_regexSpecialDrive.IsMatch(d.RootDirectory))
|
|
|
|
|
.Select(d => d.RootDirectory);
|
2013-10-16 04:11:45 +00:00
|
|
|
|
}
|
|
|
|
|
|
2015-10-03 17:45:26 +00:00
|
|
|
|
private IEnumerable<DiskSpace> GetDiskSpace(IEnumerable<string> paths, bool suppressWarnings = false)
|
2013-10-16 04:11:45 +00:00
|
|
|
|
{
|
|
|
|
|
foreach (var path in paths)
|
|
|
|
|
{
|
|
|
|
|
DiskSpace diskSpace = null;
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
2014-02-26 16:08:04 +00:00
|
|
|
|
var freeSpace = _diskProvider.GetAvailableSpace(path);
|
|
|
|
|
var totalSpace = _diskProvider.GetTotalSize(path);
|
|
|
|
|
|
|
|
|
|
if (!freeSpace.HasValue || !totalSpace.HasValue)
|
|
|
|
|
{
|
|
|
|
|
continue;
|
|
|
|
|
}
|
2013-10-16 04:11:45 +00:00
|
|
|
|
|
|
|
|
|
diskSpace = new DiskSpace
|
|
|
|
|
{
|
|
|
|
|
Path = path,
|
2014-02-26 16:08:04 +00:00
|
|
|
|
FreeSpace = freeSpace.Value,
|
|
|
|
|
TotalSpace = totalSpace.Value
|
2013-10-16 04:11:45 +00:00
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
diskSpace.Label = _diskProvider.GetVolumeLabel(path);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception ex)
|
|
|
|
|
{
|
2015-06-03 05:59:30 +00:00
|
|
|
|
if (!suppressWarnings)
|
|
|
|
|
{
|
2016-02-11 21:13:42 +00:00
|
|
|
|
_logger.Warn(ex, "Unable to get free space for: " + path);
|
2015-06-03 05:59:30 +00:00
|
|
|
|
}
|
2013-10-16 04:11:45 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (diskSpace != null)
|
|
|
|
|
{
|
|
|
|
|
yield return diskSpace;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|