Optional light colored background, for those who use black on white terminals.
[asterisk/asterisk.git] / main / term.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 1999 - 2005, Digium, Inc.
5  *
6  * Mark Spencer <markster@digium.com>
7  *
8  * See http://www.asterisk.org for more information about
9  * the Asterisk project. Please do not directly contact
10  * any of the maintainers of this project for assistance;
11  * the project provides a web site, mailing lists and IRC
12  * channels for your use.
13  *
14  * This program is free software, distributed under the terms of
15  * the GNU General Public License Version 2. See the LICENSE file
16  * at the top of the source tree.
17  */
18
19 /*! \file
20  *
21  * \brief Terminal Routines 
22  *
23  * \author Mark Spencer <markster@digium.com> 
24  */
25
26 #include "asterisk.h"
27
28 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
29
30 #include "asterisk/_private.h"
31 #include <sys/time.h>
32 #include <signal.h>
33 #include <sys/stat.h>
34 #include <fcntl.h>
35
36 #include "asterisk/term.h"
37 #include "asterisk/lock.h"
38 #include "asterisk/utils.h"
39
40 static int vt100compat;
41
42 static char prepdata[80] = "";
43 static char enddata[80] = "";
44 static char quitdata[80] = "";
45
46 static const char *termpath[] = {
47         "/usr/share/terminfo",
48         "/usr/local/share/misc/terminfo",
49         "/usr/lib/terminfo",
50         NULL
51         };
52
53 static int opposite(int color)
54 {
55         int lookup[] = {
56                 /* BLACK */ COLOR_BLACK,
57                 /* RED */ COLOR_MAGENTA,
58                 /* GREEN */ COLOR_GREEN,
59                 /* BROWN */ COLOR_BROWN,
60                 /* BLUE */ COLOR_CYAN,
61                 /* MAGENTA */ COLOR_RED,
62                 /* CYAN */ COLOR_BLUE,
63                 /* WHITE */ COLOR_BLACK };
64         return color ? lookup[color - 30] : 0;
65 }
66
67 /* Ripped off from Ross Ridge, but it's public domain code (libmytinfo) */
68 static short convshort(char *s)
69 {
70         register int a, b;
71
72         a = (int) s[0] & 0377;
73         b = (int) s[1] & 0377;
74
75         if (a == 0377 && b == 0377)
76                 return -1;
77         if (a == 0376 && b == 0377)
78                 return -2;
79
80         return a + b * 256;
81 }
82
83 int ast_term_init(void)
84 {
85         char *term = getenv("TERM");
86         char termfile[256] = "";
87         char buffer[512] = "";
88         int termfd = -1, parseokay = 0, i;
89
90         if (!term)
91                 return 0;
92         if (!ast_opt_console || ast_opt_no_color || !ast_opt_no_fork)
93                 return 0;
94
95         for (i=0 ;; i++) {
96                 if (termpath[i] == NULL) {
97                         break;
98                 }
99                 snprintf(termfile, sizeof(termfile), "%s/%c/%s", termpath[i], *term, term);
100                 termfd = open(termfile, O_RDONLY);
101                 if (termfd > -1) {
102                         break;
103                 }
104         }
105         if (termfd > -1) {
106                 int actsize = read(termfd, buffer, sizeof(buffer) - 1);
107                 short sz_names = convshort(buffer + 2);
108                 short sz_bools = convshort(buffer + 4);
109                 short n_nums   = convshort(buffer + 6);
110
111                 /* if ((sz_names + sz_bools) & 1)
112                         sz_bools++; */
113
114                 if (sz_names + sz_bools + n_nums < actsize) {
115                         /* Offset 13 is defined in /usr/include/term.h, though we do not
116                          * include it here, as it conflicts with include/asterisk/term.h */
117                         short max_colors = convshort(buffer + 12 + sz_names + sz_bools + 13 * 2);
118                         if (max_colors > 0) {
119                                 vt100compat = 1;
120                         }
121                         parseokay = 1;
122                 }
123                 close(termfd);
124         }
125
126         if (!parseokay) {
127                 /* These comparisons should not be substrings nor case-insensitive, as
128                  * terminal types are very particular about how they treat suffixes and
129                  * capitalization.  For example, terminal type 'linux-m' does NOT
130                  * support color, while 'linux' does.  Not even all vt100* terminals
131                  * support color, either (e.g. 'vt100+fnkeys'). */
132                 if (!strcmp(term, "linux")) {
133                         vt100compat = 1;
134                 } else if (!strcmp(term, "xterm")) {
135                         vt100compat = 1;
136                 } else if (!strcmp(term, "xterm-color")) {
137                         vt100compat = 1;
138                 } else if (!strncmp(term, "Eterm", 5)) {
139                         /* Both entries which start with Eterm support color */
140                         vt100compat = 1;
141                 } else if (!strcmp(term, "vt100")) {
142                         vt100compat = 1;
143                 } else if (!strncmp(term, "crt", 3)) {
144                         /* Both crt terminals support color */
145                         vt100compat = 1;
146                 }
147         }
148
149         if (vt100compat) {
150                 /* Make commands show up in nice colors */
151                 if (ast_opt_light_background) {
152                         snprintf(prepdata, sizeof(prepdata), "%c[%dm", ESC, COLOR_BROWN);
153                         snprintf(enddata, sizeof(enddata), "%c[%dm", ESC, COLOR_BLACK);
154                         snprintf(quitdata, sizeof(quitdata), "%c[0m", ESC);
155                 } else {
156                         snprintf(prepdata, sizeof(prepdata), "%c[%d;%dm", ESC, ATTR_BRIGHT, COLOR_BROWN);
157                         snprintf(enddata, sizeof(enddata), "%c[%d;%dm", ESC, ATTR_RESET, COLOR_WHITE);
158                         snprintf(quitdata, sizeof(quitdata), "%c[0m", ESC);
159                 }
160         }
161         return 0;
162 }
163
164 char *term_color(char *outbuf, const char *inbuf, int fgcolor, int bgcolor, int maxout)
165 {
166         int attr = 0;
167
168         if (!vt100compat) {
169                 ast_copy_string(outbuf, inbuf, maxout);
170                 return outbuf;
171         }
172         if (!fgcolor) {
173                 ast_copy_string(outbuf, inbuf, maxout);
174                 return outbuf;
175         }
176
177         if (fgcolor & 128) {
178                 attr = ast_opt_light_background ? 0 : ATTR_BRIGHT;
179                 fgcolor &= ~128;
180         }
181
182         if (ast_opt_light_background) {
183                 fgcolor = opposite(fgcolor);
184         }
185
186         snprintf(outbuf, maxout, "%c[%d;%dm%s%c[0m", ESC, attr, fgcolor, inbuf, ESC);
187         return outbuf;
188 }
189
190 char *term_color_code(char *outbuf, int fgcolor, int bgcolor, int maxout)
191 {
192         int attr = 0;
193         if ((!vt100compat) || (!fgcolor)) {
194                 *outbuf = '\0';
195                 return outbuf;
196         }
197
198         if (fgcolor & 128) {
199                 attr = ast_opt_light_background ? 0 : ATTR_BRIGHT;
200                 fgcolor &= ~128;
201         }
202
203         if (ast_opt_light_background) {
204                 fgcolor = opposite(fgcolor);
205         }
206
207         snprintf(outbuf, maxout, "%c[%d;%dm", ESC, attr, fgcolor);
208         return outbuf;
209 }
210
211 char *term_strip(char *outbuf, char *inbuf, int maxout)
212 {
213         char *outbuf_ptr = outbuf, *inbuf_ptr = inbuf;
214
215         while (outbuf_ptr < outbuf + maxout) {
216                 switch (*inbuf_ptr) {
217                         case ESC:
218                                 while (*inbuf_ptr && (*inbuf_ptr != 'm'))
219                                         inbuf_ptr++;
220                                 break;
221                         default:
222                                 *outbuf_ptr = *inbuf_ptr;
223                                 outbuf_ptr++;
224                 }
225                 if (! *inbuf_ptr)
226                         break;
227                 inbuf_ptr++;
228         }
229         return outbuf;
230 }
231
232 char *term_prompt(char *outbuf, const char *inbuf, int maxout)
233 {
234         if (!vt100compat) {
235                 ast_copy_string(outbuf, inbuf, maxout);
236                 return outbuf;
237         }
238         if (ast_opt_light_background) {
239                 snprintf(outbuf, maxout, "%c[%d;0m%c%c[%d;0m%s",
240                         ESC, COLOR_BLUE,
241                         inbuf[0],
242                         ESC, COLOR_BLACK,
243                         inbuf + 1);
244         } else {
245                 snprintf(outbuf, maxout, "%c[%d;%d;0m%c%c[%d;%d;0m%s",
246                         ESC, ATTR_BRIGHT, COLOR_BLUE,
247                         inbuf[0],
248                         ESC, 0, COLOR_WHITE,
249                         inbuf + 1);
250         }
251         return outbuf;
252 }
253
254 /* filter escape sequences */
255 void term_filter_escapes(char *line)
256 {
257         int i;
258         int len = strlen(line);
259
260         for (i = 0; i < len; i++) {
261                 if (line[i] != ESC)
262                         continue;
263                 if ((i < (len - 2)) &&
264                     (line[i + 1] == 0x5B)) {
265                         switch (line[i + 2]) {
266                         case 0x30:
267                         case 0x31:
268                         case 0x33:
269                                 continue;
270                         }
271                 }
272                 /* replace ESC with a space */
273                 line[i] = ' ';
274         }
275 }
276
277 char *term_prep(void)
278 {
279         return prepdata;
280 }
281
282 char *term_end(void)
283 {
284         return enddata;
285 }
286
287 char *term_quit(void)
288 {
289         return quitdata;
290 }