AgentApi.cs

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Waher.Events;
using Waher.IoTGateway;
using Waher.Networking.HTTP;
using Waher.Networking.XMPP.Server;
using Waher.Persistence;
using Waher.Persistence.Filters;
using Waher.Runtime.Inventory;
using Waher.Security.JWT;
using Waher.Security.LoginMonitor;
using Waher.Service.IoTBroker.DataStorage;
 
namespace Waher.Service.IoTBroker.WebServices.Agent
{
    public class AgentApi : IModule
    {
        private IAgentResource[] resources;
        private IPersistentDictionary nonceValues;
        private readonly LoginAuditor auditor;
 
        public AgentApi()
            : this(null, null)
        {
        }
 
        public AgentApi(JwtFactory Factory, LoginAuditor Auditor)
        {
            AgentResource.Factory = Factory;
            this.auditor = Auditor ?? Gateway.LoginAuditor;
        }
 
        public LoginAuditor Auditor => this.auditor;
 
        public Task Start()
        {
            return this.Start(Gateway.HttpServer);
        }
 
        public async Task Start(HttpServer WebServer)
        {
            this.nonceValues = await Database.GetDictionary("Nonces");
            List<IAgentResource> Resources = new List<IAgentResource>();
 
            foreach (Type T in Types.GetTypesImplementingInterface(typeof(IAgentResource)))
            {
                if (T.IsAbstract)
                    continue;
 
                try
                {
                    IAgentResource Resource = (IAgentResource)Types.Instantiate(T);
 
                    await Resource.Register(WebServer, this);
                }
                catch (Exception ex)
                {
                    Log.Exception(ex, T.FullName);
                }
            }
 
            this.resources = Resources.ToArray();
        }
 
        public Task Stop()
        {
            return this.Stop(Gateway.HttpServer);
        }
 
        public async Task Stop(HttpServer WebServer)
        {
            if (!(this.resources is null))
            {
                foreach (IAgentResource Resource in this.resources)
                    await Resource.Unregister(WebServer);
            }
        }
 
        public Task<bool> HasNonceBeenUsed(string Nonce)
        {
            return this.nonceValues.ContainsKeyAsync(Nonce);
        }
 
        public Task RegisterNonceValue(string Nonce)
        {
            return this.nonceValues.AddAsync(Nonce, true);
        }
 
        public async Task<ApiKey> GetApiKey(string ApiKey)
        {
            PersistenceLayer PersistenceLayer = XmppServerModule.PersistenceLayer ?? new PersistenceLayer();
            return await PersistenceLayer.GetApiKey(ApiKey);
        }
 
        public async Task<ApiKey> GetAgentApiApiKey()
        {
            return await Database.FindFirstDeleteRest<ApiKey>(
                new FilterFieldEqualTo("Owner", "Agent API"));
        }
    }
}