summaryrefslogtreecommitdiff
path: root/internal/inventory/inventory.go
blob: 4fe3be0d8ac2d4fb0d263b746166204df806e1d1 (plain)
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
package inventory

type Inventory struct {
	Items []Item
}

type Item struct {
	Title    string
	Quantity int
}

func (m *Inventory) Add(title string) {
	for _, v := range m.Items {
		if v.Title == title {
			v.Quantity += 1
			return
		}
	}
	m.Items = append(m.Items, Item{Title: title, Quantity: 1})
}

func (m *Inventory) Del(title string) int {
	for i, v := range m.Items {
		if v.Title == title {
			count := v.Quantity
			m.Items = append(m.Items[:i], m.Items[i+1:]...)
			return count
		}
	}
	return 0
}