GroupSource.cs

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Waher.Events;
using Waher.Groups.NodeTypes;
using Waher.Persistence;
using Waher.Persistence.Filters;
using Waher.Runtime.Inventory;
using Waher.Runtime.Language;
using Waher.Runtime.Settings;
using Waher.Things;
using Waher.Things.SourceEvents;
 
namespace Waher.Groups
{
    public class GroupSource : IDataSource
    {
        public const string SourceID = "Groups";
 
        private static readonly Dictionary<string, IDataSource> sources = new Dictionary<string, IDataSource>();
        private static readonly Dictionary<string, GroupNode> nodes = new Dictionary<string, GroupNode>();
        private static Root root = null;
        private static GroupSource instance = null;
        private static IDataSources dataSources = null;
 
        private DateTime lastChanged;
 
        public GroupSource()
        {
            this.lastChanged = RuntimeSettings.Get(SourceID + ".LastChanged", DateTime.MinValue);
            instance ??= this;
        }
 
        string IDataSource.SourceID => SourceID;
 
        public IEnumerable<IDataSource> ChildSources => null;
 
        public bool HasChildren => false;
 
        public DateTime LastChanged
        {
            get => this.lastChanged;
            internal set
            {
                if (this.lastChanged != value)
                {
                    this.lastChanged = value;
                    Task _ = RuntimeSettings.SetAsync("GroupSource.LastChanged", value);
                }
            }
        }
 
        public Task<string> GetNameAsync(Language Language)
        {
            return Language.GetStringAsync(typeof(GroupSource), 1, "Groups");
        }
 
        public async Task<INode> GetNodeAsync(IThingReference NodeRef)
        {
            return await GetNode(NodeRef);
        }
 
        public static Task<GroupNode> GetNode(string NodeId)
        {
            return GetNode(new ThingReference(NodeId, SourceID));
        }
 
        public static async Task<GroupNode> GetNode(IThingReference NodeRef)
        {
            if (NodeRef is null || NodeRef.SourceId != SourceID || !string.IsNullOrEmpty(NodeRef.Partition))
                return null;
 
            lock (nodes)
            {
                if (nodes.TryGetValue(NodeRef.NodeId, out GroupNode Node))
                    return Node;
            }
 
            foreach (GroupNode Node2 in await Database.Find<GroupNode>(new FilterFieldEqualTo("NodeId", NodeRef.NodeId)))
            {
                lock (nodes)
                {
                    if (nodes.TryGetValue(NodeRef.NodeId, out GroupNode Node))
                        return Node;
                    else
                    {
                        nodes[NodeRef.NodeId] = Node2;
                        return Node2;
                    }
                }
            }
 
            return null;
        }
 
        internal static GroupNode RegisterNode(GroupNode Node)
        {
            if (Node.SourceId == SourceID && string.IsNullOrEmpty(Node.Partition))
            {
                lock (nodes)
                {
                    if (nodes.TryGetValue(Node.NodeId, out GroupNode Node2))
                        return Node2;
                    else
                    {
                        nodes[Node.NodeId] = Node;
                        return Node;
                    }
                }
            }
            else
                return Node;
        }
 
        internal static GroupNode RegisterNewNodeId(GroupNode Node, string OldId)
        {
            if (Node.SourceId != SourceID || !string.IsNullOrEmpty(Node.Partition))
                return Node;
 
            lock (nodes)
            {
                if (!nodes.TryGetValue(OldId, out GroupNode Node2) || Node2 != Node)
                    return Node;
 
                if (nodes.TryGetValue(Node.NodeId, out Node2))
                    return Node2;
 
                nodes.Remove(OldId);
                nodes[Node.NodeId] = Node;
 
                return Node;
            }
        }
 
        internal static void UnregisterNode(GroupNode Node)
        {
            if (Node.SourceId == SourceID && string.IsNullOrEmpty(Node.Partition))
            {
                lock (nodes)
                {
                    if (nodes.TryGetValue(Node.NodeId, out GroupNode Node2) && Node == Node2)
                        nodes.Remove(Node.NodeId);
                }
            }
        }
 
        public Task<bool> CanViewAsync(RequestOrigin Caller)
        {
            return Task.FromResult(Caller.HasPrivilege("Source." + SourceID + ".View"));
        }
 
        public IEnumerable<INode> RootNodes
        {
            get
            {
                if (root is null)
                    LoadRoot().Wait();
 
                return new INode[] { root };
            }
        }
 
