ufund-api/src/main/java/com/ufund/api/ufundapi/Cupboard.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

package com.ufund.api.ufundapi;

import java.util.ArrayList;

public class Cupboard {
    ArrayList<Need> needs;

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

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

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

    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);
    }

}