Cage-Rabbits/2019

import java.util.*;
import java.util.stream.Collectors;

public class Cage {

    private String name;
    private int capacity;
    private List<Rabbit> data;

    public Cage(String name, int capacity) {
        this.name = name;
        this.capacity = capacity;
        this.data = new ArrayList<>();
    }

    public void removeSpecies(String species){
        this.data=this.data.stream().filter(e->!e.getSpecies().equals(species)).collect(Collectors.toList());
    }

    public Rabbit sellRabbit(String name){
        this.data.stream().filter(e->e.getName().equals(name)).findFirst().ifPresent(v->v.setAvailable(false));
        return this.data.stream().filter(e->e.getName().equals(name)).findFirst().orElse(null);
    }

    public List<Rabbit> sellRabbitBySpecies(String species){
        this.data.stream().filter(e->e.getSpecies().equals(species)).forEach(v->v.setAvailable(false));
        return this.data.stream().filter(e->e.getSpecies().equals(species)).collect(Collectors.toList());
    }

    public String report(){
        StringBuilder output=new StringBuilder("Rabbits available at ");
        output.append(getName()).append(":").append(System.lineSeparator());
        this.data.stream().filter(Rabbit::isAvailable).forEach(v->output.append(v).append(System.lineSeparator()));
        return output.toString();
    }


    public boolean removeRabbit(String name){
        return this.data.removeIf(element->element.getName().equals(name));
    }

    public int count(){
        return this.data.size();
    }

    public void add(Rabbit rabbit){
        if(this.data.size()<this.capacity){
            this.data.add(rabbit);
        }
    }

    public String getName() {
        return name;
    }

    public int getCapacity() {
        return capacity;
    }
}