servuo/Scripts/Misc/WeakEntityCollection.cs

#region References
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
#endregion

namespace Server
{
    public static class WeakEntityCollection
    {
        public sealed class EntityCollection : List<IEntity>
        {
            public IEnumerable<Item> Items => this.OfType<Item>();
            public IEnumerable<Mobile> Mobiles => this.OfType<Mobile>();

            public EntityCollection()
                : this(0x400)
            { }

            public EntityCollection(int capacity)
                : base(capacity)
            { }
        }

        private static readonly string _FilePath = Path.Combine("Saves", "WeakEntityCollection", "WeakEntityCollection.bin");

        private static readonly Dictionary<string, EntityCollection> _Collections =
            new Dictionary<string, EntityCollection>(StringComparer.OrdinalIgnoreCase);

        public static void Configure()
        {
            EventSink.WorldSave += OnWorldSave;
            EventSink.WorldLoad += OnWorldLoad;
        }

        private static void OnWorldSave(WorldSaveEventArgs e)
        {
            Save();
        }

        private static void OnWorldLoad()
        {
            Load();
        }

        public static void Save()
        {
            Persistence.Serialize(
                _FilePath,
                writer =>
                {
                    writer.Write(1); // Version

                    writer.Write(_Collections.Count);

                    foreach (KeyValuePair<string, EntityCollection> kv in _Collections)
                    {
                        writer.Write(kv.Key);

                        kv.Value.RemoveAll(ent => ent == null || ent.Deleted);

                        writer.Write(kv.Value.Count);

                        foreach (IEntity ent in kv.Value)
                        {
                            writer.Write(ent.Serial);
                        }
                    }
                });
        }

        public static void Load()
        {
            Persistence.Deserialize(
                _FilePath,
                reader =>
                {
                    int version = reader.ReadInt();

                    switch (version)
                    {
                        case 1:
                            {
                                int entries = reader.ReadInt();

                                while (--entries >= 0)
                                {
                                    string key = reader.ReadString();

                                    int ents = reader.ReadInt();

                                    EntityCollection col = new EntityCollection(ents);

                                    IEntity ent;

                                    while (--ents >= 0)
                                    {
                                        ent = World.FindEntity(reader.ReadInt());

                                        if (ent != null && !ent.Deleted)
                                        {
                                            col.Add(ent);
                                        }
                                    }

                                    _Collections[key] = col;
                                }
                            }
                            break;
                        case 0:
                            {
                                int entries = reader.ReadInt();

                                while (--entries >= 0)
                                {
                                    string key = reader.ReadString();

                                    List<Item> items = reader.ReadStrongItemList();
                                    List<Mobile> mobiles = reader.ReadStrongMobileList();

                                    EntityCollection col = new EntityCollection(items.Count + mobiles.Count);

                                    col.AddRange(items);
                                    col.AddRange(mobiles);

                                    _Collections[key] = col;
                                }
                            }
                            break;
                    }
                });
        }

        public static EntityCollection GetCollection(string name)
        {
            EntityCollection col;

            if (!_Collections.TryGetValue(name, out col) || col == null)
            {
                _Collections[name] = col = new EntityCollection();
            }

            return col;
        }

        public static bool HasCollection(string name)
        {
            return name != null && _Collections.ContainsKey(name);
        }

        public static void Add(string key, IEntity entity)
        {
            if (entity == null || entity.Deleted)
            {
                return;
            }

            EntityCollection col = GetCollection(key);

            if (col != null && !col.Contains(entity))
            {
                col.Add(entity);
            }
        }

        public static bool Remove(string key, IEntity entity)
        {
            if (entity == null)
            {
                return false;
            }

            EntityCollection col = GetCollection(key);

            if (col != null)
            {
                return col.Remove(entity);
            }

            return false;
        }

        public static int Clean(string key)
        {
            int removed = 0;

            EntityCollection col = GetCollection(key);

            if (col != null)
            {
                int ents = col.Count;

                while (--ents >= 0)
                {
                    if (ents < col.Count && col[ents].Deleted)
                    {
                        col.RemoveAt(ents);

                        ++removed;
                    }
                }
            }

            return removed;
        }

        public static int Delete(string key)
        {
            int deleted = 0;

            EntityCollection col = GetCollection(key);

            if (col != null)
            {
                int ents = col.Count;

                while (--ents >= 0)
                {
                    if (ents < col.Count)
                    {
                        col[ents].Delete();

                        ++deleted;
                    }
                }

                col.Clear();
            }

            _Collections.Remove(key);

            return deleted;
        }
    }
}