mirror of https://github.com/Radarr/Radarr
179 lines
6.5 KiB
C#
179 lines
6.5 KiB
C#
using System;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Security.AccessControl;
|
|
using System.Security.Principal;
|
|
using NLog;
|
|
using NzbDrone.Common.Disk;
|
|
using NzbDrone.Common.Exceptions;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Common.Instrumentation;
|
|
|
|
namespace NzbDrone.Common.EnvironmentInfo
|
|
{
|
|
public interface IAppFolderFactory
|
|
{
|
|
void Register();
|
|
}
|
|
|
|
public class AppFolderFactory : IAppFolderFactory
|
|
{
|
|
private readonly IAppFolderInfo _appFolderInfo;
|
|
private readonly IStartupContext _startupContext;
|
|
private readonly IDiskProvider _diskProvider;
|
|
private readonly IDiskTransferService _diskTransferService;
|
|
private readonly Logger _logger;
|
|
|
|
public AppFolderFactory(IAppFolderInfo appFolderInfo,
|
|
IStartupContext startupContext,
|
|
IDiskProvider diskProvider,
|
|
IDiskTransferService diskTransferService)
|
|
{
|
|
_appFolderInfo = appFolderInfo;
|
|
_startupContext = startupContext;
|
|
_diskProvider = diskProvider;
|
|
_diskTransferService = diskTransferService;
|
|
_logger = NzbDroneLogger.GetLogger(this);
|
|
}
|
|
|
|
public void Register()
|
|
{
|
|
try
|
|
{
|
|
MigrateAppDataFolder();
|
|
_diskProvider.EnsureFolder(_appFolderInfo.AppDataFolder);
|
|
}
|
|
catch (UnauthorizedAccessException)
|
|
{
|
|
throw new RadarrStartupException("Cannot create AppFolder, Access to the path {0} is denied", _appFolderInfo.AppDataFolder);
|
|
}
|
|
|
|
if (OsInfo.IsWindows)
|
|
{
|
|
SetPermissions();
|
|
}
|
|
|
|
if (!_diskProvider.FolderWritable(_appFolderInfo.AppDataFolder))
|
|
{
|
|
throw new RadarrStartupException("AppFolder {0} is not writable", _appFolderInfo.AppDataFolder);
|
|
}
|
|
|
|
InitializeMonoApplicationData();
|
|
}
|
|
|
|
private void SetPermissions()
|
|
{
|
|
try
|
|
{
|
|
_diskProvider.SetEveryonePermissions(_appFolderInfo.AppDataFolder);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.Warn(ex, "Coudn't set app folder permission");
|
|
}
|
|
}
|
|
|
|
private void MigrateAppDataFolder()
|
|
{
|
|
try
|
|
{
|
|
var oldDbFile = Path.Combine(_appFolderInfo.AppDataFolder, "nzbdrone.db");
|
|
|
|
if (_startupContext.Args.ContainsKey(StartupContext.APPDATA))
|
|
{
|
|
if (_diskProvider.FileExists(_appFolderInfo.GetDatabase()))
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (!_diskProvider.FileExists(oldDbFile))
|
|
{
|
|
return;
|
|
}
|
|
|
|
_diskProvider.MoveFile(oldDbFile, _appFolderInfo.GetDatabase());
|
|
CleanupSqLiteRollbackFiles();
|
|
RemovePidFile();
|
|
}
|
|
|
|
// Exit if a radarr.db already exists
|
|
if (_diskProvider.FileExists(_appFolderInfo.GetDatabase()))
|
|
{
|
|
return;
|
|
}
|
|
|
|
// Rename the DB file
|
|
if (_diskProvider.FileExists(oldDbFile))
|
|
{
|
|
_diskProvider.MoveFile(oldDbFile, _appFolderInfo.GetDatabase());
|
|
}
|
|
|
|
// Remove SQLite rollback files
|
|
CleanupSqLiteRollbackFiles();
|
|
|
|
// Remove Old PID file
|
|
RemovePidFile();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.Debug(ex, ex.Message);
|
|
throw new RadarrStartupException("Unable to migrate DB from nzbdrone.db to {0}. Migrate manually", _appFolderInfo.GetDatabase());
|
|
}
|
|
}
|
|
|
|
private void InitializeMonoApplicationData()
|
|
{
|
|
if (OsInfo.IsWindows)
|
|
{
|
|
return;
|
|
}
|
|
|
|
try
|
|
{
|
|
// It seems that DoNotVerify is the mono behaviour even though .net docs specify a blank string
|
|
// should be returned if the data doesn't exist. For compatibility with .net core, explicitly
|
|
// set DoNotVerify (which makes sense given we're explicitly checking that the folder exists)
|
|
var configHome = Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData, Environment.SpecialFolderOption.DoNotVerify);
|
|
if (configHome.IsNullOrWhiteSpace() ||
|
|
configHome == "/.config" ||
|
|
(configHome.EndsWith("/.config") && !_diskProvider.FolderExists(configHome.GetParentPath())) ||
|
|
!_diskProvider.FolderExists(configHome))
|
|
{
|
|
// Tell mono/netcore to use appData/.config as ApplicationData folder.
|
|
Environment.SetEnvironmentVariable("XDG_CONFIG_HOME", Path.Combine(_appFolderInfo.AppDataFolder, ".config"));
|
|
}
|
|
|
|
var dataHome = Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData, Environment.SpecialFolderOption.DoNotVerify);
|
|
if (dataHome.IsNullOrWhiteSpace() ||
|
|
dataHome == "/.local/share" ||
|
|
(dataHome.EndsWith("/.local/share") && !_diskProvider.FolderExists(dataHome.GetParentPath().GetParentPath())) ||
|
|
!_diskProvider.FolderExists(dataHome))
|
|
{
|
|
// Tell mono/netcore to use appData/.config/share as LocalApplicationData folder.
|
|
Environment.SetEnvironmentVariable("XDG_DATA_HOME", Path.Combine(_appFolderInfo.AppDataFolder, ".config/share"));
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.Warn(ex, "Failed to initialize the mono config directory.");
|
|
}
|
|
}
|
|
|
|
private void CleanupSqLiteRollbackFiles()
|
|
{
|
|
_diskProvider.GetFiles(_appFolderInfo.AppDataFolder, SearchOption.TopDirectoryOnly)
|
|
.Where(f => Path.GetFileName(f).StartsWith("nzbdrone.db"))
|
|
.ToList()
|
|
.ForEach(_diskProvider.DeleteFile);
|
|
}
|
|
|
|
private void RemovePidFile()
|
|
{
|
|
if (OsInfo.IsNotWindows)
|
|
{
|
|
_diskProvider.DeleteFile(Path.Combine(_appFolderInfo.AppDataFolder, "radarr.pid"));
|
|
}
|
|
}
|
|
}
|
|
}
|