Initial import from FreeBSD RELENG_4:
[dragonfly.git] / usr.bin / tput / tput.c
1 /*-
2  * Copyright (c) 1980, 1988, 1993
3  *      The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *      This product includes software developed by the University of
16  *      California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  */
33
34 #ifndef lint
35 static char copyright[] =
36 "@(#) Copyright (c) 1980, 1988, 1993\n\
37         The Regents of the University of California.  All rights reserved.\n";
38 #endif /* not lint */
39
40 #ifndef lint
41 #if 0
42 static char sccsid[] = "@(#)tput.c      8.2 (Berkeley) 3/19/94";
43 #endif
44 static char rcsid[] =
45 "$FreeBSD: src/usr.bin/tput/tput.c,v 1.7.6.3 2002/08/17 14:52:50 tjr Exp $";
46 #endif /* not lint */
47
48 #include <termios.h>
49
50 #include <err.h>
51 #include <termcap.h>
52 #include <stdio.h>
53 #include <stdlib.h>
54 #include <string.h>
55 #include <unistd.h>
56
57 #undef putchar
58 #define outc putchar
59
60 static void   prlongname __P((char *));
61 static void   usage __P((void));
62 static char **process __P((char *, char *, char **));
63
64 int
65 main(argc, argv)
66         int argc;
67         char **argv;
68 {
69         int ch, exitval, n;
70         char *cptr, *p, *term, buf[1024], tbuf[1024];
71
72         term = NULL;
73         while ((ch = getopt(argc, argv, "T:")) != -1)
74                 switch(ch) {
75                 case 'T':
76                         term = optarg;
77                         break;
78                 case '?':
79                 default:
80                         usage();
81                 }
82         argc -= optind;
83         argv += optind;
84
85         if (!term && !(term = getenv("TERM")))
86 errx(2, "no terminal type specified and no TERM environmental variable.");
87         if (tgetent(tbuf, term) != 1)
88                 err(2, "tgetent failure");
89         for (exitval = 0; (p = *argv) != NULL; ++argv) {
90                 switch (*p) {
91                 case 'c':
92                         if (!strcmp(p, "clear"))
93                                 p = "cl";
94                         break;
95                 case 'i':
96                         if (!strcmp(p, "init"))
97                                 p = "is";
98                         break;
99                 case 'l':
100                         if (!strcmp(p, "longname")) {
101                                 prlongname(tbuf);
102                                 continue;
103                         }
104                         break;
105                 case 'r':
106                         if (!strcmp(p, "reset"))
107                                 p = "rs";
108                         break;
109                 }
110                 cptr = buf;
111                 if (tgetstr(p, &cptr))
112                         argv = process(p, buf, argv);
113                 else if ((n = tgetnum(p)) != -1)
114                         (void)printf("%d\n", n);
115                 else
116                         exitval = !tgetflag(p);
117         }
118         exit(exitval);
119 }
120
121 static void
122 prlongname(buf)
123         char *buf;
124 {
125         int savech;
126         char *p, *savep;
127
128         for (p = buf; *p && *p != ':'; ++p);
129         savech = *(savep = p);
130         for (*p = '\0'; p >= buf && *p != '|'; --p);
131         (void)printf("%s\n", p + 1);
132         *savep = savech;
133 }
134
135 static char **
136 process(cap, str, argv)
137         char *cap, *str, **argv;
138 {
139         static char errfew[] =
140             "not enough arguments (%d) for capability `%s'";
141         static char errmany[] =
142             "too many arguments (%d) for capability `%s'";
143         static char erresc[] =
144             "unknown %% escape `%c' for capability `%s'";
145         char *cp;
146         int arg_need, arg_rows, arg_cols;
147
148         /* Count how many values we need for this capability. */
149         for (cp = str, arg_need = 0; *cp != '\0'; cp++)
150                 if (*cp == '%')
151                             switch (*++cp) {
152                             case 'd':
153                             case '2':
154                             case '3':
155                             case '.':
156                             case '+':
157                                     arg_need++;
158                                     break;
159                             case '%':
160                             case '>':
161                             case 'i':
162                             case 'r':
163                             case 'n':
164                             case 'B':
165                             case 'D':
166                                     break;
167                             case 'p':
168                                     if (cp[1]) {
169                                         cp++;
170                                         break;
171                                     }
172                             default:
173                                 /*
174                                  * hpux has lot's of them, but we complain
175                                  */
176                                  warnx(erresc, *cp, cap);
177                             }
178
179         /* And print them. */
180         switch (arg_need) {
181         case 0:
182                 (void)tputs(str, 1, outc);
183                 break;
184         case 1:
185                 arg_cols = 0;
186
187                 if (*++argv == NULL || *argv[0] == '\0')
188                         errx(2, errfew, 1, cap);
189                 arg_rows = atoi(*argv);
190
191                 (void)tputs(tgoto(str, arg_cols, arg_rows), 1, outc);
192                 break;
193         case 2:
194                 if (*++argv == NULL || *argv[0] == '\0')
195                         errx(2, errfew, 2, cap);
196                 arg_cols = atoi(*argv);
197
198                 if (*++argv == NULL || *argv[0] == '\0')
199                         errx(2, errfew, 2, cap);
200                 arg_rows = atoi(*argv);
201
202                 (void) tputs(tgoto(str, arg_cols, arg_rows), arg_rows, outc);
203                 break;
204
205         default:
206                 errx(2, errmany, arg_need, cap);
207         }
208         return (argv);
209 }
210
211 static void
212 usage()
213 {
214         (void)fprintf(stderr, "usage: tput [-T term] attribute ...\n");
215         exit(1);
216 }