SetRosterItem.cs

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using System.Xml;
using Waher.Content;
using Waher.IoTGateway;
using Waher.Networking.HTTP;
using Waher.Networking.XMPP;
using Waher.Networking.XMPP.Server;
using Waher.Runtime.Inventory;
using Waher.Script;
using Waher.Script.Abstraction.Elements;
using Waher.Service.IoTBroker.DataStorage;
 
namespace Waher.Service.IoTBroker.WebServices.Agent.Xmpp
{
    public class SetRosterItem : AgentXmppPostResource
    {
        public SetRosterItem()
            : base("Xmpp/SetRosterItem",
                new KeyValuePair<Type, Expression>(typeof(Dictionary<string, object>), new Expression(jsonPattern)),
                new KeyValuePair<Type, Expression>(typeof(XmlDocument), new Expression(xmlPattern)))
        {
        }
 
        private static readonly string jsonPattern = Resources.LoadResourceAsText(typeof(SetRosterItem).Namespace + ".JSON.SetRosterItem.req");
        private static readonly string xmlPattern = Resources.LoadResourceAsText(typeof(SetRosterItem).Namespace + ".XML.SetRosterItem.req");
 
        public override async Task POST(HttpRequest Request, HttpResponse Response, Dictionary<string, IElement> Parameters)
        {
            AccountUser User = AssertUserAuthenticated(Request);
            string BareJid = (string)Parameters["PBareJid"].AssociatedObjectValue;
            string Name = (string)Parameters["PName"].AssociatedObjectValue;
            string[] Groups;
 
            if (Parameters.TryGetValue("PGroups", out IElement E) && E?.AssociatedObjectValue is object[] GroupObjs)
            {
                int i, c = GroupObjs.Length;
 
                Groups = new string[c];
 
                for (i = 0; i < c; i++)
                    Groups[i] = GroupObjs[i].ToString();
            }
            else
                Groups = new string[0];
 
            if (string.IsNullOrEmpty(BareJid))
                throw new BadRequestException("Empty Bare JID.");
 
            if (!IsValidBareJID(BareJid))
                throw new BadRequestException("Invalid Bare JID.");
 
            if (XmppServerModule.Server is null)
            {
                XmppClient Client = Gateway.XmppClient;
 
                if (Client is null)
                {
                    if (Types.TryGetModuleParameter("XMPP", out object Obj) && Obj is XmppClient Client2)
                        Client = Client2;
                }
 
                if (Client is null)
                    throw new ServiceUnavailableException("XMPP Client not available.");
 
                Networking.XMPP.RosterItem Item = Client[BareJid];
                if (Item is null)
                {
                    Item = new Networking.XMPP.RosterItem(BareJid, Name, Groups);
                    await Client.AddRosterItem(Item);
                }
                else
                {
                    await Client.UpdateRosterItem(BareJid, Name, Groups);
                    Item = Client[BareJid];
                }
 
                await Response.Return(new NamedDictionary<string, object>("RosterItem", AgentNamespace)
                {
                    { "bareJid", Item.BareJid },
                    { "pendingSubscription", Item.PendingSubscription == PendingSubscription.Subscribe },
                    { "status", Item.State },
                    { "name", Item.Name },
                    { "Groups", Item.Groups }
                });
            }
            else
            {
                PersistenceLayer PersistenceLayer = XmppServerModule.PersistenceLayer ?? new PersistenceLayer();
                IRosterItem Item = await PersistenceLayer.SetRosterItem(User.UserName, BareJid, Name, null, null, Groups)
                    ?? throw new ServiceUnavailableException("XMPP Server not available.");
 
                await Response.Return(new NamedDictionary<string, object>("RosterItem", AgentNamespace)
                {
                    { "bareJid", Item.BareJid.Value },
                    { "pendingSubscription", Item.PendingSubscription },
                    { "status", Item.Subscription },
                    { "name", Item.Name },
                    { "Groups", Item.Groups }
                });
            }
        }
    }
}