        public static Root Root
        {
            get
            {
                if (root is null)
                    LoadRoot().Wait();
 
                return root;
            }
        }
 
        private static async Task LoadRoot()
        {
            Root Result = null;
 
            foreach (GroupNode Node in await Database.Find<GroupNode>(new FilterFieldEqualTo("ParentId", Guid.Empty)))
            {
                if (Node is Root Root)
                {
                    if (Result is null)
                        Result = Root;
                    else
                        await Database.Delete(Node);
                }
            }
 
            if (Result is null)
            {
                Result = new Root()
                {
                    NodeId = await (await Translator.GetDefaultLanguageAsync()).GetStringAsync(typeof(Root), 12, "Root")
                };
 
                await Database.Insert(Result);
 
                Language Language = await Translator.GetDefaultLanguageAsync();
                await NewEvent(new NodeAdded()
                {
                    Parameters = await Result.GetDisplayableParameterAraryAsync(Language, RequestOrigin.Empty),
                    NodeType = Result.GetType().FullName,
                    Sniffable = false,
                    DisplayName = await Result.GetTypeNameAsync(Language),
                    HasChildren = Result.HasChildren,
                    ChildrenOrdered = Result.ChildrenOrdered,
                    IsReadable = Result.IsReadable,
                    IsControllable = Result.IsControllable,
                    HasCommands = Result.HasCommands,
                    ParentId = string.Empty,
                    ParentPartition = string.Empty,
                    Updated = Result.Updated,
                    State = Result.State,
                    NodeId = Result.NodeId,
                    Partition = Result.Partition,
                    LogId = NodeAdded.EmptyIfSame(Result.LogId, Result.NodeId),
                    LocalId = NodeAdded.EmptyIfSame(Result.LocalId, Result.NodeId),
                    SourceId = Result.SourceId,
                    Timestamp = DateTime.UtcNow
                });
            }
 
            lock (nodes)
            {
                nodes[Result.NodeId] = Result;
            }
 
            root = Result;
        }
 
        public static async Task<int> DeleteOldEvents(TimeSpan MaxAge)
        {
            if (MaxAge <= TimeSpan.Zero)
                throw new ArgumentException("Age must be positive.", nameof(MaxAge));
 
            DateTime Limit = DateTime.Now.Subtract(MaxAge);
            int NrEvents = await Database.Delete<SourceEvent>(new FilterAnd(
                new FilterFieldEqualTo("SourceId", SourceID),
                new FilterFieldLesserOrEqualTo("Timestamp", Limit)));
 
            if (NrEvents > 0)
            {
                KeyValuePair<string, object>[] Tags = new KeyValuePair<string, object>[]
                {
                    new KeyValuePair<string, object>("Limit", Limit),
                    new KeyValuePair<string, object>("NrEvents", NrEvents)
                };
 
                if (NrEvents == 1)
                    Log.Informational("Deleting 1 logical group event from the database.", SourceID, Tags);
                else
                    Log.Informational("Deleting " + NrEvents.ToString() + " logical group events from the database.", SourceID, Tags);
            }
 
            return NrEvents;
        }
 
        public event EventHandlerAsync<SourceEvent> OnEvent = null;
 
        internal static async Task NewEvent(SourceEvent Event)
        {
            await Database.Insert(Event);
            await (instance?.OnEvent?.Raise(instance, Event) ?? Task.CompletedTask);
        }
 
        public static async Task<int> DeleteOrphans()
        {
            int Result = 0;
 
            foreach (GroupNode Node in await Database.Find<GroupNode>())
            {
                if (Node.ParentId == Guid.Empty)
                    continue;
 
                GroupNode ParentNode = await Database.TryLoadObject<GroupNode>(Node.ParentId);
                if (ParentNode is null)
                {
                    await Database.Delete(Node);
                    Result++;
 
                    lock (nodes)
                    {
                        nodes.Remove(Node.NodeId);
                    }
                }
            }
 
            return Result;
        }
 
        public static bool TryGetDataSource(string SourceId, out IDataSource Source)
        {
            lock (sources)
            {
                if (sources.TryGetValue(SourceId, out Source))
                    return !(Source is null);
            }
 
            if (dataSources is null)
                Types.TryGetModuleParameter("Concentrator", out dataSources);
 
            if (dataSources is null || !dataSources.TryGetDataSource(SourceId, out Source))
                Source = null;
 
            lock (sources)
            {
                sources[SourceId] = Source;
            }
 
            return !(Source is null);
        }
    }
}