// (c) Copyright Ascensio System SIA 2010-2022
//
// This program is a free software product.
// You can redistribute it and/or modify it under the terms
// of the GNU Affero General Public License (AGPL) version 3 as published by the Free Software
// Foundation. In accordance with Section 7(a) of the GNU AGPL its Section 15 shall be amended
// to the effect that Ascensio System SIA expressly excludes the warranty of non-infringement of
// any third-party rights.
//
// This program is distributed WITHOUT ANY WARRANTY, without even the implied warranty
// of MERCHANTABILITY or FITNESS FOR A PARTICULAR  PURPOSE. For details, see
// the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
//
// You can contact Ascensio System SIA at Lubanas st. 125a-25, Riga, Latvia, EU, LV-1021.
//
// The  interactive user interfaces in modified source and object code versions of the Program must
// display Appropriate Legal Notices, as required under Section 5 of the GNU AGPL version 3.
//
// Pursuant to Section 7(b) of the License you must retain the original Product logo when
// distributing the program. Pursuant to Section 7(e) we decline to grant you any rights under
// trademark law for use of our trademarks.
//
// All the Product's GUI elements, including illustrations and icon sets, as well as technical writing
// content are licensed under the terms of the Creative Commons Attribution-ShareAlike 4.0
// International. See the License terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode

using HttpContext = Microsoft.AspNetCore.Http.HttpContext;

namespace System.Web;

public static class HttpRequestExtensions
{
    public static readonly string UrlRewriterHeader = "X-REWRITER-URL";

    public static Uri GetUrlRewriter(this HttpRequest request)
    {
        return request != null ? GetUrlRewriter(request.Headers, request) : null;
    }

    public static Uri Url(this HttpRequest request)
    {
        return request != null ? new Uri(request.GetDisplayUrl()) : null;
    }

    /*public static Uri GetUrlRewriter(this HttpRequestBase request)
    {
        return request != null ? GetUrlRewriter(request.Headers, request.Url) : null;
    }*/

    public static Uri GetUrlRewriter(IHeaderDictionary headers, HttpRequest request)
    {
        if (headers != null)
        {
            var h = headers[UrlRewriterHeader];
            var rewriterUri = !string.IsNullOrEmpty(h) ? ParseRewriterUrl(h) : null;

            if (request != null && rewriterUri != null)
            {
                var result = new UriBuilder()
                {
                    Scheme = rewriterUri.Scheme,
                    Host = rewriterUri.Host,
                    Port = rewriterUri.Port
                };

                result.Query = request.QueryString.Value;
                result.Path = request.Path.Value;

                return result.Uri;
            }
        }

        if (request != null && request.Query != null)
        {
            var h = request.Query[UrlRewriterHeader];
            var rewriterUri = !string.IsNullOrEmpty(h) ? ParseRewriterUrl(h) : null;
            if (rewriterUri != null)
            {
                var result = new UriBuilder()
                {
                    Scheme = rewriterUri.Scheme,
                    Host = rewriterUri.Host,
                    Port = rewriterUri.Port
                };
                result.Query = request.QueryString.Value;
                result.Path = request.Path.Value;

                return result.Uri;
            }
        }

        return request.Url();
    }

    public static Uri PushRewritenUri(this HttpContext context)
    {
        return context != null ? PushRewritenUri(context, GetUrlRewriter(context.Request)) : null;
    }

    public static Uri PushRewritenUri(this HttpContext context, Uri rewrittenUri)
    {
        Uri oldUri = null;

        if (context != null)
        {
            var request = context.Request;

            var url = new Uri(request.GetDisplayUrl());

            if (url != rewrittenUri)
            {
                var requestUri = url;
                try
                {
                    //Push it
                    request.Headers.SetCommaSeparatedValues("HTTPS", rewrittenUri.Scheme.Equals("https", StringComparison.OrdinalIgnoreCase) ? "on" : "off");
                    request.Headers.SetCommaSeparatedValues("SERVER_NAME", rewrittenUri.Host);
                    request.Headers.SetCommaSeparatedValues("SERVER_PORT",
                                                rewrittenUri.Port.ToString(CultureInfo.InvariantCulture));

                    if (rewrittenUri.IsDefaultPort)
                    {
                        request.Headers.SetCommaSeparatedValues("HTTP_HOST",
                                                    rewrittenUri.Host);
                    }
                    else
                    {
                        request.Headers.SetCommaSeparatedValues("HTTP_HOST",
                                                    rewrittenUri.Host + ":" + requestUri.Port);
                    }
                    //Hack:
                    typeof(HttpRequest).InvokeMember("_url",
                                                      BindingFlags.NonPublic | BindingFlags.SetField |
                                                      BindingFlags.Instance,
                                                      null, request,
                                                      new object[] { null });
                    oldUri = requestUri;
                    context.Items["oldUri"] = oldUri;

                }
                catch (Exception) { }
            }
        }

        return oldUri;
    }

    public static Uri PopRewritenUri(this HttpContext context)
    {
        if (context != null && context.Items["oldUri"] != null)
        {
            var rewriteTo = context.Items["oldUri"] as Uri;

            if (rewriteTo != null)
            {
                return PushRewritenUri(context, rewriteTo);
            }
        }

        return null;
    }

    public static bool DesktopApp(this HttpRequest request)
    {
        return request != null
            && (!string.IsNullOrEmpty(request.Query["desktop"])
                || !string.IsNullOrEmpty(request.Headers[HeaderNames.UserAgent]) && request.Headers[HeaderNames.UserAgent].ToString().Contains("AscDesktopEditor"));
    }

    public static bool SailfishApp(this HttpRequest request)
    {
        return request != null
               && (!string.IsNullOrEmpty(request.Headers["sailfish"])
                   || !string.IsNullOrEmpty(request.Headers[HeaderNames.UserAgent]) && request.Headers[HeaderNames.UserAgent].ToString().Contains("SailfishOS"));
    }

    public static string GetUserHostAddress(this HttpRequest request)
    {
        return request.HttpContext.Features.Get<IHttpConnectionFeature>()?.RemoteIpAddress.ToString();
    }

    private static Uri ParseRewriterUrl(string s)
    {
        if (string.IsNullOrEmpty(s))
        {
            return null;
        }

        const StringComparison cmp = StringComparison.OrdinalIgnoreCase;
        if (0 < s.Length && s.StartsWith('0'))
        {
            s = Uri.UriSchemeHttp + s.Substring(1);
        }
        else if (3 < s.Length && s.StartsWith("OFF", cmp))
        {
            s = Uri.UriSchemeHttp + s.Substring(3);
        }
        else if (0 < s.Length && s.StartsWith('1'))
        {
            s = Uri.UriSchemeHttps + s.Substring(1);
        }
        else if (2 < s.Length && s.StartsWith("ON", cmp))
        {
            s = Uri.UriSchemeHttps + s.Substring(2);
        }
        else if (s.StartsWith(Uri.UriSchemeHttp + "%3A%2F%2F", cmp) || s.StartsWith(Uri.UriSchemeHttps + "%3A%2F%2F", cmp))
        {
            s = HttpUtility.UrlDecode(s);
        }

        Uri.TryCreate(s, UriKind.Absolute, out var result);

        return result;
    }
}