esiur-dotnet/Esiur/Resource/Warehouse.cs

/*
 
Copyright (c) 2017 Ahmed Kh. Zamil

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.

*/

using Esiur.Data;
using Esiur.Core;
using Esiur.Proxy;
using Esiur.Resource.Template;
using Esiur.Security.Permissions;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace Esiur.Resource
{
    // Centeral Resource Issuer
    public static class Warehouse
    {
        //static byte prefixCounter;

        static AutoList<IResource, Instance> stores = new AutoList<IResource, Instance>(null);
        static Dictionary<uint, IResource> resources = new Dictionary<uint, IResource>();
        static uint resourceCounter = 0;

        static KeyList<Guid, ResourceTemplate> templates = new KeyList<Guid, ResourceTemplate>();

        static bool storeIsOpen = false;

        public delegate void StoreConnectedEvent(IStore store, string name);
        public delegate void StoreDisconnectedEvent(IStore store);
        
        public static event StoreConnectedEvent StoreConnected;
        public static event StoreDisconnectedEvent StoreDisconnected;

        public static KeyList<string, Func<IStore>> Protocols { get; } = new KeyList<string, Func<IStore>>();

        /// <summary>
        /// Get a store by its name.
        /// </summary>
        /// <param name="name">Store instance name</param>
        /// <returns></returns>
        public static IStore GetStore(string name)
        {
            foreach (var s in stores)
                if (s.Instance.Name == name)
                    return s as IStore;
            return null;
        }

        /// <summary>
        /// Get a resource by instance Id.
        /// </summary>
        /// <param name="id">Instance Id</param>
        /// <returns></returns>
        public static AsyncReply<IResource> Get(uint id)
        {
            if (resources.ContainsKey(id))
                return new AsyncReply<IResource>(resources[id]);
            else
                return new AsyncReply<IResource>(null);
        }

        /// <summary>
        /// Open the warehouse.
        /// This function issues the initialize trigger to all stores and resources.
        /// </summary>
        /// <returns>True, if no problem occurred.</returns>
        public static AsyncReply<bool> Open()
        {
            var bag = new AsyncBag<bool>();

            foreach (var store in stores)
                bag.Add(store.Trigger(ResourceTrigger.Initialize));


            bag.Seal();

            var rt = new AsyncReply<bool>();
            bag.Then((x) =>
            {
                foreach (var b in x)
                    if (!b)
                    {
                        rt.Trigger(false);
                        return;
                    }

                var rBag = new AsyncBag<bool>();
                foreach (var rk in resources)
                    rBag.Add(rk.Value.Trigger(ResourceTrigger.SystemInitialized));

                rBag.Seal();

                rBag.Then(y =>
                {
                    foreach (var b in y)
                        if (!b)
                        {
                            rt.Trigger(false);
                            return;
                        }

                    rt.Trigger(true);
                    storeIsOpen = true;
                });

            });

            return rt;
        }

        /// <summary>
        /// Close the warehouse.
        /// This function issues terminate trigger to all resources and stores.
        /// </summary>
        /// <returns>True, if no problem occurred.</returns>
        public static AsyncReply<bool> Close()
        {

            var bag = new AsyncBag<bool>();

            foreach (var resource in resources.Values)
                if (!(resource is IStore))
                    bag.Add(resource.Trigger(ResourceTrigger.Terminate));

            foreach (var store in stores)
                bag.Add(store.Trigger(ResourceTrigger.Terminate));

            foreach (var resource in resources.Values)
                if (!(resource is IStore))
                    bag.Add(resource.Trigger(ResourceTrigger.SystemTerminated));

            foreach (var store in stores)
                bag.Add(store.Trigger(ResourceTrigger.SystemTerminated));

            bag.Seal();

            var rt = new AsyncReply<bool>();
            bag.Then((x) =>
            {
                foreach (var b in x)
                    if (!b)
                    {
                        rt.Trigger(false);
                        return;
                    }

                rt.Trigger(true);
            });

            return rt;
        }


        private static IResource[] QureyIn(string[] path, int index, AutoList<IResource, Instance> resources)
        {
            var rt = new List<IResource>();

            if (index == path.Length - 1)
            {
                if (path[index] == "")
                    foreach (IResource child in resources)
                      rt.Add(child);
                 else
                    foreach (IResource child in resources)
                        if (child.Instance.Name == path[index])
                            rt.Add(child);
            }
            else
                foreach (IResource child in resources)
                    if (child.Instance.Name == path[index])
                        rt.AddRange(QureyIn(path, index+1, child.Instance.Children));

            return rt.ToArray();
        }

        public static AsyncReply<IResource[]> Query(string path)
        {


            if (path == null || path == "")
            {
                var roots = stores.Where(s => s.Instance.Parents.Count == 0).ToArray();
                return new AsyncReply<IResource[]>(roots);
            }
            else
            {
                var rt = new AsyncReply<IResource[]>();
                Get(path).Then(x =>
                {
                    var p = path.Split('/');

                    if (x == null)
                    {
                        rt.Trigger(QureyIn(p, 0, stores));
                    }
                    else
                    {
                        var ar = QureyIn(p, 0, stores).Where(r => r != x).ToList();
                        ar.Insert(0, x);
                        rt.Trigger(ar.ToArray());
                    }
                });

                return rt;

            }

        }

        /// <summary>
        /// Get a resource by its path.
        /// Resource path is sperated by '/' character, e.g. "system/http".
        /// </summary>
        /// <param name="path"></param>
        /// <returns>Resource instance.</returns>
        public static AsyncReply<IResource> Get(string path, Structure attributes = null, IResource parent = null, IPermissionsManager manager = null)
        {

            var p = path.Split('/');
            IResource res;

            foreach(IStore d in stores)
                if (p[0] == d.Instance.Name)
                {
                    var i = 1;
                    res = d;
                    while(p.Length > i)
                    {
                        var si = i;

                        foreach (IResource r in res.Instance.Children)
                            if (r.Instance.Name == p[i])
                            {
                                i++;
                                res = r;
                                break;
                            }

                        if (si == i)
                            // not found, ask the store
                            return d.Get(path.Substring(p[0].Length + 1));
                    }

                    return new AsyncReply<IResource>(res);
                }

            // Should we create a new store ?
            if (path.Contains("://"))
            {
                var url = path.Split(new string[] { "://" }, 2, StringSplitOptions.None);
                var hostname = url[1].Split(new char[] { '/' }, 2)[0];
                var pathname = string.Join("/", url[1].Split(new char[] { '/' }).Skip(1));


                var rt = new AsyncReply<IResource>();

                if (Protocols.ContainsKey(url[0]))
                {
                    var handler = Protocols[url[0]];

                    var store = handler();// Activator.CreateInstance(handler.GetType()) as IStore;
                    Put(store, url[0] + "://" + hostname, null, parent, null, 0, manager, attributes);


                    store.Trigger(ResourceTrigger.Open).Then(x => {
                        if (pathname.Length > 0 && pathname != "")
                            store.Get(pathname).Then(r => {
                                rt.Trigger(r);
                            }).Error(e => rt.TriggerError(e));
                        else
                            rt.Trigger(store);
                    }).Error(e => {
                        rt.TriggerError(e);
                        Warehouse.Remove(store);
                    });
                }

                return rt;
            }


            return new AsyncReply<IResource>(null);
        }

        /// <summary>
        /// Put a resource in the warehouse.
        /// </summary>
        /// <param name="resource">Resource instance.</param>
        /// <param name="name">Resource name.</param>
        /// <param name="store">IStore that manages the resource. Can be null if the resource is a store.</param>
        /// <param name="parent">Parent resource. if not presented the store becomes the parent for the resource.</param>
        public static void Put(IResource resource, string name, IStore store = null, IResource parent = null, ResourceTemplate customTemplate = null, ulong age = 0, IPermissionsManager manager = null, Structure attributes = null)
        {
            resource.Instance = new Instance(resourceCounter++, name, resource, store, customTemplate, age);

            if (attributes != null)
                resource.Instance.SetAttributes(attributes);

            if (manager != null)
                resource.Instance.Managers.Add(manager);

            if (store == parent)
                parent = null;

            if (parent == null)
            {
                if (!(resource is IStore))
                    store.Instance.Children.Add(resource);
            }
            else
                parent.Instance.Children.Add(resource);

                

            if (resource is IStore)
            {
                stores.Add(resource as IStore);
                StoreConnected?.Invoke(resource as IStore, name);
            }
            else
                store.Put(resource);

            resources.Add(resource.Instance.Id, resource);

            if (storeIsOpen)
                 resource.Trigger(ResourceTrigger.Initialize);

        }

        public static T New<T>(string name, IStore store = null, IResource parent = null, IPermissionsManager manager = null, Structure attributes = null)
            where T:IResource
        {
            var type = ResourceProxy.GetProxy<T>();
            var res = Activator.CreateInstance(type) as IResource;
            Put(res, name, store, parent, null, 0, manager, attributes);
            return (T)res;
        }

        /// <summary>
        /// Put a resource template in the templates warehouse.
        /// </summary>
        /// <param name="template">Resource template.</param>
        public static void PutTemplate(ResourceTemplate template)
        {
            if (!templates.ContainsKey(template.ClassId))
                templates.Add(template.ClassId, template);
        }


        /// <summary>
        /// Get a template by type from the templates warehouse. If not in the warehouse, a new ResourceTemplate is created and added to the warehouse.
        /// </summary>
        /// <param name="type">.Net type.</param>
        /// <returns>Resource template.</returns>
        public static ResourceTemplate GetTemplate(Type type)
        {
            // loaded ?
            foreach (var t in templates.Values)
                if (t.ClassName == type.FullName)
                    return t;

            var template = new ResourceTemplate(type);
            templates.Add(template.ClassId, template);

            return template;
        }

        /// <summary>
        /// Get a template by class Id from the templates warehouse. If not in the warehouse, a new ResourceTemplate is created and added to the warehouse.
        /// </summary>
        /// <param name="classId">Class Id.</param>
        /// <returns>Resource template.</returns>
        public static AsyncReply<ResourceTemplate> GetTemplate(Guid classId)
        {
            if (templates.ContainsKey(classId))
                return new AsyncReply<ResourceTemplate>(templates[classId]);
            return null;
        }

        /// <summary>
        /// Get a template by class name from the templates warehouse. If not in the warehouse, a new ResourceTemplate is created and added to the warehouse.
        /// </summary>
        /// <param name="className">Class name.</param>
        /// <returns>Resource template.</returns>
        public static AsyncReply<ResourceTemplate> GetTemplate(string className)
        {
            foreach (var t in templates.Values)
                if (t.ClassName == className)
                    return new AsyncReply<ResourceTemplate>(t);

            return null; 
        }

        public static bool Remove(IResource resource)
        {

            if (resource.Instance == null)
                return false;

            if (resources.ContainsKey(resource.Instance.Id)) 
                resources.Remove(resource.Instance.Id);
            else
                return false;

            if (resource is IStore)
            {
                stores.Remove(resource as IStore);

                // remove all objects associated with the store
                var toBeRemoved = resources.Values.Where(x => x.Instance.Store == resource);
                foreach (var o in toBeRemoved)
                    Remove(o);

                StoreDisconnected?.Invoke(resource as IStore);
            }
            
            if (resource.Instance.Store != null)
                resource.Instance.Store.Remove(resource);

            resource.Destroy();

            return true;
         }
    }
}