BuildFeed/BuildFeed.Model/MetaItem.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.ComponentModel.DataAnnotations;
using System.Linq;
using System.Threading.Tasks;
using MongoDB.Bson;
using MongoDB.Bson.Serialization.Attributes;
using MongoDB.Driver;
using Required = System.ComponentModel.DataAnnotations.RequiredAttribute;

namespace BuildFeed.Model
{
    [DataObject]
    public class MetaItemModel
    {
        [Key]
        [BsonId]
        [@Required]
        public MetaItemKey Id { get; set; }

        [DisplayName("Meta Description")]
        public string MetaDescription { get; set; }

        [DisplayName("Page Content")]
        public string PageContent { get; set; }
    }

    public class MetaItem
    {
        private const string META_COLLECTION_NAME = "metaitem";
        private readonly BuildRepository _bModel;

        private readonly IMongoCollection<MetaItemModel> _metaCollection;

        public MetaItem(MongoConfig config)
        {
            var settings = new MongoClientSettings
            {
                Server = new MongoServerAddress(config.Host, config.Port)
            };

            if (!string.IsNullOrEmpty(config.Username) && !string.IsNullOrEmpty(config.Password))
            {
                settings.Credential =
                    MongoCredential.CreateCredential(config.Database, config.Username, config.Password);
            }

            var dbClient = new MongoClient(settings);

            _metaCollection = dbClient.GetDatabase(config.Database)
                .GetCollection<MetaItemModel>(META_COLLECTION_NAME);
            _bModel = new BuildRepository(config);
        }

        [DataObjectMethod(DataObjectMethodType.Select, false)]
        public async Task<IEnumerable<MetaItemModel>> Select()
            => await _metaCollection.Find(new BsonDocument()).ToListAsync();

        [DataObjectMethod(DataObjectMethodType.Select, true)]
        public async Task<IEnumerable<MetaItemModel>> SelectByType(MetaType type)
        {
            return await _metaCollection.Find(f => f.Id.Type == type).ToListAsync();
        }

        [DataObjectMethod(DataObjectMethodType.Select, false)]
        public async Task<MetaItemModel> SelectById(MetaItemKey id)
        {
            return await _metaCollection.Find(f => f.Id.Type == id.Type && f.Id.Value == id.Value)
                .SingleOrDefaultAsync();
        }

        [DataObjectMethod(DataObjectMethodType.Select, false)]
        public async Task<IEnumerable<string>> SelectUnusedLabs()
        {
            var labs = await _bModel.SelectAllLabs();

            var usedLabs = await _metaCollection.Find(f => f.Id.Type == MetaType.Lab).ToListAsync();

            return from l in labs
                where usedLabs.All(ul => ul.Id.Value != l)
                select l;
        }

        [DataObjectMethod(DataObjectMethodType.Select, false)]
        public async Task<IEnumerable<string>> SelectUnusedVersions()
        {
            var versions = await _bModel.SelectAllVersions();

            var usedVersions = await _metaCollection.Find(f => f.Id.Type == MetaType.Version).ToListAsync();

            return from v in versions
                where usedVersions.All(ul => ul.Id.Value != v.ToString())
                select v.ToString();
        }

        [DataObjectMethod(DataObjectMethodType.Select, false)]
        public async Task<IEnumerable<string>> SelectUnusedYears()
        {
            var years = await _bModel.SelectAllYears();

            var usedYears = await _metaCollection.Find(f => f.Id.Type == MetaType.Year).ToListAsync();

            return from y in years
                where usedYears.All(ul => ul.Id.Value != y.ToString())
                select y.ToString();
        }

        [DataObjectMethod(DataObjectMethodType.Select, false)]
        public async Task<IEnumerable<string>> SelectUnusedFamilies()
        {
            var families = await _bModel.SelectAllFamilies();

            var usedFamilies = await _metaCollection.Find(f => f.Id.Type == MetaType.Family).ToListAsync();

            return from y in families
                where usedFamilies.All(ul => ul.Id.Value != y.ToString())
                select y.ToString();
        }

        [DataObjectMethod(DataObjectMethodType.Insert, true)]
        public async Task Insert(MetaItemModel item)
        {
            await _metaCollection.InsertOneAsync(item);
        }

        [DataObjectMethod(DataObjectMethodType.Update, true)]
        public async Task Update(MetaItemModel item)
        {
            await _metaCollection.ReplaceOneAsync(f => f.Id.Type == item.Id.Type && f.Id.Value == item.Id.Value, item);
        }

        [DataObjectMethod(DataObjectMethodType.Insert, false)]
        public async Task InsertAll(IEnumerable<MetaItemModel> items)
        {
            await _metaCollection.InsertManyAsync(items);
        }

        [DataObjectMethod(DataObjectMethodType.Delete, true)]
        public async Task DeleteById(MetaItemKey id)
        {
            await _metaCollection.DeleteOneAsync(f => f.Id.Type == id.Type && f.Id.Value == id.Value);
        }
    }

    public class MetaItemKey
    {
        public string Value { get; set; }
        public MetaType Type { get; set; }

        public MetaItemKey()
        {
        }

        public MetaItemKey(string id)
        {
            var items = id.Split(':');
            Type = (MetaType)Enum.Parse(typeof(MetaType), items[0]);
            Value = items[1];
        }

        public override string ToString() => $"{Type}:{Value}";
    }

    public enum MetaType
    {
        Lab,
        Version,
        Source,
        Year,
        Family
    }
}