7b061daf98074d1acf7acce2acf91b188ed5b0e8
[shellsnippets/shellsnippets.git] / mksh / progress-bar
1 # -*- mode: sh -*-
2 #-
3 # Copyright © 2015
4 #       mirabilos <thorsten.glaser@teckids.org>
5 #
6 # Provided that these terms and disclaimer and all copyright notices
7 # are retained or reproduced in an accompanying document, permission
8 # is granted to deal in this work without restriction, including un‐
9 # limited rights to use, publicly perform, distribute, sell, modify,
10 # merge, give away, or sublicence.
11 #
12 # This work is provided “AS IS” and WITHOUT WARRANTY of any kind, to
13 # the utmost extent permitted by applicable law, neither express nor
14 # implied; without malicious intent or gross negligence. In no event
15 # may a licensor, author or contributor be held liable for indirect,
16 # direct, other damage, loss, or other issues arising in any way out
17 # of dealing in the work, even if advised of the possibility of such
18 # damage or existence of a defect, except proven that it results out
19 # of said person’s immediate fault when using the work as intended.
20 #-
21 # Shell library for easy display of a progress bar
22 #
23 # Usage:
24 # – before:   init_progress_bar $n
25 # – $n times: draw_progress_bar
26 # – after:    done_progress_bar
27 #
28 # init_progress_bar trashes the EXIT signal handler!
29
30 # global variables used by this library
31 _cnt_progress_bar=0
32 _cur_progress_bar=0
33 isin_progress_bar=0
34 nlin_progress_bar=0
35
36 # args: $1 = number of draw_progress_bar calls to make up 100%
37 function init_progress_bar {
38         global -i _cnt_progress_bar=$1 _cur_progress_bar=0
39         global -i nlin_progress_bar=$LINES isin_progress_bar=1
40
41         trap 'done_progress_bar' EXIT
42         # newline; up one line (to ensure we are not in the last line);
43         # save position; set scrolling region; restore position
44         print -n "\\n\\e[A\\e7\\e[1;$((# nlin_progress_bar - 1))r\\e8"
45 }
46
47 function done_progress_bar {
48         (( isin_progress_bar )) || return 0
49         # save position; clear scrolling region;
50         # go to last line; delete line; restore position
51         print "\\e7\\e[0;0r\\e[$nlin_progress_bar;0H\\e[M\\e8"
52         isin_progress_bar=0
53 }
54
55 function draw_progress_bar {
56         local bar num w=$COLUMNS
57
58         ((# num = (++_cur_progress_bar * w * 8) / _cnt_progress_bar ))
59         while ((# num >= 8 )); do
60                 bar+=█
61                 ((# num -= 8 ))
62         done
63         case $num {
64         (7) bar+=▉ ;;
65         (6) bar+=▊ ;;
66         (5) bar+=▋ ;;
67         (4) bar+=▌ ;;
68         (3) bar+=▍ ;;
69         (2) bar+=▎ ;;
70         (1) bar+=▏ ;;
71         }
72         # fill complete line, right-align completion percentage display
73         local -R$w spc="$((# _cur_progress_bar * 100 / _cnt_progress_bar))%"
74         # elide percentage when it stops fitting
75         ((# (_cur_progress_bar * w / _cnt_progress_bar) > (w - 4) )) && spc=
76         # save position; go to last line; set colours;
77         # output a line full of spaces (and completion percentage);
78         # jump to first column; output bar (line præfix); restore position
79         print -n -- "\\e7\\e[$nlin_progress_bar;0H\\e[0;1;33;44m$spc\\r$bar\\e8"
80 }