Artemis/src/Artemis.Storage/Repositories/ProfileRepository.cs

using System;
using System.Collections.Generic;
using Artemis.Storage.Entities.Profile;
using Artemis.Storage.Repositories.Interfaces;
using LiteDB;

namespace Artemis.Storage.Repositories
{
    public class ProfileRepository : IProfileRepository
    {
        private readonly LiteRepository _repository;

        internal ProfileRepository(LiteRepository repository)
        {
            _repository = repository;
            _repository.Database.GetCollection<ProfileEntity>().EnsureIndex(s => s.Name);
        }

        public void Add(ProfileEntity profileEntity)
        {
            _repository.Insert(profileEntity);
        }

        public void Remove(ProfileEntity profileEntity)
        {
            _repository.Delete<ProfileEntity>(profileEntity.Id);
        }

        public List<ProfileEntity> GetAll()
        {
            return _repository.Query<ProfileEntity>().ToList();
        }

        public ProfileEntity Get(Guid id)
        {
            return _repository.FirstOrDefault<ProfileEntity>(p => p.Id == id);
        }

        public List<ProfileEntity> GetByPluginGuid(Guid pluginGuid)
        {
            return _repository.Query<ProfileEntity>()
                .Include(p => p.Folders)
                .Include(p => p.Layers)
                .Where(s => s.PluginGuid == pluginGuid)
                .ToList();
        }

        public void Save(ProfileEntity profileEntity)
        {
            _repository.Upsert(profileEntity);
        }
    }
}