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
|
/*
* hbar
*
* create a horizontal progres bar across the screen
*/
#include <sys/ioctl.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <getopt.h>
#include "colors.h"
#define DEFAULT_COLOR BLACK BG_WHITE
#define DEFAULT_RESET WHITE BG_BLACK
int main (int argc, char **argv) {
struct winsize w;
ioctl(STDOUT_FILENO, TIOCGWINSZ, &w);
int width = w.ws_col;
char *text = "";
char *unit = "";
int total = 0;
int completed = 0;
char *color = DEFAULT_COLOR;
char *reset = DEFAULT_RESET;
int opt;
int option_index = 0;
const char *optstring = "T:u:c:r:";
static const struct option opts[] = {
{"text", required_argument, 0, 'T'},
{"unit", optional_argument, 0, 'u'},
{"color", optional_argument, 0, 'c'},
{"reset", optional_argument, 0, 'r'}
};
while ((opt = getopt_long(argc, argv, optstring, opts, &option_index)) != -1) {
switch (opt) {
case 'T':
text = optarg;
break;
case 'c':
printf("%s color is \n", optarg);
color = optarg;
break;
case 'r':
reset = optarg;
break;
case 'u':
unit = optarg;
break;
}
}
if (argc < optind + 2) {
printf(RESET "\n");
return 1;
}
completed = atoi(argv[optind]);
total = atoi(argv[optind+1]);
char *count = malloc(width);
sprintf(count, "[%d%s/%d%s]", completed, unit, total, unit);
printf(color);
for (int i = 0; i < width; i++) {
int reset_at = 0;
if (total > 0) {
float percent = (float) completed / (float) total;
reset_at = percent * width;
}
if (i == reset_at) {
printf(reset);
}
if (text && i < strlen(text)) {
printf("%c", text[i]);
} else if (i + 1 > width - strlen(count)) {
printf("%c", count[i - width + strlen(count)]);
} else {
printf(" ");
}
}
printf(RESET "\r");
return 0;
}
|
