squidex/backend/src/Squidex.Domain.Apps.Core.Operations/Scripting/Extensions/HttpJintExtension.cs

// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using System;
using System.Net.Http;
using System.Threading.Tasks;
using Jint;
using Jint.Native;
using Jint.Native.Json;
using Jint.Runtime;
using Squidex.Infrastructure.Tasks;

namespace Squidex.Domain.Apps.Core.Scripting.Extensions
{
    public sealed class HttpJintExtension : IJintExtension
    {
        private delegate void GetJsonDelegate(string url, Action<JsValue> callback, JsValue? headers = null);
        private readonly IHttpClientFactory httpClientFactory;

        public HttpJintExtension(IHttpClientFactory httpClientFactory)
        {
            this.httpClientFactory = httpClientFactory;
        }

        public void ExtendAsync(ExecutionContext context)
        {
            var action = new GetJsonDelegate((url, callback, headers) => GetJson(context, url, callback, headers));

            context.Engine.SetValue("getJSON", action);
        }

        private void GetJson(ExecutionContext context, string url, Action<JsValue> callback, JsValue? headers)
        {
            GetJsonAsync(context, url, callback, headers).Forget();
        }

        private async Task GetJsonAsync(ExecutionContext context, string url, Action<JsValue> callback, JsValue? headers)
        {
            if (callback == null)
            {
                context.Fail(new JavaScriptException("Callback cannot be null."));
                return;
            }

            if (!Uri.IsWellFormedUriString(url, UriKind.Absolute))
            {
                context.Fail(new JavaScriptException("URL is not valid."));
                return;
            }

            context.MarkAsync();

            try
            {
                using (var httpClient = httpClientFactory.CreateClient())
                {
                    using (var request = CreateRequest(url, headers))
                    {
                        using (var response = await httpClient.SendAsync(request, context.CancellationToken))
                        {
                            response.EnsureSuccessStatusCode();

                            var responseObject = await ParseResponse(context, response);

                            context.Engine.ResetConstraints();

                            callback(responseObject);
                        }
                    }
                }
            }
            catch (Exception ex)
            {
                context.Fail(ex);
            }
        }

        private static HttpRequestMessage CreateRequest(string url, JsValue? headers)
        {
            if (!Uri.TryCreate(url, UriKind.Absolute, out var uri))
            {
                throw new ArgumentException("Url must be an absolute URL");
            }

            var request = new HttpRequestMessage(HttpMethod.Get, uri);

            if (headers != null && headers.Type == Types.Object)
            {
                var obj = headers.AsObject();

                foreach (var (key, property) in obj.GetOwnProperties())
                {
                    var value = TypeConverter.ToString(property.Value);

                    var keyString = key.AsString();

                    if (!string.IsNullOrWhiteSpace(keyString))
                    {
                        request.Headers.TryAddWithoutValidation(keyString, value ?? string.Empty);
                    }
                }
            }

            return request;
        }

        private static async Task<JsValue> ParseResponse(ExecutionContext context, HttpResponseMessage response)
        {
            var responseString = await response.Content.ReadAsStringAsync();

            context.CancellationToken.ThrowIfCancellationRequested();

            var jsonParser = new JsonParser(context.Engine);
            var jsonValue = jsonParser.Parse(responseString);

            context.CancellationToken.ThrowIfCancellationRequested();

            return jsonValue;
        }
    }
}