TopologyQuery.cs

using System;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using SkiaSharp;
using Waher.Content.Xml;
using Waher.IoTGateway;
using Waher.Script;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Abstraction.Sets;
using Waher.Script.Objects.Matrices;
using Waher.Script.Operators.Matrices;
 
namespace Waher.Service.IoTBroker.Admin.Topology
{
    public class TopologyQuery : CommandRegEx
    {
        public TopologyQuery()
            : base(@"^topologyquery\s*.*$")
        {
        }
 
        public override string Name => "TopologyQuery";
 
        public override async Task Execute(ChatState State, string[] Arguments, string OrgMessage, Match Details,
            ResponseCallbackHandler ResponseCallback)
        {
            string Script = OrgMessage.Substring(13).Trim();
            object Result;
            SKColor Color;
 
            if (string.IsNullOrEmpty(Script))
            {
                Result = null;
                Color = SKColors.Empty;
            }
            else
            {
                Expression Exp = new Expression(Script);
                Result = await Exp.EvaluateAsync(State.Session);
                Result = Simplify(Result, out Color);
            }
 
            StringBuilder sb = new StringBuilder();
            string Title = Gateway.HasDomain ? Gateway.Domain.Value : Gateway.XmppClient.BareJID;
 
            sb.AppendLine("```dot");
            sb.AppendLine("digraph G {");
            sb.AppendLine("rankdir=LR");
 
            EmbedResult(sb, Title, Result, Color);
 
            sb.Append('"');
            sb.Append(IoTGateway.Setup.XmppConfiguration.Instance.Host);
            sb.Append("\" -> \"");
            sb.Append(Title);
            sb.AppendLine("\" [dir=back];");
            sb.AppendLine("}");
            sb.AppendLine("```");
 
            await ResponseCallback(sb.ToString(), string.Empty);
        }
 
        public static object Simplify(object Result, out SKColor Color)
        {
            int i, j, c;
            
            Color = SKColors.Empty;
 
            while (true)
            {
                if (Result is IToMatrix ToMatrix)
                    Result = ToMatrix.ToMatrix();
 
                if (Result is IMatrix M2)
                {
                    if (M2 is ObjectMatrix OM2 && OM2.HasColumnNames)
                        return Result;
 
                    if (M2.Columns == 1)
                        Result = M2.GetColumn(0);
                    else if (M2.Rows == 1)
                        Result = M2.GetRow(0);
                    else
                        return Result;
                }
                else if (Result is Array V2)
                {
                    if ((c = V2.Length) == 1)
                        Result = V2.GetValue(0);
                    else
                    {
                        for (i = 0; i < c; i++)
                        {
                            if (V2.GetValue(i) is SKColor Color2)
                            {
                                Color = Color2;
 
                                j = 0;
                                object[] Elements2 = new object[c - 1];
 
                                foreach (object Element in V2)
                                {
                                    if (j < i)
                                        Elements2[j] = Element;
                                    else if (j > i)
                                        Elements2[j - 1] = Element;
 
                                    j++;
                                }
 
                                Result = Elements2;
                                break;
                            }
                        }
 
                        if (i == c)
                            return Result;
                    }
                }
                else if (Result is SKColor Color2)
                {
                    Color = Color2;
                    return null;
                }
                else
                    return Result;
            }
        }
 
