Czoko_Smieciarka/Trunk/Components/CzokoŚmieciarka.DataModels/Models/Steps/CollectStep.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using CzokoŚmieciarka.DataModels.Interfaces;
using CzokoŚmieciarka.DataModels.Interfaces.TrashCans;

namespace CzokoŚmieciarka.DataModels.Models.Steps
{
    public class CollectStep : IStep
    {
        public CollectStep(IGarbageCollector garbageCollector, IGarbageLocalization garbageLocalization, ITypeOfGarbage typeOfGarbage)
        {
            this._garbageCollector = garbageCollector;
            this._garbageLocalization = garbageLocalization;
            this._typeOfGarbage = typeOfGarbage;
        }

        private IGarbageCollector _garbageCollector;
        private ITypeOfGarbage _typeOfGarbage;
        private IGarbageLocalization _garbageLocalization;

        public void Invoke()
        {
            if(_garbageCollector.Position != _garbageLocalization.Coords)
                throw new WrongPositionException("Śmieciarka nie jest w miejscu oderbania śmieci");

            var trashCans = _garbageLocalization.TrashCans.Where(t => t.TypeOfGarbage == _typeOfGarbage);
            var garbage = trashCans.Select(t => t.TakeGarbage()).Aggregate((a,b)=>a+b);

            if (_garbageCollector.TrashContainers.All(c => c.TypeOfGarbage != _typeOfGarbage))
                throw new TrashContainerNotFound($"Nie znaleziono kontenera na {_typeOfGarbage.GarbageType}.");

            _garbageCollector.TrashContainers.First(t => t.TypeOfGarbage == _typeOfGarbage).AddGarbage(garbage);
        }
    }
}