using System;
using System.Collections.Generic;
using System.Text;
using System.Threading.Tasks;
using Tango.BLL.Objects;
using Tango.DAL;
using Tango.DAL.Entities;
using System.Linq;
using Tango.BLL.Mappers;

namespace Tango.BLL.Services
{
    public class OrganizationsService : ServiceBase<Organization>
    {
        private IRepository<OrganizationEntity> _organizationsRepository;
        private IRepository<MachineEntity> _machinesRepository;
        private OrganizationToOrganizationEntityMapper _mapper;

        public OrganizationsService(IRepository<OrganizationEntity> organizationsRepository, IRepository<MachineEntity> machinesRepository)
        {
            _organizationsRepository = organizationsRepository;
            _machinesRepository = machinesRepository;
            _mapper = new OrganizationToOrganizationEntityMapper();
        }

        public async Task<List<Organization>> GetAllOrganizations()
        {
            var entities = await _organizationsRepository.GetAllAsync();

            var organizations = entities.Select(x => _mapper.Create(x)).ToList();

            return organizations;
        }

        public async Task<long> DeleteAllOrganizations()
        {
            var orgs = await GetAllOrganizations();

            long count = 0;

            foreach (var org in orgs)
            {
                count += await DeleteOrganizationByID(org.ID);
            }

            return count;
        }

        public async Task<Organization> AddOrganization(Organization organization)
        {
            OrganizationEntity entity = _mapper.Create(organization);

            entity = await _organizationsRepository.Insert(entity);

            _mapper.Map(entity, organization);

            return organization;
        }

        public async Task<Organization> GetOrganizationByID(String id)
        {
            var entity = (await _organizationsRepository.GetAsync(x => x.ID == id)).ToList().SingleOrDefault();

            if (entity == null)
            {
                throw new KeyNotFoundException($"Could not find organization with id {id}.");
            }

            return _mapper.Create(entity);
        }

        public async Task<long> DeleteOrganizationByID(String id)
        {
            await _machinesRepository.DeleteAsync(x => x.OrganizationID == id);
            long count = await _organizationsRepository.DeleteAsync(x => x.ID == id);
            return count;
        }

        public async Task<long> Count()
        {
            return await _organizationsRepository.Count();
        }
    }
}