1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0
2*4882a593Smuzhiyun #include <linux/kernel.h>
3*4882a593Smuzhiyun #include "../progress.h"
4*4882a593Smuzhiyun #include "../libslang.h"
5*4882a593Smuzhiyun #include "../ui.h"
6*4882a593Smuzhiyun #include "tui.h"
7*4882a593Smuzhiyun #include "units.h"
8*4882a593Smuzhiyun #include "../browser.h"
9*4882a593Smuzhiyun
__tui_progress__init(struct ui_progress * p)10*4882a593Smuzhiyun static void __tui_progress__init(struct ui_progress *p)
11*4882a593Smuzhiyun {
12*4882a593Smuzhiyun p->next = p->step = p->total / (SLtt_Screen_Cols - 2) ?: 1;
13*4882a593Smuzhiyun }
14*4882a593Smuzhiyun
get_title(struct ui_progress * p,char * buf,size_t size)15*4882a593Smuzhiyun static int get_title(struct ui_progress *p, char *buf, size_t size)
16*4882a593Smuzhiyun {
17*4882a593Smuzhiyun char buf_cur[20];
18*4882a593Smuzhiyun char buf_tot[20];
19*4882a593Smuzhiyun int ret;
20*4882a593Smuzhiyun
21*4882a593Smuzhiyun ret = unit_number__scnprintf(buf_cur, sizeof(buf_cur), p->curr);
22*4882a593Smuzhiyun ret += unit_number__scnprintf(buf_tot, sizeof(buf_tot), p->total);
23*4882a593Smuzhiyun
24*4882a593Smuzhiyun return ret + scnprintf(buf, size, "%s [%s/%s]",
25*4882a593Smuzhiyun p->title, buf_cur, buf_tot);
26*4882a593Smuzhiyun }
27*4882a593Smuzhiyun
tui_progress__update(struct ui_progress * p)28*4882a593Smuzhiyun static void tui_progress__update(struct ui_progress *p)
29*4882a593Smuzhiyun {
30*4882a593Smuzhiyun char buf[100], *title = (char *) p->title;
31*4882a593Smuzhiyun int bar, y;
32*4882a593Smuzhiyun /*
33*4882a593Smuzhiyun * FIXME: We should have a per UI backend way of showing progress,
34*4882a593Smuzhiyun * stdio will just show a percentage as NN%, etc.
35*4882a593Smuzhiyun */
36*4882a593Smuzhiyun if (use_browser <= 0)
37*4882a593Smuzhiyun return;
38*4882a593Smuzhiyun
39*4882a593Smuzhiyun if (p->total == 0)
40*4882a593Smuzhiyun return;
41*4882a593Smuzhiyun
42*4882a593Smuzhiyun if (p->size) {
43*4882a593Smuzhiyun get_title(p, buf, sizeof(buf));
44*4882a593Smuzhiyun title = buf;
45*4882a593Smuzhiyun }
46*4882a593Smuzhiyun
47*4882a593Smuzhiyun ui__refresh_dimensions(false);
48*4882a593Smuzhiyun pthread_mutex_lock(&ui__lock);
49*4882a593Smuzhiyun y = SLtt_Screen_Rows / 2 - 2;
50*4882a593Smuzhiyun SLsmg_set_color(0);
51*4882a593Smuzhiyun SLsmg_draw_box(y, 0, 3, SLtt_Screen_Cols);
52*4882a593Smuzhiyun SLsmg_gotorc(y++, 1);
53*4882a593Smuzhiyun SLsmg_write_string(title);
54*4882a593Smuzhiyun SLsmg_fill_region(y, 1, 1, SLtt_Screen_Cols - 2, ' ');
55*4882a593Smuzhiyun SLsmg_set_color(HE_COLORSET_SELECTED);
56*4882a593Smuzhiyun bar = ((SLtt_Screen_Cols - 2) * p->curr) / p->total;
57*4882a593Smuzhiyun SLsmg_fill_region(y, 1, 1, bar, ' ');
58*4882a593Smuzhiyun SLsmg_refresh();
59*4882a593Smuzhiyun pthread_mutex_unlock(&ui__lock);
60*4882a593Smuzhiyun }
61*4882a593Smuzhiyun
tui_progress__finish(void)62*4882a593Smuzhiyun static void tui_progress__finish(void)
63*4882a593Smuzhiyun {
64*4882a593Smuzhiyun int y;
65*4882a593Smuzhiyun
66*4882a593Smuzhiyun if (use_browser <= 0)
67*4882a593Smuzhiyun return;
68*4882a593Smuzhiyun
69*4882a593Smuzhiyun ui__refresh_dimensions(false);
70*4882a593Smuzhiyun pthread_mutex_lock(&ui__lock);
71*4882a593Smuzhiyun y = SLtt_Screen_Rows / 2 - 2;
72*4882a593Smuzhiyun SLsmg_set_color(0);
73*4882a593Smuzhiyun SLsmg_fill_region(y, 0, 3, SLtt_Screen_Cols, ' ');
74*4882a593Smuzhiyun SLsmg_refresh();
75*4882a593Smuzhiyun pthread_mutex_unlock(&ui__lock);
76*4882a593Smuzhiyun }
77*4882a593Smuzhiyun
78*4882a593Smuzhiyun static struct ui_progress_ops tui_progress__ops = {
79*4882a593Smuzhiyun .init = __tui_progress__init,
80*4882a593Smuzhiyun .update = tui_progress__update,
81*4882a593Smuzhiyun .finish = tui_progress__finish,
82*4882a593Smuzhiyun };
83*4882a593Smuzhiyun
tui_progress__init(void)84*4882a593Smuzhiyun void tui_progress__init(void)
85*4882a593Smuzhiyun {
86*4882a593Smuzhiyun ui_progress__ops = &tui_progress__ops;
87*4882a593Smuzhiyun }
88