GetRoster.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 GetRoster : AgentXmppPostResource
    {
        public GetRoster()
            : base("Xmpp/GetRoster",
                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(GetRoster).Namespace + ".JSON.GetRoster.req");
        private static readonly string xmlPattern = Resources.LoadResourceAsText(typeof(GetRoster).Namespace + ".XML.GetRoster.req");
 
        public override async Task POST(HttpRequest Request, HttpResponse Response, Dictionary<string, IElement> Parameters)
        {
            AccountUser User = AssertUserAuthenticated(Request);
 
            int Offset = 0;
            int MaxCount = int.MaxValue;
 
            if (Parameters.TryGetValue("POffset", out IElement E) && !(E is null) &&
                E.AssociatedObjectValue is double d)
            {
                Offset = (int)d;
            }
 
            if (Parameters.TryGetValue("PMaxCount", out E) && !(E is null) &&
                E.AssociatedObjectValue is double d2)
            {
                MaxCount = (int)d2;
            }
 
            List<Dictionary<string, object>> Roster = new List<Dictionary<string, object>>();
 
            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.");
 
                foreach (Networking.XMPP.RosterItem Item in Client.Roster)
                {
                    if (Offset > 0)
                    {
                        Offset--;
                        continue;
                    }
 
                    Roster.Add(new Dictionary<string, object>()
                    {
                        { "bareJid", Item.BareJid },
                        { "pendingSubscription", Item.PendingSubscription == PendingSubscription.Subscribe },
                        { "status", Item.State },
                        { "name", Item.Name },
                        { "Groups", Item.Groups }
                    });
 
                    MaxCount--;
                    if (MaxCount <= 0)
                        break;
                }
            }
            else
            {
                PersistenceLayer PersistenceLayer = XmppServerModule.PersistenceLayer ?? new PersistenceLayer();
                IEnumerable<IRosterItem> Items = await PersistenceLayer.GetRoster(User.UserName);
 
                if (!(Items is null))
                {
                    foreach (IRosterItem Item in Items)
                    {
                        if (Offset > 0)
                        {
                            Offset--;
                            continue;
                        }
 
                        Roster.Add(new Dictionary<string, object>()
                        {
                            { "bareJid", Item.BareJid.Value },
                            { "pendingSubscription", Item.PendingSubscription },
                            { "status", Item.Subscription },
                            { "name", Item.Name },
                            { "Groups", Item.Groups }
                        });
 
                        MaxCount--;
                        if (MaxCount <= 0)
                            break;
                    }
                }
            }
 
            await Response.Return(new NamedDictionary<string, object>("Roster", AgentNamespace)
            {
                { "RosterItems", Roster.ToArray() }
            });
        }
    }
}