xref: /dragonfly/usr.bin/tput/tput.c (revision f02303f9)
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  * @(#) Copyright (c) 1980, 1988, 1993 The Regents of the University of California.  All rights reserved.
34  * @(#)tput.c	8.2 (Berkeley) 3/19/94
35  * $FreeBSD: src/usr.bin/tput/tput.c,v 1.7.6.3 2002/08/17 14:52:50 tjr Exp $
36  * $DragonFly: src/usr.bin/tput/tput.c,v 1.5 2005/07/30 11:53:52 liamfoy Exp $
37  */
38 
39 #include <termios.h>
40 
41 #include <err.h>
42 #include <termcap.h>
43 #include <stdio.h>
44 #include <stdlib.h>
45 #include <string.h>
46 #include <unistd.h>
47 
48 #undef putchar
49 #define outc putchar
50 
51 static void   prlongname(char *);
52 static void   usage(void);
53 static char **process(const char *, const char *, char **);
54 
55 int
56 main(int argc, char **argv)
57 {
58 	int ch, exitval, n;
59 	char *cptr, *term, buf[1024], tbuf[1024];
60 	const char *p;
61 
62 	term = NULL;
63 	while ((ch = getopt(argc, argv, "T:")) != -1)
64 		switch(ch) {
65 		case 'T':
66 			term = optarg;
67 			break;
68 		case '?':
69 		default:
70 			usage();
71 		}
72 	argc -= optind;
73 	argv += optind;
74 
75 	if (!term && !(term = getenv("TERM")))
76 errx(2, "no terminal type specified and no TERM environmental variable.");
77 	if (tgetent(tbuf, term) != 1)
78 		err(3, "tgetent failure");
79 	for (exitval = 0; (p = *argv) != NULL; ++argv) {
80 		switch (*p) {
81 		case 'c':
82 			if (!strcmp(p, "clear"))
83 				p = "cl";
84 			break;
85 		case 'i':
86 			if (!strcmp(p, "init"))
87 				p = "is";
88 			break;
89 		case 'l':
90 			if (!strcmp(p, "longname")) {
91 				prlongname(tbuf);
92 				continue;
93 			}
94 			break;
95 		case 'r':
96 			if (!strcmp(p, "reset"))
97 				p = "rs";
98 			break;
99 		}
100 		cptr = buf;
101 		if (tgetstr(p, &cptr))
102 			argv = process(p, buf, argv);
103 		else if ((n = tgetnum(p)) != -1)
104 			(void)printf("%d\n", n);
105 		else
106 			exitval = !tgetflag(p);
107 	}
108 	exit(exitval);
109 }
110 
111 static void
112 prlongname(char *buf)
113 {
114 	int savech;
115 	char *p, *savep;
116 
117 	for (p = buf; *p && *p != ':'; ++p);
118 	savech = *(savep = p);
119 	for (*p = '\0'; p >= buf && *p != '|'; --p);
120 	(void)printf("%s\n", p + 1);
121 	*savep = savech;
122 }
123 
124 static char **
125 process(const char *cap, const char *str, char **argv)
126 {
127 	static char errfew[] =
128 	    "not enough arguments (%d) for capability `%s'";
129 	static char errmany[] =
130 	    "too many arguments (%d) for capability `%s'";
131 	static char erresc[] =
132 	    "unknown %% escape `%c' for capability `%s'";
133 	const char *cp;
134 	int arg_need, arg_rows, arg_cols;
135 
136 	/* Count how many values we need for this capability. */
137 	for (cp = str, arg_need = 0; *cp != '\0'; cp++)
138 		if (*cp == '%')
139 			    switch (*++cp) {
140 			    case 'd':
141 			    case '2':
142 			    case '3':
143 			    case '.':
144 			    case '+':
145 				    arg_need++;
146 				    break;
147 			    case '%':
148 			    case '>':
149 			    case 'i':
150 			    case 'r':
151 			    case 'n':
152 			    case 'B':
153 			    case 'D':
154 				    break;
155 			    case 'p':
156 				    if (cp[1]) {
157 					cp++;
158 					break;
159 				    }
160 			    default:
161 				/*
162 				 * hpux has lot's of them, but we complain
163 				 */
164 				 warnx(erresc, *cp, cap);
165 			    }
166 
167 	/* And print them. */
168 	switch (arg_need) {
169 	case 0:
170 		(void)tputs(str, 1, outc);
171 		break;
172 	case 1:
173 		arg_cols = 0;
174 
175 		if (*++argv == NULL || *argv[0] == '\0')
176 			errx(2, errfew, 1, cap);
177 		arg_rows = atoi(*argv);
178 
179 		(void)tputs(tgoto(str, arg_cols, arg_rows), 1, outc);
180 		break;
181 	case 2:
182 		if (*++argv == NULL || *argv[0] == '\0')
183 			errx(2, errfew, 2, cap);
184 		arg_cols = atoi(*argv);
185 
186 		if (*++argv == NULL || *argv[0] == '\0')
187 			errx(2, errfew, 2, cap);
188 		arg_rows = atoi(*argv);
189 
190 		(void) tputs(tgoto(str, arg_cols, arg_rows), arg_rows, outc);
191 		break;
192 
193 	default:
194 		errx(2, errmany, arg_need, cap);
195 	}
196 	return (argv);
197 }
198 
199 static void
200 usage(void)
201 {
202 	(void)fprintf(stderr, "usage: tput [-T term] attribute ...\n");
203 	exit(2);
204 }
205