        public static void EmbedResult(StringBuilder sb, string Title, object Result, SKColor Color)
        {
            bool First = true;
            int i, j, c, d;
            string s;
 
            sb.Append('"');
            sb.Append(XML.Encode(Title));
            sb.Append("\" [");
 
            if (Color != SKColors.Empty)
            {
                sb.Append("style=filled, fillcolor=\"#");
                sb.Append(Color.Red.ToString("X2"));
                sb.Append(Color.Green.ToString("X2"));
                sb.Append(Color.Blue.ToString("X2"));
 
                if (Color.Alpha != 255)
                    sb.Append(Color.Alpha.ToString("X2"));
 
                sb.Append('"');
                First = false;
            }
 
            if (Result is IToMatrix ToMatrix)
                Result = ToMatrix.ToMatrix();
 
            if (Result is IMatrix || Result is Array || Result is ISet)
            {
                if (!First)
                    sb.Append(", ");
                else
                    First = false;
 
                sb.Append("shape=plaintext, margin=0");
            }
 
            if (!First)
                sb.Append(", ");
 
            sb.Append("label=<");
 
            if (Result is IMatrix M)
            {
                c = M.Rows;
                d = M.Columns;
 
                sb.Append("<table border=\"1\" cellborder=\"0\" cellspacing=\"1\">");
                sb.Append("<tr><td align=\"center\" colspan=\"");
                sb.Append(d.ToString());
                sb.Append("\"><b>");
                sb.Append(Title);
                sb.Append("</b></td></tr>");
 
                string[] Alignment = new string[d];
 
                for (i = 0; i < c; i++)
                {
                    for (j = 0; j < d; j++)
                    {
                        IElement E = M.GetElement(j, i);
                        object Obj = E.AssociatedObjectValue;
 
                        if (!(Obj is null))
                        {
                            if (Obj is double || Obj is float || Obj is int || Obj is long)
                                s = "right";
                            else if (Obj is bool)
                                s = "center";
                            else if (Obj is string || Obj is Enum)
                                s = "left";
                            else
                                s = string.Empty;
 
                            if (Alignment[j] is null)
                                Alignment[j] = s;
                            else if (Alignment[j] != s)
                                Alignment[j] = string.Empty;
                        }
                    }
                }
 
                if (Result is ObjectMatrix OM && OM.HasColumnNames)
                {
                    sb.Append("<tr>");
 
                    j = 0;
                    foreach (string Name in OM.ColumnNames)
                    {
                        sb.Append("<td align=\"");
                        sb.Append(string.IsNullOrEmpty(s = Alignment[j++]) ? "center" : s);
                        sb.Append("\"><b>");
                        sb.Append(string.IsNullOrEmpty(Name) ? " " : XML.Encode(Name));
                        sb.Append("</b></td>");
                    }
 
                    sb.Append("</tr>");
                }
 
                for (i = 0; i < c; i++)
                {
                    sb.Append("<tr>");
 
                    for (j = 0; j < d; j++)
                    {
                        IElement E = M.GetElement(j, i);
                        object Obj = E.AssociatedObjectValue;
 
                        s = Alignment[j];
                        sb.Append("<td");
 
                        if (!string.IsNullOrEmpty(s))
                        {
                            sb.Append(" align=\"");
                            sb.Append(s);
                            sb.Append('"');
                        }
 
                        sb.Append('>');
                        sb.Append(Obj?.ToString() ?? string.Empty);
                        sb.Append("</td>");
                    }
 
                    sb.Append("</tr>");
                }
 
                sb.Append("</table>");
            }
            else if (Result is Array V)
            {
                d = V.Length;
 
                sb.Append("<table border=\"1\" cellborder=\"1\" cellspacing=\"1\">");
                sb.Append("<tr><td align=\"center\" colspan=\"");
                sb.Append(d.ToString());
                sb.Append("\"><b>");
                sb.Append(Title);
                sb.Append("</b></td></tr><tr>");
 
                foreach (object Obj in V)
                {
                    sb.Append("<td>");
                    sb.Append(Obj?.ToString() ?? string.Empty);
                    sb.Append("</td>");
                }
 
                sb.Append("</tr></table>");
            }
            else if (Result is ISet S)
            {
                StringBuilder sb2 = new StringBuilder();
                d = 0;
 
                foreach (IElement E in S.ChildElements)
                {
                    object Obj = E.AssociatedObjectValue;
 
                    sb2.Append("<td>");
                    sb2.Append(Obj?.ToString() ?? string.Empty);
                    sb2.Append("</td>");
 
                    d++;
                }
 
                sb.Append("<table border=\"1\" cellborder=\"1\" cellspacing=\"1\">");
                sb.Append("<tr><td align=\"center\" colspan=\"");
                sb.Append(d.ToString());
                sb.Append("\"><b>");
                sb.Append(Title);
                sb.Append("</b></td></tr><tr>");
                sb.Append(sb2.ToString());
                sb.Append("</tr></table>");
            }
            else
            {
                sb.Append(XML.Encode(Title));
 
                if (!(Result is null))
                {
                    sb.Append("<br/><FONT POINT-SIZE=\"5\"><br/></FONT><FONT POINT-SIZE=\"20\"><b>");
                    sb.Append(Result?.ToString() ?? " ");
                    sb.Append("</b></FONT>");
                }
            }
 
            sb.AppendLine(">];");
        }
 
        public override HelpItem[] GetHelp()
        {
            return new HelpItem[]
            {
                new HelpItem("TopologyQuery[ SCRIPT]", "Executes script (if provided), and returns the result in the form of a GraphViz graph of the current broker connected to its parent broker. Consolidating responses from multiple brokers generates a topology graph of all brokers, where each node contains the result of executing the script.")
            };
        }
    }
}