ServiceRegistrationComponent.cs

using System;
using System.Collections.Generic;
using System.Xml;
using System.Threading.Tasks;
using Waher.Content.Xml;
using Waher.Events;
using Waher.Networking.XMPP.Server;
using Waher.Persistence;
using Waher.Persistence.Filters;
 
namespace Waher.Service.IoTBroker.ServiceRegistrations
{
    public class ServiceRegistrationComponent : Component
    {
        public const string NamespaceServiceRegistration = "http://waher.se/Schema/ServiceRegistration.xsd";
 
        public ServiceRegistrationComponent(XmppServer Server, CaseInsensitiveString Subdomain, string Name)
            : base(Server, Subdomain, Name)
        {
            this.RegisterMessageHandler("register", NamespaceServiceRegistration, this.RegistrationMessageHandler, true);
            this.RegisterIqSetHandler("register", NamespaceServiceRegistration, this.RegistrationIqSetHandler, false);
        }
 
        public override void Dispose()
        {
            this.UnregisterMessageHandler("register", NamespaceServiceRegistration, this.RegistrationMessageHandler, true);
        }
 
        public override bool SupportsAccounts => false;
 
        private Task RegistrationMessageHandler(object Sender, MessageEventArgs e)
        {
            return this.RegistrationHandler(e.Content, e.From);
        }
 
        private async Task RegistrationIqSetHandler(object Sender, IqEventArgs e)
        {
            try
            {
                await this.RegistrationHandler(e.Query, e.From);
                await e.IqResult(string.Empty, e.To);
            }
            catch (Exception ex)
            {
                await e.IqError(ex, e.To);
            }
        }
 
        private async Task RegistrationHandler(XmlElement E, XmppAddress From)
        {
            CaseInsensitiveString BareJid = XML.Attribute(E, "jid").ToLower();
            if (string.IsNullOrEmpty(BareJid) || BareJid != From.BareJid)
            {
                Log.Error("Invalid registration received:\r\r\n" + E.OuterXml, BareJid, From.BareJid);
                return;
            }
 
            ServiceRegistration Registration = await Database.FindFirstDeleteRest<ServiceRegistration>(new FilterFieldEqualTo("BareJid", BareJid));
            bool Created = Registration is null;
 
            if (Created)
            {
                Registration = new ServiceRegistration()
                {
                    BareJid = BareJid
                };
            }
 
            Registration.ClientName = XML.Attribute(E, "clientName");
            Registration.ClientVersion = XML.Attribute(E, "clientVersion");
            Registration.ClientOS = XML.Attribute(E, "clientOS");
            Registration.Language = XML.Attribute(E, "language");
            Registration.Host = XML.Attribute(E, "host");
            Registration.Domain = XML.Attribute(E, "domain");
            Registration.Created = XML.Attribute(E, "created", DateTime.MinValue);
            Registration.Updated = XML.Attribute(E, "updated", DateTime.MinValue);
 
            List<string> Features = new List<string>();
            List<string> Assemblies = new List<string>();
            List<Annotation> Annotations = new List<Annotation>();
 
            foreach (XmlNode N in E.ChildNodes)
            {
                switch (N.LocalName)
                {
                    case "Feature":
                        Features.Add(N.InnerText);
                        break;
 
                    case "Assembly":
                        Assemblies.Add(N.InnerText);
                        break;
 
                    case "Annotation":
                        XmlElement E2 = (XmlElement)N;
                        string Tag = XML.Attribute(E2, "tag");
                        string Value = N.InnerText;
 
                        Annotations.Add(new Annotation(Tag, Value));
                        break;
                }
            }
 
            Registration.Features = Features.ToArray();
            Registration.Assemblies = Assemblies.ToArray();
            Registration.Annotations = Annotations.ToArray();
 
            if (Created)
                await Database.InsertLazy(Registration);
            else
                await Database.UpdateLazy(Registration);
        }
 
        public static ServiceRegistration GetRegistration(CaseInsensitiveString BareJid)
        {
            return GetRegistrationAsync(BareJid).Result;
        }
 
        public static async Task<ServiceRegistration> GetRegistrationAsync(CaseInsensitiveString BareJid)
        {
            return await Database.FindFirstDeleteRest<ServiceRegistration>(new FilterFieldEqualTo("BareJid", BareJid));
        }
    }
}