package com.ufund.api.ufundapi.model;

import java.util.ArrayList;

public class Cupboard {
    ArrayList<Need> needs;

    public Cupboard() {
        needs = new ArrayList<>();
    }

    public void addNeed(Need need) {
        needs.add(need);
    }

    public void createNeed(String name, double max, Need.GoalType type) {
        int id = needs.size();
        Need need = new Need(name, id, max, type);
        addNeed(need);
    }

    public void updateID(int id) {
        for (int i = 0; i < needs.size(); i++) {
            needs.get(i).setID(i);
        }
    }

    public void removeNeed(Need need) {
        needs.remove(need);
    }

    public void updateNeed(Need need) {
        for (int i = 0; i < needs.size(); i++) {
            if (needs.get(i).getID() == need.getID()) {
                needs.set(i, need);
                return;
            }
        }
    }

    public void removeNeed(String name) {
        for (int i = 0; i < needs.size(); i++) {
            if (needs.get(i).getName().equals(name)) {
                needs.remove(i);
                return;
            }
        }
    }

    public Need[] getNeeds() {
        return needs.toArray(new Need[0]);
    }

    public Need getNeed(int index) {
        return needs.get(index);
    }

}