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  * @(#) 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.3 2003/10/04 20:36:53 hmp 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(char *, char *, char **);
  54
  55 int
  56 main(int argc, char **argv)
  57 {
  58         int ch, exitval, n;
  59         char *cptr, *p, *term, buf[1024], tbuf[1024];
  60
  61         term = NULL;
  62         while ((ch = getopt(argc, argv, "T:")) != -1)
  63                 switch(ch) {
  64                 case 'T':
  65                         term = optarg;
  66                         break;
  67                 case '?':
  68                 default:
  69                         usage();
  70                 }
  71         argc -= optind;
  72         argv += optind;
  73
  74         if (!term && !(term = getenv("TERM")))
  75 errx(2, "no terminal type specified and no TERM environmental variable.");
  76         if (tgetent(tbuf, term) != 1)
  77                 err(2, "tgetent failure");
  78         for (exitval = 0; (p = *argv) != NULL; ++argv) {
  79                 switch (*p) {
  80                 case 'c':
  81                         if (!strcmp(p, "clear"))
  82                                 p = "cl";
  83                         break;
  84                 case 'i':
  85                         if (!strcmp(p, "init"))
  86                                 p = "is";
  87                         break;
  88                 case 'l':
  89                         if (!strcmp(p, "longname")) {
  90                                 prlongname(tbuf);
  91                                 continue;
  92                         }
  93                         break;
  94                 case 'r':
  95                         if (!strcmp(p, "reset"))
  96                                 p = "rs";
  97                         break;
  98                 }
  99                 cptr = buf;
 100                 if (tgetstr(p, &cptr))
 101                         argv = process(p, buf, argv);
 102                 else if ((n = tgetnum(p)) != -1)
 103                         (void)printf("%d\n", n);
 104                 else
 105                         exitval = !tgetflag(p);
 106         }
 107         exit(exitval);
 108 }
 109
 110 static void
 111 prlongname(char *buf)
 112 {
 113         int savech;
 114         char *p, *savep;
 115
 116         for (p = buf; *p && *p != ':'; ++p);
 117         savech = *(savep = p);
 118         for (*p = '\0'; p >= buf && *p != '|'; --p);
 119         (void)printf("%s\n", p + 1);
 120         *savep = savech;
 121 }
 122
 123 static char **
 124 process(char *cap, char *str, char **argv)
 125 {
 126         static char errfew[] =
 127             "not enough arguments (%d) for capability `%s'";
 128         static char errmany[] =
 129             "too many arguments (%d) for capability `%s'";
 130         static char erresc[] =
 131             "unknown %% escape `%c' for capability `%s'";
 132         char *cp;
 133         int arg_need, arg_rows, arg_cols;
 134
 135         /* Count how many values we need for this capability. */
 136         for (cp = str, arg_need = 0; *cp != '\0'; cp++)
 137                 if (*cp == '%')
 138                             switch (*++cp) {
 139                             case 'd':
 140                             case '2':
 141                             case '3':
 142                             case '.':
 143                             case '+':
 144                                     arg_need++;
 145                                     break;
 146                             case '%':
 147                             case '>':
 148                             case 'i':
 149                             case 'r':
 150                             case 'n':
 151                             case 'B':
 152                             case 'D':
 153                                     break;
 154                             case 'p':
 155                                     if (cp[1]) {
 156                                         cp++;
 157                                         break;
 158                                     }
 159                             default:
 160                                 /*
 161                                  * hpux has lot's of them, but we complain
 162                                  */
 163                                  warnx(erresc, *cp, cap);
 164                             }
 165
 166         /* And print them. */
 167         switch (arg_need) {
 168         case 0:
 169                 (void)tputs(str, 1, outc);
 170                 break;
 171         case 1:
 172                 arg_cols = 0;
 173
 174                 if (*++argv == NULL || *argv[0] == '\0')
 175                         errx(2, errfew, 1, cap);
 176                 arg_rows = atoi(*argv);
 177
 178                 (void)tputs(tgoto(str, arg_cols, arg_rows), 1, outc);
 179                 break;
 180         case 2:
 181                 if (*++argv == NULL || *argv[0] == '\0')
 182                         errx(2, errfew, 2, cap);
 183                 arg_cols = atoi(*argv);
 184
 185                 if (*++argv == NULL || *argv[0] == '\0')
 186                         errx(2, errfew, 2, cap);
 187                 arg_rows = atoi(*argv);
 188
 189                 (void) tputs(tgoto(str, arg_cols, arg_rows), arg_rows, outc);
 190                 break;
 191
 192         default:
 193                 errx(2, errmany, arg_need, cap);
 194         }
 195         return (argv);
 196 }
 197
 198 static void
 199 usage(void)
 200 {
 201         (void)fprintf(stderr, "usage: tput [-T term] attribute ...\n");
 202         exit(1);
 203 }