summaryrefslogtreecommitdiff
path: root/state.lua
blob: 8bf4ddd8a2eced9c28723fdaf811c4f5774360d9 (plain) (blame)
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
-- Copyright 2025 David Vazgenovich Shakaryan

local util = require('util')

local state = {}
local mt = {}
mt.__index = mt

local menu_mt = {}
menu_mt.__index = menu_mt

function state.new()
	return setmetatable({
		menus = {},
		depth = 0,

		favourites = {},
		playing_id = nil,
	}, mt)
end

function mt:menu()
	return self.menus[self.depth]
end

function mt:push_menu(t)
	local menu = setmetatable({
		options = {},
		cursor = 1,
		view_top = 1,
	}, menu_mt)

	for k, v in pairs(t) do
		menu[k] = v
	end

	if menu.type == 'search' then
		menu.search_options = menu.options
		menu.search_text = menu.search_text or ''
		menu.search_cursor = menu.search_cursor or
			#menu.search_text + 1
		menu:update_search_matches()
	end

	self.depth = self.depth + 1
	self.menus[self.depth] = menu

	return menu
end

-- returns index if found
function mt:favourited(id)
	for i, v in ipairs(self.favourites) do
		if v == id then
			return i
		end
	end
end

function mt:add_favourite(id)
	self.favourites[#self.favourites+1] = id
end

function mt:remove_favourite_at(i)
	table.remove(self.favourites, i)
end

-- inserts the given id into the favourites array before the next favourited
-- menu option, starting from the next cursor position, or the end if no such
-- option is found. this is meant for in-place favouriting from the favourites
-- menu.
function mt:insert_favourite_before_next_in_menu(id)
	local menu = self:menu()
	for i = menu.cursor+1, #menu.options do
		local ind = self:favourited(menu.options[i].id)
		if ind then
			table.insert(self.favourites, ind, id)
			return
		end
	end

	self:add_favourite(id)
end

function menu_mt:set_cursor(pos, lines, opts)
	local pos = math.max(1, math.min(pos, #self.options))
	local top = self.view_top

	if not lines then
		top = pos
		goto update
	end

	if opts and opts.centre then
		top = pos - math.floor((lines - 1) / 2)
	elseif opts and opts.keep_offset then
		top = top + pos - self.cursor
	end

	-- move view to keep selected option visible
	if pos < top then
		top = pos
	elseif pos > top + lines - 1 then
		top = pos - lines + 1
	end

	top = math.max(1, math.min(top, #self.options - lines + 1))

	::update::
	if pos == self.cursor and top == self.view_top then
		return false
	end
	self.cursor = pos
	self.view_top = top
	return true
end

function menu_mt:set_sort(bool, f)
	if not self.sorted == not bool then
		return
	end

	local key = self.type == 'search' and 'search_options' or 'options'
	if bool then
		self['orig_' .. key] = self[key]
		self[key] = util.copy_table(self[key])
		f(self[key])
	else
		self[key] = self['orig_' .. key]
		self['orig_' .. key] = nil
	end

	if self.type == 'search' then
		self:update_search_matches()
	end
	self.sorted = bool
end

function menu_mt:set_search_cursor(pos)
	local pos = math.max(1, math.min(#self.search_text + 1, pos))
	if pos == self.search_cursor then
		return false
	end

	self.search_cursor = pos
	return true
end

function menu_mt:set_search_text(str)
	self.search_text = str
	self:update_search_matches()
end

function menu_mt:update_search_matches()
	if #self.search_text == 0 then
		self.options = self.search_options
		return
	end

	-- no utf8 :(
	local case_sensitive = not not self.search_text:find('%u')

	local options = {}
	for _, v in ipairs(self.search_options) do
		local matches = {}

		local name = v.name
		if not case_sensitive then
			name = name:lower()
		end

		local i, j = 0, 0
		while true do
			i, j = name:find(self.search_text, j + 1, true)
			if not i then
				break
			end
			matches[#matches+1] = {start = i, stop = j}
		end

		if #matches > 0 then
			local t = util.copy_table(v)
			t.matches = matches
			options[#options+1] = t
		end
	end

	self.options = options
end

return state