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 Exception("Śmieciarka nie jest w miejscu odebrania ś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.Any(c => c.TypeOfGarbage == _typeOfGarbage)) throw new Exception("Ta śmieciarka nie może zebrać żądanych śmieci."); _garbageCollector.TrashContainers.FirstOrDefault(t => t.TypeOfGarbage == _typeOfGarbage).AddGarbage(garbage); } } }