]> mj.ucw.cz Git - libucw.git/blob - ucw/opt-help.c
5c4b44e850f7d94a4902130122ecc9de178a3220
[libucw.git] / ucw / opt-help.c
1 /*
2  *      UCW Library -- Formatting of command-line option help
3  *
4  *      (c) 2013 Jan Moskyto Matejka <mq@ucw.cz>
5  *      (c) 2014 Martin Mares <mj@ucw.cz>
6  *
7  *      This software may be freely distributed and used according to the terms
8  *      of the GNU Lesser General Public License.
9  */
10
11 #include <ucw/lib.h>
12 #include <ucw/opt.h>
13 #include <ucw/opt-internal.h>
14 #include <ucw/mempool.h>
15 #include <ucw/gary.h>
16
17 #include <string.h>
18
19 struct help {
20   struct mempool *pool;
21   struct help_line *lines;                      // A growing array of lines
22 };
23
24 struct help_line {
25   const char *extra;
26   char *fields[3];
27 };
28
29 static void opt_help_scan_item(struct help *h, struct opt_precomputed *opt)
30 {
31   struct opt_item *item = opt->item;
32
33   if (opt->flags & OPT_NO_HELP)
34     return;
35
36   if (item->cls == OPT_CL_HELP) {
37     struct help_line *l = GARY_PUSH(h->lines, 1);
38     l->extra = item->help ? : "";
39     return;
40   }
41
42   if (item->letter >= OPT_POSITIONAL_TAIL)
43     return;
44
45   struct help_line *first = GARY_PUSH(h->lines, 1);
46   if (item->help) {
47     char *text = mp_strdup(h->pool, item->help);
48     struct help_line *l = first;
49     while (text) {
50       char *eol = strchr(text, '\n');
51       if (eol)
52         *eol++ = 0;
53
54       int field = (l == first ? 1 : 0);
55       char *f = text;
56       while (f) {
57         char *tab = strchr(f, '\t');
58         if (tab)
59           *tab++ = 0;
60         if (field < 3)
61           l->fields[field++] = f;
62         f = tab;
63       }
64
65       text = eol;
66       if (text)
67         l = GARY_PUSH(h->lines, 1);
68     }
69   }
70
71   if (item->name) {
72     char *val = first->fields[1] ? : "";
73     if (opt->flags & OPT_REQUIRED_VALUE)
74       val = mp_printf(h->pool, "=%s", val);
75     else if (!(opt->flags & OPT_NO_VALUE))
76       val = mp_printf(h->pool, "[=%s]", val);
77     first->fields[1] = mp_printf(h->pool, "--%s%s", item->name, val);
78   }
79
80   if (item->letter) {
81     if (item->name)
82       first->fields[0] = mp_printf(h->pool, "-%c, ", item->letter);
83     else {
84       char *val = first->fields[1] ? : "";
85       if (!(opt->flags & OPT_REQUIRED_VALUE) && !(opt->flags & OPT_NO_VALUE))
86         val = mp_printf(h->pool, "[%s]", val);
87       first->fields[0] = mp_printf(h->pool, "-%c%s", item->letter, val);
88       first->fields[1] = NULL;
89     }
90   }
91 }
92
93 static void opt_help_scan(struct help *h, const struct opt_section *sec)
94 {
95   for (struct opt_item * item = sec->opt; item->cls != OPT_CL_END; item++) {
96     if (item->cls == OPT_CL_SECTION)
97       opt_help_scan(h, item->u.section);
98     else {
99       struct opt_precomputed opt;
100       opt_precompute(&opt, item);
101       opt_help_scan_item(h, &opt);
102     }
103   }
104 }
105
106 void opt_help(const struct opt_section * sec) {
107   // Prepare help text
108   struct help h;
109   h.pool = mp_new(4096);
110   GARY_INIT_ZERO(h.lines, 0);
111   opt_help_scan(&h, sec);
112
113   // Calculate natural width of each column
114   uns n = GARY_SIZE(h.lines);
115   uns widths[3] = { 0, 0, 0 };
116   for (uns i=0; i<n; i++) {
117     struct help_line *l = &h.lines[i];
118     for (uns f=0; f<3; f++) {
119       if (!l->fields[f])
120         l->fields[f] = "";
121       uns w = strlen(l->fields[f]);
122       widths[f] = MAX(widths[f], w);
123     }
124   }
125   if (widths[0] > 4) {
126     /*
127      *  This is tricky: if there are short options, which have an argument,
128      *  but no long variant, we are willing to let column 0 overflow to column 1.
129      */
130     widths[1] = MAX(widths[1], widths[0] - 4);
131     widths[0] = 4;
132   }
133   widths[1] += 4;
134
135   // Print columns
136   for (uns i=0; i<n; i++) {
137     struct help_line *l = &h.lines[i];
138     if (l->extra)
139       puts(l->extra);
140     else {
141       int t = 0;
142       for (uns f=0; f<3; f++) {
143         t += widths[f];
144         t -= printf("%s", l->fields[f]);
145         while (t > 0) {
146           putchar(' ');
147           t--;
148         }
149       }
150       putchar('\n');
151     }
152   }
153
154   // Clean up
155   GARY_FREE(h.lines);
156   mp_delete(h.pool);
157 }
158
159 void opt_handle_help(struct opt_item * opt UNUSED, const char * value UNUSED, void * data)
160 {
161   opt_help(data);
162   exit(0);
163 }