| | | 1 | | using System; |
| | | 2 | | using System.Globalization; |
| | | 3 | | using System.IO; |
| | | 4 | | using MediaBrowser.Common.Configuration; |
| | | 5 | | using MediaBrowser.Model.System; |
| | | 6 | | using Microsoft.Extensions.Logging; |
| | | 7 | | |
| | | 8 | | namespace Jellyfin.Server.Implementations.StorageHelpers; |
| | | 9 | | |
| | | 10 | | /// <summary> |
| | | 11 | | /// Contains methods to help with checking for storage and returning storage data for jellyfin folders. |
| | | 12 | | /// </summary> |
| | | 13 | | public static class StorageHelper |
| | | 14 | | { |
| | | 15 | | private const long TwoGigabyte = 2_147_483_647L; |
| | 0 | 16 | | private static readonly string[] _byteHumanizedSuffixes = ["B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB"]; |
| | | 17 | | |
| | | 18 | | /// <summary> |
| | | 19 | | /// Tests the available storage capacity on the jellyfin paths with estimated minimum values. |
| | | 20 | | /// </summary> |
| | | 21 | | /// <param name="applicationPaths">The application paths.</param> |
| | | 22 | | /// <param name="logger">Logger.</param> |
| | | 23 | | public static void TestCommonPathsForStorageCapacity(IApplicationPaths applicationPaths, ILogger logger) |
| | | 24 | | { |
| | 0 | 25 | | TestDataDirectorySize(applicationPaths.DataPath, logger, TwoGigabyte); |
| | 0 | 26 | | TestDataDirectorySize(applicationPaths.CachePath, logger, TwoGigabyte); |
| | 0 | 27 | | TestDataDirectorySize(applicationPaths.ProgramDataPath, logger, TwoGigabyte); |
| | 0 | 28 | | } |
| | | 29 | | |
| | | 30 | | /// <summary> |
| | | 31 | | /// Gets the free space of a specific directory. |
| | | 32 | | /// </summary> |
| | | 33 | | /// <param name="path">Path to a folder.</param> |
| | | 34 | | /// <returns>The number of bytes available space.</returns> |
| | | 35 | | public static FolderStorageInfo GetFreeSpaceOf(string path) |
| | | 36 | | { |
| | | 37 | | try |
| | | 38 | | { |
| | 0 | 39 | | var driveInfo = new DriveInfo(path); |
| | 0 | 40 | | return new FolderStorageInfo() |
| | 0 | 41 | | { |
| | 0 | 42 | | Path = path, |
| | 0 | 43 | | FreeSpace = driveInfo.AvailableFreeSpace, |
| | 0 | 44 | | UsedSpace = driveInfo.TotalSize - driveInfo.AvailableFreeSpace, |
| | 0 | 45 | | StorageType = driveInfo.DriveType.ToString(), |
| | 0 | 46 | | DeviceId = driveInfo.Name, |
| | 0 | 47 | | }; |
| | | 48 | | } |
| | 0 | 49 | | catch |
| | | 50 | | { |
| | 0 | 51 | | return new FolderStorageInfo() |
| | 0 | 52 | | { |
| | 0 | 53 | | Path = path, |
| | 0 | 54 | | FreeSpace = -1, |
| | 0 | 55 | | UsedSpace = -1, |
| | 0 | 56 | | StorageType = null, |
| | 0 | 57 | | DeviceId = null |
| | 0 | 58 | | }; |
| | | 59 | | } |
| | 0 | 60 | | } |
| | | 61 | | |
| | | 62 | | /// <summary> |
| | | 63 | | /// Gets the underlying drive data from a given path and checks if the available storage capacity matches the thresh |
| | | 64 | | /// </summary> |
| | | 65 | | /// <param name="path">The path to a folder to evaluate.</param> |
| | | 66 | | /// <param name="logger">The logger.</param> |
| | | 67 | | /// <param name="threshold">The threshold to check for or -1 to just log the data.</param> |
| | | 68 | | /// <exception cref="InvalidOperationException">Thrown when the threshold is not available on the underlying storage |
| | | 69 | | private static void TestDataDirectorySize(string path, ILogger logger, long threshold = -1) |
| | | 70 | | { |
| | 0 | 71 | | logger.LogDebug("Check path {TestPath} for storage capacity", path); |
| | 0 | 72 | | Directory.CreateDirectory(path); |
| | | 73 | | |
| | 0 | 74 | | var drive = new DriveInfo(path); |
| | 0 | 75 | | if (threshold != -1 && drive.AvailableFreeSpace < threshold) |
| | | 76 | | { |
| | 0 | 77 | | throw new InvalidOperationException($"The path `{path}` has insufficient free space. Available: {HumanizeSto |
| | | 78 | | } |
| | | 79 | | |
| | 0 | 80 | | logger.LogInformation( |
| | 0 | 81 | | "Storage path `{TestPath}` ({StorageType}) successfully checked with {FreeSpace} free which is over the mini |
| | 0 | 82 | | path, |
| | 0 | 83 | | drive.DriveType, |
| | 0 | 84 | | HumanizeStorageSize(drive.AvailableFreeSpace), |
| | 0 | 85 | | HumanizeStorageSize(threshold)); |
| | 0 | 86 | | } |
| | | 87 | | |
| | | 88 | | /// <summary> |
| | | 89 | | /// Formats a size in bytes into a common human readable form. |
| | | 90 | | /// </summary> |
| | | 91 | | /// <remarks> |
| | | 92 | | /// Taken and slightly modified from https://stackoverflow.com/a/4975942/1786007 . |
| | | 93 | | /// </remarks> |
| | | 94 | | /// <param name="byteCount">The size in bytes.</param> |
| | | 95 | | /// <returns>A human readable approximate representation of the argument.</returns> |
| | | 96 | | public static string HumanizeStorageSize(long byteCount) |
| | | 97 | | { |
| | 0 | 98 | | if (byteCount == 0) |
| | | 99 | | { |
| | 0 | 100 | | return $"0{_byteHumanizedSuffixes[0]}"; |
| | | 101 | | } |
| | | 102 | | |
| | 0 | 103 | | var bytes = Math.Abs(byteCount); |
| | 0 | 104 | | var place = Convert.ToInt32(Math.Floor(Math.Log(bytes, 1024))); |
| | 0 | 105 | | var num = Math.Round(bytes / Math.Pow(1024, place), 1); |
| | 0 | 106 | | return (Math.Sign(byteCount) * num).ToString(CultureInfo.InvariantCulture) + _byteHumanizedSuffixes[place]; |
| | | 107 | | } |
| | | 108 | | } |