You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
453 lines
14 KiB
453 lines
14 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Security.AccessControl;
|
|
using System.Security.Principal;
|
|
using NLog;
|
|
using NzbDrone.Common.EnsureThat;
|
|
using NzbDrone.Common.EnvironmentInfo;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Common.Instrumentation;
|
|
using NzbDrone.Common.Instrumentation.Extensions;
|
|
|
|
namespace NzbDrone.Common.Disk
|
|
{
|
|
public abstract class DiskProviderBase : IDiskProvider
|
|
{
|
|
private static readonly Logger Logger = NzbDroneLogger.GetLogger(typeof(DiskProviderBase));
|
|
|
|
public abstract long? GetAvailableSpace(string path);
|
|
public abstract void InheritFolderPermissions(string filename);
|
|
public abstract void SetPermissions(string path, string mask, string user, string group);
|
|
public abstract long? GetTotalSize(string path);
|
|
|
|
public DateTime FolderGetCreationTime(string path)
|
|
{
|
|
CheckFolderExists(path);
|
|
|
|
return new DirectoryInfo(path).CreationTimeUtc;
|
|
}
|
|
|
|
public DateTime FolderGetLastWrite(string path)
|
|
{
|
|
CheckFolderExists(path);
|
|
|
|
var dirFiles = GetFiles(path, SearchOption.AllDirectories).ToList();
|
|
|
|
if (!dirFiles.Any())
|
|
{
|
|
return new DirectoryInfo(path).LastWriteTimeUtc;
|
|
}
|
|
|
|
return dirFiles.Select(f => new FileInfo(f)).Max(c => c.LastWriteTimeUtc);
|
|
}
|
|
|
|
public DateTime FileGetLastWrite(string path)
|
|
{
|
|
CheckFileExists(path);
|
|
|
|
return new FileInfo(path).LastWriteTimeUtc;
|
|
}
|
|
|
|
private void CheckFolderExists(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
if (!FolderExists(path))
|
|
{
|
|
throw new DirectoryNotFoundException("Directory doesn't exist. " + path);
|
|
}
|
|
}
|
|
|
|
private void CheckFileExists(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
if (!FileExists(path))
|
|
{
|
|
throw new FileNotFoundException("File doesn't exist: " + path);
|
|
}
|
|
}
|
|
|
|
public void EnsureFolder(string path)
|
|
{
|
|
if (!FolderExists(path))
|
|
{
|
|
CreateFolder(path);
|
|
}
|
|
}
|
|
|
|
public bool FolderExists(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
return Directory.Exists(path);
|
|
}
|
|
|
|
public bool FileExists(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
return FileExists(path, OsInfo.PathStringComparison);
|
|
}
|
|
|
|
public bool FileExists(string path, StringComparison stringComparison)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
switch (stringComparison)
|
|
{
|
|
case StringComparison.CurrentCulture:
|
|
case StringComparison.InvariantCulture:
|
|
case StringComparison.Ordinal:
|
|
{
|
|
return File.Exists(path) && path == path.GetActualCasing();
|
|
}
|
|
default:
|
|
{
|
|
return File.Exists(path);
|
|
}
|
|
}
|
|
}
|
|
|
|
public string[] GetDirectories(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
return Directory.GetDirectories(path);
|
|
}
|
|
|
|
public string[] GetFiles(string path, SearchOption searchOption)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
return Directory.GetFiles(path, "*.*", searchOption);
|
|
}
|
|
|
|
public long GetFolderSize(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
return GetFiles(path, SearchOption.AllDirectories).Sum(e => new FileInfo(e).Length);
|
|
}
|
|
|
|
public long GetFileSize(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
if (!FileExists(path))
|
|
{
|
|
throw new FileNotFoundException("File doesn't exist: " + path);
|
|
}
|
|
|
|
var fi = new FileInfo(path);
|
|
return fi.Length;
|
|
}
|
|
|
|
public void CreateFolder(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
Directory.CreateDirectory(path);
|
|
}
|
|
|
|
public void CopyFolder(string source, string destination)
|
|
{
|
|
Ensure.That(source, () => source).IsValidPath();
|
|
Ensure.That(destination, () => destination).IsValidPath();
|
|
|
|
TransferFolder(source, destination, TransferMode.Copy);
|
|
}
|
|
|
|
public void MoveFolder(string source, string destination)
|
|
{
|
|
Ensure.That(source, () => source).IsValidPath();
|
|
Ensure.That(destination, () => destination).IsValidPath();
|
|
|
|
try
|
|
{
|
|
TransferFolder(source, destination, TransferMode.Move);
|
|
DeleteFolder(source, true);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
e.Data.Add("Source", source);
|
|
e.Data.Add("Destination", destination);
|
|
throw;
|
|
}
|
|
}
|
|
|
|
public void TransferFolder(string source, string destination, TransferMode mode)
|
|
{
|
|
Ensure.That(source, () => source).IsValidPath();
|
|
Ensure.That(destination, () => destination).IsValidPath();
|
|
|
|
Logger.ProgressDebug("{0} {1} -> {2}", mode, source, destination);
|
|
|
|
var sourceFolder = new DirectoryInfo(source);
|
|
var targetFolder = new DirectoryInfo(destination);
|
|
|
|
if (!targetFolder.Exists)
|
|
{
|
|
targetFolder.Create();
|
|
}
|
|
|
|
foreach (var subDir in sourceFolder.GetDirectories())
|
|
{
|
|
TransferFolder(subDir.FullName, Path.Combine(destination, subDir.Name), mode);
|
|
}
|
|
|
|
foreach (var sourceFile in sourceFolder.GetFiles("*.*", SearchOption.TopDirectoryOnly))
|
|
{
|
|
var destFile = Path.Combine(destination, sourceFile.Name);
|
|
|
|
Logger.ProgressDebug("{0} {1} -> {2}", mode, sourceFile, destFile);
|
|
|
|
TransferFile(sourceFile.FullName, destFile, mode, true);
|
|
}
|
|
}
|
|
|
|
public void DeleteFile(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
Logger.Trace("Deleting file: {0}", path);
|
|
|
|
RemoveReadOnly(path);
|
|
|
|
File.Delete(path);
|
|
}
|
|
|
|
public void CopyFile(string source, string destination, bool overwrite = false)
|
|
{
|
|
TransferFile(source, destination, TransferMode.Copy, overwrite);
|
|
}
|
|
|
|
public void MoveFile(string source, string destination, bool overwrite = false)
|
|
{
|
|
TransferFile(source, destination, TransferMode.Move, overwrite);
|
|
}
|
|
|
|
public TransferMode TransferFile(string source, string destination, TransferMode mode, bool overwrite)
|
|
{
|
|
Ensure.That(source, () => source).IsValidPath();
|
|
Ensure.That(destination, () => destination).IsValidPath();
|
|
|
|
if (source.PathEquals(destination))
|
|
{
|
|
Logger.Warn("Source and destination can't be the same {0}", source);
|
|
return TransferMode.None;
|
|
}
|
|
|
|
if (FileExists(destination) && overwrite)
|
|
{
|
|
DeleteFile(destination);
|
|
}
|
|
|
|
if (mode.HasFlag(TransferMode.HardLink))
|
|
{
|
|
bool createdHardlink = TryCreateHardLink(source, destination);
|
|
if (createdHardlink)
|
|
{
|
|
return TransferMode.HardLink;
|
|
}
|
|
if (!mode.HasFlag(TransferMode.Copy))
|
|
{
|
|
throw new IOException("Hardlinking from '" + source + "' to '" + destination + "' failed.");
|
|
}
|
|
}
|
|
|
|
if (mode.HasFlag(TransferMode.Copy))
|
|
{
|
|
File.Copy(source, destination, overwrite);
|
|
return TransferMode.Copy;
|
|
}
|
|
|
|
if (mode.HasFlag(TransferMode.Move))
|
|
{
|
|
RemoveReadOnly(source);
|
|
File.Move(source, destination);
|
|
return TransferMode.Move;
|
|
}
|
|
|
|
return TransferMode.None;
|
|
}
|
|
|
|
public abstract bool TryCreateHardLink(string source, string destination);
|
|
|
|
public void DeleteFolder(string path, bool recursive)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
var files = Directory.GetFiles(path, "*.*", recursive ? SearchOption.AllDirectories : SearchOption.TopDirectoryOnly);
|
|
Array.ForEach(files, RemoveReadOnly);
|
|
|
|
Directory.Delete(path, recursive);
|
|
}
|
|
|
|
public string ReadAllText(string filePath)
|
|
{
|
|
Ensure.That(filePath, () => filePath).IsValidPath();
|
|
|
|
return File.ReadAllText(filePath);
|
|
}
|
|
|
|
public void WriteAllText(string filename, string contents)
|
|
{
|
|
Ensure.That(filename, () => filename).IsValidPath();
|
|
RemoveReadOnly(filename);
|
|
File.WriteAllText(filename, contents);
|
|
}
|
|
|
|
public void FolderSetLastWriteTime(string path, DateTime dateTime)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
Directory.SetLastWriteTimeUtc(path, dateTime);
|
|
}
|
|
|
|
public void FileSetLastWriteTime(string path, DateTime dateTime)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
File.SetLastWriteTime(path, dateTime);
|
|
}
|
|
|
|
public bool IsFileLocked(string file)
|
|
{
|
|
try
|
|
{
|
|
using (File.Open(file, FileMode.Open, FileAccess.Read, FileShare.None))
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
catch (IOException)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
public string GetPathRoot(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
return Path.GetPathRoot(path);
|
|
}
|
|
|
|
public string GetParentFolder(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
var parent = Directory.GetParent(path);
|
|
|
|
if (parent == null)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
return parent.FullName;
|
|
}
|
|
|
|
public void SetPermissions(string filename, WellKnownSidType accountSid, FileSystemRights rights, AccessControlType controlType)
|
|
{
|
|
try
|
|
{
|
|
var sid = new SecurityIdentifier(accountSid, null);
|
|
|
|
var directoryInfo = new DirectoryInfo(filename);
|
|
var directorySecurity = directoryInfo.GetAccessControl();
|
|
|
|
var accessRule = new FileSystemAccessRule(sid, rights,
|
|
InheritanceFlags.ContainerInherit | InheritanceFlags.ObjectInherit,
|
|
PropagationFlags.None, controlType);
|
|
|
|
directorySecurity.AddAccessRule(accessRule);
|
|
directoryInfo.SetAccessControl(directorySecurity);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.WarnException(string.Format("Couldn't set permission for {0}. account:{1} rights:{2} accessControlType:{3}", filename, accountSid, rights, controlType), e);
|
|
throw;
|
|
}
|
|
|
|
}
|
|
|
|
private static void RemoveReadOnly(string path)
|
|
{
|
|
if (File.Exists(path))
|
|
{
|
|
var attributes = File.GetAttributes(path);
|
|
|
|
if (attributes.HasFlag(FileAttributes.ReadOnly))
|
|
{
|
|
var newAttributes = attributes & ~(FileAttributes.ReadOnly);
|
|
File.SetAttributes(path, newAttributes);
|
|
}
|
|
}
|
|
}
|
|
|
|
public FileAttributes GetFileAttributes(string path)
|
|
{
|
|
return File.GetAttributes(path);
|
|
}
|
|
|
|
public void EmptyFolder(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
foreach (var file in GetFiles(path, SearchOption.TopDirectoryOnly))
|
|
{
|
|
DeleteFile(file);
|
|
}
|
|
|
|
foreach (var directory in GetDirectories(path))
|
|
{
|
|
DeleteFolder(directory, true);
|
|
}
|
|
}
|
|
|
|
public string[] GetFixedDrives()
|
|
{
|
|
return (DriveInfo.GetDrives().Where(x => x.DriveType == DriveType.Fixed).Select(x => x.Name)).ToArray();
|
|
}
|
|
|
|
public string GetVolumeLabel(string path)
|
|
{
|
|
var driveInfo = DriveInfo.GetDrives().SingleOrDefault(d => d.Name == path);
|
|
|
|
if (driveInfo == null)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
return driveInfo.VolumeLabel;
|
|
}
|
|
|
|
public FileStream StreamFile(string path)
|
|
{
|
|
if (!FileExists(path))
|
|
{
|
|
throw new FileNotFoundException("Unable to find file: " + path, path);
|
|
}
|
|
|
|
return new FileStream(path, FileMode.Open, FileAccess.Read);
|
|
}
|
|
|
|
public List<DirectoryInfo> GetDirectoryInfos(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
var di = new DirectoryInfo(path);
|
|
|
|
return di.GetDirectories().ToList();
|
|
}
|
|
|
|
public List<FileInfo> GetFileInfos(string path)
|
|
{
|
|
Ensure.That(path, () => path).IsValidPath();
|
|
|
|
var di = new DirectoryInfo(path);
|
|
|
|
return di.GetFiles().ToList();
|
|
}
|
|
}
|
|
}
|