S2sGraph.cs

using System.Text;
using System.Threading.Tasks;
using SkiaSharp;
using Waher.Script;
using Waher.Networking.XMPP.Server.Statistics;
using System.Text.RegularExpressions;
using Waher.IoTGateway;
 
namespace Waher.Service.IoTBroker.Admin.Topology
{
    public class S2sGraph : CommandRegEx
    {
        public S2sGraph()
            : base(@"^s2sgraph\s*.*$")
        {
        }
 
        public override string Name => "S2SGraph";
 
        public override async Task Execute(ChatState State, string[] Arguments, string OrgMessage, Match Details,
            ResponseCallbackHandler ResponseCallback)
        {
            string Script = OrgMessage.Substring(8).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 = TopologyQuery.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");
 
            TopologyQuery.EmbedResult(sb, Title, Result, Color);
 
            foreach (EndpointStatistics Stat in XmppServerModule.GetServerConnectionStatistics())
            {
                sb.Append('"');
                sb.Append(Title);
                sb.Append("\" -> \"");
                sb.Append(Stat.Name);
                sb.Append('"');
 
                if (Stat.Type != "XMPP")
                {
                    sb.Append(" [label=\"");
                    sb.Append(Stat.Type);
                    sb.Append("\"]");
                }
 
                sb.AppendLine(";");
            }
 
            sb.AppendLine("}");
            sb.AppendLine("```");
 
            await ResponseCallback(sb.ToString(), string.Empty);
        }
 
        public override HelpItem[] GetHelp()
        {
            return new HelpItem[]
            {
                new HelpItem("S2SGraph[ SCRIPT]", "Returns a GraphViz graph of the current broker and how it is connected to other federated brokers. Such a graph illustrates the federated network. Optional script can be provided, and its result embedded in the graph. Consolidating responses from multiple brokers generates a S2S graph of all brokers.")
            };
        }
    }
}