ContractGetter.cs

using System;
using System.Collections.Generic;
using System.Collections.Specialized;
using System.Runtime.ExceptionServices;
using System.Security.Cryptography.X509Certificates;
using System.Text;
using System.Threading.Tasks;
using System.Web;
using Waher.Content;
using Waher.IoTGateway;
using Waher.Networking.XMPP.Contracts;
using Waher.Networking.XMPP.Contracts.EventArguments;
using Waher.Runtime.Inventory;
using Waher.Runtime.Temporary;
 
namespace Waher.Service.IoTBroker.Getters
{
    public class ContractGetter : IContentGetter, IDisposable
    {
        private static string defaultPurpose = string.Empty;
        private static int defaultTimeoutMilliseconds = 60000;
 
        private readonly Dictionary<string, TaskCompletionSource<ContractPetitionResponseEventArgs>> contractPetitions = new Dictionary<string, TaskCompletionSource<ContractPetitionResponseEventArgs>>();
        private readonly object syncObj = new object();
        private bool handlerAdded = false;
 
        public ContractGetter()
        {
        }
 
        public static string DefaultPurpose
        {
            get => defaultPurpose;
            set
            {
                if (string.IsNullOrEmpty(value))
                    throw new ArgumentException("Default purpose cannot be empty.", nameof(DefaultPurpose));
 
                defaultPurpose = value;
            }
        }
 
        public static int DefaultTimeoutMilliseconds
        {
            get => defaultTimeoutMilliseconds;
            set
            {
                if (value <= 0)
                    throw new ArgumentException("Default timeout must be positivt.", nameof(DefaultTimeoutMilliseconds));
 
                defaultTimeoutMilliseconds = value;
            }
        }
 
        public string[] UriSchemes => new string[] { "iotsc" };
 
        public bool CanGet(Uri Uri, out Grade Grade)
        {
            if (string.Compare(Uri.Scheme, "iotsc", true) == 0)
            {
                Grade = Grade.Ok;
                return true;
            }
            else
            {
                Grade = Grade.NotAtAll;
                return false;
            }
        }
 
        public void Dispose()
        {
            lock (this.syncObj)
            {
                if (this.handlerAdded)
                {
                    Gateway.ContractsClient.PetitionedContractResponseReceived -= this.Client_PetitionedContractResponseReceived;
                    this.handlerAdded = false;
                }
            }
        }
 
        public Task<object> GetAsync(Uri Uri, X509Certificate Certificate,
            RemoteCertificateEventHandler RemoteCertificateValidator, params KeyValuePair<string, string>[] Headers)
        {
            return this.GetAsync(Uri, Certificate, RemoteCertificateValidator, defaultTimeoutMilliseconds, Headers);
        }
 
        public async Task<object> GetAsync(Uri Uri, X509Certificate Certificate,
            RemoteCertificateEventHandler RemoteCertificateValidator, int TimeoutMs, params KeyValuePair<string, string>[] Headers)
        {
            ContractsClient Client = Gateway.ContractsClient
                ?? throw new NotSupportedException("Contracts feature not supported or activated on this system.");
 
            lock (this.syncObj)
            {
                if (!this.handlerAdded)
                {
                    Client.PetitionedContractResponseReceived += this.Client_PetitionedContractResponseReceived;
                    this.handlerAdded = true;
                }
            }
 
            string ContractId = Uri.AbsolutePath;
 
            try
            {
                return await Client.GetContractAsync(ContractId);
            }
            catch (Exception)
            {
                TaskCompletionSource<ContractPetitionResponseEventArgs> Result = new TaskCompletionSource<ContractPetitionResponseEventArgs>();
                string PetitionId = Guid.NewGuid().ToString();
                string Purpose = null;
 
                if (!string.IsNullOrEmpty(Uri.Query))
                {
                    NameValueCollection Parameters = HttpUtility.ParseQueryString(Uri.Query);
 
                    foreach (string Parameter in Parameters.AllKeys)
                    {
                        switch (Parameter.ToLower())
                        {
                            case "p":
                            case "purpose":
                                Purpose = Parameters[Parameter];
                                break;
                        }
                    }
                }
 
                if (string.IsNullOrEmpty(Purpose))
                    Purpose = "Processing referenced contract.";
 
                lock (this.syncObj)
                {
                    this.contractPetitions[PetitionId] = Result;
                }
 
                try
                {
                    await Client.PetitionContractAsync(ContractId, PetitionId, Purpose);
 
                    Task _ = Task.Delay(TimeoutMs).ContinueWith((T) =>
                    {
                        Result.TrySetException(new TimeoutException("Response to petition not received within allotted time."));
                        return Task.CompletedTask;
                    });
                }
                catch (Exception ex)
                {
                    lock (this.syncObj)
                    {
                        this.contractPetitions.Remove(PetitionId);
                    }
 
                    ExceptionDispatchInfo.Capture(ex).Throw();
                }
 
                ContractPetitionResponseEventArgs e = await Result.Task;
 
                return e.RequestedContract;
            }
        }
 
        private Task Client_PetitionedContractResponseReceived(object Sender, ContractPetitionResponseEventArgs e)
        {
            lock (this.syncObj)
            {
                if (this.contractPetitions.TryGetValue(e.PetitionId, out TaskCompletionSource<ContractPetitionResponseEventArgs> Result))
                    Result.TrySetResult(e);
            }
 
            return Task.CompletedTask;
        }
 
        public Task<KeyValuePair<string, TemporaryStream>> GetTempStreamAsync(Uri Uri, X509Certificate Certificate,
            RemoteCertificateEventHandler RemoteCertificateValidator, params KeyValuePair<string, string>[] Headers)
        {
            return this.GetTempStreamAsync(Uri, Certificate, RemoteCertificateValidator, defaultTimeoutMilliseconds, Headers);
        }
 
        public async Task<KeyValuePair<string, TemporaryStream>> GetTempStreamAsync(Uri Uri, X509Certificate Certificate,
            RemoteCertificateEventHandler RemoteCertificateValidator, int TimeoutMs, params KeyValuePair<string, string>[] Headers)
        {
            Contract Contract = (Contract)await this.GetAsync(Uri, Certificate, RemoteCertificateValidator, TimeoutMs, Headers);
            StringBuilder Xml = new StringBuilder();
            Contract.Serialize(Xml, true, true, true, true, true, true, true);
            byte[] Bin = Encoding.UTF8.GetBytes(Xml.ToString());
 
            TemporaryStream Result = new TemporaryStream();
            try
            {
                await Result.WriteAsync(Bin, 0, Bin.Length);
            }
            catch (Exception ex)
            {
                Result.Dispose();
                ExceptionDispatchInfo.Capture(ex).Throw();
                Result = null;
            }
 
            return new KeyValuePair<string, TemporaryStream>("application/xml; charset=utf-8", Result);
        }
    }
}