1 #include "libslang.h"
2 #include "ui.h"
3 #include <linux/compiler.h>
4 #include <linux/list.h>
5 #include <linux/rbtree.h>
6 #include <stdlib.h>
7 #include <sys/ttydefaults.h>
8 #include "browser.h"
9 #include "helpline.h"
10 #include "../color.h"
11 #include "../util.h"
12 #include <stdio.h>
13 
ui_browser__percent_color(double percent,bool current)14 static int ui_browser__percent_color(double percent, bool current)
15 {
16 	if (current)
17 		return HE_COLORSET_SELECTED;
18 	if (percent >= MIN_RED)
19 		return HE_COLORSET_TOP;
20 	if (percent >= MIN_GREEN)
21 		return HE_COLORSET_MEDIUM;
22 	return HE_COLORSET_NORMAL;
23 }
24 
ui_browser__set_color(struct ui_browser * self __used,int color)25 void ui_browser__set_color(struct ui_browser *self __used, int color)
26 {
27 	SLsmg_set_color(color);
28 }
29 
ui_browser__set_percent_color(struct ui_browser * self,double percent,bool current)30 void ui_browser__set_percent_color(struct ui_browser *self,
31 				   double percent, bool current)
32 {
33 	 int color = ui_browser__percent_color(percent, current);
34 	 ui_browser__set_color(self, color);
35 }
36 
ui_browser__gotorc(struct ui_browser * self,int y,int x)37 void ui_browser__gotorc(struct ui_browser *self, int y, int x)
38 {
39 	SLsmg_gotorc(self->y + y, self->x + x);
40 }
41 
ui_browser__list_head_seek(struct ui_browser * self,off_t offset,int whence)42 void ui_browser__list_head_seek(struct ui_browser *self, off_t offset, int whence)
43 {
44 	struct list_head *head = self->entries;
45 	struct list_head *pos;
46 
47 	switch (whence) {
48 	case SEEK_SET:
49 		pos = head->next;
50 		break;
51 	case SEEK_CUR:
52 		pos = self->top;
53 		break;
54 	case SEEK_END:
55 		pos = head->prev;
56 		break;
57 	default:
58 		return;
59 	}
60 
61 	if (offset > 0) {
62 		while (offset-- != 0)
63 			pos = pos->next;
64 	} else {
65 		while (offset++ != 0)
66 			pos = pos->prev;
67 	}
68 
69 	self->top = pos;
70 }
71 
ui_browser__rb_tree_seek(struct ui_browser * self,off_t offset,int whence)72 void ui_browser__rb_tree_seek(struct ui_browser *self, off_t offset, int whence)
73 {
74 	struct rb_root *root = self->entries;
75 	struct rb_node *nd;
76 
77 	switch (whence) {
78 	case SEEK_SET:
79 		nd = rb_first(root);
80 		break;
81 	case SEEK_CUR:
82 		nd = self->top;
83 		break;
84 	case SEEK_END:
85 		nd = rb_last(root);
86 		break;
87 	default:
88 		return;
89 	}
90 
91 	if (offset > 0) {
92 		while (offset-- != 0)
93 			nd = rb_next(nd);
94 	} else {
95 		while (offset++ != 0)
96 			nd = rb_prev(nd);
97 	}
98 
99 	self->top = nd;
100 }
101 
ui_browser__rb_tree_refresh(struct ui_browser * self)102 unsigned int ui_browser__rb_tree_refresh(struct ui_browser *self)
103 {
104 	struct rb_node *nd;
105 	int row = 0;
106 
107 	if (self->top == NULL)
108                 self->top = rb_first(self->entries);
109 
110 	nd = self->top;
111 
112 	while (nd != NULL) {
113 		ui_browser__gotorc(self, row, 0);
114 		self->write(self, nd, row);
115 		if (++row == self->height)
116 			break;
117 		nd = rb_next(nd);
118 	}
119 
120 	return row;
121 }
122 
ui_browser__is_current_entry(struct ui_browser * self,unsigned row)123 bool ui_browser__is_current_entry(struct ui_browser *self, unsigned row)
124 {
125 	return self->top_idx + row == self->index;
126 }
127 
ui_browser__refresh_dimensions(struct ui_browser * self)128 void ui_browser__refresh_dimensions(struct ui_browser *self)
129 {
130 	int cols, rows;
131 	newtGetScreenSize(&cols, &rows);
132 
133 	self->width = cols - 1;
134 	self->height = rows - 2;
135 	self->y = 1;
136 	self->x = 0;
137 }
138 
ui_browser__reset_index(struct ui_browser * self)139 void ui_browser__reset_index(struct ui_browser *self)
140 {
141 	self->index = self->top_idx = 0;
142 	self->seek(self, 0, SEEK_SET);
143 }
144 
ui_browser__add_exit_key(struct ui_browser * self,int key)145 void ui_browser__add_exit_key(struct ui_browser *self, int key)
146 {
147 	newtFormAddHotKey(self->form, key);
148 }
149 
ui_browser__add_exit_keys(struct ui_browser * self,int keys[])150 void ui_browser__add_exit_keys(struct ui_browser *self, int keys[])
151 {
152 	int i = 0;
153 
154 	while (keys[i] && i < 64) {
155 		ui_browser__add_exit_key(self, keys[i]);
156 		++i;
157 	}
158 }
159 
__ui_browser__show_title(struct ui_browser * browser,const char * title)160 void __ui_browser__show_title(struct ui_browser *browser, const char *title)
161 {
162 	SLsmg_gotorc(0, 0);
163 	ui_browser__set_color(browser, NEWT_COLORSET_ROOT);
164 	slsmg_write_nstring(title, browser->width);
165 }
166 
ui_browser__show_title(struct ui_browser * browser,const char * title)167 void ui_browser__show_title(struct ui_browser *browser, const char *title)
168 {
169 	pthread_mutex_lock(&ui__lock);
170 	__ui_browser__show_title(browser, title);
171 	pthread_mutex_unlock(&ui__lock);
172 }
173 
ui_browser__show(struct ui_browser * self,const char * title,const char * helpline,...)174 int ui_browser__show(struct ui_browser *self, const char *title,
175 		     const char *helpline, ...)
176 {
177 	va_list ap;
178 	int keys[] = { NEWT_KEY_UP, NEWT_KEY_DOWN, NEWT_KEY_PGUP,
179 		       NEWT_KEY_PGDN, NEWT_KEY_HOME, NEWT_KEY_END, ' ',
180 		       NEWT_KEY_LEFT, NEWT_KEY_ESCAPE, 'q', CTRL('c'), 0 };
181 
182 	if (self->form != NULL)
183 		newtFormDestroy(self->form);
184 
185 	ui_browser__refresh_dimensions(self);
186 	self->form = newtForm(NULL, NULL, 0);
187 	if (self->form == NULL)
188 		return -1;
189 
190 	self->sb = newtVerticalScrollbar(self->width, 1, self->height,
191 					 HE_COLORSET_NORMAL,
192 					 HE_COLORSET_SELECTED);
193 	if (self->sb == NULL)
194 		return -1;
195 
196 	pthread_mutex_lock(&ui__lock);
197 	__ui_browser__show_title(self, title);
198 
199 	ui_browser__add_exit_keys(self, keys);
200 	newtFormAddComponent(self->form, self->sb);
201 
202 	va_start(ap, helpline);
203 	ui_helpline__vpush(helpline, ap);
204 	va_end(ap);
205 	pthread_mutex_unlock(&ui__lock);
206 	return 0;
207 }
208 
ui_browser__hide(struct ui_browser * self)209 void ui_browser__hide(struct ui_browser *self)
210 {
211 	pthread_mutex_lock(&ui__lock);
212 	newtFormDestroy(self->form);
213 	self->form = NULL;
214 	ui_helpline__pop();
215 	pthread_mutex_unlock(&ui__lock);
216 }
217 
ui_browser__refresh(struct ui_browser * self)218 int ui_browser__refresh(struct ui_browser *self)
219 {
220 	int row;
221 
222 	pthread_mutex_lock(&ui__lock);
223 	newtScrollbarSet(self->sb, self->index, self->nr_entries - 1);
224 	row = self->refresh(self);
225 	ui_browser__set_color(self, HE_COLORSET_NORMAL);
226 	SLsmg_fill_region(self->y + row, self->x,
227 			  self->height - row, self->width, ' ');
228 	pthread_mutex_unlock(&ui__lock);
229 
230 	return 0;
231 }
232 
ui_browser__run(struct ui_browser * self)233 int ui_browser__run(struct ui_browser *self)
234 {
235 	struct newtExitStruct es;
236 
237 	if (ui_browser__refresh(self) < 0)
238 		return -1;
239 
240 	while (1) {
241 		off_t offset;
242 
243 		newtFormRun(self->form, &es);
244 
245 		if (es.reason != NEWT_EXIT_HOTKEY)
246 			break;
247 		switch (es.u.key) {
248 		case NEWT_KEY_DOWN:
249 			if (self->index == self->nr_entries - 1)
250 				break;
251 			++self->index;
252 			if (self->index == self->top_idx + self->height) {
253 				++self->top_idx;
254 				self->seek(self, +1, SEEK_CUR);
255 			}
256 			break;
257 		case NEWT_KEY_UP:
258 			if (self->index == 0)
259 				break;
260 			--self->index;
261 			if (self->index < self->top_idx) {
262 				--self->top_idx;
263 				self->seek(self, -1, SEEK_CUR);
264 			}
265 			break;
266 		case NEWT_KEY_PGDN:
267 		case ' ':
268 			if (self->top_idx + self->height > self->nr_entries - 1)
269 				break;
270 
271 			offset = self->height;
272 			if (self->index + offset > self->nr_entries - 1)
273 				offset = self->nr_entries - 1 - self->index;
274 			self->index += offset;
275 			self->top_idx += offset;
276 			self->seek(self, +offset, SEEK_CUR);
277 			break;
278 		case NEWT_KEY_PGUP:
279 			if (self->top_idx == 0)
280 				break;
281 
282 			if (self->top_idx < self->height)
283 				offset = self->top_idx;
284 			else
285 				offset = self->height;
286 
287 			self->index -= offset;
288 			self->top_idx -= offset;
289 			self->seek(self, -offset, SEEK_CUR);
290 			break;
291 		case NEWT_KEY_HOME:
292 			ui_browser__reset_index(self);
293 			break;
294 		case NEWT_KEY_END:
295 			offset = self->height - 1;
296 			if (offset >= self->nr_entries)
297 				offset = self->nr_entries - 1;
298 
299 			self->index = self->nr_entries - 1;
300 			self->top_idx = self->index - offset;
301 			self->seek(self, -offset, SEEK_END);
302 			break;
303 		default:
304 			return es.u.key;
305 		}
306 		if (ui_browser__refresh(self) < 0)
307 			return -1;
308 	}
309 	return -1;
310 }
311 
ui_browser__list_head_refresh(struct ui_browser * self)312 unsigned int ui_browser__list_head_refresh(struct ui_browser *self)
313 {
314 	struct list_head *pos;
315 	struct list_head *head = self->entries;
316 	int row = 0;
317 
318 	if (self->top == NULL || self->top == self->entries)
319                 self->top = head->next;
320 
321 	pos = self->top;
322 
323 	list_for_each_from(pos, head) {
324 		ui_browser__gotorc(self, row, 0);
325 		self->write(self, pos, row);
326 		if (++row == self->height)
327 			break;
328 	}
329 
330 	return row;
331 }
332 
333 static struct newtPercentTreeColors {
334 	const char *topColorFg, *topColorBg;
335 	const char *mediumColorFg, *mediumColorBg;
336 	const char *normalColorFg, *normalColorBg;
337 	const char *selColorFg, *selColorBg;
338 	const char *codeColorFg, *codeColorBg;
339 } defaultPercentTreeColors = {
340 	"red",       "lightgray",
341 	"green",     "lightgray",
342 	"black",     "lightgray",
343 	"lightgray", "magenta",
344 	"blue",	     "lightgray",
345 };
346 
ui_browser__init(void)347 void ui_browser__init(void)
348 {
349 	struct newtPercentTreeColors *c = &defaultPercentTreeColors;
350 
351 	sltt_set_color(HE_COLORSET_TOP, NULL, c->topColorFg, c->topColorBg);
352 	sltt_set_color(HE_COLORSET_MEDIUM, NULL, c->mediumColorFg, c->mediumColorBg);
353 	sltt_set_color(HE_COLORSET_NORMAL, NULL, c->normalColorFg, c->normalColorBg);
354 	sltt_set_color(HE_COLORSET_SELECTED, NULL, c->selColorFg, c->selColorBg);
355 	sltt_set_color(HE_COLORSET_CODE, NULL, c->codeColorFg, c->codeColorBg);
356 }
357