FuseTry-Archive/Mtp2Dav/WebDAVSharp.Server._1d2086a502937936ebc6bfe19cfa15d855be1c31/WebDAVExtensions.cs

using System;
using System.Linq;
using System.Net;
using System.Web;
using Mtp2Dav.WebDAVSharp.Server._1d2086a502937936ebc6bfe19cfa15d855be1c31.Adapters;
using Mtp2Dav.WebDAVSharp.Server._1d2086a502937936ebc6bfe19cfa15d855be1c31.Exceptions;
using Mtp2Dav.WebDAVSharp.Server._1d2086a502937936ebc6bfe19cfa15d855be1c31.Stores;

namespace Mtp2Dav.WebDAVSharp.Server._1d2086a502937936ebc6bfe19cfa15d855be1c31
{
    /// <summary>
    ///     This class holds extension methods for various types related to WebDAV#.
    /// </summary>
    internal static class WebDavExtensions
    {
        /// <summary>
        ///     Gets the Uri to the parent object.
        /// </summary>
        /// <param name="uri">The <see cref="Uri" /> of a resource, for which the parent Uri should be retrieved.</param>
        /// <returns>
        ///     The parent <see cref="Uri" />.
        /// </returns>
        /// <exception cref="System.ArgumentNullException">uri</exception>
        /// <exception cref="System.InvalidOperationException">Cannot get parent of root</exception>
        /// <exception cref="ArgumentNullException"><paramref name="uri" /> is <c>null</c>.</exception>
        /// <exception cref="InvalidOperationException"><paramref name="uri" /> has no parent, it refers to a root resource.</exception>
        public static Uri GetParentUri(this Uri uri)
        {
            if (uri == null)
                throw new ArgumentNullException(nameof(uri));
            if (uri.Segments.Length == 1)
                throw new InvalidOperationException("Cannot get parent of root");

            var url = uri.ToString();
            var index = url.Length - 1;
            if (url[index] == '/')
                index--;
            while (url[index] != '/')
                index--;
            return new Uri(url.Substring(0, index + 1));
        }

        /// <summary>
        ///     Sends a simple response with a specified HTTP status code but no content.
        /// </summary>
        /// <param name="context">The <see cref="IHttpListenerContext" /> to send the response through.</param>
        /// <param name="statusCode">The HTTP status code for the response.</param>
        /// <exception cref="System.ArgumentNullException">context</exception>
        /// <exception cref="ArgumentNullException"><paramref name="context" /> is <c>null</c>.</exception>
        public static void SendSimpleResponse(this IHttpListenerContext context,
            HttpStatusCode statusCode = HttpStatusCode.OK)
        {
            if (context == null)
                throw new ArgumentNullException(nameof(context));

            context.Response.StatusCode = (int) statusCode;
            context.Response.StatusDescription = HttpWorkerRequest.GetStatusDescription((int) statusCode);
            context.Response.Close();
        }

        /// <summary>
        ///     Gets the prefix <see cref="Uri" /> that matches the specified <see cref="Uri" />.
        /// </summary>
        /// <param name="uri">The <see cref="Uri" /> to find the most specific prefix <see cref="Uri" /> for.</param>
        /// <param name="server">
        ///     The
        ///     <see cref="WebDavServer" /> that hosts the WebDAV server and holds the collection
        ///     of known prefixes.
        /// </param>
        /// <returns>
        ///     The most specific <see cref="Uri" /> for the given <paramref name="uri" />.
        /// </returns>
        /// <exception cref="WebDavInternalServerException">Unable to find correct server root</exception>
        /// <exception cref="WebDavInternalServerException">
        ///     <paramref name="uri" /> specifies a <see cref="Uri" /> that is not
        ///     known to the <paramref name="server" />.
        /// </exception>
        public static Uri GetPrefixUri(this Uri uri, WebDavServer server)
        {
            var url = uri.ToString();
            foreach (
                var prefix in
                    server.Listener.Prefixes.Where(
                        prefix => url.StartsWith(uri.ToString(), StringComparison.OrdinalIgnoreCase)))
                return new Uri(prefix);
            throw new WebDavInternalServerException("Unable to find correct server root");
        }

        /// <summary>
        ///     Retrieves a store item through the specified
        ///     <see cref="Uri" /> from the
        ///     specified
        ///     <see cref="WebDavServer" /> and
        ///     <see cref="IWebDavStore" />.
        /// </summary>
        /// <param name="uri">The <see cref="Uri" /> to retrieve the store item for.</param>
        /// <param name="server">The <see cref="WebDavServer" /> that hosts the <paramref name="store" />.</param>
        /// <param name="store">The <see cref="IWebDavStore" /> from which to retrieve the store item.</param>
        /// <returns>
        ///     The retrieved store item.
        /// </returns>
        /// <exception cref="System.ArgumentNullException">
        ///     <para>
        ///         <paramref name="uri" /> is <c>null</c>.
        ///     </para>
        ///     <para>
        ///         <paramref name="server" /> is <c>null</c>.
        ///     </para>
        ///     <para>
        ///         <paramref name="store" /> is <c>null</c>.
        ///     </para>
        /// </exception>
        /// <exception cref="WebDavNotFoundException">If the item was not found.</exception>
        /// <exception cref="WebDavConflictException">
        ///     <paramref name="uri" /> refers to a document in a collection, where the
        ///     collection does not exist.
        /// </exception>
        /// <exception cref="WebDavNotFoundException"><paramref name="uri" /> refers to a document that does not exist.</exception>
        public static IWebDavStoreItem GetItem(this Uri uri, WebDavServer server, IWebDavStore store)
        {
            if (uri == null)
                throw new ArgumentNullException(nameof(uri));
            if (server == null)
                throw new ArgumentNullException(nameof(server));
            if (store == null)
                throw new ArgumentNullException(nameof(store));

            var prefixUri = uri.GetPrefixUri(server);
            var collection = store.Root;

            IWebDavStoreItem item = null;
            if (prefixUri.Segments.Length == uri.Segments.Length)
                return collection;

            for (var index = prefixUri.Segments.Length; index < uri.Segments.Length; index++)
            {
                var segmentName = Uri.UnescapeDataString(uri.Segments[index]);
                var nextItem = collection.GetItemByName(segmentName.TrimEnd('/', '\\'));
                if (nextItem == null)
                    throw new WebDavNotFoundException(uri.ToString()); //throw new WebDavConflictException();

                if (index == uri.Segments.Length - 1)
                    item = nextItem;
                else
                {
                    collection = nextItem as IWebDavStoreCollection;
                    if (collection == null)
                        throw new WebDavNotFoundException();
                }
            }

            if (item == null)
                throw new WebDavNotFoundException();

            return item;
        }
    }
}