jellyfin-server/MediaBrowser.Model/Net/MimeTypes.cs

#pragma warning disable CS1591

using System;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using System.IO;
using System.Linq;
using Jellyfin.Extensions;

namespace MediaBrowser.Model.Net
{
    /// <summary>
    /// Class MimeTypes.
    /// </summary>
    ///
    /// http://en.wikipedia.org/wiki/Internet_media_type
    /// https://developer.mozilla.org/en-US/docs/Web/HTTP/Basics_of_HTTP/MIME_types/Common_types
    /// http://www.iana.org/assignments/media-types/media-types.xhtml
    public static class MimeTypes
    {
        /// <summary>
        /// Any extension in this list is considered a video file.
        /// </summary>
        private static readonly HashSet<string> _videoFileExtensions = new HashSet<string>(StringComparer.OrdinalIgnoreCase)
        {
            ".3gp",
            ".asf",
            ".avi",
            ".divx",
            ".dvr-ms",
            ".f4v",
            ".flv",
            ".img",
            ".iso",
            ".m2t",
            ".m2ts",
            ".m2v",
            ".m4v",
            ".mk3d",
            ".mkv",
            ".mov",
            ".mp4",
            ".mpg",
            ".mpeg",
            ".mts",
            ".ogg",
            ".ogm",
            ".ogv",
            ".rec",
            ".ts",
            ".rmvb",
            ".webm",
            ".wmv",
            ".wtv",
        };

        /// <summary>
        /// Used for extensions not in <see cref="Model.MimeTypes"/> or to override them.
        /// </summary>
        private static readonly Dictionary<string, string> _mimeTypeLookup = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase)
        {
            // Type application
            { ".azw3", "application/vnd.amazon.ebook" },

            // Type image
            { ".tbn", "image/jpeg" },

            // Type text
            { ".ass", "text/x-ssa" },
            { ".ssa", "text/x-ssa" },
            { ".edl", "text/plain" },
            { ".html", "text/html; charset=UTF-8" },
            { ".htm", "text/html; charset=UTF-8" },

            // Type video
            { ".mpegts", "video/mp2t" },

            // Type audio
            { ".aac", "audio/aac" },
            { ".ac3", "audio/ac3" },
            { ".ape", "audio/x-ape" },
            { ".dsf", "audio/dsf" },
            { ".dsp", "audio/dsp" },
            { ".flac", "audio/flac" },
            { ".m4b", "audio/m4b" },
            { ".mp3", "audio/mpeg" },
            { ".vorbis", "audio/vorbis" },
            { ".webma", "audio/webm" },
            { ".wv", "audio/x-wavpack" },
            { ".xsp", "audio/xsp" },
        };

        private static readonly Dictionary<string, string> _extensionLookup = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase)
        {
            // Type application
            { "application/x-cbz", ".cbz" },
            { "application/x-javascript", ".js" },
            { "application/xml", ".xml" },
            { "application/x-mpegURL", ".m3u8" },

            // Type audio
            { "audio/aac", ".aac" },
            { "audio/ac3", ".ac3" },
            { "audio/dsf", ".dsf" },
            { "audio/dsp", ".dsp" },
            { "audio/flac", ".flac" },
            { "audio/m4b", ".m4b" },
            { "audio/vorbis", ".vorbis" },
            { "audio/x-ape", ".ape" },
            { "audio/xsp", ".xsp" },
            { "audio/x-wavpack", ".wv" },

            // Type image
            { "image/jpg", ".jpg" },
            { "image/x-png", ".png" },

            // Type text
            { "text/plain", ".txt" },
            { "text/rtf", ".rtf" },
            { "text/x-ssa", ".ssa" },

            // Type video
            { "video/vnd.mpeg.dash.mpd", ".mpd" },
            { "video/x-matroska", ".mkv" },
        };

        public static string GetMimeType(string path) => GetMimeType(path, "application/octet-stream");

        /// <summary>
        /// Gets the type of the MIME.
        /// </summary>
        /// <param name="filename">The filename to find the MIME type of.</param>
        /// <param name="defaultValue">The default value to return if no fitting MIME type is found.</param>
        /// <returns>The correct MIME type for the given filename, or <paramref name="defaultValue"/> if it wasn't found.</returns>
        [return: NotNullIfNotNullAttribute("defaultValue")]
        public static string? GetMimeType(string filename, string? defaultValue = null)
        {
            if (filename.Length == 0)
            {
                throw new ArgumentException("String can't be empty.", nameof(filename));
            }

            var ext = Path.GetExtension(filename);

            if (_mimeTypeLookup.TryGetValue(ext, out string? result))
            {
                return result;
            }

            if (Model.MimeTypes.TryGetMimeType(filename, out var mimeType))
            {
                return mimeType;
            }

            // Catch-all for all video types that don't require specific mime types
            if (_videoFileExtensions.Contains(ext))
            {
                return string.Concat("video/", ext.AsSpan(1));
            }

            return defaultValue;
        }

        public static string? ToExtension(string mimeType)
        {
            if (mimeType.Length == 0)
            {
                throw new ArgumentException("String can't be empty.", nameof(mimeType));
            }

            // handle text/html; charset=UTF-8
            mimeType = mimeType.AsSpan().LeftPart(';').ToString();

            if (_extensionLookup.TryGetValue(mimeType, out string? result))
            {
                return result;
            }

            var extensions = Model.MimeTypes.GetMimeTypeExtensions(mimeType);
            if (extensions.Any())
            {
                return "." + extensions.First();
            }

            return null;
        }
    }
}