sonarr-repo-only/NzbDrone.Common/PathExtensions.cs

using System;
using System.IO;

namespace NzbDrone.Common
{
    public static class PathExtensions
    {
        private const string WEB_FOLDER = "NzbDrone.Web\\";
        private const string APP_DATA = "App_Data\\";
        private const string WEB_BIN = "bin\\";

        private const string LOG_CONFIG_FILE = "log.config";
        private const string APP_CONFIG_FILE = "config.xml";

        public const string NZBDRONE_EXE = "NzbDrone.exe";

        public const string OBJ_DB_FOLDER = "objDb";
        public const string NZBDRONE_DB = "nzbdrone.db";

        private const string BACKUP_ZIP_FILE = "NzbDrone_Backup.zip";

        private const string UPDATE_SANDBOX_FOLDER_NAME = "nzbdrone_update\\";
        private const string UPDATE_PACKAGE_FOLDER_NAME = "nzbdrone\\";
        private const string UPDATE_BACKUP_FOLDER_NAME = "nzbdrone_backup\\";
        private const string UPDATE_CLIENT_EXE = "nzbdrone.update.exe";
        private const string UPDATE_CLIENT_FOLDER_NAME = "NzbDrone.Update\\";
        private const string UPDATE_LOG_FOLDER_NAME = "UpdateLogs\\";

        public static string NormalizePath(this string path)
        {
            if (String.IsNullOrWhiteSpace(path))
                throw new ArgumentException("Path can not be null or empty");

            var info = new FileInfo(path);

            if (info.FullName.StartsWith(@"\\")) //UNC
            {
                return info.FullName.TrimEnd('/', '\\', ' ');
            }

            return info.FullName.Trim('/', '\\', ' ');
        }

        public static string GetWebRoot(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.ApplicationPath, WEB_FOLDER);
        }

        public static string GetAppDataPath(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetWebRoot(), APP_DATA);
        }

        public static string GetWebBinPath(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetWebRoot(), WEB_BIN);
        }

        public static string GetNlogConfigPath(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetWebRoot(), LOG_CONFIG_FILE);
        }

        public static string GetConfigPath(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.ApplicationPath, APP_CONFIG_FILE);
        }

        public static string GetObjectDbFolder(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetAppDataPath(), OBJ_DB_FOLDER);
        }

        public static string GetMediaCoverPath(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetWebRoot(), "MediaCover");
        }

        public static string GetBannerPath(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetMediaCoverPath(), "Banners");
        }

        public static string GetFanArtPath(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetMediaCoverPath(), "Fanarts");
        }

        public static string GetCacheFolder(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetWebRoot(), "Cache");
        }

        public static string GetUpdateLogFolder(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.ApplicationPath, UPDATE_LOG_FOLDER_NAME);
        }

        public static string GetUpdateSandboxFolder(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.SystemTemp, UPDATE_SANDBOX_FOLDER_NAME);
        }

        public static string GetUpdateBackUpFolder(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetUpdateSandboxFolder(), UPDATE_BACKUP_FOLDER_NAME);
        }

        public static string GetUpdatePackageFolder(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetUpdateSandboxFolder(), UPDATE_PACKAGE_FOLDER_NAME);
        }

        public static string GetUpdateClientFolder(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetUpdatePackageFolder(), UPDATE_CLIENT_FOLDER_NAME);
        }

        public static string GetUpdateClientExePath(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetUpdateSandboxFolder(), UPDATE_CLIENT_EXE);
        }

        public static string GetSandboxLogFolder(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetUpdateSandboxFolder(), UPDATE_LOG_FOLDER_NAME);
        }

        public static string GetConfigBackupFile(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetAppDataPath(), BACKUP_ZIP_FILE);
        }

        public static string GetNzbDroneExe(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.ApplicationPath, NZBDRONE_EXE);
        }

        public static string GetNzbDroneDatabase(this EnvironmentProvider environmentProvider)
        {
            return Path.Combine(environmentProvider.GetAppDataPath(), NZBDRONE_DB);
        }
    }
}